#include "generaldialog.h"
#include "ui_generaldialog.h"
#include "functions.h"
#include "reader.h"
#include "mainwindow.h"

#include <QFile>
#include <QDebug>
#include <QProcess>
#include <QScreen>
#include <QTimer>
#include <QMessageBox>
#include <QDirIterator>
#include <QStringListModel>
#include <QListView>

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

    // Preventing outside interaction
    this->setModal(true);

    // Stylesheet, style & misc.
    QFile stylesheetFile(":/resources/eink.qss");
    stylesheetFile.open(QFile::ReadOnly);
    this->setStyleSheet(stylesheetFile.readAll());
    stylesheetFile.close();

    ui->okBtn->setProperty("type", "borderless");
    ui->cancelBtn->setProperty("type", "borderless");
    ui->acceptBtn->setProperty("type", "borderless");

    ui->okBtn->setStyleSheet("font-size: 9pt; padding: 10px; font-weight: bold; background: lightGrey");
    ui->cancelBtn->setStyleSheet("font-size: 9pt; padding: 10px; font-weight: bold; background: lightGrey");
    ui->acceptBtn->setStyleSheet("font-size: 9pt; padding: 10px; font-weight: bold; background: lightGrey");
    ui->bodyLabel->setStyleSheet("font-size: 9pt");
    ui->searchComboBox->setStyleSheet("font-size: 9pt");

    if(QFile::exists("/inkbox/searchComboBoxFunction") == true) {
        string_checkconfig_ro("/inkbox/searchComboBoxFunction");
        if(checkconfig_str_val == "Dictionary") {
            ui->searchComboBox->setCurrentIndex(0);
        }
        else if(checkconfig_str_val == "Local storage") {
            ui->searchComboBox->setCurrentIndex(1);
        }
        else if(checkconfig_str_val == "Online library") {
            ui->searchComboBox->setCurrentIndex(2);
        }
        else {
            ui->searchComboBox->setCurrentIndex(0);
        }
    }

    if(checkconfig("/inkbox/resetDialog") == true) {
        if(checkconfig("/opt/inkbox_genuine") == true) {
            resetDialog = true;
            ui->okBtn->setText("Proceed");
            ui->cancelBtn->setText("Go back");
            ui->bodyLabel->setText("This will erase any books you have on the device. Settings will be reset.");
            ui->headerLabel->setText("Warning");
            this->adjustSize();
            string_writeconfig("/inkbox/resetDialog", "false");
        }
        else {
            resetDialog = true;
            ui->okBtn->setText("Proceed");
            ui->cancelBtn->setText("Go back");
            ui->bodyLabel->setText("Settings will be reset.");
            ui->headerLabel->setText("Warning");
            this->adjustSize();
            string_writeconfig("/inkbox/resetDialog", "false");
        }
    }
    else if(checkconfig("/inkbox/updateDialog") == true) {
        updateDialog = true;
        ui->okBtn->setText("Update");
        ui->cancelBtn->setText("Not now");
        ui->bodyLabel->setText("Do you want to update InkBox now?");
        ui->headerLabel->setText("Update available");
        this->adjustSize();
        string_writeconfig("/inkbox/updateDialog", "false");
    }
    else if(global::settings::settingsRebootDialog == true) {
        settingsRebootDialog = true;
        ui->stackedWidget->setCurrentIndex(1);
        if(global::kobox::koboxSettingsRebootDialog == true or global::encfs::enableStorageEncryptionDialog) {
            koboxSettingsRebootDialog = true;
            ui->bodyLabel->setText("The device will reboot now, since the settings you chose require it to work properly.");
        }
        else {
            ui->bodyLabel->setText("The settings you chose might require a complete reboot of the device for them to work properly.");
        }
        ui->headerLabel->setText("Information");
        this->adjustSize();
    }
    else if(global::mainwindow::lowBatteryDialog == true) {
        lowBatteryDialog = true;
        ui->stackedWidget->setCurrentIndex(1);
        get_battery_level();
        QString message = "The battery's level is low. Please charge your eReader.\nCurrent level: ";
        message.append(batt_level);
        ui->bodyLabel->setText(message);
        ui->headerLabel->setText("Low battery");
        this->adjustSize();
        string_writeconfig("/inkbox/lowBatteryDialog", "false");
    }
    else if(global::usbms::usbmsDialog == true) {
        usbmsDialog = true;
        ui->okBtn->setText("Connect");
        ui->cancelBtn->setText("Cancel");
        ui->bodyLabel->setText("Do you want to connect your device to a computer to manage books?");
        ui->headerLabel->setText("USB cable connected");
        this->adjustSize();
    }
    else if(global::text::textBrowserDialog == true) {
        textBrowserDialog = true;
        textwidgetWindow = new textwidget();
        ui->headerLabel->setText("Information");
        ui->stackedWidget->setCurrentIndex(1);
        ui->mainStackedWidget->insertWidget(1, textwidgetWindow);
        ui->mainStackedWidget->setCurrentIndex(1);
        this->adjustSize();
    }
    else if(global::kobox::resetKoboxDialog == true) {
        resetKoboxDialog = true;
        ui->headerLabel->setText("Warning");
        ui->okBtn->setText("Proceed");
        ui->cancelBtn->setText("Go back");
        ui->bodyLabel->setText("This will erase all KoBox user data. KoBox settings will be reset.");
        this->adjustSize();
    }
    else if(global::keyboard::keyboardDialog == true) {
        setupKeyboardDialog();
    }
    else if(global::keyboard::keypadDialog == true) {
        keypadDialog = true;
        keypadWidget = new virtualkeypad();
        ui->headerLabel->setText("Enter a number");
        ui->okBtn->setText("OK");
        ui->cancelBtn->setText("Cancel");
        ui->mainStackedWidget->insertWidget(1, keypadWidget);
        ui->mainStackedWidget->setCurrentIndex(1);
        this->adjustSize();
    }
    else if(global::encfs::disableStorageEncryptionDialog == true) {
        ui->headerLabel->setText("Warning");
        ui->okBtn->setText("Proceed");
        ui->cancelBtn->setText("Go back");
        ui->bodyLabel->setText("This will delete all the files you have encrypted. Are you sure you want to continue?");
        this->adjustSize();
    }
    else if(global::encfs::errorNoBooksInDropboxDialog == true) {
        ui->stackedWidget->setCurrentIndex(1);
        ui->headerLabel->setText("Error");
        ui->bodyLabel->setText("Please put books in the 'encfs-dropbox' folder to repack your encrypted storage.");
        this->adjustSize();
    }
    else if(global::encfs::repackDialog == true) {
        ui->headerLabel->setText("Information");
        ui->okBtn->setText("Proceed");
        ui->cancelBtn->setText("Not now");
        ui->bodyLabel->setText("New files have been found in the 'encfs-dropbox' folder. Would you want to repack your encrypted storage with them?");
        this->adjustSize();
    }
    else {
        // We shouldn't be there ;)
        ;
    }

    // Centering dialog
    QRect screenGeometry = QGuiApplication::screens()[0]->geometry();
    int x = (screenGeometry.width() - this->width()) / 2;
    int y = (screenGeometry.height() - this->height()) / 2;
    this->move(x, y);
}

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

void generalDialog::on_cancelBtn_clicked()
{
    if(updateDialog == true) {
        string_writeconfig("/tmp/cancelUpdateDialog", "true");
        generalDialog::close();
    }
    else if(usbmsDialog == true) {
        global::usbms::usbmsDialog = false;
        generalDialog::close();
    }
    else if(resetKoboxDialog == true) {
        global::kobox::resetKoboxDialog = false;
        generalDialog::close();
    }
    else if(global::encfs::disableStorageEncryptionDialog == true) {
        emit cancelDisableStorageEncryption();
        global::encfs::disableStorageEncryptionDialog = false;
        generalDialog::close();
    }
    else {
        if(global::keyboard::searchDialog == true) {
            global::keyboard::searchDialog = false;
            global::forbidOpenSearchDialog = true;
            global::keyboard::keyboardDialog = false;
            global::keyboard::keyboardText = "";
            global::library::librarySearchDialog = false;
        }
        else if(global::keyboard::vncDialog == true) {
            global::keyboard::vncDialog = false;
            global::keyboard::keyboardDialog = false;
            global::keyboard::keyboardText = "";
        }
        else if(global::keyboard::encfsDialog == true) {
            global::keyboard::encfsDialog = false;
            global::encfs::cancelSetup = true;
            global::keyboard::keyboardText = "";
        }
        else if(global::encfs::errorNoBooksInDropboxDialog == true) {
            global::encfs::errorNoBooksInDropboxDialog = false;
        }
        else if(global::encfs::repackDialog == true) {
            global::encfs::repackDialog = false;
        }
        generalDialog::close();
    }
}

void generalDialog::on_okBtn_clicked()
{
    if(resetDialog == true) {
        if(checkconfig("/opt/inkbox_genuine") == true) {
            // Soft-reset the device
            // We set a custom boot flag and reboot silently in Diagnostics
            string_writeconfig("/external_root/boot/flags/DIAGS_BOOT", "true");
            string_writeconfig("/external_root/boot/flags/DO_SOFT_RESET", "true");
            reboot(false);
        }
        else {
            // Restore default settings, we're not on InkBox OS
            QString prog ("sh");
            QStringList args;
            args << "reset-config.sh";
            QProcess *proc = new QProcess();
            proc->start(prog, args);
            proc->waitForFinished();
            proc->deleteLater();

            // Relaunching InkBox
            QProcess process;
            process.startDetached("inkbox.sh", QStringList());
            qApp->quit();
        }
    }
    if(updateDialog == true) {
        if(global::otaUpdate::isUpdateOta == true) {
            this->hide();

            global::otaUpdate::downloadOta = true;
            otaManagerWindow = new otaManager(this);
            connect(otaManagerWindow, SIGNAL(downloadedOtaUpdate(bool)), SLOT(startOtaUpdate(bool)));
            otaManagerWindow->setAttribute(Qt::WA_DeleteOnClose);

            global::toast::indefiniteToast = true;
            global::toast::modalToast = true;
            emit showToast("Downloading update");
        }
        else {
            installUpdate();
        }
    }
    if(usbmsDialog == true) {
        global::usbms::usbmsDialog = false;
        global::usbms::launchUsbms = true;

        usbmsWindow = new usbms_splash();
        usbmsWindow->setAttribute(Qt::WA_DeleteOnClose);
        usbmsWindow->setGeometry(QRect(QPoint(0,0), screen()->geometry ().size()));
        usbmsWindow->show();
    }
    if(resetKoboxDialog == true) {
        resetKoboxUserData();
    }
    if(keypadDialog == true) {
        int number = global::keyboard::keypadText.toInt();
        emit gotoPageSelected(number);
        global::keyboard::keypadDialog = false;
        generalDialog::close();
    }
    if(keyboardDialog == true) {
        if(global::keyboard::searchDialog == true) {
            if(!global::keyboard::keyboardText.isEmpty()) {
                if(ui->searchComboBox->currentText() == "Dictionary") {
                    string_writeconfig("/inkbox/searchComboBoxFunction", "Dictionary");
                    for(int i = ui->mainStackedWidget->count(); i >= 0; i--) {
                        QWidget * widget = ui->mainStackedWidget->widget(i);
                        ui->mainStackedWidget->removeWidget(widget);
                        widget->deleteLater();
                    }
                    ui->topStackedWidget->setVisible(false);
                    ui->stackedWidget->setVisible(false);
                    dictionaryWidgetWindow = new dictionaryWidget();
                    dictionaryWidgetWindow->setAttribute(Qt::WA_DeleteOnClose);
                    connect(dictionaryWidgetWindow, SIGNAL(refreshScreen()), SLOT(refreshScreenNative()));
                    connect(dictionaryWidgetWindow, SIGNAL(destroyed(QObject*)), SLOT(restartSearchDialog()));
                    ui->mainStackedWidget->insertWidget(1, dictionaryWidgetWindow);
                }
                else if(ui->searchComboBox->currentText() == "Local storage") {
                    string_writeconfig("/inkbox/searchComboBoxFunction", "Local storage");
                    QString onboardPath;
                    QStringList storageSearchResults;
                    if(checkconfig("/opt/inkbox_genuine") == true) {
                        onboardPath = "/mnt/onboard/onboard/";
                    }
                    else {
                        onboardPath = "/mnt/onboard/";
                    }
                    QDirIterator dirIt(onboardPath, QDirIterator::Subdirectories);
                    while(dirIt.hasNext()) {
                        dirIt.next();
                        if(QFileInfo(dirIt.filePath()).isFile()) {
                            QString suffix = QFileInfo(dirIt.filePath()).suffix();
                            if(suffix == "txt" or suffix == "TXT" or suffix == "epub" or suffix == "pdf" or suffix == "PDF") {
                                if(dirIt.fileName().contains(global::keyboard::keyboardText) == true) {
                                    storageSearchResults.append(dirIt.fileName());
                                }
                            }
                        }
                    }
                    if(!storageSearchResults.isEmpty()) {
                        for(int i = ui->mainStackedWidget->count(); i >= 0; i--) {
                            QWidget * widget = ui->mainStackedWidget->widget(i);
                            ui->mainStackedWidget->removeWidget(widget);
                            widget->deleteLater();
                        }
                        ui->topStackedWidget->setVisible(false);
                        ui->stackedWidget->setVisible(false);
                        searchResultsWidgetWindow = new searchResultsWidget(this);
                        searchResultsWidgetWindow->setAttribute(Qt::WA_DeleteOnClose);
                        global::forbidOpenSearchDialog = true;
                        connect(searchResultsWidgetWindow, SIGNAL(destroyed(QObject*)), SLOT(restartSearchDialog()));
                        connect(searchResultsWidgetWindow, SIGNAL(openBookFile(QString, bool)), SLOT(openBookFileNative(QString, bool)));
                        searchResultsWidgetWindow->setListViewContents(storageSearchResults);
                        ui->mainStackedWidget->insertWidget(1, searchResultsWidgetWindow);
                    }
                    else {
                        global::toast::delay = 3000;
                        emit showToast("No results found");
                        keyboardWidget->clearLineEdit();
                        global::keyboard::keyboardText = "";
                    }
                }
                else if(ui->searchComboBox->currentText() == "Online library") {
                    string_writeconfig("/inkbox/searchComboBoxFunction", "Online library");
                    string_writeconfig("/inkbox/gutenberg_search_request", global::keyboard::keyboardText.toStdString());
                    string_writeconfig("/opt/ibxd", "gutenberg_search\n");
                    while(true) {
                        if(QFile::exists("/inkbox/gutenberg-search/search_done")) {
                            if(checkconfig("/inkbox/gutenberg-search/search_done") == true) {
                                QStringList searchResults = readFile("/inkbox/gutenberg-search/search_results_titles").split("\n");
                                global::library::libraryResults = true;

                                for(int i = ui->mainStackedWidget->count(); i >= 0; i--) {
                                    QWidget * widget = ui->mainStackedWidget->widget(i);
                                    ui->mainStackedWidget->removeWidget(widget);
                                    widget->deleteLater();
                                }
                                ui->topStackedWidget->setVisible(false);
                                ui->stackedWidget->setVisible(false);
                                searchResultsWidgetWindow = new searchResultsWidget(this);
                                searchResultsWidgetWindow->setAttribute(Qt::WA_DeleteOnClose);
                                global::forbidOpenSearchDialog = true;
                                connect(searchResultsWidgetWindow, SIGNAL(destroyed(QObject*)), SLOT(restartSearchDialog()));
                                connect(searchResultsWidgetWindow, SIGNAL(showToast(QString)), SLOT(showToastNative(QString)));
                                connect(searchResultsWidgetWindow, SIGNAL(closeIndefiniteToast()), SLOT(closeIndefiniteToastNative()));
                                connect(searchResultsWidgetWindow, SIGNAL(hideDialog()), SLOT(hide()));
                                searchResultsWidgetWindow->setListViewContents(searchResults);
                                ui->mainStackedWidget->insertWidget(1, searchResultsWidgetWindow);
                                QFile::remove("/inkbox/gutenberg-search/search_done");
                                break;
                            }
                            else {
                                global::toast::delay = 3000;
                                emit showToast("No results found");
                                keyboardWidget->clearLineEdit();
                                global::keyboard::keyboardText = "";
                                QFile::remove("/inkbox/gutenberg-search/search_done");
                                break;
                            }
                        }
                    }
                }
                else {
                    ;
                }
            }
            else {
                QMessageBox::critical(this, tr("Invalid argument"), tr("Please type in a search term."));
            }
        }
        else if(global::keyboard::vncDialog == true) {
            if(!global::keyboard::keyboardText.isEmpty()) {
                if(vncServerSet != true) {
                    vncServerAddress = global::keyboard::keyboardText;
                    vncServerSet = true;
                    keyboardWidget->clearLineEdit();
                    ui->headerLabel->setText("Enter the server's password");
                }
                else if(vncPasswordSet != true) {
                    vncServerPassword = global::keyboard::keyboardText;
                    vncPasswordSet = true;
                    keyboardWidget->clearLineEdit();
                    ui->headerLabel->setText("Enter the server's port");
                }
                else {
                    vncServerPort = global::keyboard::keyboardText;
                    global::keyboard::vncDialog = false;
                    global::keyboard::keyboardDialog = false;
                    startVNC(vncServerAddress, vncServerPassword, vncServerPort);
                    generalDialog::close();
                }
            }
            else {
                QMessageBox::critical(this, tr("Invalid argument"), tr("Please type in the required argument."));
            }
        }
        else if(global::keyboard::wifiPassphraseDialog == true) {
            if(!global::keyboard::keyboardText.isEmpty()) {
                this->hide();
                wifiPassphrase = global::keyboard::keyboardText;
                global::toast::indefiniteToast = true;
                global::toast::modalToast = true;
                emit showToast("Connecting");
                QTimer::singleShot(100, this, SLOT(connectToNetworkSlot()));
                global::keyboard::wifiPassphraseDialog = false;
                global::keyboard::keyboardDialog = false;
            }
            else {
                QMessageBox::critical(this, tr("Invalid argument"), tr("Please type in the required argument."));
            }
        }
        else if(global::keyboard::encfsDialog == true) {
            if(!global::keyboard::keyboardText.isEmpty()) {
                global::encfs::passphrase = global::keyboard::keyboardText;
                global::keyboard::encfsDialog = false;
                global::keyboard::keyboardText = "";
                global::keyboard::keyboardDialog = false;
                this->close();
            }
            else {
                QMessageBox::critical(this, tr("Invalid argument"), tr("Please type in the required argument."));
            }
        }
        else {
            global::keyboard::keyboardDialog = false;
            generalDialog::close();
        }
    }
    if(global::encfs::disableStorageEncryptionDialog == true) {
        global::encfs::disableStorageEncryptionDialog = false;
        emit disableStorageEncryption();
        this->close();
    }
    if(global::encfs::repackDialog == true) {
        global::encfs::repackDialog = false;
        string_writeconfig("/external_root/run/encfs_repack", "true");
        quit_restart();
    }
}
void generalDialog::on_acceptBtn_clicked()
{
    if(lowBatteryDialog == true) {
        global::mainwindow::lowBatteryDialog = false;
        global::battery::batteryAlertLock = false;
        global::battery::showLowBatteryDialog = false;
    }

    if(settingsRebootDialog == true) {
        if(koboxSettingsRebootDialog == true or global::encfs::enableStorageEncryptionDialog) {
            reboot(true);
        }
        else {
            QProcess process;
            process.startDetached("inkbox.sh", QStringList());
            qApp->quit();
        }
    }
    if(textBrowserDialog == true) {
        global::text::textBrowserContents = "";
        global::text::textBrowserDialog = false;
    }

    // We don't have any other option ;p
    generalDialog::close();
}

void generalDialog::adjust_size() {
    this->adjustSize();
    QRect screenGeometry = QGuiApplication::screens()[0]->geometry();
    int x = (screenGeometry.width() - this->width()) / 2;
    int y = (screenGeometry.height() - this->height()) / 2;
    this->move(x, y);
    emit refreshScreen();
}

void generalDialog::restartSearchDialog() {
    generalDialog::close();
}

void generalDialog::setupKeyboardDialog() {
    keyboardDialog = true;
    ui->stackedWidget->setVisible(true);
    if(global::keyboard::searchDialog == true) {
        ui->topStackedWidget->setCurrentIndex(1);
        ui->searchHeaderLabel->setText("Search");
        ui->okBtn->setText("Search");
        ui->cancelBtn->setText("Close");
    }
    else if(global::keyboard::vncDialog == true) {
        ui->headerLabel->setText("Enter the server's IP address");
        ui->okBtn->setText("OK");
        ui->cancelBtn->setText("Cancel");
    }
    else if(global::keyboard::wifiPassphraseDialog == true) {
        ui->headerLabel->setText("Enter the network's passphrase");
        ui->okBtn->setText("Connect");
        ui->cancelBtn->setText("Cancel");
    }
    else if(global::keyboard::encfsDialog == true) {
        ui->headerLabel->setText("Enter your encrypted storage's passphrase");
        ui->okBtn->setText("OK");
        ui->cancelBtn->setText("Cancel");
    }
    else {
        ui->headerLabel->setText("Enter a string");
        ui->okBtn->setText("OK");
        ui->cancelBtn->setText("Cancel");
    }
    keyboardWidget = new virtualkeyboard();
    connect(keyboardWidget, SIGNAL(adjust_size()), SLOT(adjust_size()));
    ui->mainStackedWidget->insertWidget(1, keyboardWidget);
    ui->mainStackedWidget->setCurrentIndex(1);
    QTimer::singleShot(1000, this, SLOT(adjust_size()));
}

void generalDialog::refreshScreenNative() {
    emit refreshScreen();
}

void generalDialog::startVNC(QString server, QString password, QString port) {
    std::string server_str = server.toStdString();
    std::string password_str = password.toStdString();
    std::string port_str = port.toStdString();
    string_writeconfig("/external_root/tmp/app_vnc_server", server_str);
    string_writeconfig("/external_root/tmp/app_vnc_password", password_str);
    string_writeconfig("/external_root/tmp/app_vnc_port", port_str);
    string_writeconfig("/opt/ibxd", "app_start_vnc\n");
    qApp->quit();
}

void generalDialog::connectToNetworkSlot() {
    if(connectToNetwork(wifiEssid, wifiPassphrase) == true) {
        emit updateWifiIcon(3);
        emit closeIndefiniteToast();
        emit showToast("Connection successful");
    }
    else {
        emit updateWifiIcon(2);
        emit closeIndefiniteToast();
        emit showToast("Connection failed");
    }
    generalDialog::close();
}

void generalDialog::startOtaUpdate(bool wasDownloadSuccessful) {
    emit closeIndefiniteToast();
    if(wasDownloadSuccessful == true) {
        global::otaUpdate::isUpdateOta = false;
        installUpdate();
    }
    else {
        emit showToast("Download failed");
        global::otaUpdate::isUpdateOta = false;
    }
    generalDialog::close();
}

void generalDialog::openBookFileNative(QString book, bool relativePath) {
    emit openBookFile(book, relativePath);
}

void generalDialog::showToastNative(QString messageToDisplay) {
    emit showToast(messageToDisplay);
}

void generalDialog::closeIndefiniteToastNative() {
    emit closeIndefiniteToast();
}

void generalDialog::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();
}