import json, os, time
from datetime import datetime as dt
from common.Logger import logger
from quickfaas.global_dict import get as get_params
from quickfaas.global_dict import _set_global_args

def reboot(reason="Rebooting for config file update"):
    #basic = Basic()
    logger.info("!" * 10 + "REBOOTING DEVICE" + "!"*10)
    logger.info(reason)
    r = os.popen("kill -s SIGHUP `cat /var/run/python/supervisord.pid`").read()
    logger.info(f"REBOOT : {r}")

def checkFileExist(filename):
    path = "/var/user/files"
    try:
        if not os.path.exists(path):
            logger.debug("no folder making files folder in var/user")
            os.makedirs(path)
            with open(path + "/" + filename, "a") as f:
                json.dump({}, f)
    except Exception as e:
        logger.error(f"Something went wrong in checkFileExist while making folder: {e}")
    
    try:
        if not os.path.exists(path + "/" + filename):
            logger.debug("no creds file making creds file")
            with open(path + "/" + filename, "a") as f:
                json.dump({}, f)
    except Exception as e:
        logger.error(f"Something went wrong in checkFileExist wihle making file: {e}")

def convertDStoJSON(ds):
    j = dict()
    try:
        for x in ds:
            j[x["key"]] = x["value"]
    except Exception as e:
        logger.error(f"Something went wrong in convertDStoJSON: {e}")
    return j

def convertJSONtoDS(j):
    d = []
    try:
        for key in j.keys():
            d.append({"key": key, "value": j[key]})
    except Exception as e:
        logger.error(f"Something went wrong in convertJSONtoDS: {e}")
    return d

def checkCredentialConfig():
    logger.debug("CHECKING CONFIG")
    cfgpath = "/var/user/cfg/device_supervisor/device_supervisor.cfg"
    credspath = "/var/user/files/creds.json"
    cfg = dict()
    with open(cfgpath, "r") as f:
        try:
            cfg = json.load(f)
            clouds = cfg.get("clouds")
            logger.debug(clouds)
            #if not configured then try to configure from stored values
            if clouds[0]["args"]["clientId"] == "unknown" or clouds[0]["args"]["username"] == "unknown" or not clouds[0]["args"]["passwd"] or clouds[0]["args"]["passwd"] == "unknown":
                try:
                    checkFileExist("creds.json")
                except Exception as e:
                    logger.error(f"Error in checkFileExist: {e}")
                with open(credspath, "r") as c:
                    try:
                        creds = json.load(c)
                        if creds:
                            logger.debug("updating config with stored data")
                            clouds[0]["args"]["clientId"] = creds["clientId"]
                            clouds[0]["args"]["username"] = creds["userName"]
                            clouds[0]["args"]["passwd"] = creds["password"]
                            cfg["clouds"] = clouds
                            cfg = checkParameterConfig(cfg)
                            with open(cfgpath, "w", encoding='utf-8') as n:
                                json.dump(cfg, n, indent=1, ensure_ascii=False)
                            reboot()
                    except Exception as e:
                        logger.error(f"Error trying to load credentials from file: {e}")
            else:
                #assuming clouds is filled out, if data is different then assume someone typed in something new and store it, if creds is empty fill with clouds' data
                checkFileExist("creds.json")
                with open(credspath, "r") as c:
                    logger.debug("updating stored file with new data")
                    cfg = checkParameterConfig(cfg)
                    with open(cfgpath, "w", encoding='utf-8') as n:
                        json.dump(cfg, n, indent=1, ensure_ascii=False)
                    creds = json.load(c)
                    if creds:
                        if creds["clientId"] != clouds[0]["args"]["clientId"]:
                            creds["clientId"] = clouds[0]["args"]["clientId"]
                        if creds["userName"] != clouds[0]["args"]["username"]:
                            creds["userName"] = clouds[0]["args"]["username"]
                        if creds["password"] != clouds[0]["args"]["passwd"]:
                            creds["password"] = clouds[0]["args"]["passwd"]
                    else:
                        creds["clientId"] = clouds[0]["args"]["clientId"]
                        creds["userName"] = clouds[0]["args"]["username"]
                        creds["password"] = clouds[0]["args"]["passwd"]
                    with open(credspath, "w") as cw:
                        json.dump(creds,cw)
        except Exception as e:
            logger.error(f"Somethign went wrong in checkCredentialConfig: {e}")

def checkParameterConfig(cfg):
    try:
        logger.debug("Checking Parameters!!!!")
        paramspath = "/var/user/files/params.json"
        cfgparams = convertDStoJSON(cfg.get("labels"))
        #check stored values 
        checkFileExist("params.json")
        with open(paramspath, "r") as f:
            logger.debug("Opened param storage file")
            params = json.load(f)
            if params:
                if cfgparams != params:
                    #go through each param
                    #if not "unknown" and cfg and params aren't the same take from cfg likely updated manually
                    #if key in cfg but not in params copy to params
                    logger.debug("equalizing params between cfg and stored")
                    for key in cfgparams.keys():
                        try:
                            if cfgparams[key] != params[key] and cfgparams[key] != "unknown":
                                params[key] = cfgparams[key]
                        except:
                            params[key] = cfgparams[key]
                    cfg["labels"] = convertJSONtoDS(params)
                    _set_global_args(convertJSONtoDS(params))
                    with open(paramspath, "w") as p:
                        json.dump(params, p)
            else:
                with open(paramspath, "w") as p:
                    logger.debug("initializing param file with params in memory")
                    json.dump(convertDStoJSON(get_params()), p)
                    cfg["labels"] = get_params()
        
        return cfg
    except Exception as e:
        logger.error(f"Something went wrong in checkParameterConfig: {e}")
        os.system(f'rm {paramspath}')
        return cfg
    
# Helper function to split the payload into chunks
def chunk_payload(payload, chunk_size=20):
    chunked_values = list(payload["values"].items())
    for i in range(0, len(chunked_values), chunk_size):
        yield {
            "ts": payload["ts"],
            "values": dict(chunked_values[i:i+chunk_size])
        }

def sendData(message):
    payload = {"ts": (round(dt.timestamp(dt.now())/600)*600)*1000, "values": {}}

    for chunk in chunk_payload(payload=payload):
        publish(__topic__, json.dumps(chunk), __qos__)
        time.sleep(2)
    
    publish("v1/devices/me/attributes", json.dumps({"latestReportTime": (round(dt.timestamp(dt.now())/600)*600)*1000}), __qos__)