#include <queue>
#include <AsyncMqttClient.h>
#include <ArduinoJson.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 tPublish(TASK_SECOND, TASK_FOREVER, publishCommand, &ts);
    Task tPublishBmp(5 * TASK_MINUTE, TASK_FOREVER, publishBmp280, &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 publishConfig() {
        StaticJsonDocument<255> jsonDoc;
        JsonObject device = jsonDoc.createNestedObject("device");
        device["name"] = "ESP Clock";

        JsonObject sensor = jsonDoc.createNestedObject("sensor");
        sensor["device_class"] = "temperature";
        sensor["name"] = "Livingroom Temperature";
        sensor["unique_id"] = "livingroom_temperature";
        sensor["unit_of_measurement"] = "°C";
        sensor["state_topic"] = bmpTopic;
        sensor["value_template"] = "{{ value_json.temperature}}";
        sensor["device"] = device;
        char message[255];
        serializeJson(jsonDoc, message);
        client.publish("homeassistant/sensor/esp_clock_temperature/config", 0, false, message);

        jsonDoc.clear();
        sensor.clear();
        sensor = jsonDoc.createNestedObject("sensor");
        sensor["device_class"] = "pressure";
        sensor["name"] = "Livingroom Pressure";
        sensor["unique_id"] = "livingroom_pressure";
        sensor["unit_of_measurement"] = "hPa";
        sensor["state_topic"] = bmpTopic;
        sensor["value_template"] = "{{ value_json.pressure}}";        
        sensor["device"] = device;
        char message1[255];
        serializeJson(jsonDoc, message);
        client.publish("homeassistant/sensor/esp_clock_pressure/config", 0, false, message1);
    }

    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 publishBmp280() {
        Bmp::data.readAll();
        StaticJsonDocument<255> jsonDoc;
        JsonObject sensor = jsonDoc.createNestedObject("sensor");
        sensor["temperature"] = Bmp::data.temp;
        sensor["pressure"] = Bmp::data.pressure;
        char message[255];
        serializeJson(jsonDoc, message);
        client.publish(bmpTopic, 0, false, message);
    }

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