#include "mainwindow.h"
#include <QMessageBox>
#include <QPalette>

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent), socket(new QTcpSocket(this)) {
    setupUI();
    statusTimer = new QTimer(this);
    connect(statusTimer, &QTimer::timeout, this, &MainWindow::updateConnectionStatus);
    statusTimer->start(1000);
    connect(socket, &QTcpSocket::readyRead, this, &MainWindow::readServerOutput);
}

MainWindow::~MainWindow() {
    if (socket->state() == QAbstractSocket::ConnectedState) {
        socket->write("exit\n");
        socket->disconnectFromHost();
        if (socket->state() != QAbstractSocket::UnconnectedState) {
            socket->waitForDisconnected(1000);
        }
    }
}

void MainWindow::attemptConnection() {
    socket->abort();
    isAuthenticated = false;
    socket->connectToHost(serverEdit->text(), portEdit->text().toUInt());
    if (!socket->waitForConnected(3000)) {
        QMessageBox::critical(this, "Connection Failed", "Could not connect to server.");
    }
    updateConnectionStatus();
}

void MainWindow::readServerOutput() {
    QByteArray response = socket->readAll();
    QString text = QString::fromUtf8(response);
    serverLogTextEdit->append(text);

    if (text.contains("Please enter password:", Qt::CaseInsensitive)) {
        socket->write(passwordEdit->text().toUtf8() + "\n");
        socket->waitForBytesWritten(1000);
        return;
    }

    if (!text.contains("Please enter password:", Qt::CaseInsensitive)) {
        isAuthenticated = true;
    }
}

void MainWindow::sendTelnetCommand(const QString &command) {
    if (socket->state() != QAbstractSocket::ConnectedState) {
        QMessageBox::warning(this, "Not Connected", "Please connect to the server first.");
        return;
    }
    if (!isAuthenticated) {
        QMessageBox::warning(this, "Not Authenticated", "Server is waiting for a password. Please wait or reconnect.");
        return;
    }
    socket->write(command.toUtf8() + "\n");
    socket->waitForBytesWritten(1000);
    historyTextEdit->append(command);
}

void MainWindow::sendTimeCommand() {
    QString cmd = QString("st %1 %2 %3")
        .arg(dayEdit->text())
        .arg(hourEdit->text())
        .arg(minuteEdit->text());
    sendTelnetCommand(cmd);
}

void MainWindow::sendSpeedCommand() {
    static const QMap<int, int> values = {
        {1, 1}, {2, 4}, {3, 6}, {4, 12}, {5, 500}
    };
    int val = values.value(speedSlider->value(), 6);
    QString cmd = QString("setgamestat TimeOfDayIncPerSec %1").arg(val);
    sendTelnetCommand(cmd);
}

void MainWindow::sendCustomCommand() {
    QString cmd = customCommandEdit->text().trimmed();
    if (cmd.isEmpty()) {
        QMessageBox::warning(this, "Input Error", "Please enter a custom command.");
        return;
    }
    sendTelnetCommand(cmd);
    customCommandEdit->clear();
}

void MainWindow::updateConnectionStatus() {
    QPalette palette;
    if (socket->state() == QAbstractSocket::ConnectedState) {
        connectionStatusLabel->setText("Connected");
        palette.setColor(QPalette::WindowText, Qt::darkGreen);
    } else {
        connectionStatusLabel->setText("Disconnected");
        palette.setColor(QPalette::WindowText, Qt::red);
    }
    connectionStatusLabel->setPalette(palette);
}

void MainWindow::disconnectFromServer() {
    if (socket->state() == QAbstractSocket::ConnectedState) {
        socket->write("exit\n");
        socket->disconnectFromHost();
    }
    updateConnectionStatus();
}

void MainWindow::updateSliderLabel(int value) {
    static const QMap<int, QString> labels = {
        {1, "Slow-Mo"}, {2, "90"}, {3, "60"}, {4, "Double"}, {5, "Giga"}
    };
    sliderLabel->setText(labels.value(value, QString::number(value)));
}

void MainWindow::setupUI() {
    tabWidget = new QTabWidget(this);
    setCentralWidget(tabWidget);

    QWidget *mainTab = new QWidget;
    QVBoxLayout *mainLayout = new QVBoxLayout(mainTab);

    QGroupBox *serverBox = new QGroupBox("Server Info");
    QVBoxLayout *serverVBox = new QVBoxLayout(serverBox);
    QHBoxLayout *serverLayout = new QHBoxLayout();
    serverEdit = new QLineEdit();
    portEdit = new QLineEdit();
    passwordEdit = new QLineEdit();
    passwordEdit->setEchoMode(QLineEdit::Password);
    serverLayout->addWidget(new QLabel("Server:"));
    serverLayout->addWidget(serverEdit);
    serverLayout->addWidget(new QLabel("Port:"));
    serverLayout->addWidget(portEdit);
    serverLayout->addWidget(new QLabel("Password:"));
    serverLayout->addWidget(passwordEdit);
    serverVBox->addLayout(serverLayout);

    QHBoxLayout *statusLayout = new QHBoxLayout();
    connectionStatusLabel = new QLabel("Disconnected");
    QPalette palette;
    palette.setColor(QPalette::WindowText, Qt::red);
    connectionStatusLabel->setPalette(palette);
    statusLayout->addWidget(connectionStatusLabel);
    statusLayout->addStretch();
    connectButton = new QPushButton("Connect");
    disconnectButton = new QPushButton("Disconnect");
    connect(connectButton, &QPushButton::clicked, this, &MainWindow::attemptConnection);
    connect(disconnectButton, &QPushButton::clicked, this, &MainWindow::disconnectFromServer);
    statusLayout->addWidget(connectButton);
    statusLayout->addWidget(disconnectButton);
    serverVBox->addLayout(statusLayout);
    mainLayout->addWidget(serverBox);

    QGroupBox *timeBox = new QGroupBox("Set Server Time");
    QHBoxLayout *timeLayout = new QHBoxLayout(timeBox);
    dayEdit = new QLineEdit(); dayEdit->setMaxLength(2);
    hourEdit = new QLineEdit(); hourEdit->setMaxLength(2);
    minuteEdit = new QLineEdit(); minuteEdit->setMaxLength(2);
    QPushButton *timeButton = new QPushButton("Set Server Time");
    connect(timeButton, &QPushButton::clicked, this, &MainWindow::sendTimeCommand);
    timeLayout->addWidget(new QLabel("Day:"));
    timeLayout->addWidget(dayEdit);
    timeLayout->addWidget(new QLabel("Hour:"));
    timeLayout->addWidget(hourEdit);
    timeLayout->addWidget(new QLabel("Minute:"));
    timeLayout->addWidget(minuteEdit);
    timeLayout->addStretch();
    timeLayout->addWidget(timeButton);
    mainLayout->addWidget(timeBox);

    QGroupBox *speedBox = new QGroupBox("Set Game Speed");
    QVBoxLayout *speedLayout = new QVBoxLayout(speedBox);
    QHBoxLayout *sliderLayout = new QHBoxLayout();
    speedSlider = new QSlider(Qt::Horizontal);
    speedSlider->setMinimum(1);
    speedSlider->setMaximum(5);
    speedSlider->setTickInterval(1);
    speedSlider->setTickPosition(QSlider::TicksBelow);
    connect(speedSlider, &QSlider::valueChanged, this, &MainWindow::updateSliderLabel);
    sliderLayout->addWidget(speedSlider);
    QPushButton *speedButton = new QPushButton("Set Time Speed");
    connect(speedButton, &QPushButton::clicked, this, &MainWindow::sendSpeedCommand);
    sliderLayout->addWidget(speedButton);
    sliderLabel = new QLabel("60");
    sliderLabel->setAlignment(Qt::AlignCenter);
    speedLayout->addLayout(sliderLayout);
    speedLayout->addWidget(sliderLabel);
    mainLayout->addWidget(speedBox);

    tabWidget->addTab(mainTab, "Time");

    QWidget *historyTab = new QWidget;
    QVBoxLayout *historyLayout = new QVBoxLayout(historyTab);
    historyTextEdit = new QTextEdit();
    historyTextEdit->setReadOnly(true);
    historyTextEdit->setTextInteractionFlags(Qt::TextSelectableByMouse);
    historyLayout->addWidget(historyTextEdit);
    tabWidget->addTab(historyTab, "History");

    QWidget *logTab = new QWidget;
    QVBoxLayout *logLayout = new QVBoxLayout(logTab);
    serverLogTextEdit = new QTextEdit();
    serverLogTextEdit->setReadOnly(true);
    logLayout->addWidget(serverLogTextEdit);

    QGroupBox *customBox = new QGroupBox();
    QHBoxLayout *customLayout = new QHBoxLayout(customBox);
    customCommandEdit = new QLineEdit();
    QPushButton *customButton = new QPushButton("Send Command");
    connect(customButton, &QPushButton::clicked, this, &MainWindow::sendCustomCommand);
    connect(customCommandEdit, &QLineEdit::returnPressed, this, &MainWindow::sendCustomCommand);
    customLayout->addWidget(customCommandEdit);
    customLayout->addWidget(customButton);
    logLayout->addWidget(customBox);

    tabWidget->addTab(logTab, "Console");

    updateSliderLabel(speedSlider->value());
}