#include <QApplication>
#include <QHBoxLayout>
#include <QLabel>
#include <QLineEdit>
#include <QPushButton>
#include <QTabWidget>
#include <QTextEdit>
#include <QTreeWidgetItem>
#include <QVBoxLayout>
#include <QWidget>

#include "GamesPanel.h"
#include "SettingsTree.h"
#include "VersionsPanel.h"
#include "WebSocketController.h"

static QWidget *makeGamesTab(WebSocketController *ctrl, QWidget *parent)
{
    auto *panel = new GamesPanel(parent);
    ctrl->setGamesPanel(panel);
    return panel;
}

static QWidget *makeVersionsTab(WebSocketController *ctrl, QWidget *parent)
{
    auto *panel = new VersionsPanel(parent);
    ctrl->setVersionsPanel(panel);
    return panel;
}

static QWidget *makeSettingsTab(WebSocketController *ctrl, QWidget *parent)
{
    auto *page   = new QWidget(parent);
    auto *layout = new QVBoxLayout(page);
    layout->setContentsMargins(2, 2, 2, 2);
    auto *tree = new SettingsTree(page);
    auto *ph   = new QTreeWidgetItem(tree);
    ph->setText(0, "Connect to load settings...");
    layout->addWidget(tree, 1);
    ctrl->setSettingsTree(tree);
    return page;
}

static QWidget *makeManualTab(WebSocketController *ctrl, QWidget *parent)
{
    auto *page    = new QWidget(parent);
    auto *layout  = new QVBoxLayout(page);
    auto *row     = new QHBoxLayout();
    auto *cmdEdit = new QLineEdit(page);
    cmdEdit->setPlaceholderText("Enter command to send...");
    auto *sendBtn = new QPushButton("Send", page);
    row->addWidget(new QLabel("Command:", page));
    row->addWidget(cmdEdit, 1);
    row->addWidget(sendBtn);
    auto *log = new QTextEdit(page);
    log->setReadOnly(true);
    layout->addLayout(row);
    layout->addWidget(log, 1);
    ctrl->addLogView(log);

    auto send = [cmdEdit, ctrl]() {
        const QString cmd = cmdEdit->text().trimmed();
        if (!cmd.isEmpty()) { ctrl->sendCommand(cmd); cmdEdit->clear(); }
    };
    QObject::connect(sendBtn, &QPushButton::clicked,     page, send);
    QObject::connect(cmdEdit, &QLineEdit::returnPressed, page, send);
    return page;
}

static QWidget *makePanelsTab(WebSocketController *ctrl, QWidget *parent)
{
    auto *page   = new QWidget(parent);
    auto *layout = new QVBoxLayout(page);
    auto *log    = new QTextEdit(page);
    log->setReadOnly(true);
    layout->addWidget(log, 1);
    ctrl->addLogView(log);
    return page;
}

static QWidget *makePlaceholder(const QString &text, QWidget *parent)
{
    auto *page   = new QWidget(parent);
    auto *layout = new QVBoxLayout(page);
    auto *label  = new QLabel(text, page);
    label->setAlignment(Qt::AlignCenter);
    layout->addWidget(label);
    return page;
}

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

    QWidget window;
    window.setWindowTitle("ESA WebSocket Client");

    auto *mainLayout = new QVBoxLayout(&window);
    mainLayout->setContentsMargins(6, 6, 6, 6);
    mainLayout->setSpacing(4);

    // --- Header bar ---
    auto *headerRow    = new QHBoxLayout();
    auto *urlEdit      = new QLineEdit(&window);
    urlEdit->setPlaceholderText("ws://127.0.0.1:3491/");
    urlEdit->setText("ws://127.0.0.1:3491/");
    auto *connectBtn   = new QPushButton("Connect",    &window);
    auto *disconnectBtn= new QPushButton("Disconnect", &window);
    disconnectBtn->setEnabled(false);
    auto *statusLabel  = new QLabel("Disconnected", &window);

    headerRow->addWidget(new QLabel("WebSocket URL:", &window));
    headerRow->addWidget(urlEdit, 1);
    headerRow->addWidget(connectBtn);
    headerRow->addWidget(disconnectBtn);
    headerRow->addWidget(statusLabel);
    mainLayout->addLayout(headerRow);

    // --- Controller ---
    auto *ctrl = new WebSocketController(urlEdit, statusLabel, &window);

    // --- Tabs ---
    auto *tabs = new QTabWidget(&window);
    tabs->addTab(makeGamesTab   (ctrl, &window), "games");
    tabs->addTab(makeVersionsTab(ctrl, &window), "versions");
    tabs->addTab(makeManualTab  (ctrl, &window), "manual");
    tabs->addTab(makeSettingsTab(ctrl, &window), "settings");
    tabs->addTab(makePlaceholder("Power",  &window), "power");
    tabs->addTab(makePanelsTab  (ctrl, &window), "panels");
    mainLayout->addWidget(tabs, 1);

    // --- Wire up buttons ---
    QObject::connect(connectBtn,    &QPushButton::clicked, ctrl, &WebSocketController::startConnection);
    QObject::connect(disconnectBtn, &QPushButton::clicked, ctrl, &WebSocketController::closeConnection);

    QObject::connect(ctrl->socket(), &QWebSocket::connected,            ctrl, &WebSocketController::onConnected);
    QObject::connect(ctrl->socket(), &QWebSocket::disconnected,         ctrl, &WebSocketController::onDisconnected);
    QObject::connect(ctrl->socket(), &QWebSocket::textMessageReceived,  ctrl, &WebSocketController::onTextMessageReceived);
    QObject::connect(ctrl->socket(), &QWebSocket::errorOccurred,        ctrl, &WebSocketController::onErrorOccurred);

    QObject::connect(ctrl->socket(), &QWebSocket::connected,    &window, [connectBtn, disconnectBtn]() {
        connectBtn->setEnabled(false); disconnectBtn->setEnabled(true);
    });
    QObject::connect(ctrl->socket(), &QWebSocket::disconnected, &window, [connectBtn, disconnectBtn]() {
        connectBtn->setEnabled(true);  disconnectBtn->setEnabled(false);
    });

    window.resize(1050, 620);
    window.show();
    return app.exec();
}