/**
 * This file is part of GPGFrontend.
 *
 * GPGFrontend is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Foobar is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Foobar.  If not, see <https://www.gnu.org/licenses/>.
 *
 * The initial version of the source code is inherited from gpg4usb-team.
 * Their source code version also complies with GNU General Public License.
 *
 * The source code version of this software was modified and released
 * by Saturneric<eric@bktus.com> starting on May 12, 2021.
 *
 */

#include "MainWindow.h"

void MainWindow::slotAbout() {
    new AboutDialog(this);
}

void MainWindow::slotSetStatusBarText(const QString &text) {
    statusBar()->showMessage(text, 20000);
}

void MainWindow::slotStartWizard() {
    auto *wizard = new Wizard(mCtx, keyMgmt, this);
    wizard->show();
    wizard->setModal(true);
}


void MainWindow::slotCheckAttachmentFolder() {
    // TODO: always check?
    if (!settings.value("mime/parseMime").toBool()) {
        return;
    }

    QString attachmentDir = qApp->applicationDirPath() + "/attachments/";
    // filenum minus . and ..
    uint filenum = QDir(attachmentDir).count() - 2;
    if (filenum > 0) {
        QString statusText;
        if (filenum == 1) {
            statusText = tr("There is one unencrypted file in attachment folder");
        } else {
            statusText = tr("There are ") + QString::number(filenum) + tr(" unencrypted files in attachment folder");
        }
        statusBarIcon->setStatusTip(statusText);
        statusBarIcon->show();
    } else {
        statusBarIcon->hide();
    }
}

void MainWindow::slotImportKeyFromEdit() {
    if (edit->tabCount() == 0 || edit->slotCurPageTextEdit() == nullptr)
        return;
    keyMgmt->slotImportKeys(edit->curTextPage()->toPlainText().toUtf8());
}

void MainWindow::slotOpenKeyManagement() {
    keyMgmt->show();
    keyMgmt->raise();
    keyMgmt->activateWindow();
}

void MainWindow::slotOpenFileTab() {
    edit->slotNewFileTab();
}

void MainWindow::slotDisableTabActions(int number) {
    bool disable;

    if (number == -1)
        disable = true;
    else
        disable = false;

    if(edit->curFilePage() != nullptr) {
        disable = true;
    }

    printAct->setDisabled(disable);
    saveAct->setDisabled(disable);
    saveAsAct->setDisabled(disable);
    quoteAct->setDisabled(disable);
    cutAct->setDisabled(disable);
    copyAct->setDisabled(disable);
    pasteAct->setDisabled(disable);
    closeTabAct->setDisabled(disable);
    selectAllAct->setDisabled(disable);
    findAct->setDisabled(disable);
    verifyAct->setDisabled(disable);
    signAct->setDisabled(disable);
    encryptAct->setDisabled(disable);
    encryptSignAct->setDisabled(disable);
    decryptAct->setDisabled(disable);
    decryptVerifyAct->setDisabled(disable);

    redoAct->setDisabled(disable);
    undoAct->setDisabled(disable);
    zoomOutAct->setDisabled(disable);
    zoomInAct->setDisabled(disable);
    cleanDoubleLinebreaksAct->setDisabled(disable);
    quoteAct->setDisabled(disable);
    appendSelectedKeysAct->setDisabled(disable);
    importKeyFromEditAct->setDisabled(disable);

    cutPgpHeaderAct->setDisabled(disable);
    addPgpHeaderAct->setDisabled(disable);
}

void MainWindow::slotOpenSettingsDialog() {

    auto dialog = new SettingsDialog(mCtx, this);

    connect(dialog, &SettingsDialog::finished, this, [&] () -> void {

        qDebug() << "Setting Dialog Finished";

        // Iconsize
        QSize iconSize = settings.value("toolbar/iconsize", QSize(32, 32)).toSize();
        this->setIconSize(iconSize);
        importButton->setIconSize(iconSize);
        fileEncButton->setIconSize(iconSize);

        // Iconstyle
        Qt::ToolButtonStyle buttonStyle = static_cast<Qt::ToolButtonStyle>(settings.value("toolbar/iconstyle",
                                                                                          Qt::ToolButtonTextUnderIcon).toUInt());
        this->setToolButtonStyle(buttonStyle);
        importButton->setToolButtonStyle(buttonStyle);
        fileEncButton->setToolButtonStyle(buttonStyle);

        // Mime-settings
        if (settings.value("mime/parseMime").toBool()) {
            createAttachmentDock();
        } else if (attachmentDockCreated) {
            closeAttachmentDock();
        }

        // restart mainwindow if necessary
        if (getRestartNeeded()) {
            if (edit->maybeSaveAnyTab()) {
                saveSettings();
                qApp->exit(RESTART_CODE);
            }
        }

        // steganography hide/show
        if (!settings.value("advanced/steganography").toBool()) {
            this->menuBar()->removeAction(steganoMenu->menuAction());
        } else {
            this->menuBar()->insertAction(viewMenu->menuAction(), steganoMenu->menuAction());
        }
    });

}

void MainWindow::slotCleanDoubleLinebreaks() {
    if (edit->tabCount() == 0 || edit->slotCurPageTextEdit() == nullptr) {
        return;
    }

    QString content = edit->curTextPage()->toPlainText();
    content.replace("\n\n", "\n");
    edit->slotFillTextEditWithText(content);
}

void MainWindow::slotAddPgpHeader() {
    if (edit->tabCount() == 0 || edit->slotCurPageTextEdit() == nullptr) {
        return;
    }

    QString content = edit->curTextPage()->toPlainText().trimmed();

    content.prepend("\n\n").prepend(GpgConstants::PGP_CRYPT_BEGIN);
    content.append("\n").append(GpgConstants::PGP_CRYPT_END);

    edit->slotFillTextEditWithText(content);
}

void MainWindow::slotCutPgpHeader() {

    if (edit->tabCount() == 0 || edit->slotCurPageTextEdit() == nullptr) {
        return;
    }

    QString content = edit->curTextPage()->toPlainText();
    int start = content.indexOf(GpgConstants::PGP_CRYPT_BEGIN);
    int end = content.indexOf(GpgConstants::PGP_CRYPT_END);

    if (start < 0 || end < 0) {
        return;
    }

    // remove head
    int headEnd = content.indexOf("\n\n", start) + 2;
    content.remove(start, headEnd - start);

    // remove tail
    end = content.indexOf(GpgConstants::PGP_CRYPT_END);
    content.remove(end, QString(GpgConstants::PGP_CRYPT_END).size());

    edit->slotFillTextEditWithText(content.trimmed());
}

void MainWindow::slotSetRestartNeeded(bool needed) {
    this->restartNeeded = needed;
}

bool MainWindow::getRestartNeeded() const {
    return this->restartNeeded;
}