mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 10:59:30 -04:00
93 lines
3.2 KiB
C++
93 lines
3.2 KiB
C++
// Copyright (C) 2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#include "ToolsRegistration.hpp"
|
|
|
|
#include <LLMQore/ToolsManager.hpp>
|
|
|
|
#include <settings/ToolsSettings.hpp>
|
|
#include <utils/aspects.h>
|
|
|
|
#include "BuildProjectTool.hpp"
|
|
#include "CreateNewFileTool.hpp"
|
|
#include "EditFileTool.hpp"
|
|
#include "ExecuteTerminalCommandTool.hpp"
|
|
#include "FindFileTool.hpp"
|
|
#include "GetIssuesListTool.hpp"
|
|
#include "ListProjectFilesTool.hpp"
|
|
#include "ProjectSearchTool.hpp"
|
|
#include "ReadFileTool.hpp"
|
|
#include "ReadOriginalHistoryTool.hpp"
|
|
#include "SkillTool.hpp"
|
|
#include "TodoTool.hpp"
|
|
|
|
namespace QodeAssist::Tools {
|
|
|
|
namespace {
|
|
|
|
template<typename ToolT>
|
|
void wireTool(::LLMQore::ToolsManager *manager,
|
|
Utils::BoolAspect &aspect,
|
|
const QString &toolId)
|
|
{
|
|
auto sync = [manager, toolId, &aspect]() {
|
|
const bool wanted = aspect.volatileValue();
|
|
const bool present = manager->tool(toolId) != nullptr;
|
|
if (wanted && !present) {
|
|
manager->addTool(new ToolT(manager));
|
|
} else if (!wanted && present) {
|
|
manager->removeTool(toolId);
|
|
}
|
|
};
|
|
|
|
sync();
|
|
|
|
QObject::connect(&aspect, &Utils::BoolAspect::volatileValueChanged, manager, sync);
|
|
QObject::connect(&aspect, &Utils::BaseAspect::changed, manager, sync);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
void registerQodeAssistTools(::LLMQore::ToolsManager *manager)
|
|
{
|
|
auto &s = Settings::toolsSettings();
|
|
|
|
wireTool<ListProjectFilesTool>(manager, s.enableListProjectFilesTool, "list_project_files");
|
|
wireTool<FindFileTool>(manager, s.enableFindFileTool, "find_file");
|
|
wireTool<ReadFileTool>(manager, s.enableReadFileTool, "read_file");
|
|
wireTool<ProjectSearchTool>(manager, s.enableProjectSearchTool, "search_project");
|
|
wireTool<CreateNewFileTool>(manager, s.enableCreateNewFileTool, "create_new_file");
|
|
wireTool<EditFileTool>(manager, s.enableEditFileTool, "edit_file");
|
|
wireTool<BuildProjectTool>(manager, s.enableBuildProjectTool, "build_project");
|
|
wireTool<GetIssuesListTool>(manager, s.enableGetIssuesListTool, "get_issues_list");
|
|
wireTool<ExecuteTerminalCommandTool>(
|
|
manager, s.enableTerminalCommandTool, "execute_terminal_command");
|
|
wireTool<TodoTool>(manager, s.enableTodoTool, "todo_tool");
|
|
wireTool<ReadOriginalHistoryTool>(
|
|
manager, s.enableReadOriginalHistoryTool, "read_original_history");
|
|
}
|
|
|
|
void registerSkillTool(
|
|
::LLMQore::ToolsManager *manager, Skills::SkillsManager *skillsManager)
|
|
{
|
|
Utils::BoolAspect &aspect = Settings::toolsSettings().enableSkillTool;
|
|
const QString toolId = QStringLiteral("load_skill");
|
|
|
|
auto sync = [manager, toolId, &aspect, skillsManager]() {
|
|
const bool wanted = aspect.volatileValue();
|
|
const bool present = manager->tool(toolId) != nullptr;
|
|
if (wanted && !present) {
|
|
manager->addTool(new SkillTool(skillsManager, manager));
|
|
} else if (!wanted && present) {
|
|
manager->removeTool(toolId);
|
|
}
|
|
};
|
|
|
|
sync();
|
|
|
|
QObject::connect(&aspect, &Utils::BoolAspect::volatileValueChanged, manager, sync);
|
|
QObject::connect(&aspect, &Utils::BaseAspect::changed, manager, sync);
|
|
}
|
|
|
|
} // namespace QodeAssist::Tools
|