#include <queue>
#include <AsyncMqttClient.h>
#include <ArduinoJson.h>
#include "ha.h"

#define MQTT_HOST IPAddress(192, 168, 5, 138)
#define MQTT_PORT 1883

namespace Mqtt {

    void publishConfig();
    void publishCommand();
    void publishBmp280();
    Task tPublishConfig(TASK_IMMEDIATE, TASK_ONCE, publishConfig, &ts);
    Task tPublishBmp(5 * TASK_MINUTE, TASK_FOREVER, publishBmp280, &ts);
    Task tPublishCommand(TASK_SECOND, TASK_FOREVER, publishCommand, &ts);

    AsyncMqttClient client;

    struct {
        const char* topic = "esp_clock/sensor/ir/value";
        std::queue<uint8_t> queue;
        uint8_t getCurrent() {
            return queue.empty() ? 0 : queue.front();
        }
        void push(uint8_t el) {
            if (client.connected()) queue.push(el);
        }
        void pop() {
            queue.pop();
        }
    } commands;

    const char* bmpTopic = "homeassistant/sensor/esp_clock/state";

    void publishTempConfig() {
        char message[JSON_SIZE];
        Ha::buildSensorConfig(message, Ha::TemperatureConfig{"Livingroom Temperature", "livingroom_temperature", bmpTopic});
        client.publish("homeassistant/sensor/esp_clock/temperature/config", 0, true, message);
    }

    void publishPressureConfig() {
        char message[JSON_SIZE];
        Ha::buildSensorConfig(message, Ha::PressureConfig{"Livingroom Pressure", "livingroom_pressure", bmpTopic});
        client.publish("homeassistant/sensor/esp_clock/pressure/config", 0, true, message);
    }

    void publishAltitudeConfig() {
        char message[JSON_SIZE];
        Ha::buildSensorConfig(message, Ha::AltitudeConfig{"Livingroom Altitude", "livingroom_altitude", bmpTopic});
        client.publish("homeassistant/sensor/esp_clock/altitude/config", 0, true, message);
    }

    void publishConfig() {
        publishTempConfig();
        publishPressureConfig();
        publishAltitudeConfig();
        ts.deleteTask(tPublishConfig);
    }

    void publishBmp280() {
        Bmp::data.readAll();
        StaticJsonDocument<255> jsonDoc;
        jsonDoc["temperature"] = Bmp::data.temp;
        jsonDoc["pressure"] = Bmp::data.pressure;
        jsonDoc["altitude"] = Bmp::data.altitude;
        char message[255];
        serializeJson(jsonDoc, message);
        client.publish(bmpTopic, 0, true, message);
    }

    void publishCommand() {
        if (uint8_t cmd = commands.getCurrent()) {
            char message[32];
            sprintf(message, "%X", cmd);
            if (client.publish(commands.topic, 0, true, message) != 0) {
                Serial.print(cmd, HEX);
                Serial.println();
                commands.queue.pop();
            }
        }
    }

    void setup() {
        client.onConnect([](bool sessionPresent) {
            tPublishConfig.enable();
            tPublishBmp.enableIfNot();
            tPublishCommand.enableDelayed();
            Serial.println("Connected to MQTT");
            });
        client.onDisconnect([](AsyncMqttClientDisconnectReason reason) {
            tPublishCommand.disable();
            tPublishBmp.disable();
            Serial.println("Disconnected from MQTT");
            });
        client.setServer(MQTT_HOST, MQTT_PORT);
        Serial.println("Connecting to MQTT...");
    }
}