#include "alert.h"
#include "ui_alert.h"
#include "functions.h"

#include <QPixmap>
#include <QScreen>
#include <QFile>
#include <QProcess>

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

    // Getting the screen's size
    float sW = QGuiApplication::screens()[0]->size().width();
    float sH = QGuiApplication::screens()[0]->size().height();
    // Defining what the "Alert" icon size will be
    float stdIconWidth = sW / 1.7;
    float stdIconHeight = sH / 1.7;

    // General stylesheet
    QFile stylesheetFile(":/resources/eink-dark.qss");
    stylesheetFile.open(QFile::ReadOnly);
    this->setStyleSheet(stylesheetFile.readAll());
    stylesheetFile.close();

    // Default icon in case none of the conditions below are met
    QPixmap pixmap(":/resources/alert.png");
    QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
    ui->alertIconLabel->setPixmap(scaledPixmap);

    // Checking if the update's signature is untrusted. The signature error will always take precedence over the downgrade one (c.f. update.sh script)
    if(checkconfig("/external_root/boot/flags/ALERT_SIGN") == true) {
        QPixmap pixmap(":/resources/alert.png");
        QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
        ui->alertIconLabel->setPixmap(scaledPixmap);

        signatureError = true;
        ui->securityLabel->setText("Failed to update InkBox.");
        ui->messageLabel->setText("The digital signature of the update is untrusted.\nFor security reasons, it cannot be installed.");
        ui->stackedWidget->setCurrentIndex(1);
    }
    if(checkconfig("/external_root/boot/flags/ALERT_DOWNGRADE") == true) {
        QPixmap pixmap(":/resources/alert.png");
        QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
        ui->alertIconLabel->setPixmap(scaledPixmap);

        downgradeError = true;
        ui->securityLabel->setText("Failed to update InkBox.");
        ui->messageLabel->setText("An error occured during the update process.\nThe update package's version is lower than the actual installed version.");
        ui->stackedWidget->setCurrentIndex(1);
    }
    if(global::encfs::lockdown == true) {
        ui->stackedWidget->setVisible(false);
        ui->stackedWidget->deleteLater();
        QPixmap pixmap(":/resources/alert.png");
        QPixmap scaledPixmap = pixmap.scaled(stdIconWidth, stdIconHeight, Qt::KeepAspectRatio);
        ui->alertIconLabel->setPixmap(scaledPixmap);

        ui->warningLabel->setText("Fatal error");
        ui->securityLabel->setText("Device lockdown");
        QString message = "Due to multiple incorrect passphrase attempts, this device is locked down until\n" + global::encfs::unlockTime + "\nand won't boot.";
        ui->messageLabel->setText(message);
    }
    if(global::battery::showCriticalBatteryAlert == true) {
        global::battery::showCriticalBatteryAlert = false;
        ui->stackedWidget->setVisible(false);
        ui->stackedWidget->deleteLater();

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

        criticalBattery = true;
        ui->warningLabel->setText("Please charge your eReader.");
        ui->securityLabel->setText("The battery's charge level is critical.");
        ui->messageLabel->setText("To prevent damage, your device has been turned off.\nPlease consider charging it.");
        poweroff(false);
    }

    ui->warningLabel->setStyleSheet("QLabel { background-color : black; color : white; font-size: 16pt}");
    ui->messageLabel->setStyleSheet("QLabel { background-color : black; color : white; font-size: 9pt}");
    ui->securityLabel->setStyleSheet("QLabel { background-color : black; color : white; font-size: 11pt}");
    ui->continueBtn->setProperty("type", "borderless");
    ui->resetBtn->setProperty("type", "borderless");
    ui->continueBtn->setStyleSheet("padding: 20px");
    ui->resetBtn->setStyleSheet("padding: 20px");
    ui->continue2Btn->setProperty("type", "borderless");
    ui->continue2Btn->setStyleSheet("padding: 20px");
}

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

void alert::on_continueBtn_clicked()
{
    // We continue anyway and re-set the ALERT flag
    string_writeconfig("/external_root/boot/flags/ALERT", "false");
    string_writeconfig("/external_root/boot/flags/ALERT_SIGN", "false");
    updateReset();
    QProcess process;
    process.startDetached("inkbox", QStringList());
    qApp->quit();
}

void alert::on_resetBtn_clicked()
{
    // We set the DO_FACTORY_RESET flag and we restart the Kobo
    string_writeconfig("/external_root/boot/flags/DO_FACTORY_RESET", "true");
    string_writeconfig("/external_root/boot/flags/DIAGS_BOOT", "true");
    QString reboot_prog ("/sbin/reboot");
    QStringList reboot_args;
    reboot_args << "no_splash";
    QProcess *reboot_proc = new QProcess();
    reboot_proc->start(reboot_prog, reboot_args);
    reboot_proc->waitForFinished();
    reboot_proc->deleteLater();
}

void alert::on_continue2Btn_clicked()
{
    // We continue anyway and re-set the ALERT flag
    string_writeconfig("/external_root/boot/flags/ALERT", "false");
    updateReset();

    if(signatureError == true) {
        string_writeconfig("/external_root/boot/flags/ALERT_SIGN", "false");
        QProcess process;
        process.startDetached("inkbox", QStringList());
        qApp->quit();
    }
    if(downgradeError == true) {
        string_writeconfig("/external_root/boot/flags/ALERT_DOWNGRADE", "false");
        QProcess process;
        process.startDetached("inkbox", QStringList());
        qApp->quit();
    }
}

void alert::updateReset() {
    string_writeconfig("/mnt/onboard/onboard/.inkbox/can_really_update", "false");
    string_writeconfig("/mnt/onboard/onboard/.inkbox/can_update", "false");
}