mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "QMessageBox"
#include "Qstring"
#include "ticket.h"
#include "addwidget.h"
#include "QCloseEvent"

// Глобальні змінні
CTicketList listOfTickets;
int iPriceInCents = 80000;

// --- Конструктор та Деструктор ---

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    // Завантаження даних при старті
    ifstream file("listInformationStorage.txt");
    try {
        file >> listOfTickets;
    }
    catch (std::runtime_error& err) {
        QMessageBox errorBox;
        errorBox.setIcon(QMessageBox::Critical);
        errorBox.setText(QString::fromStdString(err.what()));
        errorBox.exec();
    }
    file.close();

    displayedList = &listOfTickets;
    refreshTicketList(displayedList);
}

MainWindow::~MainWindow()
{
    delete ui;
}

// --- Події та Логіка ---

// Обробка закриття програми (запит на збереження)
void MainWindow::closeEvent(QCloseEvent *event)
{
    QMessageBox::StandardButton reply;
    reply = QMessageBox::question(this, "Вихід",
                                  "Бажаєте зберегти зміни перед виходом?",
                                  QMessageBox::Yes | QMessageBox::No | QMessageBox::Cancel);

    if (reply == QMessageBox::Yes) {
        on_saveListToFileBtn_clicked();
        event->accept();
    }
    else if (reply == QMessageBox::No) {
        event->accept();
    }
    else {
        event->ignore();
    }
}

// Перемальовує список квитків у ScrollArea
void MainWindow::refreshTicketList(CTicketList* listToDraw)
{
    // Очищення старих віджетів
    QLayoutItem* item;
    while ((item = ui->widgetBox->layout()->takeAt(0)) != nullptr) {
        delete item->widget();
        delete item;
    }

    // Додавання нових
    if (listToDraw) {
        for(auto it = listToDraw->begin(); it != listToDraw->end(); it++)
        {
            Ticket* ticketPtr = new Ticket();
            ticketPtr->fillTicket(*it);
            ui->widgetBox->addWidget(ticketPtr);
            ticketPtr->show();

            // Підключення сигналів від квитка до головного вікна
            connect(ticketPtr, &Ticket::removeRequested, this, &MainWindow::onTicketDelete);
            connect(ticketPtr, &Ticket::refresh, this, [this](){
                this->refreshTicketList(this->displayedList);
            });
        }
    }
}

// --- Слоти ---

// Збереження у файл
void MainWindow::on_saveListToFileBtn_clicked()
{
    ofstream file("listInformationStorage.txt");
    try {
        file << listOfTickets;
    }
    catch (std::runtime_error& err) {
        QMessageBox errorBox;
        errorBox.setIcon(QMessageBox::Critical);
        errorBox.setText(QString::fromStdString(err.what()));
        errorBox.exec();
    }
    file.close();
}

// Фільтрація списку
void MainWindow::on_FilterBtn_clicked()
{
    // Очищення попереднього тимчасового списку (якщо це не основний)
    if (displayedList != nullptr && displayedList != &listOfTickets) {
        delete displayedList;
        displayedList = nullptr;
    }

    QString sSortingChoice = ui->filterBox->currentText();
    string searchKey = ui->optionsBox->currentText().toStdString();

    if (sSortingChoice == "За містом") {
        displayedList = listOfTickets.findTicketsByCity(searchKey);
    }
    else if (sSortingChoice == "За типом транспорту") {
        displayedList = listOfTickets.findTicketsByTransportType(searchKey);
    }
    else if (sSortingChoice == "За категорією громадян") {
        displayedList = listOfTickets.findTicketsByCitizenCategory(searchKey);
    }
    else {
        displayedList = &listOfTickets;
    }

    refreshTicketList(displayedList);
}

// Оновлення опцій у випадаючому списку при зміні типу фільтру
void MainWindow::on_filterBox_activated(int index)
{
    ui->optionsBox->clear();
    ui->optionsBox->addItem("Не вибрано");

    QString selectedOption = ui->filterBox->currentText();

    if (selectedOption == "За містом") {
        for (const auto& city : listOfTickets.citiesList) {
            ui->optionsBox->addItem(QString::fromStdString(city));
        }
    }
    else if (selectedOption == "За типом транспорту") {
        vector<string> transportTypeList = { "Метро", "Автобус", "Тролейбус", "Трамвай" };
        for (const auto& type : transportTypeList) {
            ui->optionsBox->addItem(QString::fromStdString(type));
        }
    }
    else if (selectedOption == "За категорією громадян") {
        vector<string> categoriesList = { "Громадянин", "Школяр", "Студент", "Пенсіонер", "Ветеран", "ВПО", "Соціальна категорія" };
        for (const auto& category : categoriesList) {
            ui->optionsBox->addItem(QString::fromStdString(category));
        }
    }
    this->displayedList = &listOfTickets;
    this->refreshTicketList(displayedList);
}

// Сортування списку
void MainWindow::on_sortByTransportBtn_clicked()
{
    listOfTickets.mergeSortTicketsByTransportType();
    this->displayedList = &listOfTickets;
    this->refreshTicketList(displayedList);
}

// Видалення квитка
void MainWindow::onTicketDelete(int extId)
{
    try {
        CSeasonTicket* ticketToRemove = listOfTickets.findTicketByID(extId);
        if (ticketToRemove) {
            listOfTickets.removeTicket(*ticketToRemove);
            displayedList = &listOfTickets;
            refreshTicketList(displayedList);
        }
    }
    catch (std::runtime_error& err) {
        QMessageBox errorBox;
        errorBox.setIcon(QMessageBox::Critical);
        errorBox.setText(QString::fromStdString(err.what()));
        errorBox.exec();
        return;
    }
}

// Додавання нового квитка
void MainWindow::on_AddNewBtn_clicked()
{
    addWidget* widget = new addWidget();
    widget->show();

    connect(widget, &addWidget::ticketAdded, this, [this](){
        this->displayedList = &listOfTickets;
        this->refreshTicketList(displayedList);
        this->displayedList = &listOfTickets;
    });
}