#include "AppData.h"
#include "common/Utils.h"
#include "app/Station.h"
#include "app/Device.h"
#include "app/Policy.h"
#include "database/Dao.h"
#include "common/JsonN.h"
#include "common/Snowflake.h"
#include "common/Spdlogger.h"
#include "protocol/MqttEntity.h"

void ElectPeriod::parse(std::string jsonstr)
{
    njson jsonroot;
    JSON::parse(jsonstr, jsonroot);

    JSON::read(jsonroot, "price_super_peak", this->priceSuperPeak);
    JSON::read(jsonroot, "price_peak", this->pricePeak);
    JSON::read(jsonroot, "price_shoulder", this->priceShoulder);
    JSON::read(jsonroot, "price_off_peak", this->priceOffPeak);
    JSON::read<std::vector<std::vector<std::string>>>(jsonroot, "periods", this->vecPeriods);
}

std::string ElectPeriod::dump()
{
    njson jsonroot;
    jsonroot["price_super_peak"] = this->priceSuperPeak;
    jsonroot["price_peak"] = this->pricePeak;
    jsonroot["price_shoulder"] = this->priceShoulder;
    jsonroot["price_off_peak"] = this->priceOffPeak;
    jsonroot["periods"] = this->vecPeriods;
    return jsonroot.dump();
}




void AppData::initFromDB()
{
    auto dao = DaoEntity::create("");
    if (!dao->isConnected())
    {
        spdlog::error("Init app data failed, database connected error.");
        return;
    }

    std::string str;
    std::vector<Fields> result;

    { // 数据库读取工作模式定义
        str = "", result.clear();
        DAO::queryWorkModeDef(dao, result);
        for (auto& fields: result)
        {
            int workModeId = fields.get<int>(DMDefWorkMode::WORK_MODE_ID);
            std::string name = fields.value(DMDefWorkMode::NAME);
            this->mapping.workMode.push_back({std::to_string(workModeId), name});
            this->mapWorkMode[workModeId] = name;
            str += ("工作模式: {" + std::to_string(workModeId)+":" + name + "},");
        }
        spdlog::info(str);
    }
    { // 数据库读取策略类型定义
        str = "", result.clear();
        DAO::queryPolicyTypeDef(dao, result);
        for (auto& fields: result)
        {
            int policyTypeId = fields.get<int>(DMDefPolicyType::POLICY_TYPE_ID);
            std::string name = fields.value(DMDefWorkMode::NAME);
            this->mapping.policyType.push_back({std::to_string(policyTypeId), name});
            this->mapPolicyType[policyTypeId] = name;
            str += ("策略类型: {" + std::to_string(policyTypeId) + ":" + name + "},");
        }
        spdlog::info(str);
    }
    { // 数据库读取设备类型定义
        str = "", result.clear();
        DAO::queryDeviceTypeDef(dao, result);
        for (auto& fields: result)
        {
            auto item = std::make_shared<DeviceType>();
            item->typeId = fields.get<int>(DMDefDeviceType::DEVICE_TYPE_ID);
            item->name = fields.value(DMDefDeviceType::NAME);
            item->group = fields.value(DMDefDeviceType::GROUP);
            item->attr = fields.value(DMDefDeviceType::ATTRS);
            item->fieldsAttr.parseJson(item->attr);
            mapDeviceType[item->typeId] = item;
            mapping.deviceType.push_back({std::to_string(item->typeId), item->name});
            str += ("设备类型: {" + std::to_string(item->typeId) + ":" + item->name + "},");
        }
        spdlog::info(str);
    }
    { // 数据库读取角色定义
        str = "", result.clear();
        this->mapping.role.clear();
        DAO::queryRoleList(dao, result);
        for (auto& fields : result)
        {
            auto item = std::make_shared<Role>();
            item->roleId = fields.get<int>(DMRole::ROLE_ID);
            item->name = fields.value(DMRole::NAME);
            item->isOpen = fields.get<int>(DMRole::IS_OPEN);
            mapRole[item->roleId] = item;
            mapping.role.push_back({std::to_string(item->roleId), item->name});
            str += ("角色: {" + std::to_string(item->roleId) + ":" + item->name + "},");
        }
        spdlog::info(str);
    }
    { // 数据库读取场站信息
        str = "", result.clear();
        DAO::queryStationList(dao, result);
        for (auto& fields: result)
        {
            auto station = std::make_shared<Station>();
            station->setFields(fields);
            this->mapStation[station->id] = station;
            mapping.stationName.push_back({std::to_string(station->id), station->name});
            str += ("场站: {" + std::to_string(station->id) + ":" + station->name + "},");
        }
        spdlog::info(str);
    }
    { // 数据库读取设备信息
        str = "", result.clear();
        DAO::queryDeviceList(dao, result);
        for (auto& fields: result)
        {
            int deviceId = fields.get<int>(DMDevice::DEVICE_ID);
            int stationId = fields.get<int>(DMDevice::STATION_ID);
            auto station = this->getStation(stationId);
            if (station)
            {
                auto device = Device::create(fields);
                station->addDevice(deviceId, device);
            }
            else
            {
                spdlog::error("init device error: unknown station_id:, device_id=",  stationId, deviceId);
            }
        }
    }
    { // 数据库读取策略信息
        str = "", result.clear();
        DAO::queryPolicyList(dao, result);
        for (auto& fields: result)
        {
            auto policy = std::make_shared<MyPolicy>();
            policy->policyId = fields.get<int>(DMPolicy::POLICY_ID);
            policy->type = fields.get<int>(DMPolicy::TYPE);
            policy->name = fields.value(DMPolicy::NAME);
            policy->value = fields.value(DMPolicy::VALUE);
            this->mapPolicy[policy->policyId] = policy;
        }
    }
    { // 数据库读取电价分段信息
        result.clear();
        vecElectPeriods.resize(12);
        DAO::exec(dao, "SELECT * FROM configure;", result);
        
        Fields info;
        for (auto& fields: result)
        {
            auto k = fields.value("key");
            auto v = fields.value("val");
            info.set(k, v);
        }

        for (int month = 1; month<=12; month++)
        {
            if (month-1 < vecElectPeriods.size())
            {
                auto& vecItems = vecElectPeriods[month-1];
                std::string str = info.value("period_" + std::to_string(month));
                std::vector<std::string> vec;
                Utils::split(str, ",", vecItems);
            }
        }
        electPriceSuperPeak = info.get<double>("price_super_peak");
        electPricePeak = info.get<double>("price_peak");
        electPriceShoulder = info.get<double>("price_shoulder");
        electPriceOffPeak = info.get<double>("price_off_peak");
    }
    { // 数据库读取统计数据
        vector<Fields> result;
        std::string curDate = Utils::dateStr();
        DAO::queryStatDataList(dao, curDate, curDate, result);
        for (auto& fields: result)
        {
            std::string dt = fields.value(DMStatStation::DT);
            int stationId = fields.get<int>(DMStatStation::STATION_ID);
            auto station = this->getStation(stationId);
            if (station)
            {
                station->storageIn = fields.get<double>(DMStatStation::STORAGE_ELECT_IN);
                station->storageOut = fields.get<double>(DMStatStation::STORAGE_ELECT_OUT);
                //station->storageNumIn = fields.getFloat(DMStatStation::STORAGE_NUM);
                //station->storageNumOut = fields.getFloat(DMStatStation::STORAGE_NUM);
                station->storageNumErr = fields.get<int>(DMStatStation::STORAGE_NUM_ERR);

                station->solarGen = fields.get<double>(DMStatStation::SOLAR_ELECT_GEN);
                station->solarGrid = fields.get<double>(DMStatStation::SOLAR_ELECT_GRID);
                station->solarNumErr = fields.get<int>(DMStatStation::SOLAR_NUM_ERR);

                station->chargeElect = fields.get<double>(DMStatStation::CHARGE_ELECT);
                station->chargeNum = fields.get<int>(DMStatStation::CHARGE_NUM);
                station->chargeNumErr = fields.get<int>(DMStatStation::CHARGE_NUM_ERR);
            }
            else
            {
                spdlog::error("init staticis data error: unknown station_id:{}, dt={}", stationId, dt);
            }
        }
    }
}

void AppData::init()
{
    this->initFromDB();

    auto& optionMqtt = Config::option.mqtt;
    if (!optionMqtt.host.empty())
    {
        for (auto& item : mapStation)
        {
            auto& station = item.second;
            // "tcp://localhost:1883"
            station->mqttCli->init(optionMqtt.host, station->code, optionMqtt.username, optionMqtt.password);
        }
    }
}

std::shared_ptr<Station> AppData::getStation(int stationId)
{
    auto iter = mapStation.find(stationId);
    if (iter!=mapStation.end())
    {
        return iter->second;
    }
    return nullptr;
}
int AppData::getStationCount()
{
    return mapStation.size();
}

std::shared_ptr<Station> AppData::getStationByName(std::string name)
{
    for (auto iter = mapStation.begin(); iter!=mapStation.end(); ++iter)
    {
        if (iter->second->name == name)
        {
            return iter->second;
        }
    }
    return nullptr;
}

std::shared_ptr<Station> AppData::getStationByCode(std::string code)
{
    for (auto iter = mapStation.begin(); iter!=mapStation.end(); ++iter)
    {
        if (iter->second->code == code)
        {
            return iter->second;
        }
    }
    return nullptr;
}

std::shared_ptr<Device> AppData::getDevice(int stationId, int deviceId)
{
    auto station = getStation(stationId);
    if (station)
    {
        return station->getDevice(deviceId);
    }
    return nullptr;
}

std::shared_ptr<Device> AppData::getDeviceByType(int stationId, int deviceType, std::string code)
{
    auto station = getStation(stationId);
    if (station)
    {
        return station->getDeviceByType(deviceType, code);
    }
    return nullptr;
}

std::string AppData::getDeviceNameById(int typeId)
{
    auto iter = mapDeviceType.find(typeId);
    if (iter != mapDeviceType.end())
    {
        return iter->second->name;
    }
    return "";
}

std::shared_ptr<DeviceType> AppData::getDeviceTypeDef(int typeId)
{
    auto iter = mapDeviceType.find(typeId);
    if (iter != mapDeviceType.end())
    {
        return iter->second;
    }
    return nullptr;
}

void AppData::loadStatData()
{
}

static std::map<std::string, User> g_mapUser;
std::string AppData::userLogin(std::string userId, std::string account)
{
    for (auto iter = g_mapUser.begin(); iter!=g_mapUser.end(); ++iter)
    {
        if (iter->second.userId == userId) // 重复登录
        {
            g_mapUser.erase(iter);
            break;
        }
    }
    User user;
    user.userId = userId;
    user.account = account;
    user.loginTime = Utils::time();
    user.token = Snowflake::instance().getIdStr();
    g_mapUser[user.token] = user;
    return user.token;
}
User AppData::getUser(std::string token)
{
    auto iter = g_mapUser.find(token);
    if (iter != g_mapUser.end()) 
    { 
        return iter->second; 
    }
    return User();
}


int AppData::getWorkModeIdByName(std::string  name)
{
    for (auto iter = mapWorkMode.begin(); iter!=mapWorkMode.end(); ++iter)
    {
        if (iter->second == name)
        {
            return iter->first;
        }
    }
    return 0;
}

std::vector<std::string> AppData::getRoleNames()
{
    std::vector<std::string> vec(mapRole.size());
    int i = 0;
    for (auto iter = mapRole.begin(); iter!=mapRole.end(); ++iter)
    {
        vec[i] = iter->second->name;
        ++i;
    }
    return vec;
}

std::vector<std::string> AppData::getStationNames()
{
    std::vector<std::string> vec(mapStation.size());
    int i = 0;
    for (auto iter = mapStation.begin(); iter!=mapStation.end(); ++iter)
    {
        vec[i] = iter->second->name;
        ++i;
    }
    return vec;
}

std::vector<std::string> AppData::getDeviceTypeNames()
{
    std::vector<std::string> vec(mapping.deviceType.size());
    int i = 0;
    for (auto iter = mapping.deviceType.begin(); iter!=mapping.deviceType.end(); ++iter)
    {
        vec[i] = iter->second;
        ++i;
    }
    return vec;
}

std::vector<std::string> AppData::getWorkModes()
{
    std::vector<std::string> vec(mapWorkMode.size());
    int i = 0;
    for (auto iter = mapWorkMode.begin(); iter!=mapWorkMode.end(); ++iter)
    {
        vec[i] = iter->second;
        ++i;
    }
    return vec;
}

std::vector<std::string> AppData::getPolicyTypeNames()
{
    std::vector<std::string> vec(mapPolicyType.size());
    int i = 0;
    for (auto iter = mapPolicyType.begin(); iter!=mapPolicyType.end(); ++iter)
    {
        vec[i] = iter->second;
        ++i;
    }
    return vec;
}

std::vector<std::string> AppData::getPolicyNames()
{
    std::vector<std::string> vec;
    return vec;
}

int AppData::getPolicyTypeId(std::string name)
{
    for (auto iter = mapPolicyType.begin(); iter != mapPolicyType.end(); ++iter)
    {
        if (iter->second == name) { return iter->first; }
    }
    return 0;
}


std::vector<std::string> AppData::getElectPreiodVals(int month)
{
    if (month > 0 && month-1 < vecElectPeriods.size())
    {
        return vecElectPeriods[month-1];
    }
    return {};
}

std::string AppData::getElectPreiodVal(int month, int hour)
{
    if (month > 0 && month-1 < vecElectPeriods.size())
    {
        auto& vec = vecElectPeriods[month-1];
        if (hour > 0 && hour-1 < vec.size())
        {
            auto& val = vec[hour-1];
            if (val == "尖") return "尖峰";
            if (val == "峰") return "高峰";
            if (val == "平") return "平段";
            if (val == "谷") return "低谷";
            return val;
        }
    }
    return "";
}

void AppData::storeRuntimeDB()
{
    auto t = Utils::date();

    std::string valStr;
    for (auto iter=mapDataDay.begin(); iter!=mapDataDay.end(); ++iter)
    {
        auto& v = iter->second;
        if (v != 0.0)
        {
            if (!valStr.empty()) valStr += ",";
            valStr += ("[" + std::to_string(iter->first) + "," + Utils::toStr(v, 2) + "]");
        }
    }
    valStr = "[" + valStr + "]";
    Fields fields;
    fields.set("dt", Utils::dateStr(t));
    fields.set("device_id", 1);
    fields.set("datatype", 1);
    fields.set("value", valStr);
    DAO::insertRuntimeData(NULL, fields);
}