cutter/src/mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "createnewdialog.h"
#include "dialogs/commentsdialog.h"
#include "dialogs/aboutdialog.h"
#include "dialogs/renamedialog.h"
#include "helpers.h"

#include <qfont.h>
#include <qsettings.h>
#include <qlabel.h>
#include <qfile.h>
#include <qmessagebox.h>
#include <qfontdialog.h>
#include <qcompleter.h>
#include <qstringlistmodel.h>
#include <QScrollBar>
#include <QDir>
#include <QDebug>
#include <QDesktopServices>
#include <QFileDialog>
#include <QPropertyAnimation>
#include <QStyledItemDelegate>
#include <QProcess>

#include <QtGlobal>
#include <QToolTip>
#include <QShortcut>
#include <QTextCursor>
#include <QMessageBox>
#include <QStyleFactory>

#include <QLabel>
#include <QComboBox>
#include <QTreeWidgetItem>
#include <QDockWidget>
#include <QLineEdit>
#include <QSettings>
#include <QList>

#include "highlighter.h"
#include "hexascii_highlighter.h"
#include "newfiledialog.h"
#include "helpers.h"

#include "widgets/memorywidget.h"
#include "widgets/functionswidget.h"
#include "widgets/sectionswidget.h"
#include "widgets/commentswidget.h"
#include "widgets/importswidget.h"
#include "widgets/symbolswidget.h"
#include "widgets/stringswidget.h"
#include "widgets/sectionsdock.h"
#include "widgets/relocswidget.h"
#include "widgets/flagswidget.h"
#include "widgets/codegraphic.h"
#include "widgets/dashboard.h"
#include "widgets/notepad.h"
#include "widgets/sidebar.h"
#include "widgets/sdbdock.h"
#include "widgets/omnibar.h"

// graphics
#include <QGraphicsEllipseItem>
#include <QGraphicsScene>
#include <QGraphicsView>

#include <cassert>

static void registerCustomFonts()
{
    int ret = QFontDatabase::addApplicationFont(":/new/prefix1/fonts/Anonymous Pro.ttf");
    assert(-1 != ret && "unable to register Anonymous Pro.ttf");

    ret = QFontDatabase::addApplicationFont(":/new/prefix1/fonts/Inconsolata-Regular.ttf");
    assert(-1 != ret && "unable to register Inconsolata-Regular.ttf");

    // do not issue a warning in release
    Q_UNUSED(ret)
}

MainWindow::MainWindow(QWidget *parent, QRCore *kore) :
    QMainWindow(parent),
    core(kore),
    memoryDock(nullptr),
    notepadDock(nullptr),
    asmDock(nullptr),
    calcDock(nullptr),
    omnibar(nullptr),
    sideBar(nullptr),
    ui(new Ui::MainWindow),
    highlighter(nullptr),
    hex_highlighter(nullptr),
    graphicsBar(nullptr),
    functionsDock(nullptr),
    importsDock(nullptr),
    symbolsDock(nullptr),
    relocsDock(nullptr),
    commentsDock(nullptr),
    stringsDock(nullptr),
    flagsDock(nullptr),
    dashboardDock(nullptr),
    gotoEntry(nullptr),
    sdbDock(nullptr),
    sidebar_action(nullptr),
    sectionsDock(nullptr),
    webserverThread(core, this)
{
    this->start_web_server();
    ui->setupUi(this);

    doLock = false;
    this->cursor_address = core->getOffset();

    registerCustomFonts();


    /*
    * Toolbar
    */
    // Hide central tab widget tabs
    QTabBar *centralbar = ui->centralTabWidget->tabBar();
    centralbar->setVisible(false);
    // Adjust console lineedit
    ui->consoleInputLineEdit->setTextMargins(10, 0, 0, 0);
    /*
    ui->consoleOutputTextEdit->setFont(QFont("Monospace", 8));
    ui->consoleOutputTextEdit->setStyleSheet("background-color:black;color:gray;");
    ui->consoleInputLineEdit->setStyleSheet("background-color:black;color:gray;");
    */

    // Adjust text margins of consoleOutputTextEdit
    QTextDocument *console_docu = ui->consoleOutputTextEdit->document();
    console_docu->setDocumentMargin(10);

    // Sepparator between back/forward and undo/redo buttons
    QWidget *spacer4 = new QWidget();
    spacer4->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Expanding);
    spacer4->setMinimumSize(10, 10);
    ui->mainToolBar->insertWidget(ui->actionForward, spacer4);

    // Popup menu on theme toolbar button
    QToolButton *backButton = new QToolButton(this);
    backButton->setIcon(QIcon(":/new/prefix1/img/icons/arrow_left.png"));
    //backButton->setPopupMode(QToolButton::DelayedPopup);
    ui->mainToolBar->insertWidget(ui->actionForward, backButton);
    connect(backButton, SIGNAL(clicked()), this, SLOT(on_backButton_clicked()));

    // Sepparator between undo/redo and goto lineEdit
    QWidget *spacer3 = new QWidget();
    spacer3->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
    spacer3->setMinimumSize(20, 20);
    spacer3->setMaximumWidth(300);
    ui->mainToolBar->insertWidget(ui->actionShow_Hide_mainsidebar, spacer3);

    // Omnibar LineEdit
    this->omnibar = new Omnibar(this);
    ui->mainToolBar->insertWidget(ui->actionShow_Hide_mainsidebar, this->omnibar);

    // Add special sepparators to the toolbar that expand to separate groups of elements
    QWidget *spacer2 = new QWidget();
    spacer2->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
    spacer2->setMinimumSize(10, 10);
    spacer2->setMaximumWidth(300);
    ui->mainToolBar->insertWidget(ui->actionShow_Hide_mainsidebar, spacer2);

    // Sepparator between back/forward and undo/redo buttons
    QWidget *spacer = new QWidget();
    spacer->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Expanding);
    spacer->setMinimumSize(20, 20);
    ui->mainToolBar->addWidget(spacer);

    // codeGraphics tool bar
    this->graphicsBar = new GraphicsBar(this);
    this->graphicsBar->setMovable(false);
    addToolBarBreak(Qt::TopToolBarArea);
    addToolBar(graphicsBar);

    // Fix output panel font
    qhelpers::normalizeFont(ui->consoleOutputTextEdit);

    /*
     * Dock Widgets
     */

    dockWidgets.reserve(11);

    // Add Memory DockWidget
    this->memoryDock = new MemoryWidget(this);
    dockWidgets.push_back(memoryDock);
    // To use in the future when we handle more than one memory views
    // this->memoryDock->setAttribute(Qt::WA_DeleteOnClose);
    // this->add_debug_output( QString::number(this->dockList.length()) );

    // Add Sections dock panel
    this->sectionsDock = new SectionsDock(this);
    dockWidgets.push_back(sectionsDock);

    // Add functions DockWidget
    this->functionsDock = new FunctionsWidget(this);
    dockWidgets.push_back(functionsDock);

    // Add imports DockWidget
    this->importsDock = new ImportsWidget(this);
    dockWidgets.push_back(importsDock);

    // Add symbols DockWidget
    this->symbolsDock = new SymbolsWidget(this);
    dockWidgets.push_back(symbolsDock);

    // Add relocs DockWidget
    this->relocsDock = new RelocsWidget(this);
    dockWidgets.push_back(relocsDock);

    // Add comments DockWidget
    this->commentsDock = new CommentsWidget(this);
    dockWidgets.push_back(commentsDock);

    // Add strings DockWidget
    this->stringsDock = new StringsWidget(this);
    dockWidgets.push_back(stringsDock);

    // Add flags DockWidget
    this->flagsDock = new FlagsWidget(this);
    dockWidgets.push_back(flagsDock);

    // Add Notepad Dock panel
    this->notepadDock = new Notepad(this);
    dockWidgets.push_back(notepadDock);
    connect(memoryDock, SIGNAL(fontChanged(QFont)), notepadDock, SLOT(setFonts(QFont)));

    //Add Dashboard Dock panel
    this->dashboardDock = new Dashboard(this);
    dockWidgets.push_back(dashboardDock);

    // Set up dock widgets default layout
    restoreDocks();
    hideAllDocks();
    showDefaultDocks();

    // Restore saved settings
    this->readSettings();
    // TODO: Allow the user to select this option visually in the GUI settings
    // Adjust the DockWidget areas
    setCorner(Qt::TopLeftCorner, Qt::LeftDockWidgetArea);
    //setCorner( Qt::TopRightCorner, Qt::RightDockWidgetArea );
    setCorner(Qt::BottomLeftCorner, Qt::LeftDockWidgetArea);
    //setCorner( Qt::BottomRightCorner, Qt::RightDockWidgetArea );

    // Set console output context menu
    ui->consoleOutputTextEdit->setContextMenuPolicy(Qt::CustomContextMenu);
    connect(ui->consoleOutputTextEdit, SIGNAL(customContextMenuRequested(const QPoint &)),
            this, SLOT(showConsoleContextMenu(const QPoint &)));

    // Setup and hide sidebar by default
    this->sideBar = new SideBar(this);
    this->sidebar_action = ui->sideToolBar->addWidget(this->sideBar);
    ui->sideToolBar->hide();

    // Show dashboard by default
    this->dashboardDock->raise();

    //qDebug() << "FOLDER: " << QStandardPaths::writableLocation(QStandardPaths::GenericDataLocation);
    /*
     *  Some global shortcuts
     */
    // Period goes to command entry
    QShortcut *cmd_shortcut = new QShortcut(QKeySequence(Qt::Key_Period), this);
    connect(cmd_shortcut, SIGNAL(activated()), ui->consoleInputLineEdit, SLOT(setFocus()));

    // G and S goes to goto entry
    QShortcut *goto_shortcut = new QShortcut(QKeySequence(Qt::Key_G), this);
    connect(goto_shortcut, SIGNAL(activated()), this->omnibar, SLOT(setFocus()));
    QShortcut *seek_shortcut = new QShortcut(QKeySequence(Qt::Key_S), this);
    connect(seek_shortcut, SIGNAL(activated()), this->omnibar, SLOT(setFocus()));

    // : goes to goto entry
    QShortcut *commands_shortcut = new QShortcut(QKeySequence(Qt::Key_Colon), this);
    connect(commands_shortcut, SIGNAL(activated()), this->omnibar, SLOT(showCommands()));

    connect(&webserverThread, SIGNAL(finished()), this, SLOT(webserverThreadFinished()));

    QShortcut *refresh_shortcut = new QShortcut(QKeySequence(QKeySequence::Refresh), this);
    connect(refresh_shortcut, SIGNAL(activated()), this, SLOT(refreshVisibleDockWidgets()));
}

MainWindow::~MainWindow()
{
    delete ui;
    delete core;
}

void MainWindow::start_web_server()
{
    // Start web server
    webserverThread.startServer();
}

void MainWindow::webserverThreadFinished()
{
    core->core()->http_up = webserverThread.isStarted() ? R_TRUE : R_FALSE;

    // this is not true anymore, cause the webserver might have been stopped
    //if (core->core->http_up == R_FALSE) {
    //    eprintf("FAILED TO LAUNCH\n");
    //}
}

void MainWindow::setWebServerState(bool start)
{
    if (start)
    {
        webserverThread.startServer();

        // Open web interface on default browser
        // ballessay: well isn't this possible with =H&
        //QString link = "http://localhost:9090/";
        //QDesktopServices::openUrl(QUrl(link));
    }
    else
    {
        webserverThread.stopServer();
    }
}

void MainWindow::raiseMemoryDock()
{
    memoryDock->raise();
}

void MainWindow::toggleSideBarTheme()
{
    sideBar->themesButtonToggle();
}

void MainWindow::refreshOmniBar(const QStringList &flags)
{
    omnibar->refresh(flags);
}

void MainWindow::setFilename(QString fn)
{
    // Add file name to window title
    this->filename = fn;
    this->setWindowTitle("Iaito - " + fn);
}

void MainWindow::showConsoleContextMenu(const QPoint &pt)
{
    // Set console output popup menu
    QMenu *menu = ui->consoleOutputTextEdit->createStandardContextMenu();
    menu->clear();
    menu->addAction(ui->actionClear_ConsoleOutput);
    menu->addAction(ui->actionConsoleSync_with_core);
    ui->actionConsoleSync_with_core->setChecked(true);
    ui->consoleOutputTextEdit->setContextMenuPolicy(Qt::CustomContextMenu);

    menu->exec(ui->consoleOutputTextEdit->mapToGlobal(pt));
    delete menu;
}

void MainWindow::closeEvent(QCloseEvent *event)
{
    QMessageBox::StandardButton ret = QMessageBox::question(this, "Iaito",
                                      "Do you really want to exit?\nSave your project before closing!",
                                      QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel);
    //qDebug() << ret;
    if (ret == QMessageBox::Save)
    {
        QSettings settings;
        settings.setValue("geometry", saveGeometry());
        settings.setValue("size", size());
        settings.setValue("pos", pos());
        settings.setValue("state", saveState());
        core->cmd("Ps " + qhelpers::uniqueProjectName(filename));
        QString notes = this->notepadDock->textToBase64();
        //this->add_debug_output(notes);
        this->core->cmd("Pnj " + notes);
        QMainWindow::closeEvent(event);
    }
    else if (ret == QMessageBox::Discard)
    {
        QSettings settings;
        settings.setValue("geometry", saveGeometry());
        settings.setValue("size", size());
        settings.setValue("pos", pos());
        settings.setValue("state", saveState());
    }
    else
    {
        event->ignore();
    }
}

void MainWindow::readSettings()
{
    QSettings settings;
    QByteArray geo = settings.value("geometry", QByteArray()).toByteArray();
    restoreGeometry(geo);
    QByteArray state = settings.value("state", QByteArray()).toByteArray();
    restoreState(state);
    if (settings.value("dark").toBool())
    {
        this->dark();
    }
    this->responsive = settings.value("responsive").toBool();
}

void MainWindow::dark()
{
    qApp->setStyleSheet("QPlainTextEdit { background-color: rgb(64, 64, 64); color: rgb(222, 222, 222);} QTextEdit { background-color: rgb(64, 64, 64); color: rgb(222, 222, 222);} ");
    this->memoryDock->switchTheme(true);
    QSettings settings;
    settings.setValue("dark", true);
}

void MainWindow::def_theme()
{
    qApp->setStyleSheet("");
    this->memoryDock->switchTheme(false);
    QSettings settings;
    settings.setValue("dark", false);
}

/*
 * Refresh widget functions
 */

void MainWindow::refreshFunctions()
{
    functionsDock->refresh();
}

void MainWindow::refreshComments()
{
    commentsDock->refresh();
}

void MainWindow::updateFrames()
{
    if (core == NULL)
        return;

    static bool first_time = true;

    if (first_time)
    {
        for (auto w : dockWidgets)
        {
            w->setup();
        }

        first_time = false;
    }
    else
    {
        for (auto w : dockWidgets)
        {
            w->refresh();
        }
    }

    // graphicsBar->refreshColorBar();
    graphicsBar->fillData();
}

void MainWindow::on_actionLock_triggered()
{
    doLock = !doLock;
    if (doLock)
    {
        foreach (QDockWidget *dockWidget, findChildren<QDockWidget *>())
        {
            dockWidget->setFeatures(QDockWidget::NoDockWidgetFeatures);
        }
    }
    else
    {
        foreach (QDockWidget *dockWidget, findChildren<QDockWidget *>())
        {
            dockWidget->setFeatures(QDockWidget::AllDockWidgetFeatures);
        }
    }
}

void MainWindow::lockUnlock_Docks(bool what)
{
    if (what)
    {
        foreach (QDockWidget *dockWidget, findChildren<QDockWidget *>())
        {
            dockWidget->setFeatures(QDockWidget::NoDockWidgetFeatures);
        }
    }
    else
    {
        foreach (QDockWidget *dockWidget, findChildren<QDockWidget *>())
        {
            dockWidget->setFeatures(QDockWidget::AllDockWidgetFeatures);
        }
    }

}

void MainWindow::on_actionLockUnlock_triggered()
{
    if (ui->actionLockUnlock->isChecked())
    {
        foreach (QDockWidget *dockWidget, findChildren<QDockWidget *>())
        {
            dockWidget->setFeatures(QDockWidget::NoDockWidgetFeatures);
        }
        ui->actionLockUnlock->setIcon(QIcon(":/new/prefix1/lock"));
    }
    else
    {
        foreach (QDockWidget *dockWidget, findChildren<QDockWidget *>())
        {
            dockWidget->setFeatures(QDockWidget::AllDockWidgetFeatures);
        }
        ui->actionLockUnlock->setIcon(QIcon(":/new/prefix1/unlock"));
    }
}

void MainWindow::on_actionTabs_triggered()
{
    if (ui->centralTabWidget->tabPosition() == QTabWidget::South)
    {
        ui->centralTabWidget->setTabPosition(QTabWidget::North);
        this->memoryDock->memTabWidget->setTabPosition(QTabWidget::North);
        this->setTabPosition(Qt::AllDockWidgetAreas, QTabWidget::North);
    }
    else
    {
        ui->centralTabWidget->setTabPosition(QTabWidget::South);
        this->memoryDock->memTabWidget->setTabPosition(QTabWidget::South);
        this->setTabPosition(Qt::AllDockWidgetAreas, QTabWidget::South);
    }
}

void MainWindow::on_actionMem_triggered()
{
    //this->memoryDock->show();
    //this->memoryDock->raise();
    MemoryWidget *newMemDock = new MemoryWidget(this);
    this->dockWidgets << newMemDock;
    newMemDock->setAttribute(Qt::WA_DeleteOnClose);
    this->tabifyDockWidget(this->memoryDock, newMemDock);
    newMemDock->refreshDisasm();
    newMemDock->refreshHexdump();
}

void MainWindow::on_actionFunctions_triggered()
{
    if (this->functionsDock->isVisible())
    {
        this->functionsDock->close();
    }
    else
    {
        this->functionsDock->show();
        this->functionsDock->raise();
    }
}

void MainWindow::on_actionImports_triggered()
{
    if (this->importsDock->isVisible())
    {
        this->importsDock->close();
    }
    else
    {
        this->importsDock->show();
        this->importsDock->raise();
    }
}

void MainWindow::on_actionSymbols_triggered()
{
    if (this->symbolsDock->isVisible())
    {
        this->symbolsDock->close();
    }
    else
    {
        this->symbolsDock->show();
        this->symbolsDock->raise();
    }
}

void MainWindow::on_actionReloc_triggered()
{
    if (this->relocsDock->isVisible())
    {
        this->relocsDock->close();
    }
    else
    {
        this->relocsDock->show();
        this->relocsDock->raise();
    }
}

void MainWindow::on_actionStrings_triggered()
{
    if (this->stringsDock->isVisible())
    {
        this->stringsDock->close();
    }
    else
    {
        this->stringsDock->show();
        this->stringsDock->raise();
    }
}

void MainWindow::on_actionSections_triggered()
{
    if (this->sectionsDock->isVisible())
    {
        this->sectionsDock->close();
    }
    else
    {
        this->sectionsDock->show();
        this->sectionsDock->raise();
    }
}

void MainWindow::on_actionFlags_triggered()
{
    if (this->flagsDock->isVisible())
    {
        this->flagsDock->close();
    }
    else
    {
        this->flagsDock->show();
        this->flagsDock->raise();
    }
}

void MainWindow::on_actionComents_triggered()
{
    if (this->commentsDock->isVisible())
    {
        this->commentsDock->close();
    }
    else
    {
        this->commentsDock->show();
        this->commentsDock->raise();
    }
}

void MainWindow::on_actionNotepad_triggered()
{
    if (this->notepadDock->isVisible())
    {
        this->notepadDock->close();
    }
    else
    {
        this->notepadDock->show();
        this->notepadDock->raise();
    }
}

void MainWindow::on_actionAbout_triggered()
{
    AboutDialog *a = new AboutDialog(this);
    a->open();
}

void MainWindow::on_consoleInputLineEdit_returnPressed()
{
    if (this->core)
    {
        QString input = ui->consoleInputLineEdit->text();
        ui->consoleOutputTextEdit->appendPlainText(this->core->cmd(input));
        ui->consoleOutputTextEdit->verticalScrollBar()->setValue(ui->consoleOutputTextEdit->verticalScrollBar()->maximum());
        // Add new command to history
        QCompleter *completer = ui->consoleInputLineEdit->completer();
        if (completer != NULL)
        {
            QStringListModel *completerModel = (QStringListModel *)(completer->model());
            if (completerModel != NULL)
                completerModel->setStringList(completerModel->stringList() << input);
        }

        ui->consoleInputLineEdit->setText("");
    }
}

void MainWindow::on_showHistoToolButton_clicked()
{
    QCompleter *completer = ui->consoleInputLineEdit->completer();
    if (completer == NULL)
        return;

    if (ui->showHistoToolButton->isChecked())
    {
        completer->setCompletionMode(QCompleter::UnfilteredPopupCompletion);
        // Uhm... shouldn't it be called always?
        completer->complete();
    }
    else
    {
        completer->setCompletionMode(QCompleter::PopupCompletion);
    }
}

void MainWindow::on_actionClear_ConsoleOutput_triggered()
{
    ui->consoleOutputTextEdit->setPlainText("");
}

void MainWindow::on_actionRefresh_Panels_triggered()
{
    this->updateFrames();
}


void MainWindow::seek(const QString &offset, const QString &name, bool raise_memory_dock)
{
    // TODO: remove this method and use the one with RVA only!

    if(offset.length() < 2)
        return;

    bool ok;
    RVA addr = offset.mid(2).toULongLong(&ok, 16);
    if(!ok)
        return;

    seek(addr, name, raise_memory_dock);
}


void MainWindow::seek(const RVA offset, const QString &name, bool raise_memory_dock)
{
    if (name != NULL)
    {
        this->memoryDock->setWindowTitle(name);
        //this->current_address = name;
    }
    this->hexdumpTopOffset = 0;
    this->hexdumpBottomOffset = 0;
    core->seek(offset);
    setCursorAddress(offset);

    refreshMem(offset);
    this->memoryDock->disasTextEdit->setFocus();

    // Rise and shine baby!
    if(raise_memory_dock)
        this->memoryDock->raise();
}

void MainWindow::refreshMem()
{
    this->memoryDock->updateViews();

}

void MainWindow::refreshMem(RVA offset)
{
    //add_debug_output("Refreshing to: " + off);
    //graphicsBar->refreshColorBar();
    /*
    this->memoryDock->refreshDisasm(off);
    this->memoryDock->refreshHexdump(off);
    this->memoryDock->create_graph(off);
    */
    refreshMem();
    this->memoryDock->get_refs_data(RAddressString(offset));
    //this->memoryDock->setFcnName(offset);
}

void MainWindow::on_backButton_clicked()
{
    QList<RVA> seek_history = core->getSeekHistory();
    this->core->cmd("s-");
    RVA offset = this->core->getOffset();
    QString fcn = this->core->cmdFunctionAt(QString::number(offset));
    this->seek(offset, fcn);
}

void MainWindow::on_actionCalculator_triggered()
{
    if (!this->sideBar->isVisible())
    {
        this->on_actionShow_Hide_mainsidebar_triggered();
    }
}

void MainWindow::on_actionCreate_File_triggered()
{
    createNewDialog *n = new createNewDialog(this);
    n->exec();
}

void MainWindow::on_actionAssembler_triggered()
{
    if (!this->sideBar->isVisible())
    {
        this->on_actionShow_Hide_mainsidebar_triggered();
    }
}

void MainWindow::on_consoleExecButton_clicked()
{
    on_consoleInputLineEdit_returnPressed();
}

void MainWindow::on_actionStart_Web_Server_triggered()
{
    setWebServerState(ui->actionStart_Web_Server->isChecked());
}

void MainWindow::on_actionConsoleSync_with_core_triggered()
{
    if (ui->actionConsoleSync_with_core->isChecked())
    {
        //Enable core syncronization
    }
    else
    {
        // Disable core sync
    }
}

void MainWindow::on_actionDisasAdd_comment_triggered()
{
    CommentsDialog *c = new CommentsDialog(this);
    c->exec();
}

void MainWindow::restoreDocks()
{
    addDockWidget(Qt::RightDockWidgetArea, sectionsDock);
    addDockWidget(Qt::TopDockWidgetArea, this->dashboardDock);
    this->tabifyDockWidget(sectionsDock, this->commentsDock);
    this->tabifyDockWidget(this->dashboardDock, this->memoryDock);
    this->tabifyDockWidget(this->dashboardDock, this->functionsDock);
    this->tabifyDockWidget(this->dashboardDock, this->flagsDock);
    this->tabifyDockWidget(this->dashboardDock, this->stringsDock);
    this->tabifyDockWidget(this->dashboardDock, this->relocsDock);
    this->tabifyDockWidget(this->dashboardDock, this->importsDock);
    this->tabifyDockWidget(this->dashboardDock, this->symbolsDock);
    this->tabifyDockWidget(this->dashboardDock, this->notepadDock);
    this->dashboardDock->raise();
    sectionsDock->raise();
    this->functionsDock->raise();
}

void MainWindow::on_actionDefaut_triggered()
{
    hideAllDocks();
    restoreDocks();
    showDefaultDocks();
    this->dashboardDock->raise();
}

void MainWindow::hideAllDocks()
{
    for (auto w : dockWidgets)
    {
        w->hide();
    }
}

void MainWindow::showDefaultDocks()
{
    const QList<DockWidget *> defaultDocks = { sectionsDock,
                                               functionsDock,
                                               memoryDock,
                                               commentsDock,
                                               stringsDock,
                                               importsDock,
                                               symbolsDock,
                                               notepadDock,
                                               dashboardDock
                                             };

    for (auto w : dockWidgets)
    {
        if (defaultDocks.contains(w))
        {
            w->show();
        }
    }
}

void MainWindow::on_actionhide_bottomPannel_triggered()
{
    if (ui->centralWidget->isVisible())
    {
        ui->centralWidget->hide();
    }
    else
    {
        ui->centralWidget->show();
    }
}

void MainWindow::send_to_notepad(QString txt)
{
    this->notepadDock->appendPlainText("```\n" + txt + "\n```");
}

void MainWindow::on_actionFunctionsRename_triggered()
{
    RenameDialog *r = new RenameDialog(this);
    // Get function based on click position
    //r->setFunctionName(fcn_name);
    r->open();
}

void MainWindow::get_refs(const QString &offset)
{
    this->memoryDock->get_refs_data(offset);
}

void MainWindow::add_output(QString msg)
{
    ui->consoleOutputTextEdit->appendPlainText(msg);
    ui->consoleOutputTextEdit->verticalScrollBar()->setValue(ui->consoleOutputTextEdit->verticalScrollBar()->maximum());
}

void MainWindow::add_debug_output(QString msg)
{
    printf("debug output: %s\n", msg.toLocal8Bit().constData());
    ui->consoleOutputTextEdit->appendHtml("<font color=\"red\"> [DEBUG]:\t" + msg + "</font>");
    ui->consoleOutputTextEdit->verticalScrollBar()->setValue(ui->consoleOutputTextEdit->verticalScrollBar()->maximum());
}

void MainWindow::on_actionNew_triggered()
{
    if (close())
        on_actionLoad_triggered();
}

void MainWindow::on_actionSave_triggered()
{
    core->cmd("Ps " + qhelpers::uniqueProjectName(filename));
    QString notes = this->notepadDock->textToBase64();
    //this->add_debug_output(notes);
    this->core->cmd("Pnj " + notes);
    this->add_output("Project saved");
}

void MainWindow::on_actionRun_Script_triggered()
{
    QFileDialog dialog(this);
    dialog.setFileMode(QFileDialog::ExistingFile);
    dialog.setViewMode(QFileDialog::Detail);
    dialog.setDirectory(QDir::home());

    QString fileName;
    fileName = dialog.getOpenFileName(this, "Select radare2 script");
    if (!fileName.length()) //cancel was pressed
        return;

    qDebug() << "Meow: " + fileName;
    this->core->cmd(". " + fileName);
    this->refreshMem();
}

void MainWindow::on_actionDark_Theme_triggered()
{
    this->dark();
}

void MainWindow::on_actionWhite_Theme_triggered()
{
    this->def_theme();
}

void MainWindow::on_actionSDB_browser_triggered()
{
    this->sdbDock = new SdbDock(this);
    this->tabifyDockWidget(this->memoryDock, this->sdbDock);
    this->sdbDock->setFloating(true);
    this->sdbDock->show();
}

void MainWindow::on_actionLoad_triggered()
{
    QProcess process(this);
    process.setEnvironment(QProcess::systemEnvironment());
    process.startDetached(qApp->applicationFilePath());
}

void MainWindow::on_actionShow_Hide_mainsidebar_triggered()
{
    if (ui->sideToolBar->isVisible())
    {
        ui->sideToolBar->hide();
    }
    else
    {
        ui->sideToolBar->show();
    }
}

void MainWindow::on_actionDashboard_triggered()
{
    if (this->dashboardDock->isVisible())
    {
        this->dashboardDock->close();
    }
    else
    {
        this->dashboardDock->show();
        this->dashboardDock->raise();
    }
}

void MainWindow::on_actionForward_triggered()
{
    this->core->cmd("s+");
    RVA offset = core->getOffset();
    this->add_debug_output(QString::number(offset));
    this->seek(offset);
}

void MainWindow::toggleResponsive(bool maybe)
{
    this->responsive = maybe;
    // Save options in settings
    QSettings settings;
    settings.setValue("responsive", this->responsive);
}

void MainWindow::on_actionTabs_on_Top_triggered()
{
    this->on_actionTabs_triggered();
}

void MainWindow::on_actionReset_settings_triggered()
{
    QMessageBox::StandardButton ret = QMessageBox::question(this, "Iaito",
                                      "Do you really want to clear all settings?",
                                      QMessageBox::Ok | QMessageBox::Cancel);
    if (ret == QMessageBox::Ok)
    {
        // Save options in settings
        QSettings settings;
        settings.clear();
    }
}

void MainWindow::on_actionQuit_triggered()
{
    close();
}

void MainWindow::setCursorAddress(RVA addr)
{
    this->cursor_address = addr;
    emit cursorAddressChanged(addr);
}

void MainWindow::refreshVisibleDockWidgets()
{
    // There seems to be no convenience function to check if a QDockWidget
    // is really visible or hidden in a tabbed dock. So:
    auto isDockVisible = [](const QDockWidget * const pWidget)
    {
        return pWidget != nullptr && !pWidget->visibleRegion().isEmpty();
    };

    for (auto w : dockWidgets)
    {
        if (isDockVisible(w))
        {
            w->refresh();
        }
    }
}