#include "squareboot.h"

#include <QUrlQuery>
#include <QNetworkReply>
#include <QMessageBox>
#include <QPushButton>
#include <QStandardPaths>
#include <QFile>
#include <patch.h>
#include <QJsonDocument>
#include <QJsonObject>

#include "squarelauncher.h"

SquareBoot::SquareBoot(LauncherCore& window, SquareLauncher& launcher) : window(window), launcher(launcher) {

}

void SquareBoot::bootCheck(const LoginInformation& info) {
    dialog = new QProgressDialog();
    dialog->setLabelText("Checking the FINAL FANTASY XIV Updater/Launcher version.");
    dialog->show();

    QUrlQuery query;
    query.addQueryItem("time", QDateTime::currentDateTimeUtc().toString("yyyy-MM-dd-HH-mm"));

    QUrl url;
    url.setScheme("http");
    url.setHost("patch-bootver.ffxiv.com");
    url.setPath(QString("/http/win32/ffxivneo_release_boot/%1").arg(info.settings->bootVersion));
    url.setQuery(query);

    auto request = QNetworkRequest(url);
    if(info.settings->license == GameLicense::macOS) {
        request.setRawHeader("User-Agent", "FFXIV-MAC PATCH CLIENT");
    } else {
        request.setRawHeader("User-Agent", "FFXIV PATCH CLIENT");
    }

    request.setRawHeader("Host", "patch-bootver.ffxiv.com");

    auto reply = window.mgr->get(request);
    connect(reply, &QNetworkReply::finished, [=, &info] {
        const QString response = reply->readAll();

        if(response.isEmpty()) {
            dialog->hide();

            launcher.getStored(info);
        } else {
            dialog->setLabelText("Updating the FINAL FANTASY XIV Updater/Launcher version.");

            // TODO: move this out into a dedicated function, we need to use this for regular game patches later on
            // TODO: create a nice progress window like ffxivboot has
            // TODO: improve flow when updating boot, maybe do at launch ala official launcher?
            const QStringList parts = response.split(QRegExp("\n|\r\n|\r"));

            // patch list starts at line 5
            for(int i = 5; i < parts.size() - 2; i++) {
                const QStringList patchParts = parts[i].split("\t");

                const int length = patchParts[0].toInt();
                const int version = patchParts[4].toInt();
                const int hashType = patchParts[5].toInt();

                QString name = patchParts[4];
                QString url = patchParts[5];

                // TODO: show bytes recieved/total in the progress window, and speed
                dialog->setLabelText("Updating the FINAL FANTASY XIV Updater/Launcher version.\nDownloading ffxivboot - " + name);
                dialog->setMinimum(0);
                dialog->setMaximum(length);

                QNetworkRequest patchRequest(url);
                auto patchReply = window.mgr->get(patchRequest);
                connect(patchReply, &QNetworkReply::downloadProgress, [=](int recieved, int total) {
                    dialog->setValue(recieved);
                });

                connect(patchReply, &QNetworkReply::finished, [&] {
                    const QString dataDir =
                        QStandardPaths::writableLocation(QStandardPaths::TempLocation);

                    QFile file(dataDir + "/" + name + ".patch");
                    file.open(QIODevice::WriteOnly);
                    file.write(patchReply->readAll());
                    file.close();

                    // TODO: we really a dedicated .ver writing/reading class
                    QFile verFile(info.settings->gamePath + "/boot/ffxivboot.ver");
                    verFile.open(QIODevice::WriteOnly | QIODevice::Text);
                    verFile.write(name.toUtf8());
                    verFile.close();

                    processPatch((dataDir + "/" + name + ".patch").toStdString(), (info.settings->gamePath + "/boot").toStdString());

                    info.settings->bootVersion = name;

                    launcher.getStored(info);
                });
            }
        }
    });
}

void SquareBoot::checkGateStatus(LoginInformation* info) {
    QUrlQuery query;
    query.addQueryItem("", QString::number(QDateTime::currentMSecsSinceEpoch()));

    QUrl url;
    url.setUrl("https://frontier.ffxiv.com/worldStatus/gate_status.json");
    url.setQuery(query);

    QNetworkRequest request;
    request.setUrl(url);

    // TODO: really?
    window.buildRequest(*info->settings, request);

    auto reply = window.mgr->get(request);
    connect(reply, &QNetworkReply::finished, [=] {
        // I happen to run into this issue often, if I start the launcher really quickly after bootup
        // it's possible to actually check this quicker than the network is actually available,
        // causing the launcher to be stuck in "maintenace mode". so if that happens, we try to rerun this logic.
        // TODO: this selection of errors is currently guesswork, i'm assuming one of these will fit the bill of "internet is unavailable" in
        // some way.
        if(reply->error() == QNetworkReply::HostNotFoundError || reply->error() == QNetworkReply::TimeoutError || reply->error() == QNetworkReply::UnknownServerError)
            checkGateStatus(info);

        QJsonDocument document = QJsonDocument::fromJson(reply->readAll());

        const bool isGateOpen = !document.isEmpty() && document.object()["status"].toInt() != 0;

        if(isGateOpen) {
            bootCheck(*info);
        } else {
            auto messageBox = new QMessageBox(QMessageBox::Icon::Critical,
                                              "Failed to Login",
                                              "The login gate is closed, the game may be under maintenance.");

            messageBox->show();
        }
    });
}