//
// Created by misaki on 2025/9/2.
//

#pragma once

#include <string>
#include <functional>

#include <mutex>
#include <queue>
#include <condition_variable>
#include "esp_websocket_client.h"
#include "cJSON.h"
#include "ThreadManager.h"
#include "WifiConnectors.h"

// WebSocket事件类型
enum class WebSocketEvent {
    CONNECTED,
    DISCONNECTED,
    DATA_RECEIVED,
    ERROR
};

// WebSocket配置
struct WebSocketConfig {
    std::string uri;                    // WebSocket服务器URI
    int reconnect_interval = 5000;      // 重连间隔(ms)
    int heartbeat_interval = 30000;     // 心跳间隔(ms)
    int max_reconnect_attempts = 10;    // 最大重连次数
    bool auto_reconnect = true;         // 是否自动重连
};

// JSON数据回调
using JsonDataCallback = std::function<void(cJSON* json)>;
// 事件回调
using EventCallback = std::function<void(WebSocketEvent event, const std::string& message)>;

/* 回调函数示例：
    // JSON数据回调示例
    void onJsonData(cJSON* json) {
        // 处理接收到的JSON数据
        cJSON* type = cJSON_GetObjectItem(json, "type");
        if (type && cJSON_IsString(type)) {
            ESP_LOGI("App", "Received message type: %s", type->valuestring);

            if (strcmp(type->valuestring, "sensor_data") == 0) {
                cJSON* value = cJSON_GetObjectItem(json, "value");
                if (value && cJSON_IsNumber(value)) {
                    ESP_LOGI("App", "Sensor value: %.2f", value->valuedouble);
                }
            }
        }

        cJSON_Delete(json);
    }
    // 事件回调示例
    void onWebSocketEvent(WebSocketEvent event, const std::string& message) {
        switch (event) {
            case WebSocketEvent::CONNECTED:
                ESP_LOGI("App", "WebSocket connected: %s", message.c_str());
                break;
            case WebSocketEvent::DISCONNECTED:
                ESP_LOGI("App", "WebSocket disconnected: %s", message.c_str());
                break;
            case WebSocketEvent::DATA_RECEIVED:
                ESP_LOGI("App", "Received raw data: %s", message.c_str());
                break;
            case WebSocketEvent::ERROR:
                ESP_LOGE("App", "WebSocket error: %s", message.c_str());
                break;
        }
    }
 */



class WebSocketManager {
public:
    // 获取单例实例
    static WebSocketManager* getInstance();

    // 删除拷贝构造函数和赋值运算符
    WebSocketManager(const WebSocketManager&) = delete;
    WebSocketManager& operator=(const WebSocketManager&) = delete;

    // 初始化WebSocket管理器
    bool initialize(const WebSocketConfig& config);

    // 连接到WebSocket服务器
    bool connect();

    // 断开连接
    void disconnect();

    // 发送JSON数据
    bool sendJson(cJSON* json);

    // 发送原始字符串数据
    bool sendRaw(const std::string& data);

    // 设置JSON数据回调
    void setJsonCallback(JsonDataCallback callback);

    // 设置事件回调
    void setEventCallback(EventCallback callback);

    // 获取连接状态
    bool isConnected() const;

    // 获取配置信息
    WebSocketConfig getConfig() const;

    // 更新配置
    void updateConfig(const WebSocketConfig& config);

    // 获取统计信息
    struct Stats {
        uint32_t messages_sent;             /// 发送消息数
        uint32_t messages_received;         /// 接收到的消息数
        uint32_t connection_attempts;       /// 连接尝试次数
        uint32_t successful_connections;    /// 成功连接次数
    };
    Stats getStats() const;

private:
    WebSocketManager();  // 私有构造函数
    ~WebSocketManager(); // 私有析构函数

    /**
     * WebSocket事件处理函数
     * @param handler_args  处理参数
     * @param base  事件基础
     * @param event_id  事件ID
     * @param event_data  事件数据
     */
    static void websocketEventHandler(void* handler_args, esp_event_base_t base,
                                      int32_t event_id, void* event_data);

    /**
     * 处理接收到的数据
     * @param data   数据
     * @param len    数据长度
     */
    void handleReceivedData(const char* data, int len);

    // 重连线程函数
    void reconnectThread();

    // 心跳线程函数
    void heartbeatThread();

    // 发送线程函数
    void sendThread();

    // 创建WebSocket客户端
    bool createWebSocketClient();

    // 销毁WebSocket客户端
    void destroyWebSocketClient();

    // 处理WiFi事件
    static void wifiEventHandler(void* arg, esp_event_base_t event_base,
                                int32_t event_id, void* event_data);

private:
    static WebSocketManager* instance;
    static std::mutex instance_mutex;

    esp_websocket_client_handle_t client;
    WebSocketConfig config;

    // 线程相关
    std::thread reconnect_thread;           /// 重连线程
    std::thread heartbeat_thread;           /// 心跳线程
    std::thread send_thread;                /// 发送线程
    bool threads_running;                   /// 线程运行标志

    // 回调函数
    JsonDataCallback json_callback;         /// JSON数据回调
    EventCallback event_callback;           /// 事件回调

    // 状态变量
    bool connected;
    bool connecting;
    uint32_t reconnect_attempts;

    // 统计信息
    Stats stats;

    // 发送队列
    std::queue<std::string> send_queue;
    std::mutex queue_mutex;
    std::condition_variable queue_cv;

    // WiFi实例
    WifiConnectors* wifi;
};