/*
 * Copyright (C) 2020-2022 Roy Qu (royqh1979@gmail.com)
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.
 */
#include "compilersetoptionwidget.h"
#include "ui_compilersetoptionwidget.h"
#include "../settings.h"
#include "../mainwindow.h"
#include "compilersetdirectorieswidget.h"
#include <QMessageBox>
#include <QProgressDialog>
#include "../utils.h"
#include "../iconsmanager.h"
#include "../systemconsts.h"
#include <qt_utils/charsetinfo.h>
#include <QDebug>
#include <QFileDialog>
#include <QInputDialog>

CompilerSetOptionWidget::CompilerSetOptionWidget(const QString& name, const QString& group, QWidget* parent) :
    SettingsWidget(name,group,parent),
    ui(new Ui::CompilerSetOptionWidget)
{
    ui->setupUi(this);

    mBinDirWidget = new CompilerSetDirectoriesWidget();
    ui->dirTabs->addTab(mBinDirWidget,QObject::tr("Binaries"));
    mLibDirWidget = new CompilerSetDirectoriesWidget();
    ui->dirTabs->addTab(mLibDirWidget,QObject::tr("Libraries"));
    mCIncludeDirWidget = new CompilerSetDirectoriesWidget();
    ui->dirTabs->addTab(mCIncludeDirWidget,QObject::tr("C Includes"));
    mCppIncludeDirWidget = new CompilerSetDirectoriesWidget();
    ui->dirTabs->addTab(mCppIncludeDirWidget,QObject::tr("C++ Includes"));

    connect(ui->chkUseCustomCompilerParams, &QCheckBox::stateChanged,
             ui->txtCustomCompileParams, &QPlainTextEdit::setEnabled);
    connect(ui->chkUseCustomLinkParams, &QCheckBox::stateChanged,
             ui->txtCustomLinkParams, &QPlainTextEdit::setEnabled);

    updateIcons(pIconsManager->actionIconSize());
#ifdef Q_OS_WIN
    ui->txtExecutableSuffix->setReadOnly(true);
#endif
    ui->settingTabs->setCurrentWidget(ui->tabGeneral);
}

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

void CompilerSetOptionWidget::init()
{
    ui->cbEncodingDetails->setVisible(false);
    ui->cbEncoding->clear();
    ui->cbEncoding->addItem(tr("ANSI"),ENCODING_SYSTEM_DEFAULT);
    ui->cbEncoding->addItem(tr("UTF-8"),ENCODING_UTF8);
    foreach (const QString& langName, pCharsetInfoManager->languageNames()) {
        ui->cbEncoding->addItem(langName,langName);
    }
    SettingsWidget::init();
}


static void loadCompilerSetSettings(Settings::PCompilerSet pSet, Ui::CompilerSetOptionWidget* ui) {
    bool supportCharset = CompilerInfoManager::supportCovertingCharset(pSet->compilerType());
    ui->chkAutoAddCharset->setEnabled(supportCharset);
    ui->cbEncoding->setEnabled(supportCharset);
    ui->cbEncodingDetails->setEnabled(supportCharset);
    ui->panelCharset->setVisible(supportCharset);
    ui->chkAutoAddCharset->setEnabled(supportCharset);
    ui->chkAutoAddCharset->setVisible(supportCharset);

    bool supportStaticLink = CompilerInfoManager::supportStaticLink(pSet->compilerType());
    ui->chkStaticLink->setEnabled(supportStaticLink);
    ui->chkStaticLink->setVisible(supportStaticLink);

    ui->chkUseCustomCompilerParams->setChecked(pSet->useCustomCompileParams());
    ui->txtCustomCompileParams->setPlainText(pSet->customCompileParams());
    ui->txtCustomCompileParams->setEnabled(pSet->useCustomCompileParams());
    ui->chkUseCustomLinkParams->setChecked(pSet->useCustomLinkParams());
    ui->txtCustomLinkParams->setPlainText(pSet->customLinkParams());
    ui->txtCustomLinkParams->setEnabled(pSet->useCustomLinkParams());
    ui->chkAutoAddCharset->setChecked(pSet->autoAddCharsetParams());
    ui->chkStaticLink->setChecked(pSet->staticLink());
    ui->chkPersistInAutoFind->setChecked(pSet->persistInAutoFind());
    ui->chkForceEnglishOutput->setChecked(pSet->forceEnglishOutput());
    //rest tabs in the options widget

    ui->optionTabs->resetUI(pSet,pSet->compileOptions());

    ui->txtCCompiler->setText(pSet->CCompiler());
    ui->txtCppCompiler->setText(pSet->cppCompiler());
    ui->txtMake->setText(pSet->make());
    ui->txtDebugger->setText(pSet->debugger());
    ui->txtGDBServer->setText(pSet->debugServer());
    ui->txtResourceCompiler->setText(pSet->resourceCompiler());

    if (pSet->execCharset() == ENCODING_AUTO_DETECT
            || pSet->execCharset() == ENCODING_SYSTEM_DEFAULT
            || pSet->execCharset() == ENCODING_UTF8) {
        int index =ui->cbEncoding->findData(pSet->execCharset());
        ui->cbEncoding->setCurrentIndex(index);
        ui->cbEncodingDetails->clear();
        ui->cbEncodingDetails->setVisible(false);
    } else {
        QString encoding = pSet->execCharset();
        QString language = pCharsetInfoManager->findLanguageByCharsetName(encoding);
        ui->cbEncoding->setCurrentText(language);
        ui->cbEncodingDetails->setVisible(true);
        ui->cbEncodingDetails->clear();
        QList<PCharsetInfo> infos = pCharsetInfoManager->findCharsetsByLanguageName(language);
        foreach (const PCharsetInfo& info, infos) {
            ui->cbEncodingDetails->addItem(info->name);
        }
        ui->cbEncodingDetails->setCurrentText(encoding);
    }

    ui->txtPreprocessingSuffix->setText(pSet->preprocessingSuffix());
    ui->txtCompilationSuffix->setText(pSet->compilationProperSuffix());
    ui->txtExecutableSuffix->setText(pSet->executableSuffix());
    switch(pSet->compilationStage()) {
    case Settings::CompilerSet::CompilationStage::PreprocessingOnly:
        ui->rbPreprocessingOnly->setChecked(true);
        break;
    case Settings::CompilerSet::CompilationStage::CompilationProperOnly:
        ui->rbCompilationProperOnly->setChecked(true);
        break;
    default:
        ui->rbGenerateExecutable->setChecked(true);
    }
#ifdef ENABLE_SDCC
    bool isSDCC = (pSet->compilerType()==CompilerType::SDCC);
    ui->grpCompilationStages->setVisible(!isSDCC);
    ui->lbPreprocessingSuffix->setVisible(!isSDCC);
    ui->txtPreprocessingSuffix->setVisible(!isSDCC);
    ui->lbCompilingSuffix->setVisible(!isSDCC);
    ui->txtCompilationSuffix->setVisible(!isSDCC);
    ui->lbExecSuffix->setVisible(!isSDCC);
    ui->txtExecutableSuffix->setVisible(!isSDCC);
    ui->lbBinarySuffix->setVisible(isSDCC);
    ui->cbBinarySuffix->setVisible(isSDCC);
    if (isSDCC) {
        ui->cbBinarySuffix->clear();
        ui->cbBinarySuffix->addItem(SDCC_IHX_SUFFIX);
        ui->cbBinarySuffix->addItem(SDCC_HEX_SUFFIX);
        ui->cbBinarySuffix->addItem(SDCC_BIN_SUFFIX);
        ui->cbBinarySuffix->setCurrentText(pSet->executableSuffix());
    } else {
        ui->txtExecutableSuffix->setText(pSet->executableSuffix());
    }
#else
    ui->grpCompilationStages->setVisible(true);
    ui->lbPreprocessingSuffix->setVisible(true);
    ui->txtPreprocessingSuffix->setVisible(true);
    ui->lbCompilingSuffix->setVisible(true);
    ui->txtCompilationSuffix->setVisible(true);
    ui->lbExecSuffix->setVisible(true);
    ui->txtExecutableSuffix->setVisible(true);
    ui->lbBinarySuffix->setVisible(false);
    ui->cbBinarySuffix->setVisible(false);
    ui->txtExecutableSuffix->setText(pSet->executableSuffix());
#endif
}

void CompilerSetOptionWidget::doLoad()
{
    disconnectInputs();
    ui->cbCompilerSet->clear();
    if (pSettings->compilerSets().size()<=0) {
        ui->btnRenameCompilerSet->setEnabled(false);
        ui->btnRemoveCompilerSet->setEnabled(false);
        return;
    } else {
        ui->btnRenameCompilerSet->setEnabled(true);
        ui->btnRemoveCompilerSet->setEnabled(true);
    }
    int index=pSettings->compilerSets().defaultIndex();
    for (size_t i=0;i<pSettings->compilerSets().size();i++) {
        ui->cbCompilerSet->addItem(pSettings->compilerSets().getSet(i)->name());
    }
    if (index < 0 || index>=ui->cbCompilerSet->count()) {
        index = 0;
    }
    ui->cbCompilerSet->setCurrentIndex(index);
    reloadCurrentCompilerSet();
    connectInputs();
}

void CompilerSetOptionWidget::doSave()
{
    if (pSettings->compilerSets().size()>0) {
        saveCurrentCompilerSet();
    }
    //update default index timestamp
    pSettings->compilerSets().setDefaultIndex(pSettings->compilerSets().defaultIndex());
    pSettings->compilerSets().saveSets();
    pMainWindow->updateCompilerSet();
}

void CompilerSetOptionWidget::on_cbCompilerSet_currentIndexChanged(int index)
{
    if (index<0)
        return;
    setSettingsChanged();
    pSettings->compilerSets().setDefaultIndex(index);
    disconnectInputs();
    reloadCurrentCompilerSet();
    connectInputs();
}

void CompilerSetOptionWidget::reloadCurrentCompilerSet()
{
    Settings::PCompilerSet pSet = pSettings->compilerSets().defaultSet();
    loadCompilerSetSettings(pSet, ui);

    mBinDirWidget->setDirList(pSet->binDirs());
    mLibDirWidget->setDirList(pSet->libDirs());
    mCIncludeDirWidget->setDirList(pSet->CIncludeDirs());
    mCppIncludeDirWidget->setDirList(pSet->CppIncludeDirs());
}

void CompilerSetOptionWidget::saveCurrentCompilerSet()
{
    Settings::PCompilerSet pSet = pSettings->compilerSets().defaultSet();

    pSet->setUseCustomCompileParams(ui->chkUseCustomCompilerParams->isChecked());
    pSet->setCustomCompileParams(ui->txtCustomCompileParams->toPlainText().trimmed());
    pSet->setUseCustomLinkParams(ui->chkUseCustomLinkParams->isChecked());
    pSet->setCustomLinkParams(ui->txtCustomLinkParams->toPlainText().trimmed());
    pSet->setAutoAddCharsetParams(ui->chkAutoAddCharset->isChecked());
    pSet->setStaticLink(ui->chkStaticLink->isChecked());
    pSet->setPersistInAutoFind(ui->chkPersistInAutoFind->isChecked());
    pSet->setForceEnglishOutput(ui->chkForceEnglishOutput->isChecked());


    pSet->setCCompiler(ui->txtCCompiler->text().trimmed());
    pSet->setCppCompiler(ui->txtCppCompiler->text().trimmed());
    pSet->setMake(ui->txtMake->text().trimmed());
    pSet->setDebugger(ui->txtDebugger->text().trimmed());
    pSet->setDebugServer(ui->txtGDBServer->text().trimmed());
    pSet->setResourceCompiler(ui->txtResourceCompiler->text().trimmed());

    pSet->binDirs()=mBinDirWidget->dirList();

    pSet->libDirs()=mLibDirWidget->dirList();
    pSet->CIncludeDirs()=mCIncludeDirWidget->dirList();
    pSet->CppIncludeDirs()=mCppIncludeDirWidget->dirList();

    if (ui->cbEncodingDetails->isVisible()) {
        pSet->setExecCharset(ui->cbEncodingDetails->currentText());
    } else {
        pSet->setExecCharset(ui->cbEncoding->currentData().toString());
    }

    //read values in the options widget
    pSet->setCompileOptions(ui->optionTabs->arguments(false));

    if (ui->rbPreprocessingOnly->isChecked()) {
        pSet->setCompilationStage(Settings::CompilerSet::CompilationStage::PreprocessingOnly);
    } else if (ui->rbCompilationProperOnly->isChecked()) {
        pSet->setCompilationStage(Settings::CompilerSet::CompilationStage::CompilationProperOnly);
    } else if (ui->rbGenerateExecutable->isChecked()) {
        pSet->setCompilationStage(Settings::CompilerSet::CompilationStage::GenerateExecutable);
    }
    pSet->setPreprocessingSuffix(ui->txtPreprocessingSuffix->text());
    pSet->setCompilationProperSuffix(ui->txtCompilationSuffix->text());
#ifdef ENABLE_SDCC
    if (pSet->compilerType()==CompilerType::SDCC) {
        pSet->setExecutableSuffix(ui->cbBinarySuffix->currentText());
    } else {
        pSet->setExecutableSuffix(ui->txtExecutableSuffix->text());
    }
#else
    pSet->setExecutableSuffix(ui->txtExecutableSuffix->text());
#endif
}

QString CompilerSetOptionWidget::getBinDir()
{
    Settings::PCompilerSet pSet = pSettings->compilerSets().defaultSet();
    if (!pSet->binDirs().isEmpty())
        return pSet->binDirs().front();
    if (!mBinDirWidget->dirList().isEmpty())
        return mBinDirWidget->dirList().front();
    return QDir().absolutePath();
}

void CompilerSetOptionWidget::on_btnFindCompilers_clicked()
{
#ifdef Q_OS_WIN
        QString msg = tr("Red Panda C++ will clear previously found compiler list and search"
                      " for compilers in the following locations:<br /> '%1'<br /> '%2'<br />Do you really want to continue?")
                                 .arg(includeTrailingPathDelimiter(pSettings->dirs().appDir()) + "MinGW32")
                                 .arg(includeTrailingPathDelimiter(pSettings->dirs().appDir()) + "MinGW64");
#else
        QString msg = tr("Red Panda C++ will clear previously found compiler list and search"
                      " for compilers in the the PATH. <br />Do you really want to continue?");
#endif
    if (QMessageBox::warning(this,tr("Confirm"),msg,
                                 QMessageBox::Ok | QMessageBox::Cancel) != QMessageBox::Ok )
        return;
    QProgressDialog progressDlg(
                tr("Searching for compilers..."),
                tr("Abort"),
                0,
                1,
                pMainWindow);

    progressDlg.setWindowModality(Qt::WindowModal);
    progressDlg.setMaximum(2);
    progressDlg.setLabelText(tr("Searching..."));
    pSettings->compilerSets().findSets();
    progressDlg.setValue(1);
    doLoad();
    progressDlg.setValue(2);
    setSettingsChanged();
    if (pSettings->compilerSets().size()==0) {
        QMessageBox::warning(this,tr("Failed"),tr("Can't find any compiler."));
    }
}

void CompilerSetOptionWidget::on_btnAddBlankCompilerSet_clicked()
{
    QString name = QInputDialog::getText(this,tr("Compiler Set Name"),tr("Name"));
    Settings::PCompilerSet set = pSettings->compilerSets().addSet();
    pSettings->compilerSets().setDefaultIndex(pSettings->compilerSets().size()-1);
    set->setName(name);
    set->setPersistInAutoFind(true);
    doLoad();
}

void CompilerSetOptionWidget::on_btnAddCompilerSetByFolder_clicked()
{
    QString folder = QFileDialog::getExistingDirectory(this, tr("Compiler Set Folder"));
    int oldSize = pSettings->compilerSets().size();

    if (!pSettings->compilerSets().addSets(folder)) {
        pSettings->compilerSets().addSets(folder+QDir::separator()+"bin");
    }
    doLoad();
    int newSize = pSettings->compilerSets().size();
    if (oldSize == newSize) {
        QMessageBox::warning(this,tr("Failed"),tr("Can't find any compiler."));
    }
}

void CompilerSetOptionWidget::on_btnRenameCompilerSet_clicked()
{
    Settings::PCompilerSet set=pSettings->compilerSets().getSet(ui->cbCompilerSet->currentIndex());
    if (!set)
        return;
    QString name = QInputDialog::getText(this,tr("Compiler Set Name"),tr("New name"),QLineEdit::Normal,
                                         set->name());
    if (!name.isEmpty())
        set->setName(name);
    doLoad();
}

void CompilerSetOptionWidget::on_btnRemoveCompilerSet_clicked()
{
    Settings::PCompilerSet set=pSettings->compilerSets().getSet(ui->cbCompilerSet->currentIndex());
    if (!set)
        return;
    if (QMessageBox::question(this,
                              QObject::tr("Remove"),
                              QString(QObject::tr("Do you really want to remove \"%1\"?")).arg(set->name()),
                              QMessageBox::Yes|QMessageBox::No|QMessageBox::Cancel)==QMessageBox::Yes) {

        pSettings->compilerSets().deleteSet(ui->cbCompilerSet->currentIndex());
        doLoad();
    }
}

void CompilerSetOptionWidget::updateIcons(const QSize& /*size*/)
{
    pIconsManager->setIcon(ui->btnFindCompilers, IconsManager::ACTION_EDIT_SEARCH);
    pIconsManager->setIcon(ui->btnAddCompilerSetByFolder, IconsManager::ACTION_FILE_OPEN_FOLDER);
    pIconsManager->setIcon(ui->btnAddBlankCompilerSet, IconsManager::ACTION_MISC_ADD);
    pIconsManager->setIcon(ui->btnRemoveCompilerSet, IconsManager::ACTION_MISC_REMOVE);
    pIconsManager->setIcon(ui->btnRenameCompilerSet, IconsManager::ACTION_MISC_RENAME);

    pIconsManager->setIcon(ui->btnChooseCCompiler, IconsManager::ACTION_FILE_OPEN_FOLDER);
    pIconsManager->setIcon(ui->btnChooseCppCompiler, IconsManager::ACTION_FILE_OPEN_FOLDER);
    pIconsManager->setIcon(ui->btnChooseGDB, IconsManager::ACTION_FILE_OPEN_FOLDER);
    pIconsManager->setIcon(ui->btnChooseGDBServer, IconsManager::ACTION_FILE_OPEN_FOLDER);
    pIconsManager->setIcon(ui->btnChooseMake, IconsManager::ACTION_FILE_OPEN_FOLDER);
    pIconsManager->setIcon(ui->btnChooseResourceCompiler, IconsManager::ACTION_FILE_OPEN_FOLDER);
}

void CompilerSetOptionWidget::on_cbEncoding_currentTextChanged(const QString &/*arg1*/)
{
    QString userData = ui->cbEncoding->currentData().toString();
    if (userData == ENCODING_AUTO_DETECT
            || userData == ENCODING_SYSTEM_DEFAULT
            || userData == ENCODING_UTF8) {
        ui->cbEncodingDetails->setVisible(false);
        ui->cbEncodingDetails->clear();
    } else {
        ui->cbEncodingDetails->setVisible(true);
        ui->cbEncodingDetails->clear();
        QList<PCharsetInfo> infos = pCharsetInfoManager->findCharsetsByLanguageName(userData);
        foreach (const PCharsetInfo& info, infos) {
            ui->cbEncodingDetails->addItem(info->name);
        }
    }
}


void CompilerSetOptionWidget::on_cbEncodingDetails_currentTextChanged(const QString &/*arg1*/)
{

}


void CompilerSetOptionWidget::on_btnChooseCCompiler_clicked()
{
    QString fileName = QFileDialog::getOpenFileName(
                this,
                tr("Locate C Compiler"),
                getBinDir(),
                tr("Executable files (*.exe)"));
    if (fileExists(fileName))
        ui->txtCCompiler->setText(fileName);
}


void CompilerSetOptionWidget::on_btnChooseCppCompiler_clicked()
{
    QString fileName = QFileDialog::getOpenFileName(
                this,
                tr("Locate C++ Compiler"),
                getBinDir(),
                tr("Executable files (*.exe)"));
    if (fileExists(fileName))
        ui->txtCppCompiler->setText(fileName);
}


void CompilerSetOptionWidget::on_btnChooseMake_clicked()
{
    QString fileName = QFileDialog::getOpenFileName(
                this,
                tr("Locate Make"),
                getBinDir(),
                tr("Executable files (*.exe)"));
    if (fileExists(fileName))
        ui->txtMake->setText(fileName);
}


void CompilerSetOptionWidget::on_btnChooseGDB_clicked()
{
    QString fileName = QFileDialog::getOpenFileName(
                this,
                tr("Locate GDB"),
                getBinDir(),
                tr("Executable files (*.exe)"));
    if (fileExists(fileName))
        ui->txtDebugger->setText(fileName);
}


void CompilerSetOptionWidget::on_btnChooseGDBServer_clicked()
{
    QString fileName = QFileDialog::getOpenFileName(
                this,
                tr("Locate GDB Server"),
                getBinDir(),
                tr("Executable files (*.exe)"));
    if (fileExists(fileName))
        ui->txtGDBServer->setText(fileName);
}


void CompilerSetOptionWidget::on_btnChooseResourceCompiler_clicked()
{
    QString fileName = QFileDialog::getOpenFileName(
                this,
                tr("Locate windres"),
                getBinDir(),
                tr("Executable files (*.exe)"));
    if (fileExists(fileName))
        ui->txtResourceCompiler->setText(fileName);
}