#include "Device.h"

#include "common/Spdlogger.h"
#include "common/Utils.h"
#include "protocol/CommEntity.h"
#include "common/JsonN.h"

#include <unordered_set>

static std::unordered_set<int> g_setCacheDeviceType = {3, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110};
static bool CheckCacheType(int type)
{
    return g_setCacheDeviceType.find(type) != g_setCacheDeviceType.end();
}

std::shared_ptr<Device> Device::create(Fields& fields)
{
    auto device = std::make_shared<Device>();
    device->deviceId = fields.get<int>("device_id");
    device->type = fields.get<int>("type");
    device->name = fields.value("name");
    device->code = fields.value("code");
    device->isOpen = fields.get<int>("is_open");
    device->attrsJson = fields.value("attrs");
    device->category = fields.get<int>("category");

    // 解析属性的JSON字符串，转换成键值对
    njson jsonroot;
    bool ret = JSON::parse(device->attrsJson, jsonroot);
    if (!ret) // 解析错误
    {
        spdlog::error("[device] device attr json parse error, device_id={}", device->deviceId);
    }
    else
    {
        for (auto& [key, val] : jsonroot.items()) {
            std::string valType = val.type_name();
            if (valType == "string") {
                device->attrs.set(key, val.get<std::string>());
            }
            else if (valType == "number") {
                device->attrs.set(key, val.get<int>());
            }
            else {
                spdlog::error("[device] device attr unknown type: key={}, valtype={}", key, valType);
            }
        }
    }

    //int step = 600;
    //for (int i = 0; i*600<86400; ++i)
    //{
    //    double voltage = double(Utils::random(20000, 30000))*0.01;
    //    double current = double(Utils::random(1000, 2000))*0.01;
    //    device->mapCacheVoltage[i*step] = voltage;
    //    device->mapCacheCurrent[i*step] = current;
    //    device->mapCachePower[i*step] = voltage * current;
    //}

    // 启动通讯，该函数中会自动判断isOpen状态，选择是否进行通讯连接
    //device->startComm();
    return device;
}


int Device::startComm()
{
    if (!isOpen) 
    {
        if (commEntity && commEntity->alive)
        {
            commEntity->close();
        }
        return 0;
    }

    //// 从属性列表中获取通讯方式和通讯地址、端口
    //std::string commType = attrs.value("commType");
    //
    //// 如果entity的通讯协议类型当前配置不一致，需要关闭连接删除通讯后创建新的通讯
    //if (commEntity && commEntity->type != commType)
    //{
    //    commEntity->close();
    //    commEntity = nullptr;
    //}
    //// 创建新的通讯
    //if (!commEntity) 
    //{
    //    commEntity = CommEntity::create(attrs);
    //    if (!commEntity) { return -1; }
    //}
    //commEntity->start();
    return 0;
}

void  Device::getCacheVoltage(std::vector<std::string>& vec)
{
    vec.resize(mapCacheVoltage.size());
    int i = 0;
    for (auto iter = mapCacheVoltage.begin(); iter != mapCacheVoltage.end(); ++iter)
    {
        vec[i] = Utils::toStr(iter->second);
        i++;
    }
}
void  Device::getCacheCurrent(std::vector<std::string>& vec)
{
    vec.resize(mapCacheCurrent.size());
    int i = 0;
    for (auto iter = mapCacheCurrent.begin(); iter != mapCacheCurrent.end(); ++iter)
    {
        vec[i] = Utils::toStr(iter->second);
        i++;
    }
}
void  Device::getCachePower(std::vector<std::string>& vec)
{
    vec.resize(mapCachePower.size());
    int i = 0;
    for (auto iter = mapCachePower.begin(); iter != mapCachePower.end(); ++iter)
    {
        vec[i] = Utils::toStr(iter->second);
        i++;
    }
}

int64_t GetCurrentTimePos(int step)
{
    auto tp = chrono::system_clock::now();
    int64_t tTime = chrono::time_point_cast<chrono::seconds>(tp).time_since_epoch().count();
    std::time_t t = chrono::system_clock::to_time_t(tp);
    std::tm* tmlocal = localtime(&t);
    tmlocal->tm_hour = 0;
    tmlocal->tm_min = 0;
    tmlocal->tm_sec = 0;
    int64_t tDate = chrono::time_point_cast<chrono::seconds>(chrono::system_clock::from_time_t(mktime(tmlocal))).time_since_epoch().count();
    return (tTime - tDate) / step;
}

void Device::setCache(int datatype, std::vector<double>& vec)
{
    std::map<int, double>* mapptr = NULL;
    if (datatype == 1) { mapptr = &mapCacheVoltage; }
    else if (datatype == 2) { mapptr = &mapCacheCurrent; }
    else if (datatype == 3) { mapptr = &mapCachePower; }

    if (mapptr)
    {
        const int step = 600;
        const int N = 86400/step;

        int n = GetCurrentTimePos(step);
        for (int i = 0; i<N; ++i)
        {
            if (i < vec.size()) { (*mapptr)[i] = vec[i]; }
            else if (i <= n) { (*mapptr)[i] = 0; }
        }
    }
}

bool Device::cache(int npos)
{
    if (!CheckCacheType(this->type))
    {
        return false;
    }
    if (npos == 0)
    {
        mapCacheVoltage.clear();
        mapCacheCurrent.clear();
        mapCachePower.clear();
    }
    // 根据设备类型从参数（寄存器地址）中读取实时数据进行保存
    mapCacheVoltage[npos] = Utils::random(100, 200);
    mapCacheCurrent[npos] = Utils::random(100, 200);
    mapCachePower[npos] = Utils::random(100, 200);

    return true;
}

void Device::storeDB(int npos)
{
}

void Device::setParam(std::string k, std::string v)
{
    mapParams[k] = v;
}

std::string Device::getParam(std::string k, std::string defaultVal)
{
    auto iter = mapParams.find(k);
    if (iter != mapParams.end())
    {
        return iter->second;
    }
    return defaultVal;
}

void Device::getRuntimeParams(std::vector<std::pair<std::string, std::string>>& params)
{
    // 3    电表
    // 101	EMS
    // 102	PCS
    // 103	PCU
    // 104	BMS
    // 105	BCU
    // 106	充电桩
    // 109	光伏板

    if (this->type == 3)
    {
        params.push_back({"A相电压", getParam("0x000B", "0.0") + "V"});
        params.push_back({"B相电压", getParam("0x000D", "0.0") + "V"});
        params.push_back({"C相电压", getParam("0x000F", "0.0") + "V"});
        params.push_back({"A相电流", getParam("0x0011", "0.0") + "A"});
        params.push_back({"B相电流", getParam("0x0013", "0.0") + "A"});
        params.push_back({"C相电流", getParam("0x0015", "0.0") + "A"});
    }
    else if (this->type == 101)
    {
        params.push_back({"额定电压", getParam("0x0001", "0.0") + "V"});
        params.push_back({"实时电压", getParam("0x0001", "0.0") + "V"});
        params.push_back({"额定电流", getParam("0x0001", "0.0") + "A"});
        params.push_back({"实时电流", getParam("0x0001", "0.0") + "A"});
        params.push_back({"额定功率", getParam("0x0001", "0.0") + "kW"});
        params.push_back({"实时功率", getParam("0x0001", "0.0") + "A"});
    }
    //else if (this->type == 101)
    //{

    //}
    else
    {
        params.push_back({"额定电压", getParam("0x0001", "0.0") + "V"});
        params.push_back({"实时电压", getParam("0x0001", "0.0") + "V"});
        params.push_back({"额定电流", getParam("0x0001", "0.0") + "A"});
        params.push_back({"实时电流", getParam("0x0001", "0.0") + "A"});
        params.push_back({"额定功率", getParam("0x0001", "0.0") + "kW"});
        params.push_back({"实时功率", getParam("0x0001", "0.0") + "A"});
    }
}