/*
 * etcpmon_client.c - ETCP Monitor Client Network Implementation
 */

#include "etcpmon_client.h"
#include "etcpmon_graph.h"
#include <stdio.h>
#include <string.h>
#include <ws2tcpip.h>
#include <windows.h>

//#pragma comment(lib, "ws2_32.lib")

/* Log file name */
#define LOG_FILENAME "etcpmon.log"

/* Get current timestamp in milliseconds (Unix epoch) */
uint64_t get_timestamp_ms(void) {
    FILETIME ft;
    GetSystemTimeAsFileTime(&ft);
    
    /* Convert FILETIME (100-nanosecond intervals since 1601) to milliseconds */
    uint64_t time = ((uint64_t)ft.dwHighDateTime << 32) | ft.dwLowDateTime;
    /* Convert from 1601 to 1970 epoch (11644473600000 milliseconds) */
    return (time / 10000) - 11644473600000ULL;
}

/* Log hex data to file */
static void log_hex_data(FILE* log, const char* prefix, const uint8_t* data, size_t len) {
    if (!log || !data || len == 0) return;
    
    fprintf(log, "%llu: [%s] ", (unsigned long long)get_timestamp_ms(), prefix);
    for (size_t i = 0; i < len && i < 256; i++) {  /* Limit to 256 bytes per line */
        fprintf(log, "%02X ", data[i]);
    }
    if (len > 256) {
        fprintf(log, "... (%llu bytes total)", (unsigned long long)len);
    }
    fprintf(log, "\n");
    fflush(log);
}

void etcpmon_client_init(struct etcpmon_client* client) {
    if (!client) return;
    memset(client, 0, sizeof(*client));
    client->sock = INVALID_SOCKET;
    client->connected = 0;
    client->fully_connected = 0;
    client->log_file = NULL;
    client->next_seq_id = 1;  /* Start from 1, 0 = broadcast */
    memset(&client->history, 0, sizeof(client->history));
    for (int i = 0; i < GRAPH_METRIC_COUNT; i++) {
        client->history.min_val[i] = 1e9f;
        client->history.max_val[i] = -1e9f;
    }
    client->history.last_minmax_update = 0;
    
    /* Новый таймер авто-запроса метрик */
    client->last_metrics_request_ms = 0;
}

void etcpmon_client_cleanup(struct etcpmon_client* client) {
    if (!client) return;
    
    etcpmon_client_disconnect(client);
    
    /* Free pending requests */
    struct etcpmon_pending_request* req = client->pending_head;
    while (req) {
        struct etcpmon_pending_request* next = req->next;
        free(req);
        req = next;
    }
    client->pending_head = NULL;
    client->pending_tail = NULL;
    
    /* Close log file */
    if (client->log_file) {
        fprintf(client->log_file, "%llu: [LOG] Log closed\n", 
               (unsigned long long)get_timestamp_ms());
        fclose(client->log_file);
        client->log_file = NULL;
    }
    
    /* Free cached data */
    if (client->conn_list) {
        free(client->conn_list);
        client->conn_list = NULL;
    }
    if (client->socket_list) {
        free(client->socket_list);
        client->socket_list = NULL;
    }
    if (client->last_metrics) {
        free(client->last_metrics);
        client->last_metrics = NULL;
    }
    if (client->last_links) {
        free(client->last_links);
        client->last_links = NULL;
    }
}

int etcpmon_client_connect(struct etcpmon_client* client, const char* addr, uint16_t port) {
    if (!client || !addr) return -1;
    
    if (client->connected) {
        etcpmon_client_disconnect(client);
    }

    /* Open log file (truncate on start) */
    if (!client->log_file) {
        client->log_file = fopen(LOG_FILENAME, "w");
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [LOG] ETCP Monitor communication log started\n", 
                   (unsigned long long)get_timestamp_ms());
            fprintf(client->log_file, "%llu: [LOG] Connecting to %s:%d\n", 
                   (unsigned long long)get_timestamp_ms(), addr, port);
            fflush(client->log_file);
        }
    }
    
    /* Create socket */
    client->sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (client->sock == INVALID_SOCKET) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to create socket\n", 
                   (unsigned long long)get_timestamp_ms());
            fflush(client->log_file);
        }
        if (client->on_error) {
            client->on_error("Failed to create socket", client->user_data);
        }
        return -1;
    }
    
    /* Set non-blocking mode */
    u_long nonblock = 1;
    ioctlsocket(client->sock, FIONBIO, &nonblock);
    
    /* Setup address */
    struct sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);
    
    if (inet_pton(AF_INET, addr, &server_addr.sin_addr) != 1) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Invalid address: %s\n", 
                   (unsigned long long)get_timestamp_ms(), addr);
            fflush(client->log_file);
        }
        closesocket(client->sock);
        client->sock = INVALID_SOCKET;
        if (client->on_error) {
            client->on_error("Invalid address", client->user_data);
        }
        return -1;
    }
    
    /* Connect (non-blocking) */
    int result = connect(client->sock, (struct sockaddr*)&server_addr, sizeof(server_addr));
    if (result == SOCKET_ERROR) {
        int err = WSAGetLastError();
        if (err != WSAEWOULDBLOCK && err != WSAEINPROGRESS) {
            if (client->log_file) {
                fprintf(client->log_file, "%llu: [ERROR] Connection failed: %d\n", 
                       (unsigned long long)get_timestamp_ms(), err);
                fflush(client->log_file);
            }
            closesocket(client->sock);
            client->sock = INVALID_SOCKET;
            if (client->on_error) {
                char msg[256];
                snprintf(msg, sizeof(msg), "Connection failed: %d", err);
                client->on_error(msg, client->user_data);
            }
            return -1;
        }
        /* Connection in progress, will complete asynchronously */
    }
    
    strncpy(client->server_addr, addr, sizeof(client->server_addr) - 1);
    client->server_port = port;
    client->connected = 1;
    client->fully_connected = 0;
    
    if (client->log_file) {
        fprintf(client->log_file, "%llu: [LOG] Socket created, connection in progress\n", 
               (unsigned long long)get_timestamp_ms());
        fflush(client->log_file);
    }
    
    return 0;
}

void etcpmon_client_disconnect(struct etcpmon_client* client) {
    if (!client) return;
    
    if (client->sock != INVALID_SOCKET) {
        /* Send disconnect command */
        if (client->connected) {
            struct etcpmon_msg_header hdr;
            etcpmon_build_header(&hdr, 0, ETCPMON_CMD_DISCONNECT, 0);
            
            if (client->log_file) {
                log_hex_data(client->log_file, "TX", (const uint8_t*)&hdr, sizeof(hdr));
                fprintf(client->log_file, "%llu: [LOG] Sending CMD_DISCONNECT\n", 
                       (unsigned long long)get_timestamp_ms());
            }
            
            send(client->sock, (const char*)&hdr, sizeof(hdr), 0);
        }
        
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [LOG] Connection closed\n", 
                   (unsigned long long)get_timestamp_ms());
            fflush(client->log_file);
        }
        
        closesocket(client->sock);
        client->sock = INVALID_SOCKET;
    }
    
    client->connected = 0;
    client->fully_connected = 0;
    client->recv_len = 0;
    client->selected_peer_id = 0;
    
    /* Очистить очередь pending requests и сбросить seq_id */
    struct etcpmon_pending_request* req = client->pending_head;
    while (req) {
        struct etcpmon_pending_request* next = req->next;
        free(req);
        req = next;
    }
    client->pending_head = NULL;
    client->pending_tail = NULL;
    client->next_seq_id = 1;
    client->last_response_seq_id = 0;
}

int etcpmon_client_is_connected(const struct etcpmon_client* client) {
    return client ? client->connected : 0;
}

int etcpmon_client_request_list(struct etcpmon_client* client) {
    if (!client || client->sock == INVALID_SOCKET) return -1;
    
    struct etcpmon_msg_header hdr;
    etcpmon_build_header(&hdr, 0, ETCPMON_CMD_LIST_CONN, 0);  /* seq=0 = legacy режим */
    
    if (client->log_file) {
        log_hex_data(client->log_file, "TX", (const uint8_t*)&hdr, sizeof(hdr));
        fprintf(client->log_file, "%llu: [LOG] CMD_LIST_CONN (seq=0)\n", 
               (unsigned long long)get_timestamp_ms());
        fflush(client->log_file);
    }
    
    int sent = send(client->sock, (const char*)&hdr, sizeof(hdr), 0);
    if (sent != sizeof(hdr)) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to send CMD_LIST_CONN: %d\n", 
                   (unsigned long long)get_timestamp_ms(), WSAGetLastError());
            fflush(client->log_file);
        }
        return -1;
    }
    
    return 0;
}

int etcpmon_client_select_connection(struct etcpmon_client* client, uint64_t peer_node_id) {
    if (!client || client->sock == INVALID_SOCKET) return -1;
    
    /* SELECT is special - it's not a request/response, it just sets state
     * So we send with seq_id=0 (broadcast) */
    uint8_t buffer[sizeof(struct etcpmon_msg_header) + sizeof(struct etcpmon_cmd_select)];
    struct etcpmon_msg_header* hdr = (struct etcpmon_msg_header*)buffer;
    struct etcpmon_cmd_select* cmd = (struct etcpmon_cmd_select*)(buffer + sizeof(*hdr));
    
    etcpmon_build_header(hdr, sizeof(*cmd), ETCPMON_CMD_SELECT_CONN, 0);  /* seq_id=0 for SELECT */
    cmd->peer_node_id = peer_node_id;
    
    if (client->log_file) {
        log_hex_data(client->log_file, "TX", buffer, sizeof(buffer));
        fprintf(client->log_file, "%llu: [LOG] CMD_SELECT_CONN peer_id=%016llX\n", 
               (unsigned long long)get_timestamp_ms(), (unsigned long long)peer_node_id);
        fflush(client->log_file);
    }
    
    int sent = send(client->sock, (const char*)buffer, sizeof(buffer), 0);
    if (sent != sizeof(buffer)) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to send CMD_SELECT_CONN: %d\n", 
                   (unsigned long long)get_timestamp_ms(), WSAGetLastError());
            fflush(client->log_file);
        }
        return -1;
    }
    
    client->selected_peer_id = peer_node_id;
    return 0;
}

int etcpmon_client_request_socket_list(struct etcpmon_client* client) {
    if (!client || client->sock == INVALID_SOCKET) return -1;

    struct etcpmon_msg_header hdr;
    etcpmon_build_header(&hdr, 0, ETCPMON_CMD_LIST_SOCKETS, 0);

    if (client->log_file) {
        log_hex_data(client->log_file, "TX", (const uint8_t*)&hdr, sizeof(hdr));
        fprintf(client->log_file, "%llu: [LOG] CMD_LIST_SOCKETS (seq=0)\n",
               (unsigned long long)get_timestamp_ms());
        fflush(client->log_file);
    }

    int sent = send(client->sock, (const char*)&hdr, sizeof(hdr), 0);
    if (sent != sizeof(hdr)) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to send CMD_LIST_SOCKETS: %d\n",
                   (unsigned long long)get_timestamp_ms(), WSAGetLastError());
            fflush(client->log_file);
        }
        return -1;
    }

    return 0;
}

int etcpmon_client_request_metrics(struct etcpmon_client* client) {
    if (!client || client->sock == INVALID_SOCKET) return -1;
    
    struct etcpmon_msg_header hdr;
    etcpmon_build_header(&hdr, 0, ETCPMON_CMD_GET_METRICS, 0);
    
    if (client->log_file) {
        log_hex_data(client->log_file, "TX", (const uint8_t*)&hdr, sizeof(hdr));
        fprintf(client->log_file, "%llu: [LOG] CMD_GET_METRICS\n", 
               (unsigned long long)get_timestamp_ms());
        fflush(client->log_file);
    }
    
    int sent = send(client->sock, (const char*)&hdr, sizeof(hdr), 0);
    if (sent != sizeof(hdr)) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to send CMD_GET_METRICS: %d\n", 
                   (unsigned long long)get_timestamp_ms(), WSAGetLastError());
            fflush(client->log_file);
        }
        return -1;
    }
    
    return 0;
}

static void action_auto_callback(void* arg, uint8_t msg_type, const uint8_t* payload, uint16_t payload_size) {
    (void)arg; (void)msg_type; (void)payload; (void)payload_size;
    /* legacy callback is called via on_action_result in handle_response */
}

int etcpmon_client_send_action(struct etcpmon_client* client, const char* action) {
    if (!client || client->sock == INVALID_SOCKET || !action) return -1;

    struct etcpmon_cmd_action cmd;
    memset(&cmd, 0, sizeof(cmd));
    strncpy(cmd.action, action, sizeof(cmd.action) - 1);
    cmd.action[sizeof(cmd.action) - 1] = '\0';

    return etcpmon_client_send_request(client, ETCPMON_CMD_ACTION, &cmd, sizeof(cmd),
                                        action_auto_callback, NULL);
}

/* Send request with callback - queued version */
int etcpmon_client_send_request(struct etcpmon_client* client,
                                 uint8_t type,
                                 const void* payload, uint16_t payload_size,
                                 void (*callback)(void* arg, uint8_t msg_type, const uint8_t* payload, uint16_t payload_size),
                                 void* arg) {
    if (!client || client->sock == INVALID_SOCKET || !callback) return -1;
    
    /* Allocate pending request */
    struct etcpmon_pending_request* req = (struct etcpmon_pending_request*)malloc(sizeof(*req));
    if (!req) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to allocate pending request\n",
                   (unsigned long long)get_timestamp_ms());
            fflush(client->log_file);
        }
        return -1;
    }
    
    /* Assign sequence ID */
    req->seq_id = client->next_seq_id++;
    if (client->next_seq_id == 0) client->next_seq_id = 1;  /* Skip 0 (broadcast) */
    req->callback = callback;
    req->arg = arg;
    req->next = NULL;
    
    /* Add to queue */
    if (client->pending_tail) {
        client->pending_tail->next = req;
    } else {
        client->pending_head = req;
    }
    client->pending_tail = req;
    
    /* Build message with seq_id */
    uint8_t buffer[sizeof(struct etcpmon_msg_header) + ETCPMON_MAX_MSG_SIZE];
    if (payload_size > ETCPMON_MAX_MSG_SIZE) {
        free(req);
        return -1;
    }
    
    struct etcpmon_msg_header* hdr = (struct etcpmon_msg_header*)buffer;
    etcpmon_build_header(hdr, payload_size, type, req->seq_id);
    
    if (payload && payload_size > 0) {
        memcpy(buffer + sizeof(*hdr), payload, payload_size);
    }
    
    uint16_t total_size = sizeof(*hdr) + payload_size;
    
    if (client->log_file) {
        log_hex_data(client->log_file, "TX", buffer, total_size);
        fprintf(client->log_file, "%llu: [LOG] CMD 0x%02X seq=%d queued\n",
               (unsigned long long)get_timestamp_ms(), type, req->seq_id);
        fflush(client->log_file);
    }
    
    int sent = send(client->sock, (const char*)buffer, total_size, 0);
    if (sent != total_size) {
        /* Remove from queue on error */
        if (client->pending_head == req) {
            client->pending_head = req->next;
            if (!client->pending_head) client->pending_tail = NULL;
        }
        free(req);
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] Failed to send request: %d\n",
                   (unsigned long long)get_timestamp_ms(), WSAGetLastError());
            fflush(client->log_file);
        }
        return -1;
    }
    
    return 0;
}

static void handle_response(struct etcpmon_client* client, uint8_t msg_type, 
                           const uint8_t* payload, uint16_t payload_size) {
    switch (msg_type) {
        case ETCPMON_RSP_CONN_LIST: {
            if (payload_size < sizeof(struct etcpmon_rsp_conn_list)) break;
            
            struct etcpmon_rsp_conn_list* rsp = (struct etcpmon_rsp_conn_list*)payload;
            uint8_t count = rsp->count;
            
            if (payload_size < sizeof(*rsp) + count * sizeof(struct etcpmon_conn_info)) break;
            
            /* Update cached list */
            if (client->conn_list) {
                free(client->conn_list);
            }
            client->conn_list = (struct etcpmon_conn_info*)malloc(count * sizeof(struct etcpmon_conn_info));
            if (client->conn_list) {
                memcpy(client->conn_list, payload + sizeof(*rsp), count * sizeof(struct etcpmon_conn_info));
                client->conn_count = count;
                
                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [LOG] RSP_CONN_LIST count=%d\n", 
                           (unsigned long long)get_timestamp_ms(), count);
                    for (uint8_t i = 0; i < count; i++) {
                        fprintf(client->log_file, "%llu: [LOG]   [%d] peer_id=%016llX name='%s'\n", 
                               (unsigned long long)get_timestamp_ms(), i,
                               (unsigned long long)client->conn_list[i].peer_node_id,
                               client->conn_list[i].name);
                    }
                    fflush(client->log_file);
                }
                
                if (client->on_conn_list) {
                    client->on_conn_list(client->conn_list, count, client->user_data);
                }
            }
            break;
        }
        
        case ETCPMON_RSP_SOCKET_LIST: {
            if (payload_size < sizeof(struct etcpmon_rsp_socket_list)) break;

            struct etcpmon_rsp_socket_list* rsp = (struct etcpmon_rsp_socket_list*)payload;
            uint8_t count = rsp->count;

            if (count > 16) count = 16;  /* Limit to prevent buffer overflow */
            if (payload_size < sizeof(*rsp) + count * sizeof(struct etcpmon_socket_info)) break;

            if (client->socket_list) {
                free(client->socket_list);
            }
            client->socket_list = (struct etcpmon_socket_info*)malloc(count * sizeof(struct etcpmon_socket_info));
            if (client->socket_list) {
                memcpy(client->socket_list, payload + sizeof(*rsp), count * sizeof(struct etcpmon_socket_info));
                client->socket_count = count;

                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [LOG] RSP_SOCKET_LIST count=%d\n",
                           (unsigned long long)get_timestamp_ms(), count);
                    fflush(client->log_file);
                }

                if (client->on_socket_list) {
                    client->on_socket_list(client->socket_list, count, client->user_data);
                }
            }
            break;
        }

        case ETCPMON_RSP_METRICS: {
            if (payload_size < sizeof(struct etcpmon_rsp_metrics)) break;

            struct etcpmon_rsp_metrics* rsp = (struct etcpmon_rsp_metrics*)payload;
            uint8_t links_count = rsp->etcp.links_count;
            
            if (payload_size < sizeof(*rsp) + links_count * sizeof(struct etcpmon_link_metrics)) break;
            
            /* Update cached metrics */
            if (client->last_metrics) {
                free(client->last_metrics);
            }
            if (client->last_links) {
                free(client->last_links);
            }
            
            client->last_metrics = (struct etcpmon_rsp_metrics*)malloc(sizeof(*rsp));
            client->last_links = (struct etcpmon_link_metrics*)malloc(links_count * sizeof(struct etcpmon_link_metrics));
            
            if (client->last_metrics && client->last_links) {
                memcpy(client->last_metrics, rsp, sizeof(*rsp));
                memcpy(client->last_links, payload + sizeof(*rsp), 
                       links_count * sizeof(struct etcpmon_link_metrics));
                client->last_links_count = links_count;
                
                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [LOG] RSP_METRICS rtt=%u bytes=%llu links=%d\n", 
                           (unsigned long long)get_timestamp_ms(),
                           rsp->etcp.rtt_last,
                           (unsigned long long)rsp->etcp.bytes_sent_total,
                           links_count);
                    fflush(client->log_file);
                }
                
                if (client->on_metrics) {
                    client->on_metrics(client->last_metrics, client->last_links, 
                                        links_count, client->user_data);
                }
                
                etcpmon_client_add_to_history(client, client->last_metrics);
            }
            break;
        }
        
        case ETCPMON_RSP_ERROR: {
            if (payload_size >= sizeof(struct etcpmon_rsp_error)) {
                struct etcpmon_rsp_error* rsp = (struct etcpmon_rsp_error*)payload;
                const char* msg = (const char*)(payload + sizeof(*rsp));

                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [LOG] RSP_ERROR code=%d msg='%s'\n",
                           (unsigned long long)get_timestamp_ms(), rsp->error_code, msg);
                    fflush(client->log_file);
                }

                if (client->on_error) {
                    client->on_error(msg, client->user_data);
                }
            }
            break;
        }

        case ETCPMON_RSP_ACTION_RESULT: {
            if (payload_size >= sizeof(struct etcpmon_rsp_action_result)) {
                const char* text = (const char*)(payload + sizeof(struct etcpmon_rsp_action_result));

                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [LOG] RSP_ACTION_RESULT len=%u\n",
                           (unsigned long long)get_timestamp_ms(), payload_size);
                    fflush(client->log_file);
                }

                if (client->on_action_result) {
                    client->on_action_result(text, client->user_data);
                }
            }
            break;
        }

        case ETCPMON_RSP_DEBUG_CONFIG: {
            if (payload_size >= 2) {
                client->debug_global_level = payload[0];
                client->debug_category_count = payload[1];
                uint16_t names_len = (uint16_t)strlen((const char*)(payload + 2)) + 1;
                if (2 + names_len <= payload_size) {
                    size_t copy_len = names_len < sizeof(client->debug_category_names) ? names_len : sizeof(client->debug_category_names) - 1;
                    memcpy(client->debug_category_names, payload + 2, copy_len);
                    client->debug_category_names[copy_len] = '\0';
                    uint16_t levels_off = 2 + names_len;
                    uint8_t lvl_count = client->debug_category_count;
                    if (lvl_count > ETCPMON_MAX_DEBUG_CATEGORIES) lvl_count = ETCPMON_MAX_DEBUG_CATEGORIES;
                    for (uint8_t i = 0; i < lvl_count && levels_off + i < payload_size; i++)
                        client->debug_category_levels[i] = payload[levels_off + i];
                    client->debug_config_dirty = 1;
                    if (client->log_file) {
                        fprintf(client->log_file, "%llu: [LOG] RSP_DEBUG_CONFIG global=%d categories=%d\n",
                               (unsigned long long)get_timestamp_ms(), client->debug_global_level, client->debug_category_count);
                        fflush(client->log_file);
                    }
                }
            }
            break;
        }
    }
}

/* Find pending request by seq_id */
static struct etcpmon_pending_request* find_pending_request(struct etcpmon_client* client, uint8_t seq_id) {
    struct etcpmon_pending_request* req = client->pending_head;
    while (req) {
        if (req->seq_id == seq_id) {
            return req;
        }
        req = req->next;
    }
    return NULL;
}

/* Remove pending request from queue */
static void remove_pending_request(struct etcpmon_client* client, struct etcpmon_pending_request* req) {
    struct etcpmon_pending_request** curr = &client->pending_head;
    while (*curr) {
        if (*curr == req) {
            *curr = req->next;
            if (client->pending_tail == req) {
                client->pending_tail = *curr;
            }
            free(req);
            return;
        }
        curr = &(*curr)->next;
    }
}

/* Handle response - find matching request and call callback */
static void handle_response_with_callback(struct etcpmon_client* client, uint8_t msg_type,
                                           const uint8_t* payload, uint16_t payload_size) {
    uint8_t seq_id = client->last_response_seq_id;
    
    /* Find pending request */
    struct etcpmon_pending_request* req = find_pending_request(client, seq_id);
    
    if (req) {
        /* Call the callback */
        if (req->callback) {
            req->callback(req->arg, msg_type, payload, payload_size);
        }
        /* Remove from queue */
        remove_pending_request(client, req);
        
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [LOG] Response handled, seq=%d removed from queue\n",
                   (unsigned long long)get_timestamp_ms(), seq_id);
            fflush(client->log_file);
        }
    } else {
        /* No pending request - maybe broadcast/unsolicited or stale */
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [LOG] Response seq=%d - no pending request (broadcast?)\n",
                   (unsigned long long)get_timestamp_ms(), seq_id);
            fflush(client->log_file);
        }
    }
    
    /* === НОВОЕ: ВСЕГДА вызываем legacy обработчики (on_conn_list / on_metrics / on_error) ===
     * Это позволяет использовать и старый код, и новый асинхронный send_request одновременно */
    if (seq_id == 0 || msg_type == ETCPMON_RSP_CONN_LIST ||
        msg_type == ETCPMON_RSP_SOCKET_LIST || msg_type == ETCPMON_RSP_DEBUG_CONFIG ||
        msg_type == ETCPMON_RSP_METRICS || msg_type == ETCPMON_RSP_ERROR ||
        msg_type == ETCPMON_RSP_ACTION_RESULT) {
        handle_response(client, msg_type, payload, payload_size);
    }
}

static void metrics_auto_callback(void* arg, uint8_t msg_type, const uint8_t* payload, uint16_t payload_size) {
    (void)arg;
    (void)msg_type;
    (void)payload;
    (void)payload_size;
    /* Ничего не делаем — legacy on_metrics вызывается автоматически ниже */
}

int etcpmon_client_process(struct etcpmon_client* client) {
    if (!client || client->sock == INVALID_SOCKET) return -1;

    /* Проверка завершения non-blocking connect */
    if (client->connected && !client->fully_connected) {
        int error = 0;
        int len = sizeof(error);
        if (getsockopt(client->sock, SOL_SOCKET, SO_ERROR, (char*)&error, &len) == 0) {
            if (error == 0) {
                client->fully_connected = 1;
                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [LOG] Connection established\n",
                           (unsigned long long)get_timestamp_ms());
                    fflush(client->log_file);
                }
                if (client->on_connected) {
                    client->on_connected(client->user_data);
                }
                return 0;
            } else if (error == WSAEINPROGRESS || error == WSAEALREADY) {
                return 0;
            } else {
                if (client->log_file) {
                    fprintf(client->log_file, "%llu: [ERROR] Connect failed: %d\n",
                           (unsigned long long)get_timestamp_ms(), error);
                    fflush(client->log_file);
                }
                etcpmon_client_disconnect(client);
                if (client->on_disconnected) client->on_disconnected(client->user_data);
                if (client->on_error) {
                    char msg[256];
                    snprintf(msg, sizeof(msg), "Connect failed: %d", error);
                    client->on_error(msg, client->user_data);
                }
                return -1;
            }
        }
        etcpmon_client_disconnect(client);
        return -1;
    }

    /* Авто-запрос метрик каждые 100 мс (асинхронно) */
    if (client->fully_connected && client->selected_peer_id != 0) {
        uint64_t now = get_timestamp_ms();
        if (now - client->last_metrics_request_ms >= ETCPMON_UPDATE_INTERVAL_MS) {
            etcpmon_client_send_request(client, ETCPMON_CMD_GET_METRICS, NULL, 0,
                                        metrics_auto_callback, NULL);
            client->last_metrics_request_ms = now;
        }
    }

    /* Приём данных */
    uint8_t* buf = client->recv_buffer + client->recv_len;
    int buf_space = ETCPMON_MAX_MSG_SIZE - client->recv_len;
    if (buf_space <= 0) {
        client->recv_len = 0;           /* защита от переполнения */
        buf = client->recv_buffer;
        buf_space = ETCPMON_MAX_MSG_SIZE;
    }

    int received = recv(client->sock, (char*)buf, buf_space, 0);

    if (received == SOCKET_ERROR) {
        int err = WSAGetLastError();
        if (err == WSAEWOULDBLOCK) return 0;
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [ERROR] recv failed: %d\n",
                   (unsigned long long)get_timestamp_ms(), err);
            fflush(client->log_file);
        }
        etcpmon_client_disconnect(client);
        if (client->on_disconnected) client->on_disconnected(client->user_data);
        return -1;
    }

    if (received == 0) {
        if (client->log_file) {
            fprintf(client->log_file, "%llu: [LOG] Connection closed by server\n",
                   (unsigned long long)get_timestamp_ms());
            fflush(client->log_file);
        }
        etcpmon_client_disconnect(client);
        if (client->on_disconnected) client->on_disconnected(client->user_data);
        return -1;
    }

    if (client->log_file) {
        fprintf(client->log_file, "%llu: [LOG] Received %d bytes (offset=%d)\n",
               (unsigned long long)get_timestamp_ms(), received, client->recv_len);
        log_hex_data(client->log_file, "RX", buf, received);
    }

    client->recv_len += received;

    /* === ОСНОВНОЙ ПАРСЕР (ИСПРАВЛЕННЫЙ) === */
    int processed = 0;
    while (client->recv_len >= sizeof(struct etcpmon_msg_header)) {
        struct etcpmon_msg_header* hdr = (struct etcpmon_msg_header*)client->recv_buffer;

        /* Валидация */
        if (etcpmon_validate_header(hdr) != 0) {
            if (client->log_file) {
                fprintf(client->log_file, "%llu: [ERROR] Invalid header, skipping byte %02X\n",
                       (unsigned long long)get_timestamp_ms(), client->recv_buffer[0]);
                fflush(client->log_file);
            }
            memmove(client->recv_buffer, client->recv_buffer + 1, client->recv_len - 1);
            client->recv_len--;
            continue;
        }

        /* hdr->size — это ПОЛНЫЙ размер сообщения (header + payload) */
        uint16_t total_size = hdr->size;
        if (client->recv_len < total_size) {
            break; /* ждём остаток */
        }

        uint16_t payload_size = total_size - sizeof(*hdr);
        uint8_t* payload = client->recv_buffer + sizeof(*hdr);

        client->last_response_seq_id = hdr->seq_id;

        if (client->log_file) {
            fprintf(client->log_file, "%llu: [LOG] Parsed message type=0x%02X total=%d payload=%d seq=%d\n",
                   (unsigned long long)get_timestamp_ms(), hdr->type, total_size, payload_size, hdr->seq_id);
            fflush(client->log_file);
        }

        handle_response_with_callback(client, hdr->type, payload, payload_size);
        processed++;

        /* Удаляем обработанное сообщение */
        if (client->recv_len > total_size) {
            memmove(client->recv_buffer, client->recv_buffer + total_size,
                    client->recv_len - total_size);
        }
        client->recv_len -= total_size;
    }

    return processed;
}

void etcpmon_client_set_callbacks(struct etcpmon_client* client,
                                   void (*on_connected)(void*),
                                   void (*on_disconnected)(void*),
                                   void (*on_conn_list)(struct etcpmon_conn_info*, uint8_t, void*),
                                   void (*on_socket_list)(struct etcpmon_socket_info*, uint8_t, void*),
                                   void (*on_metrics)(struct etcpmon_rsp_metrics*,
                                                      struct etcpmon_link_metrics*,
                                                      uint8_t, void*),
                                   void (*on_error)(const char*, void*),
                                   void (*on_action_result)(const char*, void*),
                                   void* user_data) {
    if (!client) return;

    client->on_connected = on_connected;
    client->on_disconnected = on_disconnected;
    client->on_conn_list = on_conn_list;
    client->on_socket_list = on_socket_list;
    client->on_metrics = on_metrics;
    client->on_error = on_error;
    client->on_action_result = on_action_result;
    client->user_data = user_data;
}

const char* etcpmon_client_get_conn_name(struct etcpmon_client* client, uint64_t peer_id) {
    if (!client || !client->conn_list) return NULL;
    
    for (uint8_t i = 0; i < client->conn_count; i++) {
        if (client->conn_list[i].peer_node_id == peer_id) {
            return client->conn_list[i].name;
        }
    }
    return NULL;
}

struct metrics_history* etcpmon_client_get_history(struct etcpmon_client* client) {
    if (!client) return NULL;
    return &client->history;
}

static void debug_config_auto_callback(void* arg, uint8_t msg_type, const uint8_t* payload, uint16_t payload_size) {
    (void)arg; (void)msg_type; (void)payload; (void)payload_size;
}

int etcpmon_client_request_debug_config(struct etcpmon_client* client) {
    if (!client || client->sock == INVALID_SOCKET) return -1;
    return etcpmon_client_send_request(client, ETCPMON_CMD_GET_DEBUG_CONFIG, NULL, 0,
                                        debug_config_auto_callback, NULL);
}

int etcpmon_client_send_debug_config(struct etcpmon_client* client,
                                      uint8_t global_level, uint8_t count, const uint8_t* levels) {
    if (!client || client->sock == INVALID_SOCKET || !levels || count == 0 || count > ETCPMON_MAX_DEBUG_CATEGORIES) return -1;

    uint8_t payload[2 + ETCPMON_MAX_DEBUG_CATEGORIES];
    payload[0] = global_level;
    payload[1] = count;
    memcpy(payload + 2, levels, count);
    uint16_t payload_size = 2 + count;

    struct etcpmon_msg_header hdr;
    etcpmon_build_header(&hdr, payload_size, ETCPMON_CMD_SET_DEBUG_CONFIG, 0);

    if (client->log_file) {
        fprintf(client->log_file, "%llu: [LOG] CMD_SET_DEBUG_CONFIG global=%d count=%d\n",
               (unsigned long long)get_timestamp_ms(), global_level, count);
        fflush(client->log_file);
    }

    int sent = send(client->sock, (const char*)&hdr, sizeof(hdr), 0);
    if (sent != sizeof(hdr)) return -1;
    sent = send(client->sock, (const char*)payload, payload_size, 0);
    if (sent != payload_size) return -1;
    client->debug_global_level = global_level;
    for (uint8_t i = 0; i < count; i++) client->debug_category_levels[i] = levels[i];
    return 0;
}

void etcpmon_client_add_to_history(struct etcpmon_client* client, struct etcpmon_rsp_metrics* metrics) {
    if (!client || !metrics) return;
    
    struct metrics_history* h = &client->history;
    int idx = h->head;
    
   /* BBR per-link metrics (use first link as representative) */
    if (client->last_links_count > 0 && client->last_links) {
        struct etcpmon_link_metrics* l = &client->last_links[0];
        h->values[GRAPH_METRIC_INFLIGHT_BYTES][idx] = l->inflight_bytes / 1024.f;
        h->values[GRAPH_METRIC_INFLIGHT_LIM][idx]   = l->inflight_lim_bytes / 1024.f;
        h->values[GRAPH_METRIC_MIN_RTT][idx]        = l->bbr_min_rtt_us / 1000.f;
        h->values[GRAPH_METRIC_BW_HI][idx]          = l->bbr_bw_hi / 125.f;
        h->values[GRAPH_METRIC_PACING_RATE][idx]    = l->bbr_pacing_rate / 125.f;
        h->values[GRAPH_METRIC_BW_LO][idx]          = l->bbr_bw_lo / 125.f;
        h->values[GRAPH_METRIC_INFLIGHT_HI][idx]     = l->bbr_inflight_hi / 1024.f;
        h->values[GRAPH_METRIC_BBR_MODE][idx]        = (float)l->bbr_mode;
    }
    
    h->head = (h->head + 1) % GRAPH_HISTORY_SIZE;
    if (h->count < GRAPH_HISTORY_SIZE) h->count++;

    if (client->log_file) {
        fprintf(client->log_file, "history updated: count=%d head=%d\n", h->count, h->head);
        fflush(client->log_file);
    }
}