cutter/src/utils/ColorSchemeFileSaver.cpp

#include "utils/ColorSchemeFileSaver.h"

#include <QDir>
#include <QDebug>
#include <QColor>
#include <QJsonArray>
#include <QJsonObject>
#include <QStandardPaths>
#include "utils/Configuration.h"

static const QStringList cutterSpecificOptions = {
    "gui.main",
    "highlight",
    "gui.imports",
    "highlightPC",
    "highlightWord",
    "gui.navbar.err",
    "gui.navbar.sym",
    "gui.dataoffset",
    "gui.navbar.code",
    "gui.navbar.empty",
    "angui.navbar.str",
    "gui.disass_selected",
    "gui.breakpoint_background"
};

ColorSchemeFileSaver::ColorSchemeFileSaver(QObject *parent) : QObject (parent)
{
    customR2ThemesLocationPath = QStandardPaths::writableLocation(QStandardPaths::GenericDataLocation) +
                                 QDir::separator() +
                                 "radare2" + QDir::separator() +
                                 "cons";
    if (!QDir(customR2ThemesLocationPath).exists()) {
        QDir().mkpath(customR2ThemesLocationPath);
    }

    QDir currDir;
    QStringList dirs = QStandardPaths::standardLocations(QStandardPaths::GenericDataLocation);
    dirs.removeOne(QStandardPaths::writableLocation(QStandardPaths::GenericDataLocation));
    standardR2ThemesLocationPath = "";

    for (auto &it : dirs) {
        currDir = QDir(it).filePath("radare2");
        if (currDir.exists()) {
            break;
        }
        currDir.setPath("");
    }

    currDir.setPath(QString(r_sys_prefix(nullptr)) + QString(R_SYS_DIR) + R2_THEMES);
    if (currDir.exists()) {
        standardR2ThemesLocationPath = currDir.absolutePath();
    } else {
        standardR2ThemesLocationPath = "";
        QMessageBox mb;
        mb.setIcon(QMessageBox::Critical);
        mb.setStandardButtons(QMessageBox::Ok);
        mb.setWindowTitle(tr("Standard themes not found!"));
        mb.setText(tr("The radare2 standard themes could not be found! This probably means radare2 is not properly installed. If you think it is open an issue please."));
        mb.exec();
    }
}

QFile::FileError ColorSchemeFileSaver::copy(const QString &srcThemeName,
                                            const QString &copyThemeName) const
{
    QFile fIn(standardR2ThemesLocationPath + QDir::separator() + srcThemeName);
    QFile fOut(customR2ThemesLocationPath + QDir::separator() + copyThemeName);

    if (srcThemeName != "default" && !fIn.open(QFile::ReadOnly)) {
        fIn.setFileName(customR2ThemesLocationPath + QDir::separator() + srcThemeName);
        if (!fIn.open(QFile::ReadOnly)) {
            return fIn.error();
        }
    }

    if (!fOut.open(QFile::WriteOnly | QFile::Truncate)) {
        fIn.close();
        return fOut.error();
    }

    QStringList options = Core()->cmdj("ecj").object().keys();
    options << cutterSpecificOptions;
    QStringList src;

    if (srcThemeName == "default") {
        QString theme = Config()->getCurrentTheme();
        Core()->cmd("ecd");
        QJsonObject _obj = Core()->cmdj("ecj").object();
        Core()->cmd(QString("eco %1").arg(theme));
        QColor back = Config()->getColor(standardBackgroundOptionName);
        _obj[standardBackgroundOptionName] = QJsonArray({back.red(), back.green(), back.blue()});
        for (auto &it : _obj.keys()) {
            QJsonArray rgb = _obj[it].toArray();
            if (rgb.size() != 3) {
                continue;
            }
            src.push_back("ec " + it + " " +
                          QColor(rgb[0].toInt(), rgb[1].toInt(), rgb[2].toInt()).name().replace("#", "rgb:"));
        }
    } else {
        src = QString(fIn.readAll()).split('\n');
    }

    QStringList tmp;
    for (auto &it : src) {
        if (it.isEmpty()) {
            continue;
        }
        fOut.write(it.toUtf8() + '\n');

        tmp = it.split(' ');
        if (it.length() > 2 && it.left(2) == "#~") {
            options.removeOne(tmp[0].remove("#~").toUtf8());
        } else if (tmp.size() > 1) {
            options.removeOne(tmp.at(1));
        }
    }

    for (auto &it : options) {
        if (cutterSpecificOptions.contains(it))  {
            fOut.write("#~");
        } else {
            fOut.write("ec ");
        }
        fOut.write(QString(it + " rgb:%1\n").
                   arg(Config()->getColor(it).name().remove("#")).toUtf8());
    }

    fOut.close();
    fIn.close();

    return QFile::FileError::NoError;
}

QFile::FileError ColorSchemeFileSaver::save(const QString &scheme, const QString &schemeName) const
{
    QFile fOut(customR2ThemesLocationPath + QDir::separator() + schemeName);
    if (!fOut.open(QFile::WriteOnly | QFile::Truncate))
        return fOut.error();

    fOut.write(scheme.toUtf8());
    fOut.close();
    return QFile::FileError::NoError;
}

bool ColorSchemeFileSaver::isCustomScheme(const QString &schemeName) const
{
    for (auto &it : QDir(customR2ThemesLocationPath).entryInfoList())
        if (it.fileName() == schemeName)
            return true;
    return false;
}

bool ColorSchemeFileSaver::isNameEngaged(const QString &name) const
{
    return QFile::exists(standardR2ThemesLocationPath + QDir::separator() + name) ||
           QFile::exists(customR2ThemesLocationPath + QDir::separator() + name);
}

QMap<QString, QColor> ColorSchemeFileSaver::getCutterSpecific() const
{
    QFile f(customR2ThemesLocationPath + QDir::separator() + Config()->getCurrentTheme());
    if (!f.open(QFile::ReadOnly))
        return  QMap<QString, QColor>();

    QMap<QString, QColor> ret;
    QStringList data = QString(f.readAll()).split('\n');
    for (auto &it : data) {
        if (it.length() > 2 && it.left(2) == "#~") {
            QStringList currLine = it.split(' ');
            if (currLine.size() > 1) {
                ret.insert(currLine[0].remove("#~"), currLine[1].replace("rgb:", "#"));
            }
        }
    }

    f.close();
    return ret;
}

QStringList ColorSchemeFileSaver::getCustomSchemes() const
{
    QStringList sl;
    sl = QDir(customR2ThemesLocationPath).entryList(QDir::Files | QDir::NoDotAndDotDot, QDir::Name);
    return sl;
}

void ColorSchemeFileSaver::deleteScheme(const QString &schemeName) const
{
    if (!isCustomScheme(schemeName))
        return;
    QFile::remove(customR2ThemesLocationPath + QDir::separator() + schemeName);
}