#include "../include/ConnectionWindow.h"
#include "imgui.h"
#include <cstdio>

ConnectionWindow::ConnectionWindow() 
    : selectedPortIdx(0), connected(false), activePort(nullptr) {

    // Заполняем массив стандартными скоростями UART
    baudRates = { 9600, 19200, 38400, 57600, 115200, 230400, 460800, 921600 };

    // По умолчанию выбираем 9600 (индекс 0 в векторе)
    selectedBaudIdx = 0;
}

ConnectionWindow::~ConnectionWindow() {
    if (connected && activePort) {
        sp_close(activePort);
        sp_free_port(activePort);
    }
}

void ConnectionWindow::initialize() {
    refreshPorts();
}

void ConnectionWindow::refreshPorts() {
    ports.clear();
    selectedPortIdx = 0;

    struct sp_port** port_list;
    if (sp_list_ports(&port_list) == SP_OK) {
        for (int i = 0; port_list[i] != nullptr; i++) {
            struct sp_port* port = port_list[i];
            SerialPortInfo info;
            const char* name = sp_get_port_name(port);
            info.portName = name ? name : "Unknown";
            const char* desc = sp_get_port_description(port);
            info.description = desc ? desc : "No Description";
            ports.push_back(info);
        }
        sp_free_port_list(port_list);
    }

    if (ports.empty()) {
        ports.push_back({"None", "No serial devices detected"});
    }
}

void ConnectionWindow::forceLogMessage(const std::string& msg) {
    txLog += msg + "\n";
}

void ConnectionWindow::render(bool* p_open) {
    if (!ImGui::Begin("Connection Settings", p_open)) {
        ImGui::End();
        return;
    }

    if (ImGui::Button("Refresh Ports") && !connected) {
        refreshPorts();
    }
    ImGui::SameLine();

    // 1. Выбор COM-порта
    std::string combo_preview = ports[selectedPortIdx].portName;
    if (ports[selectedPortIdx].portName != "None") {
        combo_preview += " (" + ports[selectedPortIdx].description + ")";
    }

    ImGui::BeginDisabled(connected);
    if (ImGui::BeginCombo("Serial Port", combo_preview.c_str())) {
        for (int n = 0; n < ports.size(); n++) {
            const bool is_selected = (selectedPortIdx == n);
            std::string item_text = ports[n].portName + " - " + ports[n].description;
            if (ImGui::Selectable(item_text.c_str(), is_selected)) {
                selectedPortIdx = n;
            }
        }
        ImGui::EndCombo();
    }

    // 2. Выбор Baud Rate через выпадающий список
    std::string baud_preview = std::to_string(baudRates[selectedBaudIdx]);
    if (ImGui::BeginCombo("Baud Rate", baud_preview.c_str())) {
        for (int b = 0; b < baudRates.size(); b++) {
            const bool is_selected = (selectedBaudIdx == b);
            std::string baud_item_text = std::to_string(baudRates[b]);

            if (ImGui::Selectable(baud_item_text.c_str(), is_selected)) {
                selectedBaudIdx = b;
            }
        }
        ImGui::EndCombo();
    }
    ImGui::EndDisabled();

    // 3. Логика подключения
    if (!connected) {
        ImGui::BeginDisabled(ports[selectedPortIdx].portName == "None");
        if (ImGui::Button("Connect", ImVec2(120, 0))) {
            if (sp_get_port_by_name(ports[selectedPortIdx].portName.c_str(), &activePort) == SP_OK) {
                if (sp_open(activePort, SP_MODE_READ_WRITE) == SP_OK) {

                    // Передаем реальное значение скорости из выбранного индекса массива
                    int real_baud = baudRates[selectedBaudIdx];
                    sp_set_baudrate(activePort, real_baud);

                    sp_set_bits(activePort, 8);
                    sp_set_parity(activePort, SP_PARITY_NONE);
                    sp_set_stopbits(activePort, 1);

                    connected = true;
                    forceLogMessage("[SYSTEM] Connected to " + ports[selectedPortIdx].portName + " at " + std::to_string(real_baud) + " baud.");
                } else {
                    forceLogMessage("[SYSTEM ERROR] Could not open port " + ports[selectedPortIdx].portName);
                    sp_free_port(activePort);
                    activePort = nullptr;
                }
            }
        }
        ImGui::EndDisabled();
    } else {
        ImGui::PushStyleColor(ImGuiCol_Button, ImVec4(0.8f, 0.1f, 0.1f, 1.0f));
        if (ImGui::Button("Disconnect", ImVec2(120, 0))) { 
            if (activePort) {
                sp_close(activePort);
                sp_free_port(activePort);
                activePort = nullptr;
            }
            connected = false;
            forceLogMessage("[SYSTEM] Disconnected.");
        }
        ImGui::PopStyleColor();
    }

    ImGui::SameLine();
    ImGui::Text("Status: %s", connected ? "CONNECTED" : "DISCONNECTED");

    ImGui::End();
}