// main.cpp

#include "network/tcp_server.h"
#include "mqtt/mqtt_client.h"
#include "mqtt/mqtt_router.h"
#include "systemMonitor/system_monitor.h"
#include "spdlog/spdlog.h"
#include "deviceManager/device_manager.h" 
#include "dataCache/data_cache.h"
#include "dataCache/cache_uploader.h"

#include <boost/asio.hpp>
#include <boost/asio/steady_timer.hpp>
#include <csignal>
#include <iostream>
#include <functional>

// 用于 ASIO 服务的全局 io_context
boost::asio::io_context g_io_context;

/**
 * @brief 处理终止信号 (SIGINT, SIGTERM).
 */
void signalHandler(int signum) {
    spdlog::warn("Interrupt signal ({}) received. Shutting down.", signum);
    g_io_context.stop();
}

/**
 * @brief 周期性轮询系统状态并发布到 MQTT
 */
void poll_system_metrics(
    boost::asio::steady_timer& timer,
    SystemMonitor::SystemMonitor& monitor,
    MqttClient& mqtt_client
) {
    // ... 此函数内部逻辑保持不变 ...
    auto cpu_util = monitor.getCpuUtilization();
    auto mem_info = monitor.getMemoryInfo();
    double mem_total_gb = mem_info.total_kb / 1024.0 / 1024.0;

    std::string topic = "proxy/system_status";
    std::string payload = "{\"cpu_usage\":" + std::to_string(cpu_util.totalUsagePercentage) +
                          ",\"mem_total_gb\":" + std::to_string(mem_total_gb) + "}";
    mqtt_client.publish(topic, payload);
    spdlog::debug("System metrics published.");

    timer.expires_at(timer.expiry() + std::chrono::seconds(15));
    timer.async_wait(std::bind(poll_system_metrics, std::ref(timer), std::ref(monitor), std::ref(mqtt_client)));
}

int main(int argc, char* argv[]) {

    try {
        spdlog::set_level(spdlog::level::debug);
        spdlog::info("Edge Proxy starting up...");
    } catch (const spdlog::spdlog_ex& ex) {
        std::cerr << "Log initialization failed: " << ex.what() << std::endl;
        return 1;
    }

    signal(SIGINT, signalHandler);
    signal(SIGTERM, signalHandler);

    try {
        // --- 1. 初始化核心服务 ---
        MqttClient mqtt_client("tcp://mqtt-broker:1883", "edge-proxy-main-client");
        MqttRouter mqtt_router(mqtt_client);
        std::vector<uint16_t> listen_ports = { 8888 };
        TCPServer tcp_server(g_io_context, listen_ports, mqtt_client);
        SystemMonitor::SystemMonitor monitor;

        // <<< MODIFIED: 初始化数据缓存和上传器 >>>
        DataCache data_cache;
        if (!data_cache.open("edge_data_cache.db")) { // 数据库文件将创建在程序运行目录
            spdlog::critical("Failed to initialize data cache. Exiting.");
            return 1;
        }
        CacheUploader cache_uploader(g_io_context, mqtt_client, data_cache);

        // <<< MODIFIED: 设置MQTT连接成功的回调，用于触发断点续传 >>>
        mqtt_client.set_connected_handler([&](const std::string& cause){
            spdlog::info("MQTT client connected: {}", cause);
            // 当连接成功时，启动缓存上传
            cache_uploader.start_upload();
        });
        
        // 现在再连接MQTT
        mqtt_client.connect();
        mqtt_router.start();

        // --- 2. 启动系统状态监控定时器 ---
        monitor.getCpuUtilization(); // 首次调用以初始化
        boost::asio::steady_timer system_monitor_timer(g_io_context, std::chrono::seconds(15));
        system_monitor_timer.async_wait(std::bind(poll_system_metrics, std::ref(system_monitor_timer), std::ref(monitor), std::ref(mqtt_client)));

        // --- 3. <<< MODIFIED: 创建包含缓存逻辑的统一数据上报回调函数 >>>
        auto report_to_mqtt = [&](const UnifiedData& data) {
            if (mqtt_client.is_connected()) {
                // 网络正常，直接上报
                std::string topic = "devices/" + data.device_id + "/data";
                g_io_context.post([&, topic, payload = data.data_json]() {
                    mqtt_client.publish(topic, payload, 1, false);
                });
            } else {
                // 网络断开，写入缓存
                spdlog::warn("MQTT disconnected. Caching data for device '{}'.", data.device_id);
                data_cache.add(data);
            }
        };

        // --- 4. 实例化设备管理器并从文件加载所有设备 (使用新的回调) ---
        DeviceManager device_manager(g_io_context, report_to_mqtt);
        // 默认从程序运行目录下的 "devices.json" 文件加载
        device_manager.load_and_start("../config/devices.json");

        // --- 5. 启动主事件循环 (程序将阻塞在这里直到被信号中断) ---
        spdlog::info("All services are running. Press Ctrl+C to exit.");
        g_io_context.run();

        // --- 清理工作 ---
        spdlog::info("Shutting down MQTT client...");
        mqtt_client.disconnect();

    } catch (const std::exception& e) {
        spdlog::critical("An unhandled exception occurred: {}", e.what());
        return 1;
    }

    spdlog::info("Server has been shut down gracefully. Exiting.");
    return 0;
}