ihm/MainWindow.h

#pragma once

#include <QMainWindow>
#include <QWidget>
#include <QStackedWidget>
#include <QPixmap>
#include <TCPSocket/TCPUtils.hpp>

#include "homeButton.h"
#include "Homologation.h"
#include "InGame.h"
#include "PreparationMatch.h"
#include "TeamChooser.h"
#include "TestMode.h"
#include "tcp/MyTCPClient.h"

class MainWindow : public QMainWindow {
    Q_OBJECT
public:
    MainWindow(const char* address = "127.0.0.1", int port = 8080, QWidget* parent = nullptr) : QMainWindow(parent)
    {
        this->socketClient = new MyTCPSocket(this);
        this->socketClient->connectToServer(address, port);
        connect(this->socketClient, &MyTCPSocket::messageReceived, this, &MainWindow::handleMessage);
        this->centralWidget = new QWidget(this);
        this->setCentralWidget(centralWidget);

        this->mainLayout = new QVBoxLayout(centralWidget);
        this->homeBtn = new QPushButton("", this);
        this->homeBtn->setIcon(QPixmap(":/img/logo_without_background.png", "PNG"));
        this->homeBtn->setStyleSheet("background-color: transparent; border: none;");
        this->homeBtn->setIconSize(QSize(249, 51));
        this->homeBtn->setFixedSize(QSize(400, 51));

        connect(this->homeBtn, &QPushButton::pressed, this, &MainWindow::onHomePressed);

        this->topLayout = new QHBoxLayout;

        this->quit = new QPushButton("", this);
        this->quit->setIcon(QPixmap(":/img/close.png", "PNG"));
        this->quit->setIconSize(QSize(51, 51));
        this->quit->setStyleSheet("background-color: transparent; border: none;");

        connect(this->quit, &QPushButton::pressed, this, [=]() {
            this->close();
        });

        this->topLayout->addWidget(this->homeBtn);
        this->topLayout->addWidget(this->quit, 1, Qt::AlignTop | Qt::AlignRight);

        this->mainLayout->addLayout(this->topLayout);

        this->setFixedSize(QSize(480, 320));

        this->home = new homeButton(centralWidget);

        connect(this->home, &homeButton::homologationClicked, this, &MainWindow::onHomologationPressed);

        connect(this->home, &homeButton::jeuClicked, this, &MainWindow::onTeamChooserPressed);

        connect(this->home, &homeButton::testClicked, this, &MainWindow::onTestModePressed);

        this->homologation = new Homologation(centralWidget);
        connect(this->homologation, &Homologation::deplierClicked, this, &MainWindow::deplierRobot);
        connect(this->homologation, &Homologation::replierClicked, this, &MainWindow::replierRobot);

        this->teamChooser = new TeamChooser(centralWidget);
        connect(this->teamChooser, &TeamChooser::spawnPointChoose, this, &MainWindow::onSpawnPointChoose);

        this->preparationMatch = new PreparationMatch(centralWidget);
        connect(this->preparationMatch, &PreparationMatch::startGame, this, &MainWindow::onStartGame);
        // connect(this->preparationMatch, &PreparationMatch::askTCPServer, this, &MainWindow::broadcastTCPMessage);
        connect(this->preparationMatch, &PreparationMatch::askTCPServer, [&](const std::string& message) {
            this->socketClient->sendMessage(message.c_str());
        });

        this->testMode = new TestMode(centralWidget);
        connect(this->testMode, &TestMode::goPressed, this, &MainWindow::moveRobot);

        this->inGame = new InGame(teamChooser);

        this->stackedWidget = new QStackedWidget(centralWidget);
        this->stackedWidget->addWidget(this->home);
        this->stackedWidget->addWidget(this->homologation);
        this->stackedWidget->addWidget(this->teamChooser);
        this->stackedWidget->addWidget(this->preparationMatch);
        this->stackedWidget->addWidget(this->testMode);
        this->stackedWidget->addWidget(this->inGame);

        this->mainLayout->addWidget(this->stackedWidget);

        this->setWidgetNb(0);
    }

    void setWidgetNb(const int index)
    {
        if (index == 2)
        {
            QPalette palette;
            palette.setBrush(this->backgroundRole(), QBrush(QPixmap(":/img/table.jpg", "JPG").scaled(this->size(), Qt::IgnoreAspectRatio)));
            this->setPalette(palette);
            this->homeBtn->hide();
            this->quit->hide();
        } else
        {
            QPalette palette;
            palette.setBrush(this->backgroundRole(), QBrush(Qt::white));
            this->setPalette(palette);
            this->homeBtn->show();
            this->quit->show();
        }
        this->stackedWidget->setCurrentIndex(index);
    }

protected slots:
    void onHomePressed()
    {
        this->setWidgetNb(0);
    }

    void onHomologationPressed()
    {
        this->setWidgetNb(1);
    }

    void onTeamChooserPressed()
    {
        this->setWidgetNb(2);
    }

    void onTestModePressed()
    {
        this->setWidgetNb(4);
    }

    void onSpawnPointChoose(int nb)
    {
        this->setWidgetNb(3);
    }

    void onStartGame()
    {
        this->setWidgetNb(5);
    }

    void onDeplierRobot()
    {
        emit deplierRobot();
    }

    void onReplierRobot()
    {
        emit replierRobot();
    }


    void handleMessage(const QByteArray &message)
    {
        std::string mes = message.toStdString();
        std::vector<std::string> list = TCPSocket::split(mes, ";");

        if (TCPSocket::startWith(list[2], "pong"))
        {
            preparationMatch->responseFromPing(QString::fromStdString(mes));
        }
        if (TCPSocket::contains(list[0], "tirette") && TCPSocket::contains(list[2], "set state"))
        {
            preparationMatch->responseTiretteState(QString::fromStdString(mes));
        }
        if (TCPSocket::contains(list[0], "lidar"))
        {
            preparationMatch->responseLidar(QString::fromStdString(mes));
        }
    }


signals:
    void deplierRobot();
    void replierRobot();
    void moveRobot(int x, int y, int theta);

private:
    QVBoxLayout* mainLayout;
    QHBoxLayout* topLayout;
    QPushButton* homeBtn;
    QWidget* centralWidget;
    QStackedWidget* stackedWidget;
    QPushButton* quit;

    homeButton* home;
    Homologation* homologation;
    TeamChooser* teamChooser;
    PreparationMatch* preparationMatch;
    TestMode* testMode;
    InGame* inGame;

    MyTCPSocket* socketClient;
};