QodeAssist/settings/ConfigurationManager.cpp

/* 
 * Copyright (C) 2024-2025 Petr Mironychev
 *
 * This file is part of QodeAssist.
 *
 * QodeAssist 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.
 *
 * QodeAssist 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 QodeAssist. If not, see <https://www.gnu.org/licenses/>.
 */

#include "ConfigurationManager.hpp"

#include <QDir>
#include <QFile>
#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>
#include <QRegularExpression>
#include <QUuid>

#include <coreplugin/icore.h>

#include "Logger.hpp"

namespace QodeAssist::Settings {

ConfigurationManager::ConfigurationManager(QObject *parent)
    : QObject(parent)
{}

ConfigurationManager &ConfigurationManager::instance()
{
    static ConfigurationManager instance;
    return instance;
}

QString ConfigurationManager::configurationTypeToString(ConfigurationType type) const
{
    switch (type) {
    case ConfigurationType::CodeCompletion:
        return "code_completion";
    case ConfigurationType::Chat:
        return "chat";
    case ConfigurationType::QuickRefactor:
        return "quick_refactor";
    }
    return "unknown";
}

QString ConfigurationManager::getConfigurationDirectory(ConfigurationType type) const
{
    QString path = QString("%1/qodeassist/configurations/%2")
                       .arg(Core::ICore::userResourcePath().toFSPathString(),
                            configurationTypeToString(type));
    return path;
}

bool ConfigurationManager::ensureDirectoryExists(ConfigurationType type) const
{
    QDir dir(getConfigurationDirectory(type));
    if (!dir.exists()) {
        return dir.mkpath(".");
    }
    return true;
}

bool ConfigurationManager::loadConfigurations(ConfigurationType type)
{
    QVector<AIConfiguration> *configs = nullptr;
    switch (type) {
    case ConfigurationType::CodeCompletion:
        configs = &m_ccConfigurations;
        break;
    case ConfigurationType::Chat:
        configs = &m_caConfigurations;
        break;
    case ConfigurationType::QuickRefactor:
        configs = &m_qrConfigurations;
        break;
    }

    if (!configs) {
        return false;
    }

    configs->clear();

    if (!ensureDirectoryExists(type)) {
        LOG_MESSAGE("Failed to create configuration directory");
        return false;
    }

    QDir dir(getConfigurationDirectory(type));
    QStringList filters;
    filters << "*.json";
    QFileInfoList files = dir.entryInfoList(filters, QDir::Files);

    for (const QFileInfo &fileInfo : files) {
        QFile file(fileInfo.absoluteFilePath());
        if (!file.open(QIODevice::ReadOnly)) {
            LOG_MESSAGE(QString("Failed to open configuration file: %1").arg(fileInfo.fileName()));
            continue;
        }

        QJsonDocument doc = QJsonDocument::fromJson(file.readAll());
        file.close();

        if (!doc.isObject()) {
            LOG_MESSAGE(QString("Invalid configuration file: %1").arg(fileInfo.fileName()));
            continue;
        }

        QJsonObject obj = doc.object();
        AIConfiguration config;
        config.id = obj["id"].toString();
        config.name = obj["name"].toString();
        config.provider = obj["provider"].toString();
        config.model = obj["model"].toString();
        config.templateName = obj["template"].toString();
        config.url = obj["url"].toString();
        config.endpointMode = obj["endpointMode"].toString();
        config.customEndpoint = obj["customEndpoint"].toString();
        config.type = type;
        config.formatVersion = obj.value("formatVersion").toInt(1);

        if (config.id.isEmpty() || config.name.isEmpty()) {
            LOG_MESSAGE(QString("Invalid configuration data in file: %1").arg(fileInfo.fileName()));
            continue;
        }

        configs->append(config);
    }

    emit configurationsChanged(type);
    return true;
}

bool ConfigurationManager::saveConfiguration(const AIConfiguration &config)
{
    if (!ensureDirectoryExists(config.type)) {
        LOG_MESSAGE("Failed to create configuration directory");
        return false;
    }

    QJsonObject obj;
    obj["formatVersion"] = config.formatVersion;
    obj["id"] = config.id;
    obj["name"] = config.name;
    obj["provider"] = config.provider;
    obj["model"] = config.model;
    obj["template"] = config.templateName;
    obj["url"] = config.url;
    obj["endpointMode"] = config.endpointMode;
    obj["customEndpoint"] = config.customEndpoint;

    QString sanitizedName = config.name;
    sanitizedName.replace(" ", "_");
    sanitizedName.replace(QRegularExpression("[^a-zA-Z0-9_-]"), "");

    QString fileName = QString("%1/%2_%3.json")
                           .arg(getConfigurationDirectory(config.type), sanitizedName, config.id);

    QFile file(fileName);
    if (!file.open(QIODevice::WriteOnly)) {
        LOG_MESSAGE(QString("Failed to create configuration file: %1").arg(fileName));
        return false;
    }

    QJsonDocument doc(obj);
    file.write(doc.toJson(QJsonDocument::Indented));
    file.close();

    loadConfigurations(config.type);
    return true;
}

bool ConfigurationManager::deleteConfiguration(const QString &id, ConfigurationType type)
{
    QDir dir(getConfigurationDirectory(type));
    QStringList filters;
    filters << QString("*_%1.json").arg(id);
    QFileInfoList files = dir.entryInfoList(filters, QDir::Files);

    if (files.isEmpty()) {
        LOG_MESSAGE(QString("Configuration file not found for id: %1").arg(id));
        return false;
    }

    for (const QFileInfo &fileInfo : files) {
        QFile file(fileInfo.absoluteFilePath());
        if (!file.remove()) {
            LOG_MESSAGE(QString("Failed to delete configuration file: %1")
                            .arg(fileInfo.absoluteFilePath()));
            return false;
        }
    }

    loadConfigurations(type);
    return true;
}

QVector<AIConfiguration> ConfigurationManager::configurations(ConfigurationType type) const
{
    switch (type) {
    case ConfigurationType::CodeCompletion:
        return m_ccConfigurations;
    case ConfigurationType::Chat:
        return m_caConfigurations;
    case ConfigurationType::QuickRefactor:
        return m_qrConfigurations;
    }
    return {};
}

AIConfiguration ConfigurationManager::getConfigurationById(const QString &id,
                                                           ConfigurationType type) const
{
    const QVector<AIConfiguration> &configs = configurations(type);
    for (const AIConfiguration &config : configs) {
        if (config.id == id) {
            return config;
        }
    }
    return AIConfiguration();
}

} // namespace QodeAssist::Settings