#include "usbms_splash.h"
#include "ui_usbms_splash.h"

#include <QPixmap>
#include <QScreen>
#include <QTimer>

#include "functions.h"

usbms_splash::usbms_splash(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::usbms_splash)
{
    ui->setupUi(this);
    usbms_splash::setFont(QFont("u001"));
    ui->label->setFont(QFont("Inter"));

    // Getting the screen's size
    sW = QGuiApplication::screens()[0]->size().width();
    sH = QGuiApplication::screens()[0]->size().height();

    // Defining what the default icon size will be
    if(global::kobox::showKoboxSplash == true) {
        float stdIconWidth = sW / 1.30;
        float stdIconHeight = sH / 1.30;

        // Stylesheet
        QFile stylesheetFile("/mnt/onboard/.adds/inkbox/eink.qss");
        stylesheetFile.open(QFile::ReadOnly);
        this->setStyleSheet(stylesheetFile.readAll());
        stylesheetFile.close();

        ui->label->setFont(QFont("Inter"));
        ui->label->setText("Launching KoBox subsystem");
        ui->label->setStyleSheet("font-size: 14pt; font-weight: bold");
        ui->label_3->setText("Please wait, this could take a while.");
        if(global::deviceID == "n905\n" or global::deviceID == "kt\n") {
            ui->label_3->setStyleSheet("font-size: 11pt");
        }
        else {
            ui->label_3->setStyleSheet("font-size: 10pt");
        }

        QPixmap pixmap(":/resources/kobox-icon.png");
        QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
        ui->label_2->setPixmap(scaledPixmap);
    }
    else {
        float stdIconWidth = sW / 1.15;
        float stdIconHeight = sH / 1.15;

        this->setStyleSheet("background-color:black;");
        ui->label->setStyleSheet("QLabel { background-color : black; color : white; font-size: 15pt; font-weight: bold }");
        ui->label_3->setStyleSheet("QLabel { background-color : black; color : white; font-size: 10pt }");

        QPixmap pixmap(":/resources/usbms.png");
        QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
        ui->label_2->setPixmap(scaledPixmap);
    }

    if(global::usbms::launchUsbms == true) {
        global::usbms::launchUsbms = false;
        usbms_launch();
    }
}

void usbms_splash::usbms_launch()
{
    log("Entering USBMS session", className);
    string_writeconfig("/tmp/in_usbms", "true");
    QTimer::singleShot(1500, this, SLOT(brightnessDown()));

    if(global::usbms::koboxExportExtensions == true) {
        string_writeconfig("/opt/ibxd", "kobox_extensions_storage_unmount\n");
    }
    if(checkconfig("/external_root/run/encfs_mounted") == true) {
        string_writeconfig("/external_root/run/encfs_stop_cleanup", "true");
        string_writeconfig("/opt/ibxd", "encfs_stop\n");
        QThread::msleep(1500);
    }

    string_writeconfig("/opt/ibxd", "onboard_unmount\n");
    QThread::msleep(500);

    string_writeconfig("/opt/ibxd", "usbnet_stop\n");
    QThread::msleep(1000);

    string_writeconfig("/opt/ibxd", "gui_apps_stop\n");
    QThread::msleep(1000);

    if(global::deviceID == "n306\n" or global::deviceID == "n873\n") {
        QProcess::execute("insmod", QStringList() << "/external_root/lib/modules/fs/configfs/configfs.ko");
        QProcess::execute("insmod", QStringList() << "/external_root/lib/modules/drivers/usb/gadget/libcomposite.ko");
        QProcess::execute("insmod", QStringList() << "/external_root/lib/modules/drivers/usb/gadget/function/usb_f_mass_storage.ko");
    }
    else if(global::deviceID == "kt\n") {
        QProcess::execute("insmod", QStringList() << "/external_root/lib/modules/2.6.35-inkbox/kernel/drivers/usb/gadget/arcotg_udc.ko");
    }

    QString prog_1 ("insmod");
    QStringList args_1;

    if(global::deviceID == "kt\n") {
        massStorageModule = "/external_root/lib/modules/2.6.35-inkbox/kernel/drivers/usb/gadget/g_file_storage.ko";
    }
    else {
        massStorageModule = "/external_root/lib/modules/g_mass_storage.ko";
    }

    if(global::usbms::koboxExportExtensions == true) {
        args_1 << massStorageModule << "file=/external_root/opt/storage/X11/extensions-user.img" << "removable=y" << "stall=0";
    }
    else {
        args_1 << massStorageModule << "file=/external_root/opt/storage/onboard" << "removable=y" << "stall=0";
    }

    QProcess *proc_1 = new QProcess();
    proc_1->start(prog_1, args_1);
    proc_1->waitForFinished();
    proc_1->deleteLater();

    bool exitUsbMsDone = false;
    QTimer *usbms_t = new QTimer(this);
    usbms_t->setInterval(1000);
    connect(usbms_t, &QTimer::timeout, [&]() {
        if(exitUsbMsDone == false) {
            if(isUsbPluggedIn() == 0) {
                if(global::usbms::koboxExportExtensions == true) {
                    reboot(false);
                    qApp->quit();
                }
                else {
                    log("Exiting USBMS session", className);
                    // '<font/>' bit: because nothing else works ...
                    ui->label->setText("<font face='Inter'>Processing content</font>");
                    ui->label->setFont(QFont("Inter"));
                    ui->label_3->setText("Please wait");
                    ui->label_3->setFont(QFont("u001"));
                    ui->label->setStyleSheet("QLabel { background-color : black; color : white; font-size: 15pt; font-weight: bold }");
                    ui->label_3->setStyleSheet("QLabel { background-color : black; color : white; font-size: 11pt }");
                    ui->label->setFont(QFont("u001"));

                    float stdIconWidth = sW / 2;
                    float stdIconHeight = sH / 2;

                    QPixmap pixmap(":/resources/clock-inverted.png");
                    QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
                    ui->label_2->setPixmap(scaledPixmap);

                    this->repaint();

                    QString prog("rmmod");
                    QStringList args;
                    args << massStorageModule;
                    QProcess * proc = new QProcess();
                    proc->start(prog, args);
                    proc->waitForFinished();
                    proc->deleteLater();

                    QTimer::singleShot(500, this, SLOT(restartServices()));
                }
                exitUsbMsDone = true;
            }
            else {
                ;
            }
        }
    } );
    usbms_t->start();
}

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

void usbms_splash::brightnessDown() {
    cinematicBrightness(0, 1);
}

void usbms_splash::quit_restart() {
    // If existing, cleaning bookconfig_mount mountpoint
    string_writeconfig("/opt/ibxd", "bookconfig_unmount\n");

    // Restarting InkBox
    QProcess process;
    process.startDetached("inkbox", QStringList());
    qApp->quit();
}

void usbms_splash::restartServices() {
    // Restarting USBNet
    // NOTE: USBNet is only started if required conditions are met (see https://github.com/Kobo-InkBox/rootfs/blob/master/etc/init.d/usbnet)
    string_writeconfig("/opt/ibxd", "usbnet_start\n");
    QThread::msleep(1000);
    // Mounting onboard storage
    string_writeconfig("/opt/ibxd", "onboard_mount\n");
    QThread::msleep(1000);
    // Checking for updates
    string_writeconfig("/opt/ibxd", "update_inkbox_restart\n");
    QThread::msleep(2500);
    string_writeconfig("/tmp/in_usbms", "false");
    // GUI apps: update main JSON file
    string_writeconfig("/opt/ibxd", "gui_apps_start\n");
    while(true) {
        if(QFile::exists("/tmp/gui_apps_started")) {
            if(checkconfig("/tmp/gui_apps_started") == true) {
                QFile::remove("/tmp/gui_apps_started");
                updateUserAppsMainJsonFile();
                break;
            }
            else {
                log("GUI apps service failed to start", className);
                QFile::remove("/tmp/gui_apps_started");
                break;
            }
        }
    }
    // Re-generate local library on next launch
    QFile::remove(global::localLibrary::databasePath);

    quit_restart();
}