#include "optionsdialog.h"
#include "mainwindow.h"
#include "ui_optionsdialog.h"
#include "newfiledialog.h"

#include <QSettings>
#include <QFileInfo>

OptionsDialog::OptionsDialog(QWidget *parent):
    QDialog(parent),
    ui(new Ui::OptionsDialog)
{
    this->core = new QRCore();
    this->anal_level = 0;

    ui->setupUi(this);
    ui->progressBar->setVisible(0);
    ui->statusLabel->setVisible(0);

    // Fill the plugins combo
    QStringList plugins;
    for (auto i: this->core->getList ("asm", "plugins")) {
        this->asm_plugins.append(i);
        plugins.append(i);
    }
    ui->processorComboBox->insertItems(1, plugins);

    // Restore settings
    QSettings settings("iaito", "iaito");
    ui->bytesCheckBox->setChecked(settings.value("bytes").toBool());
    ui->attCheckBox->setChecked(settings.value("syntax").toBool());
    ui->descriptionCheckBox->setChecked(settings.value("describe").toBool());
    ui->stackCheckBox->setChecked(settings.value("stackptr").toBool());
    ui->ucaseCheckBox->setChecked(settings.value("ucase").toBool());
    ui->spacyCheckBox->setChecked(settings.value("spacy").toBool());

    ui->hideFrame->setVisible(false);

    // Add this so the dialog resizes when widgets are shown/hidden
    //this->layout()->setSizeConstraint(QLayout::SetFixedSize);
}

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

void OptionsDialog::setFilename(QString fn, QString shortfn) {
    this->filename = fn;
    this->shortfn = shortfn;
    //qDebug() << QFileInfo(fn).fileName();
    ui->programLineEdit->setText(fn);
    this->core->tryFile (fn, 1);
}

void OptionsDialog::on_closeButton_clicked()
{
    close();
}

void OptionsDialog::on_okButton_clicked()
{
    this->setEnabled(0);
    ui->logo->setEnabled(true);

    // Show Progress Bar
    ui->progressBar->setEnabled(1);
    ui->statusLabel->setEnabled(1);
    ui->progressBar->setVisible(1);
    ui->statusLabel->setVisible(1);

    ui->statusLabel->setText("Starting analysis");
    //ui->progressBar->setValue(5);

    // Close dialog and open OptionsDialog
    this->w = new MainWindow(this);
    this->w->core = this->core;

    // Fill asm plugins in hexdump combo
    this->w->memoryDock->fillPlugins(this->asm_plugins);

    bool va = ui->vaCheckBox->isChecked();
    ut64 loadaddr = 0LL;
    ut64 mapaddr = 0LL;
    int bits = 0;
    QString sel_bits = ui->bitsComboBox->currentText();
    if (sel_bits != "Auto") {
        bits = sel_bits.toInt();
    }

    // Save options in settings
    QSettings settings("iaito", "iaito");

    // Show asm bytes
    if (ui->bytesCheckBox->isChecked()) {
        this->w->core->config("asm.bytes", "true");
        this->w->core->config("asm.cmtcol", "100");
    } else {
        this->w->core->config("asm.bytes", "false");
        this->w->core->config("asm.cmtcol", "70");
    }
    settings.setValue("bytes", ui->bytesCheckBox->isChecked());

    // Show AT&T syntax
    if (ui->attCheckBox->isChecked()) {
        this->w->core->config("asm.syntax", "att");
    } else {
        this->w->core->config("asm.syntax", "intel");
    }
    settings.setValue("syntax", ui->attCheckBox->isChecked());

    // Show opcode description
    if (ui->descriptionCheckBox->isChecked()) {
        this->w->core->config("asm.describe", "true");
    } else {
        this->w->core->config("asm.describe", "false");
    }
    settings.setValue("describe", ui->descriptionCheckBox->isChecked());

    // Show stack pointer
    if (ui->stackCheckBox->isChecked()) {
        this->w->core->config("asm.stackptr", "true");
    } else {
        this->w->core->config("asm.stackptr", "false");
    }
    settings.setValue("stackptr", ui->stackCheckBox->isChecked());

    // Show uppercase dasm
    if (ui->ucaseCheckBox->isChecked()) {
        this->w->core->config("asm.ucase", "true");
    } else {
        this->w->core->config("asm.ucase", "false");
    }
    settings.setValue("ucase", ui->ucaseCheckBox->isChecked());

    // Show spaces in dasm
    if (ui->spacyCheckBox->isChecked()) {
        this->w->core->config("asm.spacy", "true");
    } else {
        this->w->core->config("asm.spacy", "false");
    }
    settings.setValue("spacy", ui->spacyCheckBox->isChecked());

    bool rw = false;
    bool load_bininfo = ui->binCheckBox->isChecked();

    if (load_bininfo) {
        if (!va) {
            va = 2;
            loadaddr = UT64_MAX;
            r_config_set_i (this->core->core->config, "bin.laddr", loadaddr);
            mapaddr = 0;
        }
    } else {
        va = false;
        loadaddr = mapaddr = 0;
    }

    //ui->progressBar->setValue(20);
    ui->statusLabel->setText("Loading binary");
    // options dialog should show the list of archs inside the given fatbin
    int binidx = 0; // index of subbin

    anal_level = ui->analCheckBox->isChecked();
    this->w->add_output(" > Loading file: " + this->filename);
    this->w->core->loadFile(this->filename.toUtf8(), loadaddr, mapaddr, rw, va, bits, binidx, load_bininfo);
    //ui->progressBar->setValue(40);
    ui->statusLabel->setText("Analysis in progress");

    // Threads stuff
    // create an instance of MyThread
    this->analThread = new AnalThread(w);

    // connect signal/slot
    connect(analThread, SIGNAL(finished()), this, SLOT(anal_finished()));
    //analThread->level = anal_level;
    if (anal_level == true) {
        analThread->level = ui->analSlider->value();
    } else {
        analThread->level = 0;
    }
    analThread->start();
}

void OptionsDialog::anal_finished()
{
    // Get opcodes
    this->w->core->getOpcodes();

    //fprintf(stderr, "anal done");
    //ui->progressBar->setValue(70);
    this->w->core->cmd("Po " + this->shortfn);
    ui->statusLabel->setText("Loading interface");
    this->w->add_output(" > Analysis finished");
    QString initial_seek = ui->entry_initialSeek->text();
    if (initial_seek.length()>0) {
        this->w->core->seek(initial_seek);
    } else {
        this->w->core->seek("entry0");
    }
    this->w->add_output(" > Populating UI");
    this->w->updateFrames();
    this->w->setFilename(this->filename);
    this->w->get_refs( this->w->core->cmd("?v entry0") );
    this->w->memoryDock->selectHexPreview();

    // Restore project notes
    QString notes = this->core->cmd("Pn");
    //qDebug() << "Notes:" << notes;
    if (notes != "") {
        QByteArray ba;
        ba.append(notes);
        this->w->notepadDock->notesTextEdit->setPlainText(QByteArray::fromBase64(ba));
    }

    //Get binary beginning/end addresses
    this->core->binStart = this->core->cmd("?v $M");
    this->core->binEnd = this->core->cmd("?v $M+$s");

    this->w->add_output(" > Finished, happy reversing :)");
    // Add fortune message
    this->w->add_output( "\n" + this->w->core->cmd("fo") );
    this->w->memoryDock->setWindowTitle("entry0");
    this->w->start_web_server();
    close();
    this->w->showMaximized();
    // Initialize syntax highlighters
    this->w->memoryDock->highlightDisasms();
    this->w->notepadDock->highlightPreview();
}

void OptionsDialog::on_cancelButton_clicked()
{
    delete this->core;
    this->core = NULL; // implicit by delete?
    // Close dialog and open OptionsDialog
    close();
    NewFileDialog* n = new NewFileDialog(this);
    n->show();
}

void OptionsDialog::on_analSlider_valueChanged(int value)
{
    ui->analLevel->setText(QString::number(value));
    if (value == 0) {
        ui->analCheckBox->setChecked(false);
    } else {
        ui->analCheckBox->setChecked(true);
    }
}

void OptionsDialog::on_AdvOptButton_clicked()
{
    if(ui->AdvOptButton->isChecked())
    {
        ui->hideFrame->setVisible(true);
        ui->AdvOptButton->setArrowType(Qt::DownArrow);
    } else {
        ui->hideFrame->setVisible(false);
        ui->AdvOptButton->setArrowType(Qt::RightArrow);

    }
}