mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 10:59:30 -04:00
feat: Add mcp client hub
This commit is contained in:
Petr Mironychev
364
mcp/McpClientsManager.cpp
Normal file
364
mcp/McpClientsManager.cpp
Normal file
@@ -0,0 +1,364 @@
|
||||
// Copyright (C) 2024-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
|
||||
#include "McpClientsManager.hpp"
|
||||
|
||||
#include <coreplugin/icore.h>
|
||||
|
||||
#include <QDir>
|
||||
#include <QFile>
|
||||
#include <QFileInfo>
|
||||
#include <QFileSystemWatcher>
|
||||
#include <QJsonDocument>
|
||||
#include <QJsonObject>
|
||||
#include <QSaveFile>
|
||||
#include <QTimer>
|
||||
|
||||
#include <logger/Logger.hpp>
|
||||
#include <pluginllmcore/Provider.hpp>
|
||||
#include <pluginllmcore/ProvidersManager.hpp>
|
||||
#include <settings/McpSettings.hpp>
|
||||
|
||||
namespace QodeAssist::Mcp {
|
||||
|
||||
namespace {
|
||||
|
||||
constexpr const char *kServersKey = "mcpServers";
|
||||
|
||||
QByteArray serializedEntry(const McpServerConfig &cfg)
|
||||
{
|
||||
return QJsonDocument(cfg.toJson()).toJson(QJsonDocument::Compact);
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
McpClientsManager &McpClientsManager::instance()
|
||||
{
|
||||
static McpClientsManager manager;
|
||||
return manager;
|
||||
}
|
||||
|
||||
QString McpClientsManager::configFilePath()
|
||||
{
|
||||
const QString base = Core::ICore::userResourcePath().toFSPathString();
|
||||
return QDir(base).filePath(QStringLiteral("qodeassist/mcp-server.json"));
|
||||
}
|
||||
|
||||
QByteArray McpClientsManager::emptyConfigTemplate()
|
||||
{
|
||||
return QByteArray(
|
||||
"{\n"
|
||||
" \"mcpServers\": {\n"
|
||||
" // Example HTTP/SSE server:\n"
|
||||
" // \"my-http-server\": {\n"
|
||||
" // \"type\": \"sse\",\n"
|
||||
" // \"url\": \"http://127.0.0.1:9000/mcp\",\n"
|
||||
" // \"enable\": true\n"
|
||||
" // },\n"
|
||||
" // Example stdio server:\n"
|
||||
" // \"my-stdio-server\": {\n"
|
||||
" // \"command\": \"/path/to/server\",\n"
|
||||
" // \"args\": [\"--flag\"],\n"
|
||||
" // \"env\": {\"KEY\": \"value\"},\n"
|
||||
" // \"enable\": false\n"
|
||||
" // }\n"
|
||||
" }\n"
|
||||
"}\n");
|
||||
}
|
||||
|
||||
McpClientsManager::McpClientsManager(QObject *parent)
|
||||
: QObject(parent)
|
||||
{}
|
||||
|
||||
McpClientsManager::~McpClientsManager()
|
||||
{
|
||||
for (auto *c : m_connections) {
|
||||
if (c)
|
||||
c->disconnectFromServer();
|
||||
}
|
||||
m_connections.clear();
|
||||
}
|
||||
|
||||
void McpClientsManager::init()
|
||||
{
|
||||
if (m_initialized)
|
||||
return;
|
||||
m_initialized = true;
|
||||
|
||||
ensureFileExists();
|
||||
setupWatcher();
|
||||
|
||||
connect(
|
||||
&Settings::mcpSettings().enableMcpClients,
|
||||
&Utils::BaseAspect::changed,
|
||||
this,
|
||||
[this]() { loadFromDisk(); });
|
||||
|
||||
connect(
|
||||
&Settings::mcpSettings().mcpClientExtraPaths,
|
||||
&Utils::BaseAspect::changed,
|
||||
this,
|
||||
[this]() {
|
||||
for (auto *c : m_connections) {
|
||||
if (c && c->config().transport == McpTransportKind::Stdio
|
||||
&& c->config().enabled
|
||||
&& Settings::mcpSettings().enableMcpClients()) {
|
||||
c->disconnectFromServer();
|
||||
c->connectToServer();
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
loadFromDisk();
|
||||
}
|
||||
|
||||
void McpClientsManager::ensureFileExists()
|
||||
{
|
||||
const QString path = configFilePath();
|
||||
QFileInfo fi(path);
|
||||
if (fi.exists())
|
||||
return;
|
||||
|
||||
QDir().mkpath(fi.absolutePath());
|
||||
QFile f(path);
|
||||
if (f.open(QIODevice::WriteOnly | QIODevice::Text)) {
|
||||
f.write(emptyConfigTemplate());
|
||||
f.close();
|
||||
LOG_MESSAGE(QString("Created empty MCP clients config: %1").arg(path));
|
||||
}
|
||||
}
|
||||
|
||||
void McpClientsManager::setupWatcher()
|
||||
{
|
||||
m_watcher = new QFileSystemWatcher(this);
|
||||
m_reloadDebounce = new QTimer(this);
|
||||
m_reloadDebounce->setSingleShot(true);
|
||||
m_reloadDebounce->setInterval(300);
|
||||
|
||||
connect(m_reloadDebounce.data(), &QTimer::timeout, this, [this]() {
|
||||
const bool suppress = m_suppressNextWatcherReload;
|
||||
m_suppressNextWatcherReload = false;
|
||||
if (!suppress)
|
||||
loadFromDisk();
|
||||
updateWatchedPaths();
|
||||
});
|
||||
connect(m_watcher.data(), &QFileSystemWatcher::fileChanged, this, [this]() {
|
||||
m_reloadDebounce->start();
|
||||
});
|
||||
connect(m_watcher.data(), &QFileSystemWatcher::directoryChanged, this, [this]() {
|
||||
m_reloadDebounce->start();
|
||||
});
|
||||
|
||||
updateWatchedPaths();
|
||||
}
|
||||
|
||||
void McpClientsManager::updateWatchedPaths()
|
||||
{
|
||||
if (!m_watcher)
|
||||
return;
|
||||
if (!m_watcher->files().isEmpty())
|
||||
m_watcher->removePaths(m_watcher->files());
|
||||
if (!m_watcher->directories().isEmpty())
|
||||
m_watcher->removePaths(m_watcher->directories());
|
||||
|
||||
const QString path = configFilePath();
|
||||
const QFileInfo info(path);
|
||||
if (info.exists())
|
||||
m_watcher->addPath(path);
|
||||
const QString dir = info.absolutePath();
|
||||
if (QFileInfo::exists(dir))
|
||||
m_watcher->addPath(dir);
|
||||
}
|
||||
|
||||
QList<McpServerConnection *> McpClientsManager::connections() const
|
||||
{
|
||||
return m_connections;
|
||||
}
|
||||
|
||||
QList<PluginLLMCore::Provider *> McpClientsManager::toolsCapableProviders() const
|
||||
{
|
||||
QList<PluginLLMCore::Provider *> out;
|
||||
auto &pm = PluginLLMCore::ProvidersManager::instance();
|
||||
for (const QString &name : pm.providersNames()) {
|
||||
auto *p = pm.getProviderByName(name);
|
||||
if (!p)
|
||||
continue;
|
||||
if (p->capabilities().testFlag(PluginLLMCore::ProviderCapability::Tools))
|
||||
out.append(p);
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
QJsonObject McpClientsManager::readRoot() const
|
||||
{
|
||||
QFile f(configFilePath());
|
||||
if (!f.open(QIODevice::ReadOnly | QIODevice::Text))
|
||||
return QJsonObject{{QLatin1String(kServersKey), QJsonObject{}}};
|
||||
QJsonParseError err;
|
||||
const QJsonDocument doc = QJsonDocument::fromJson(f.readAll(), &err);
|
||||
f.close();
|
||||
if (err.error != QJsonParseError::NoError || !doc.isObject())
|
||||
return QJsonObject{{QLatin1String(kServersKey), QJsonObject{}}};
|
||||
QJsonObject root = doc.object();
|
||||
if (!root.contains(QLatin1String(kServersKey)))
|
||||
root.insert(QLatin1String(kServersKey), QJsonObject{});
|
||||
return root;
|
||||
}
|
||||
|
||||
bool McpClientsManager::writeRoot(const QJsonObject &root)
|
||||
{
|
||||
QSaveFile out(configFilePath());
|
||||
if (!out.open(QIODevice::WriteOnly | QIODevice::Text)) {
|
||||
const QString reason = out.errorString();
|
||||
LOG_MESSAGE(
|
||||
QString("MCP clients: cannot write %1: %2").arg(configFilePath(), reason));
|
||||
emit writeFailed(reason);
|
||||
return false;
|
||||
}
|
||||
out.write(QJsonDocument(root).toJson(QJsonDocument::Indented));
|
||||
if (!out.commit()) {
|
||||
const QString reason = out.errorString();
|
||||
LOG_MESSAGE(
|
||||
QString("MCP clients: commit failed for %1: %2").arg(configFilePath(), reason));
|
||||
emit writeFailed(reason);
|
||||
return false;
|
||||
}
|
||||
m_suppressNextWatcherReload = true;
|
||||
return true;
|
||||
}
|
||||
|
||||
void McpClientsManager::reload()
|
||||
{
|
||||
loadFromDisk();
|
||||
updateWatchedPaths();
|
||||
}
|
||||
|
||||
bool McpClientsManager::setServerEnabled(const QString &name, bool enabled)
|
||||
{
|
||||
QJsonObject root = readRoot();
|
||||
QJsonObject servers = root.value(QLatin1String(kServersKey)).toObject();
|
||||
if (!servers.contains(name)) {
|
||||
LOG_MESSAGE(QString("MCP clients: setServerEnabled: no entry '%1'").arg(name));
|
||||
return false;
|
||||
}
|
||||
QJsonObject entry = servers.value(name).toObject();
|
||||
entry.insert(QStringLiteral("enable"), enabled);
|
||||
servers.insert(name, entry);
|
||||
root.insert(QLatin1String(kServersKey), servers);
|
||||
if (!writeRoot(root))
|
||||
return false;
|
||||
loadFromDisk();
|
||||
return true;
|
||||
}
|
||||
|
||||
bool McpClientsManager::addServer(const QString &name, const QJsonObject &entry)
|
||||
{
|
||||
QJsonObject root = readRoot();
|
||||
QJsonObject servers = root.value(QLatin1String(kServersKey)).toObject();
|
||||
|
||||
QString finalName = name;
|
||||
int suffix = 2;
|
||||
while (servers.contains(finalName))
|
||||
finalName = QStringLiteral("%1-%2").arg(name).arg(suffix++);
|
||||
|
||||
servers.insert(finalName, entry);
|
||||
root.insert(QLatin1String(kServersKey), servers);
|
||||
if (!writeRoot(root))
|
||||
return false;
|
||||
loadFromDisk();
|
||||
return true;
|
||||
}
|
||||
|
||||
bool McpClientsManager::removeServer(const QString &name)
|
||||
{
|
||||
QJsonObject root = readRoot();
|
||||
QJsonObject servers = root.value(QLatin1String(kServersKey)).toObject();
|
||||
if (!servers.contains(name))
|
||||
return false;
|
||||
servers.remove(name);
|
||||
root.insert(QLatin1String(kServersKey), servers);
|
||||
if (!writeRoot(root))
|
||||
return false;
|
||||
loadFromDisk();
|
||||
return true;
|
||||
}
|
||||
|
||||
void McpClientsManager::loadFromDisk()
|
||||
{
|
||||
const QJsonObject root = readRoot();
|
||||
const QJsonObject servers = root.value(QLatin1String(kServersKey)).toObject();
|
||||
|
||||
QList<McpServerConfig> newConfigs;
|
||||
for (auto it = servers.begin(); it != servers.end(); ++it) {
|
||||
if (!it.value().isObject())
|
||||
continue;
|
||||
newConfigs.append(McpServerConfig::fromJson(it.key(), it.value().toObject()));
|
||||
}
|
||||
|
||||
const auto providers = toolsCapableProviders();
|
||||
|
||||
const bool masterEnabled = Settings::mcpSettings().enableMcpClients();
|
||||
|
||||
QList<McpServerConnection *> keep;
|
||||
QList<McpServerConnection *> oldConnections = m_connections;
|
||||
m_connections.clear();
|
||||
|
||||
for (const McpServerConfig &cfg : newConfigs) {
|
||||
McpServerConnection *existing = nullptr;
|
||||
for (auto *c : oldConnections) {
|
||||
if (c && c->config().name == cfg.name) {
|
||||
existing = c;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
const bool configUnchanged
|
||||
= existing && serializedEntry(existing->config()) == serializedEntry(cfg);
|
||||
|
||||
McpServerConnection *c = nullptr;
|
||||
if (configUnchanged) {
|
||||
oldConnections.removeAll(existing);
|
||||
c = existing;
|
||||
} else {
|
||||
if (existing) {
|
||||
oldConnections.removeAll(existing);
|
||||
existing->disconnectFromServer();
|
||||
existing->deleteLater();
|
||||
}
|
||||
c = new McpServerConnection(cfg, this);
|
||||
c->setProviders(providers);
|
||||
connect(
|
||||
c,
|
||||
&McpServerConnection::stateChanged,
|
||||
this,
|
||||
&McpClientsManager::serversChanged);
|
||||
}
|
||||
keep.append(c);
|
||||
|
||||
const bool wantRunning = cfg.enabled && masterEnabled;
|
||||
const bool currentlyRunning = c->state() == McpConnectionState::Connected
|
||||
|| c->state() == McpConnectionState::Connecting;
|
||||
|
||||
if (wantRunning && !currentlyRunning)
|
||||
c->connectToServer();
|
||||
else if (!wantRunning && currentlyRunning)
|
||||
c->disconnectFromServer();
|
||||
}
|
||||
|
||||
for (auto *c : oldConnections) {
|
||||
if (!c)
|
||||
continue;
|
||||
c->disconnectFromServer();
|
||||
c->deleteLater();
|
||||
}
|
||||
|
||||
m_connections = keep;
|
||||
|
||||
LOG_MESSAGE(QString("MCP clients: loaded %1 server(s) from %2")
|
||||
.arg(m_connections.size())
|
||||
.arg(configFilePath()));
|
||||
|
||||
emit serversChanged();
|
||||
}
|
||||
|
||||
} // namespace QodeAssist::Mcp
|
||||
Reference in New Issue
Block a user