#include "mainwindow.h"
#include "ui_mainwindow.h"

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

MainWindow::~MainWindow()
{
    delete week_template;
    foreach (Year *y, years) {
        delete y;
    }
    years.clear();
    delete ui;
}

QString MainWindow::get_save_file_path() {
    QString path = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
    if (!QDir(path).exists()) {
        QDir().mkpath(path);
    }
    path += "/data/";
    if (!QDir(path).exists()) {
        QDir().mkpath(path);
    }
    path += SAVE_FILENAME;
    return path;
}

void MainWindow::init() {
    objectId.insert(ui->mon_button->objectName(), Identifier::mon);
    objectId.insert(ui->tue_button->objectName(), Identifier::tue);
    objectId.insert(ui->wed_button->objectName(), Identifier::wed);
    objectId.insert(ui->thu_button->objectName(), Identifier::thu);
    objectId.insert(ui->fri_button->objectName(), Identifier::fri);
    connect(ui->aboutButton, &QPushButton::clicked, this, &MainWindow::open_about);
    connect(ui->template_settings_button, &QPushButton::clicked, this, &MainWindow::edit_template);
    connect(ui->today_button, &QPushButton::clicked, this, &MainWindow::set_date_to_now);
    connect(ui->week_options_button, &QPushButton::clicked, this, &MainWindow::set_week_options);
    connect(ui->dateEdit, &QDateEdit::dateTimeChanged, this, &MainWindow::compute_week_number);
    connect(ui->mon_button, &QPushButton::clicked, this, &MainWindow::edit);
    connect(ui->tue_button, &QPushButton::clicked, this, &MainWindow::edit);
    connect(ui->wed_button, &QPushButton::clicked, this, &MainWindow::edit);
    connect(ui->thu_button, &QPushButton::clicked, this, &MainWindow::edit);
    connect(ui->fri_button, &QPushButton::clicked, this, &MainWindow::edit);
    todayWeekNumber = QDate::currentDate().weekNumber();
    dayOfWeek = QDate::currentDate().dayOfWeek();
    saveLoaded = false;

    if (QFile::exists(get_save_file_path())) {
        open_save();
    } else {
        week_template = new Week();
        Welcome w(week_template);
        w.exec();
        saveLoaded = true;
    }
    if (saveLoaded) {
        set_date_to_now();
        compute_time();
    }
}

void MainWindow::highlightDayOfWeek() {
    ui->monLabel->setText("Lundi");
    ui->tueLabel->setText("Mardi");
    ui->wedLabel->setText("Mercredi");
    ui->thuLabel->setText("Jeudi");
    ui->friLabel->setText("Vendredi");
    if (todayWeekNumber == current_week->getWeekNumber()) {
        ui->today_button->setDisabled(true);
        switch (dayOfWeek) {
            case 1: {
                ui->monLabel->setText(QString("%1 📆").arg(ui->monLabel->text()));
                break;
            }
            case 2: {
                ui->tueLabel->setText(QString("%1 📆").arg(ui->tueLabel->text()));
                break;
            }
            case 3: {
                ui->wedLabel->setText(QString("%1 📆").arg(ui->wedLabel->text()));
                break;
            }
            case 4: {
                ui->thuLabel->setText(QString("%1 📆").arg(ui->thuLabel->text()));
                break;
            }
            case 5: {
                ui->friLabel->setText(QString("%1 📆").arg(ui->friLabel->text()));
                break;
            }
            default: break;
        }
    } else {
        ui->today_button->setDisabled(false);
    }
}

void MainWindow::set_date_to_now() {
    auto date = QDate::currentDate();
    ui->dateEdit->setDate(date);
}

void MainWindow::open_save() {
    QFile* file = new QFile(get_save_file_path());
    file->open(QIODevice::ReadOnly);
    auto json = QString(file->readAll());
    file->close();
    delete file;

    QJsonObject obj = QJsonDocument::fromJson(json.toUtf8()).object();
    if (obj[KEY_SAVE_FILE_VERSION] != SAVE_FILE_VERSION) {
        panic_dialog("Your save file version does not match the supported version."
                     " Please update this file and restart the application");
    }
    week_template = Week::from_json(obj[KEY_TEMPLATE].toObject());
    QJsonArray arr = obj[KEY_YEARS].toArray();
    foreach (QJsonValue val, arr) {
        Year *y = Year::from_json(val.toObject());
        years[y->getNumber()] = y;
    }
    saveLoaded = true;
}

void MainWindow::panic_dialog(QString text) {
    QMessageBox::critical(this, tr("Chronos"), text,
                                   QMessageBox::Ok, QMessageBox::Ok);
    QTimer::singleShot(0, this, &MainWindow::close);
}

void MainWindow::save_to_file() {
    QJsonArray arr;
    foreach (Year *y, years) {
        arr.append(y->to_json());
    }
    QJsonObject obj {
        {KEY_TEMPLATE, week_template->to_json()},
        {KEY_YEARS, arr},
        {KEY_SAVE_FILE_VERSION, SAVE_FILE_VERSION}
    };
    QJsonDocument doc(obj);
    QFile *f = new QFile(get_save_file_path());
    if (f->open(QIODevice::WriteOnly)) {
        f->write(doc.toJson());
        f->close();
    }
    delete f;
}

void MainWindow::compute_week_number(const QDateTime &dt) {
    int n = dt.date().weekNumber();
    int y = dt.date().year();
    ui->label_semaine->setText(QString("Semaine : %1").arg(QString::number(n)));
    if (years.contains(y)) {
        current_year = years[y];
    } else {
        Year *nYear = new Year(y);
        years[y] = nYear;
        current_year = nYear;
        save_to_file();
    }

    if (current_year->hasWeek(n)) {
        current_week = current_year->getWeek(n);
    } else {
        current_week = new Week(week_template);
        current_week->setWeekNumber(n);
        current_year->addWeek(current_week);
        save_to_file();
    }
    compute_time();
    highlightDayOfWeek();
}

void MainWindow::compute_time() {
    ui->mon_time_label->setText(Tools::double_to_string_time(current_week->getMon()->get_total()));
    ui->tue_time_label->setText(Tools::double_to_string_time(current_week->getTue()->get_total()));
    ui->wed_time_label->setText(Tools::double_to_string_time(current_week->getWed()->get_total()));
    ui->thu_time_label->setText(Tools::double_to_string_time(current_week->getThu()->get_total()));
    ui->fri_time_label->setText(Tools::double_to_string_time(current_week->getFri()->get_total()));
    ui->total_time_label->setText(Tools::double_to_string_time(current_week->total()));
    updateStartLabel();
    updateBreakLabel();
    updateEndLabel();
    updateWeekTime();
    updateValidIcon();
    updateWarningIcon();

    double late = 0.0;
    double overtime = 0.0;
    int todayWeekNumber = QDate::currentDate().weekNumber();
    QMapIterator i = current_year->iterator();
    while (i.hasNext()) {
        Week *w = i.next().value();
        if (w->getWeekNumber() <= todayWeekNumber) {
            late += (week_template->total() - w->total()) - w->getTimeDeltaInHours();
            overtime += (w->total() - week_template->total()) + w->getTimeDeltaInHours();
        }
    }
    late -= current_week->getTimeDeltaInHours();
    overtime += - current_week->getTimeDeltaInHours();
    ui->late_time_label->setText(Tools::double_to_string_time((late > 0.0) ? late : 0.0));
    ui->overtime_time_label->setText(Tools::double_to_string_time((overtime > 0.0) ? overtime : 0.0));
}

void MainWindow::updateValidIcon() {
    ui->mondayValidate->setVisible(current_week->getMon()->get_validate());
    ui->tuesdayValidate->setVisible(current_week->getTue()->get_validate());
    ui->wednesdayValidate->setVisible(current_week->getWed()->get_validate());
    ui->thurdayValidate->setVisible(current_week->getThu()->get_validate());
    ui->fridayValidate->setVisible(current_week->getFri()->get_validate());
}

void MainWindow::updateWarningIcon() {
    ui->mondayWarning->setVisible(current_week->getMon()->has_warning());
    ui->tuesdayWarning->setVisible(current_week->getTue()->has_warning());
    ui->wednesdayWarning->setVisible(current_week->getWed()->has_warning());
    ui->thurdayWarning->setVisible(current_week->getThu()->has_warning());
    ui->fridayWarning->setVisible(current_week->getFri()->has_warning());
}

void MainWindow::updateWeekTime()
{
    double t = (current_week->total() + current_week->getTimeDeltaInHours()) - week_template->total();
    if (t > 0) {
        ui->overtime_time_label_week->setText(Tools::double_to_string_time(t));
        ui->late_time_label_week->setText("0h");
    } else if(t < 0) {
        ui->late_time_label_week->setText(Tools::double_to_string_time(-t));
        ui->overtime_time_label_week->setText("0h");
    } else {
        ui->overtime_time_label_week->setText("0h");
        ui->late_time_label_week->setText("0h");
    }
}

void MainWindow::updateStartLabel() {
    ui->monStartLabel->setText(current_week->getMon()->get_start().toString("HH:mm"));
    ui->tueStartLabel->setText(current_week->getTue()->get_start().toString("HH:mm"));
    ui->wedStartLabel->setText(current_week->getWed()->get_start().toString("HH:mm"));
    ui->thuStartLabel->setText(current_week->getThu()->get_start().toString("HH:mm"));
    ui->friStartLabel->setText(current_week->getFri()->get_start().toString("HH:mm"));
}

void MainWindow::updateBreakLabel() {
    ui->monBreakLabel->setText(QString("%1 min.").arg(current_week->getMon()->get_time_break()));
    ui->tueBreakLabel->setText(QString("%1 min.").arg(current_week->getTue()->get_time_break()));
    ui->wedBreakLabel->setText(QString("%1 min.").arg(current_week->getWed()->get_time_break()));
    ui->thuBreakLabel->setText(QString("%1 min.").arg(current_week->getThu()->get_time_break()));
    ui->friBreakLabel->setText(QString("%1 min.").arg(current_week->getFri()->get_time_break()));
}

void MainWindow::updateEndLabel() {
    ui->monEndLabel->setText(current_week->getMon()->get_end().toString("HH:mm"));
    ui->tueEndLabel->setText(current_week->getTue()->get_end().toString("HH:mm"));
    ui->wedEndLabel->setText(current_week->getWed()->get_end().toString("HH:mm"));
    ui->thuEndLabel->setText(current_week->getThu()->get_end().toString("HH:mm"));
    ui->friEndLabel->setText(current_week->getFri()->get_end().toString("HH:mm"));
}

void MainWindow::edit() {
    QString name = QObject::sender()->objectName();
    switch (get_identifier(name)) {
        case mon:
            modify_value(current_week->getMon());
            break;
        case tue:
            modify_value(current_week->getTue());
            break;
        case wed:
            modify_value(current_week->getWed());
            break;
        case thu:
            modify_value(current_week->getThu());
            break;
        case fri:
            modify_value(current_week->getFri());
            break;
    }
    save_to_file();
    compute_time();
}

void MainWindow::modify_value(Day *d) {
    bool isNotValidable = (current_week->getWeekNumber() > todayWeekNumber);
    SetDayDialog sdd(d, isNotValidable, this);
    sdd.exec();
}

Identifier MainWindow::get_identifier(QString objectName) {
    return objectId[objectName];
}

void MainWindow::edit_template() {
    Welcome w(week_template);
    int res = w.exec();
    if (res == QDialog::Accepted) {
        save_to_file();
        compute_time();
    }
}

void MainWindow::open_about() {
    AboutBox a(this);
    a.exec();
}

void MainWindow::set_week_options()
{
    WeekOption ow(current_week, this);
    int result = ow.exec();
    if (result == QDialog::Accepted) {
        current_week->setTimeDelta(ow.get_time_delta());
        save_to_file();
        compute_time();
    }
}