/*
 * Copyright (C) 2019-2024 Matthias Klumpp <matthias@tenstral.net>
 *
 * Licensed under the GNU Lesser General Public License Version 3
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the license, or
 * (at your option) any later version.
 *
 * This software 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this software.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "config.h"
#include "globalconfig.h"

#include <QDebug>
#include <QDir>
#include <QSettings>
#include <QStandardPaths>
#include <QCoreApplication>

#include "rtkit.h"
#include "utils/misc.h"

using namespace Syntalos;

namespace Syntalos
{
Q_LOGGING_CATEGORY(logGlobalConfig, "global.config")
}

GlobalConfig::GlobalConfig(QObject *parent)
    : QObject(parent)
{
    m_s = new QSettings("DraguhnLab", "Syntalos", this);

    m_userHome = QStandardPaths::writableLocation(QStandardPaths::HomeLocation);
    if (isInFlatpakSandbox())
        m_appDataRoot = QDir(m_userHome).filePath(".var/app/org.syntalos.syntalos/data");
    else
        m_appDataRoot = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);

    if (m_userHome.isEmpty())
        qCCritical(logGlobalConfig, "Unable to determine user home directory!");
    if (m_appDataRoot.isEmpty())
        qCCritical(logGlobalConfig, "Unable to determine application data directory!");
}

QString GlobalConfig::iconThemeName() const
{
    return m_s->value("ui/icon_theme", QStringLiteral("breeze")).toString();
}

void GlobalConfig::setIconThemeName(const QString &iconTheme)
{
    m_s->setValue("ui/icon_theme", iconTheme);
}

ColorMode GlobalConfig::appColorMode() const
{
    const auto strMode = m_s->value("ui/color_mode", QStringLiteral("system")).toString();
    return colorModeFromString(strMode);
}

void GlobalConfig::setAppColorMode(ColorMode mode)
{
    m_s->setValue("ui/color_mode", colorModeToString(mode));
}

QByteArray GlobalConfig::mainWinGeometry() const
{
    return m_s->value("ui/geometry").toByteArray();
}

void GlobalConfig::setMainWinGeometry(const QByteArray &geometry)
{
    m_s->setValue("ui/geometry", geometry);
}

QByteArray GlobalConfig::mainWinState() const
{
    return m_s->value("ui/window_state").toByteArray();
}

void GlobalConfig::setMainWinState(const QByteArray &state)
{
    m_s->setValue("ui/window_state", state);
}

int GlobalConfig::defaultThreadNice() const
{
    RtKit rtkit;
    const auto minNice = rtkit.queryMinNiceLevel();
    int nice = m_s->value("engine/default_thread_nice", -10).toInt();
    if (nice > 20)
        nice = 20;
    else if (nice < minNice)
        nice = minNice;
    return nice;
}

void GlobalConfig::setDefaultThreadNice(int nice)
{
    if (nice > 20)
        nice = 20;
    else if (nice < -19)
        nice = -19;
    m_s->setValue("engine/default_thread_nice", nice);
}

int GlobalConfig::defaultRTThreadPriority() const
{
    RtKit rtkit;
    const auto maxPrio = rtkit.queryMaxRealtimePriority();
    int prio = m_s->value("engine/default_rt_thread_priority", 20).toInt();
    if (prio > 99)
        prio = 99;
    else if (prio > maxPrio)
        prio = maxPrio;

    return prio;
}

void GlobalConfig::setDefaultRTThreadPriority(int priority)
{
    if (priority > 99)
        priority = 99;
    else if (priority < 1)
        priority = 1;
    m_s->setValue("engine/default_rt_thread_priority", priority);
}

bool GlobalConfig::explicitCoreAffinities() const
{
    return m_s->value("engine/explicit_core_affinities", false).toBool();
}

void GlobalConfig::setExplicitCoreAffinities(bool enabled)
{
    m_s->setValue("engine/explicit_core_affinities", enabled);
}

bool GlobalConfig::showDevelModules() const
{
    return m_s->value("devel/show_devel_modules", false).toBool();
}

void GlobalConfig::setShowDevelModules(bool enabled)
{
    m_s->setValue("devel/show_devel_modules", enabled);
}

bool GlobalConfig::saveExperimentDiagnostics() const
{
    return m_s->value("devel/save_diagnostics", false).toBool();
}

void GlobalConfig::setSaveExperimentDiagnostics(bool enabled)
{
    m_s->setValue("devel/save_diagnostics", enabled);
}

QString GlobalConfig::appDataLocation() const
{
    return m_appDataRoot;
}

QString GlobalConfig::userModulesDir() const
{
    return QDir(m_appDataRoot).filePath("modules");
}

QString GlobalConfig::virtualenvDir() const
{
    return QDir(m_appDataRoot).filePath("venv");
}

QString GlobalConfig::homeDevelDir() const
{
    return QDir(m_userHome).filePath("SyntalosDevel");
}

QString GlobalConfig::userCacheDir() const
{
    return QDir(m_appDataRoot).filePath("cache");
}

bool GlobalConfig::useVenvForPyScript() const
{
    return m_s->value("devel/use_venv_for_pyscript", false).toBool();
}

void GlobalConfig::setUseVenvForPyScript(bool enabled)
{
    m_s->setValue("devel/use_venv_for_pyscript", enabled);
}

bool GlobalConfig::emergencyOOMStop() const
{
    return m_s->value("engine/emergency_oom_stop", true).toBool();
}

void GlobalConfig::setEmergencyOOMStop(bool enabled)
{
    m_s->setValue("engine/emergency_oom_stop", enabled);
}

QString Syntalos::colorModeToString(ColorMode mode)
{
    switch (mode) {
    case ColorMode::BRIGHT:
        return QStringLiteral("bright");
    case ColorMode::DARK:
        return QStringLiteral("dark");
    default:
        return QStringLiteral("system");
    }
}

ColorMode Syntalos::colorModeFromString(const QString &str)
{
    if (str == "bright")
        return ColorMode::BRIGHT;
    if (str == "dark")
        return ColorMode::DARK;
    return ColorMode::SYSTEM;
}

QString Syntalos::findSyntalosPyWorkerBinary()
{
    auto workerBinary = QStringLiteral("%1/python/pyworker").arg(QCoreApplication::applicationDirPath());
    QFileInfo checkBin(workerBinary);
    if (!checkBin.exists() || workerBinary.startsWith("/usr/")) {
        workerBinary = QStringLiteral("%1/pyworker").arg(SY_LIBDIR);
        QFileInfo fi(workerBinary);
        workerBinary = fi.canonicalFilePath();
    }

    return workerBinary;
}
void Syntalos::findSyntalosLibraryPaths(QString &pkgConfigPath, QString &ldLibraryPath, QString &includePath)
{
    // check if we are running from the build directory
    QFile siFile(QStringLiteral("%1/../sy-source-info.txt").arg(QCoreApplication::applicationDirPath()));

    // If the file does not exist, we are not running from the build directory
    // and can assume the libraries are found in system search paths
    if (!siFile.exists())
        return;

    if (!siFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
        qWarning().noquote()
            << "Cannot open source-info file for reading. Syntalos library search paths will not be set.";
        return;
    }

    QTextStream in(&siFile);
    const auto srcRootKey = QStringLiteral("#define SY_SOURCE_ROOT");
    QString srcRootPath;
    while (!in.atEnd()) {
        const auto line = in.readLine();
        if (line.contains(srcRootKey)) {
            QString path = line.mid(srcRootKey.length()).trimmed();
            if (path.startsWith('/'))
                srcRootPath = path;
            break;
        }
    }
    siFile.close();

    if (srcRootPath.isEmpty()) {
        qWarning().noquote() << "Syntalos source-root path was not found. Can not set include paths.";
    } else {
        includePath = QStringLiteral("%1/src:%1/src/mlink/include:%1/src/datactl/include").arg(srcRootPath);
    }

    QFileInfo pkgCRootFi(QStringLiteral("%1/../meson-private/").arg(QCoreApplication::applicationDirPath()));
    pkgConfigPath = pkgCRootFi.canonicalFilePath();
    ldLibraryPath = QStringLiteral("%1/mlink").arg(QCoreApplication::applicationDirPath()) + ":"
                    + QStringLiteral("%1/datactl").arg(QCoreApplication::applicationDirPath()) + ":"
                    + QStringLiteral("%1/utils").arg(QCoreApplication::applicationDirPath());
}