mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 19:09:14 -04:00
42 lines
1.2 KiB
C++
42 lines
1.2 KiB
C++
// Copyright (C) 2025-2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#pragma once
|
|
|
|
#include <LLMQore/BaseTool.hpp>
|
|
#include <QObject>
|
|
|
|
namespace QodeAssist::Tools {
|
|
|
|
class ExecuteTerminalCommandTool : public ::LLMQore::BaseTool
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit ExecuteTerminalCommandTool(QObject *parent = nullptr);
|
|
|
|
QString id() const override;
|
|
QString displayName() const override;
|
|
QString description() const override;
|
|
QJsonObject parametersSchema() const override;
|
|
|
|
QFuture<LLMQore::ToolResult> executeAsync(const QJsonObject &input = QJsonObject()) override;
|
|
|
|
private:
|
|
bool isCommandAllowed(const QString &command) const;
|
|
bool isCommandSafe(const QString &command) const;
|
|
bool areArgumentsSafe(const QString &args) const;
|
|
QStringList getAllowedCommands() const;
|
|
QString getCommandDescription() const;
|
|
QString sanitizeOutput(const QString &output, qint64 maxSize) const;
|
|
|
|
int commandTimeoutMs() const;
|
|
|
|
// Constants for production safety
|
|
static constexpr qint64 MAX_OUTPUT_SIZE = 10 * 1024 * 1024; // 10 MB
|
|
static constexpr int MAX_COMMAND_LENGTH = 1024;
|
|
static constexpr int MAX_ARGS_LENGTH = 4096;
|
|
};
|
|
|
|
} // namespace QodeAssist::Tools
|
|
|