// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2022 Thomas Basler and others
 */
#include "MqttHandlePowerLimiterHass.h"
#include "Configuration.h"
#include "MqttSettings.h"
#include "NetworkSettings.h"
#include "MessageOutput.h"
#include "Utils.h"

MqttHandlePowerLimiterHassClass MqttHandlePowerLimiterHass;

void MqttHandlePowerLimiterHassClass::init(Scheduler& scheduler)
{
    scheduler.addTask(_loopTask);
    _loopTask.setCallback(std::bind(&MqttHandlePowerLimiterHassClass::loop, this));
    _loopTask.setIterations(TASK_FOREVER);
    _loopTask.enable();
}

void MqttHandlePowerLimiterHassClass::loop()
{
    if (!Configuration.get().PowerLimiter.Enabled) {
        return;
    }
    if (_updateForced) {
        publishConfig();
        _updateForced = false;
    }

    if (MqttSettings.getConnected() && !_wasConnected) {
        // Connection established
        _wasConnected = true;
        publishConfig();
    } else if (!MqttSettings.getConnected() && _wasConnected) {
        // Connection lost
        _wasConnected = false;
    }
}

void MqttHandlePowerLimiterHassClass::forceUpdate()
{
    _updateForced = true;
}

void MqttHandlePowerLimiterHassClass::publishConfig()
{
    auto const& config = Configuration.get();

    if (!config.Mqtt.Hass.Enabled) {
        return;
    }

    if (!MqttSettings.getConnected()) {
        return;
    }

    if (!config.PowerLimiter.Enabled) {
        return;
    }

    publishSelect("DPL Mode", "mdi:gauge", "config", "mode", "mode");

    if (config.PowerLimiter.IsInverterSolarPowered) {
        return;
    }

    // as this project revolves around Hoymiles inverters, 16 - 60 V is a reasonable voltage range
    publishNumber("DPL battery voltage start threshold", "mdi:battery-charging",
            "config", "threshold/voltage/start", "threshold/voltage/start", "V", 16, 60);
    publishNumber("DPL battery voltage stop threshold", "mdi:battery-charging",
            "config", "threshold/voltage/stop", "threshold/voltage/stop", "V", 16, 60);

    if (config.Vedirect.Enabled) {
        publishNumber("DPL full solar passthrough start voltage",
                "mdi:transmission-tower-import", "config",
                "threshold/voltage/full_solar_passthrough_start",
                "threshold/voltage/full_solar_passthrough_start", "V", 16, 60);
        publishNumber("DPL full solar passthrough stop voltage",
                "mdi:transmission-tower-import", "config",
                "threshold/voltage/full_solar_passthrough_stop",
                "threshold/voltage/full_solar_passthrough_stop", "V", 16, 60);
    }

    if (config.Battery.Enabled && !config.PowerLimiter.IgnoreSoc) {
        publishNumber("DPL battery SoC start threshold", "mdi:battery-charging",
                "config", "threshold/soc/start", "threshold/soc/start", "%", 0, 100);
        publishNumber("DPL battery SoC stop threshold", "mdi:battery-charging",
                "config", "threshold/soc/stop", "threshold/soc/stop", "%", 0, 100);

        if (config.Vedirect.Enabled) {
            publishNumber("DPL full solar passthrough SoC",
                    "mdi:transmission-tower-import", "config",
                    "threshold/soc/full_solar_passthrough",
                    "threshold/soc/full_solar_passthrough", "%", 0, 100);
        }
    }
}

void MqttHandlePowerLimiterHassClass::publishSelect(
    const char* caption, const char* icon, const char* category,
    const char* commandTopic, const char* stateTopic)
{

    String selectId = caption;
    selectId.replace(" ", "_");
    selectId.toLowerCase();

    const String configTopic = "select/powerlimiter/" + selectId + "/config";

    const String cmdTopic = MqttSettings.getPrefix() + "powerlimiter/cmd/" + commandTopic;
    const String statTopic = MqttSettings.getPrefix() + "powerlimiter/status/" + stateTopic;

    JsonDocument root;

    root["name"] = caption;
    root["uniq_id"] = selectId;
    if (strcmp(icon, "")) {
        root["ic"] = icon;
    }
    root["ent_cat"] = category;
    root["cmd_t"] = cmdTopic;
    root["stat_t"] = statTopic;
    JsonArray options = root["options"].to<JsonArray>();
    options.add("0");
    options.add("1");
    options.add("2");

    JsonObject deviceObj = root["dev"].to<JsonObject>();
    createDeviceInfo(deviceObj);

    if (!Utils::checkJsonAlloc(root, __FUNCTION__, __LINE__)) {
        return;
    }

    String buffer;
    serializeJson(root, buffer);
    publish(configTopic, buffer);
}

void MqttHandlePowerLimiterHassClass::publishNumber(
    const char* caption, const char* icon, const char* category,
    const char* commandTopic, const char* stateTopic, const char* unitOfMeasure,
    const int16_t min, const int16_t max)
{

    String numberId = caption;
    numberId.replace(" ", "_");
    numberId.toLowerCase();

    const String configTopic = "number/powerlimiter/" + numberId + "/config";

    const String cmdTopic = MqttSettings.getPrefix() + "powerlimiter/cmd/" + commandTopic;
    const String statTopic = MqttSettings.getPrefix() + "powerlimiter/status/" + stateTopic;

    JsonDocument root;

    root["name"] = caption;
    root["uniq_id"] = numberId;
    if (strcmp(icon, "")) {
        root["ic"] = icon;
    }
    root["ent_cat"] = category;
    root["cmd_t"] = cmdTopic;
    root["stat_t"] = statTopic;
    root["unit_of_meas"] = unitOfMeasure;
    root["min"] = min;
    root["max"] = max;
    root["mode"] = "box";

    auto const& config = Configuration.get();
    if (config.Mqtt.Hass.Expire) {
        root["exp_aft"] = config.Mqtt.PublishInterval * 3;
    }

    JsonObject deviceObj = root["dev"].to<JsonObject>();
    createDeviceInfo(deviceObj);

    if (!Utils::checkJsonAlloc(root, __FUNCTION__, __LINE__)) {
        return;
    }

    String buffer;
    serializeJson(root, buffer);
    publish(configTopic, buffer);
}

void MqttHandlePowerLimiterHassClass::createDeviceInfo(JsonObject& object)
{
    object["name"] = "Dynamic Power Limiter";
    object["ids"] = "0002";
    object["cu"] = String("http://") + NetworkSettings.localIP().toString();
    object["mf"] = "OpenDTU";
    object["mdl"] = "Dynamic Power Limiter";
    object["sw"] = AUTO_GIT_HASH;
}

void MqttHandlePowerLimiterHassClass::publish(const String& subtopic, const String& payload)
{
    String topic = Configuration.get().Mqtt.Hass.Topic;
    topic += subtopic;
    MqttSettings.publishGeneric(topic.c_str(), payload.c_str(), Configuration.get().Mqtt.Hass.Retain);
}