
import taos
import sys
import time
import socket
import os
import platform
if platform.system().lower() == 'windows':
    import wexpect as taosExpect
else:
    import pexpect as taosExpect

from util.log import *
from util.sql import *
from util.cases import *
from util.dnodes import *

def taos_command (buildPath, key, value, expectString, sqlString=''):
    if len(key) == 0:
        tdLog.exit("taos test key is null!")

    if platform.system().lower() == 'windows':
        taosCmd = buildPath + '\\build\\bin\\taos.exe '
        taosCmd = taosCmd.replace('\\','\\\\')
    else:
        taosCmd = buildPath + '/build/bin/taos '

    cfgPath = buildPath + "/../sim/psim/cfg"
    taosCmd = taosCmd + ' -c' + cfgPath + ' -' + key
    if len(value) != 0:
        taosCmd = taosCmd + ' ' + value

    tdLog.info ("taos cmd: %s" % taosCmd)

    child = taosExpect.spawn(taosCmd, timeout=20)
    #output = child.readline()
    #print (output.decode())
    if len(expectString) != 0:
        i = child.expect([expectString, taosExpect.TIMEOUT, taosExpect.EOF], timeout=20)
    else:
        i = child.expect([taosExpect.TIMEOUT, taosExpect.EOF], timeout=20)

    if platform.system().lower() == 'windows':
        retResult = child.before
    else:
        retResult = child.before.decode()
    print(retResult)
    #print(child.after.decode())
    if i == 0:
        print ('taos login success! Here can run sql, taos> ')
        return  "TAOS_OK"
    else:
        return "TAOS_FAIL"

class TDTestCase:
    #updatecfgDict = {'clientCfg': {'serverPort': 7080, 'firstEp': 'trd02:7080', 'secondEp':'trd02:7080'},\
    #                 'serverPort': 7080, 'firstEp': 'trd02:7080'}
    hostname = socket.gethostname()
    if (platform.system().lower() == 'windows' and not tdDnodes.dnodes[0].remoteIP == ""):
        try:
            config = eval(tdDnodes.dnodes[0].remoteIP)
            hostname = config["host"]
        except Exception:
            hostname = tdDnodes.dnodes[0].remoteIP
    serverPort = '7080'
    rpcDebugFlagVal = '143'
    clientCfgDict = {'serverPort': '', 'firstEp': '', 'secondEp':'', 'rpcDebugFlag':'135', 'fqdn':''}
    clientCfgDict["serverPort"]    = serverPort
    clientCfgDict["firstEp"]       = hostname + ':' + serverPort
    clientCfgDict["secondEp"]      = hostname + ':' + serverPort
    clientCfgDict["rpcDebugFlag"]  = rpcDebugFlagVal
    clientCfgDict["fqdn"] = hostname

    updatecfgDict = {'clientCfg': {}, 'serverPort': '', 'firstEp': '', 'secondEp':'', 'rpcDebugFlag':'135', 'fqdn':''}
    updatecfgDict["clientCfg"]  = clientCfgDict
    updatecfgDict["serverPort"] = serverPort
    updatecfgDict["firstEp"]    = hostname + ':' + serverPort
    updatecfgDict["secondEp"]   = hostname + ':' + serverPort
    updatecfgDict["fqdn"] = hostname

    print ("===================: ", updatecfgDict)

    def init(self, conn, logSql, replicaVar=1):
        self.replicaVar = int(replicaVar)
        tdLog.debug(f"start to excute {__file__}")
        tdSql.init(conn.cursor())

    def getBuildPath(self):
        selfPath = os.path.dirname(os.path.realpath(__file__))

        if ("community" in selfPath):
            projPath = selfPath[:selfPath.find("community")]
        else:
            projPath = selfPath[:selfPath.find("tests")]

        for root, dirs, files in os.walk(projPath):
            if ("taosd" in files or "taosd.exe" in files):
                rootRealPath = os.path.dirname(os.path.realpath(root))
                if ("packaging" not in rootRealPath):
                    buildPath = root[:len(root) - len("/build/bin")]
                    break
        return buildPath

    def run(self):  # sourcery skip: extract-duplicate-method, remove-redundant-fstring
        tdSql.prepare()
        # time.sleep(2)
        tdSql.query("create user testpy pass 'testpy'")

        buildPath = self.getBuildPath()
        if (buildPath == ""):
            tdLog.exit("taosd not found!")
        else:
            tdLog.info("taosd found in %s" % buildPath)
        cfgPath = buildPath + "/../sim/psim/cfg"
        tdLog.info("cfgPath: %s" % cfgPath)

        checkNetworkStatus = ['0: unavailable', '1: network ok', '2: service ok', '3: service degraded', '4: exiting']
        netrole            = ['client', 'server']

        keyDict = {'h':'', 'P':'6030', 'p':'testpy', 'u':'testpy', 'a':'', 'A':'', 'c':'', 'C':'', 's':'', 'r':'', 'f':'', \
                   'k':'', 't':'', 'n':'', 'l':'1024', 'N':'100', 'V':'', 'd':'db', 'w':'30', '-help':'', '-usage':'', '?':''}

        keyDict['h'] = self.hostname
        keyDict['c'] = cfgPath
        keyDict['P'] = self.serverPort

        tdSql.query("drop database if exists db1")
        tdSql.query("create database if not exists db1 vgroups 1")
        tdSql.query("use db1")
        tdSql.query("create table tba (ts timestamp, f1 binary(2))")
        tdSql.query("insert into tba values (now, '22')")
        tdSql.query("select * from tba")
        tdSql.checkData(0, 1, '22')

        keyDict['s'] = "\"alter table db1.tba modify column f1 binary(5) \""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Query OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        keyDict['s'] = "\"insert into db1.tba values (now, '55555')\""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Insert OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        tdSql.query("select * from tba order by ts")
        tdSql.checkData(0, 1, '22')
        tdSql.checkData(1, 1, '55555')



        keyDict['s'] = "\"alter table db1.tba add column f2 binary(5) \""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Query OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        tdSql.query("select * from tba order by ts")
        tdSql.query("select * from tba order by ts")
        tdSql.checkData(0, 2, None)
        tdSql.checkData(1, 2, None)




        keyDict['s'] = "\"alter table db1.tba add column f3 binary(5) \""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Query OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        tdSql.query("select f3 from tba order by ts")
        tdSql.checkData(0, 0, None)
        tdSql.checkData(1, 0, None)


        tdSql.query("create table stb (ts timestamp, f1 int, f2 binary(2)) tags (tg1 binary(2))")
        tdSql.query("create table tb1 using stb tags('bb')")
        tdSql.query("insert into tb1 values (now, 2,'22')")
        tdSql.query("select count(*) from stb group by tg1")
        tdSql.checkData(0, 0, 1)

        keyDict['s'] = "\"alter table db1.stb modify tag tg1 binary(5) \""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Query OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        keyDict['s'] = "\"create table db1.tb2 using db1.stb tags('bbbbb')\""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Create OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        keyDict['s'] = "\"insert into db1.tb2 values (now, 2,'22')\""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Insert OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        keyDict['s'] = "\"alter table db1.stb modify column f2 binary(5) \""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Query OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        keyDict['s'] = "\"insert into db1.tb2 values (now, 3,'55555')\""
        retCode = taos_command(buildPath, "s", keyDict['s'], "Insert OK", '')
        if retCode != "TAOS_OK":
            tdLog.exit("taos -s fail")

        tdSql.query("select count(*) from stb group by tg1")
        tdSql.checkData(0, 0, 2)
        tdSql.checkData(1, 0, 1)


    def stop(self):
        tdSql.close()
        tdLog.success(f"{__file__} successfully executed")

tdCases.addLinux(__file__, TDTestCase())
tdCases.addWindows(__file__, TDTestCase())
