import hmac
import base64
import random
import mqtt
from PikaStdDevice import Time


class IOT:
    def __init__(self):
        self._signMethodTable = ["hmac-md5", "hmac-sha1", "hmac-sha256"]

    def randStr(self, len):
        a = ""
        for i in range(len):
            a = a + str(random.randint(0, 9))
        return a

    def getTimeStamp(self, t):
        return int(Time.time()*1000) + t  # todo 时间戳64位整形实现

    def aliyun(self, clientId: str, productKey: str, deviceName: str, deviceSecret: str,
               signMethod="hmac-md5", regionID="cn-shanghai", ssl=False):
        if clientId == None or productKey == None or deviceName == None or deviceSecret == None:
            print("[Error]input param is None")
            return False
        if signMethod not in self._signMethodTable:
            print("[Error]not support signMethod")
            return False
        if ssl:
            securemode = "2"
            self.port = int(443)
        else:
            securemode = "3"
            self.port = int(1883)

        self.address = productKey + ".iot-as-mqtt." + regionID + ".aliyuncs.com"
        self.username = deviceName + "&" + productKey
        self.clientid = clientId + "|securemode=" + securemode + \
            ",signmethod="+signMethod.replace("-", "")+"|"
        hmac_payload = "clientId" + clientId + "deviceName" + \
            deviceName + "productKey" + productKey
        self.password = hmac.new(deviceSecret.encode(),
                                 msg=hmac_payload.encode(),
                                 digestmod=signMethod).hexdigest()
        return True

    def tencent(self, productId, deviceName, deviceSecret, signMethod="hmac-sha1", expiryTime=3600, ssl=False):
        if productId == None or deviceName == None or deviceSecret == None:
            print("[Error]input param is None")
            return False
        if signMethod not in self._signMethodTable:
            print("[Error]not support signMethod")
            return False
        connid = self.randStr(5)
        expiry = self.getTimeStamp(expiryTime)
        self.address = productId + ".iotcloud.tencentdevices.com"
        self.port = int(1883)
        self.clientid = productId + deviceName
        self.username = self.clientid + ";12010126;" + connid + ";" + str(expiry)
        token = hmac.new(base64.b64decode(deviceSecret.encode()),
                         msg=self.username.encode(),
                         digestmod=signMethod).hexdigest()
        self.password = token + ";" + signMethod.replace("-", "")
        return True

    def onenet(self): ...

    def connect(self, keepalive=600):
        self._client = mqtt.MQTT(self.address, port=self.port, clinetID=self.clientid,
                                 username=self.username, password=self.password, keepalive=keepalive)
        return self._client.connect()

    def disconnect(self):
        return self._client.disconnect()

    def subsribe(self, topic, cb, qos=0):
        return self._client.subscribe(topic, qos, cb)

    def publish(self, topic, payload, qos=0):
        return self._client.publish(qos, topic, payload)


def new():
    iot = IOT()
    return iot