
// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2022 Thomas Basler and others
 */
#include "WebApi_prometheus.h"
#include "Configuration.h"
#include "NetworkSettings.h"
#include <Hoymiles.h>

void WebApiPrometheusClass::init(AsyncWebServer* server)
{
    using std::placeholders::_1;

    _server = server;

    _server->on("/api/prometheus/metrics", HTTP_GET, std::bind(&WebApiPrometheusClass::onPrometheusMetricsGet, this, _1));
}

void WebApiPrometheusClass::loop()
{
}

void WebApiPrometheusClass::onPrometheusMetricsGet(AsyncWebServerRequest* request)
{
    auto stream = request->beginResponseStream("text/plain; charset=utf-8", 40960);

    stream->print(F("# HELP opendtu_build Build info\n"));
    stream->print(F("# TYPE opendtu_build gauge\n"));
    stream->printf("opendtu_build{name=\"%s\",id=\"%s\",version=\"%d.%d.%d\"} 1\n",
        NetworkSettings.getHostname().c_str(), AUTO_GIT_HASH, CONFIG_VERSION >> 24 & 0xff, CONFIG_VERSION >> 16 & 0xff, CONFIG_VERSION >> 8 & 0xff);

    stream->print(F("# HELP opendtu_platform Platform info\n"));
    stream->print(F("# TYPE opendtu_platform gauge\n"));
    stream->printf("opendtu_platform{arch=\"%s\",mac=\"%s\"} 1\n", ESP.getChipModel(), NetworkSettings.macAddress().c_str());

    stream->print(F("# HELP opendtu_uptime Uptime in seconds\n"));
    stream->print(F("# TYPE opendtu_uptime counter\n"));
    stream->printf("opendtu_uptime %lld\n", esp_timer_get_time() / 1000000);

    stream->print(F("# HELP opendtu_heap_size System memory size\n"));
    stream->print(F("# TYPE opendtu_heap_size gauge\n"));
    stream->printf("opendtu_heap_size %zu\n", ESP.getHeapSize());

    stream->print(F("# HELP opendtu_free_heap_size System free memory\n"));
    stream->print(F("# TYPE opendtu_free_heap_size gauge\n"));
    stream->printf("opendtu_free_heap_size %zu\n", ESP.getFreeHeap());

    stream->print(F("# HELP wifi_rssi WiFi RSSI\n"));
    stream->print(F("# TYPE wifi_rssi gauge\n"));
    stream->printf("wifi_rssi %d\n", WiFi.RSSI());

    for (uint8_t i = 0; i < Hoymiles.getNumInverters(); i++) {
        auto inv = Hoymiles.getInverterByPos(i);

        String serial = inv->serialString();
        const char* name = inv->name();
        if (i == 0) {
            stream->print(F("# HELP opendtu_last_update last update from inverter in s\n"));
            stream->print(F("# TYPE opendtu_last_update gauge\n"));
        }
        stream->printf("opendtu_last_update{serial=\"%s\",unit=\"%d\",name=\"%s\"} %d\n",
            serial.c_str(), i, name, inv->Statistics()->getLastUpdate() / 1000);

        // Loop all channels
        for (auto& t : inv->Statistics()->getChannelTypes()) {
            for (auto& c : inv->Statistics()->getChannelsByType(t)) {
                addField(stream, serial, i, inv, t, c, FLD_PAC);
                addField(stream, serial, i, inv, t, c, FLD_UAC);
                addField(stream, serial, i, inv, t, c, FLD_IAC);
                if (t == TYPE_AC) {
                    addField(stream, serial, i, inv, t, c, FLD_PDC, "PowerDC");
                } else {
                    addField(stream, serial, i, inv, t, c, FLD_PDC);
                }
                addField(stream, serial, i, inv, t, c, FLD_UDC);
                addField(stream, serial, i, inv, t, c, FLD_IDC);
                addField(stream, serial, i, inv, t, c, FLD_YD);
                addField(stream, serial, i, inv, t, c, FLD_YT);
                addField(stream, serial, i, inv, t, c, FLD_F);
                addField(stream, serial, i, inv, t, c, FLD_T);
                addField(stream, serial, i, inv, t, c, FLD_PF);
                addField(stream, serial, i, inv, t, c, FLD_PRA);
                addField(stream, serial, i, inv, t, c, FLD_EFF);
                addField(stream, serial, i, inv, t, c, FLD_IRR);
            }
        }
    }
    stream->addHeader(F("Cache-Control"), F("no-cache"));
    request->send(stream);
}

void WebApiPrometheusClass::addField(AsyncResponseStream* stream, String& serial, uint8_t idx, std::shared_ptr<InverterAbstract> inv, ChannelType_t type, ChannelNum_t channel, FieldId_t fieldId, const char* channelName)
{
    if (inv->Statistics()->hasChannelFieldValue(type, channel, fieldId)) {
        const char* chanName = (channelName == NULL) ? inv->Statistics()->getChannelFieldName(type, channel, fieldId) : channelName;
        if (idx == 0 && type == TYPE_AC && channel == 0) {
            stream->printf("# HELP opendtu_%s in %s\n", chanName, inv->Statistics()->getChannelFieldUnit(type, channel, fieldId));
            stream->printf("# TYPE opendtu_%s gauge\n", chanName);
        }
        stream->printf("opendtu_%s{serial=\"%s\",unit=\"%d\",name=\"%s\",type=\"%s\",channel=\"%d\"} %f\n",
            chanName,
            serial.c_str(),
            idx,
            inv->name(),
            inv->Statistics()->getChannelTypeName(type),
            channel,
            inv->Statistics()->getChannelFieldValue(type, channel, fieldId));
    }
}