feat: Add configuration manager (#270)

settings/ConfigurationManager.cpp

@@ -0,0 +1,237 @@
+/* 
+ * 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
+