#pragma once

/**
 *   ESP32 SvelteKit
 *
 *   A simple, secure and extensible framework for IoT projects for ESP32 platforms
 *   with responsive Sveltekit front-end built with TailwindCSS and DaisyUI.
 *   https://github.com/theelims/ESP32-sveltekit
 *
 *   Copyright (C) 2023 theelims
 *
 *   All Rights Reserved. This software may be modified and distributed under
 *   the terms of the LGPL v3 license. See the LICENSE file for details.
 **/

#include <ArduinoJson.h>
#include <ESPFS.h>
#include <EventSocket.h>
#include <TaskManager.h>
#include <WiFi.h>

#define MAX_ESP_ANALYTICS_SIZE 2024
#define EVENT_ANALYTICS "analytics"
#define ANALYTICS_INTERVAL 2000

class AnalyticsService
{
  public:
    AnalyticsService(EventSocket *socket, TaskManager *taskManager) : _socket(socket), _taskManager(taskManager){};

    void begin()
    {
        _socket->registerEvent(EVENT_ANALYTICS);
    };

    void loop()
    {
        unsigned long currentMillis = millis();

        if (!_lastUpdate || (currentMillis - _lastUpdate) >= ANALYTICS_INTERVAL)
        {
            _lastUpdate = currentMillis;
            updateAnalytics();
        }
    };
    StaticJsonDocument<MAX_ESP_ANALYTICS_SIZE> doc;
    char message[MAX_ESP_ANALYTICS_SIZE];

  private:
    EventSocket *_socket;
    TaskManager *_taskManager;
    unsigned long _lastUpdate;

    void updateAnalytics() {
        doc.clear();
        doc["uptime"] = millis() / 1000;
        doc["free_heap"] = ESP.getFreeHeap();
        doc["total_heap"] = ESP.getHeapSize();
        doc["min_free_heap"] = ESP.getMinFreeHeap();
        doc["max_alloc_heap"] = ESP.getMaxAllocHeap();
        doc["fs_used"] = ESPFS.usedBytes();
        doc["fs_total"] = ESPFS.totalBytes();
        doc["core_temp"] = temperatureRead();
        doc["cpu0_usage"] = _taskManager->getCpuUsage(0);
        doc["cpu1_usage"] = _taskManager->getCpuUsage(1);
        doc["cpu_usage"] = _taskManager->getCpuUsage();
        // Add _taskManager->getTaskNames() as a JSON array
        JsonArray tasks = doc.createNestedArray("tasks");
        for (auto const &task : _taskManager->getTasks())
        {
            JsonObject nested = tasks.createNestedObject();
            nested["name"] = task.name;
            nested["stackSize"] = task.stackSize;
            nested["priority"] = task.priority;
            nested["coreId"] = task.coreId;
        }

        serializeJson(doc, message);
        _socket->emit(EVENT_ANALYTICS, message);
    }
};