mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-07-19 09:40:56 -04:00
refactor: Restructure project into sources/tests layout and dissolve PluginLLMCore
This commit is contained in:
451
sources/refactor/QuickRefactorHandler.cpp
Normal file
451
sources/refactor/QuickRefactorHandler.cpp
Normal file
@@ -0,0 +1,451 @@
|
||||
// Copyright (C) 2025-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#include "refactor/QuickRefactorHandler.hpp"
|
||||
|
||||
#include <LLMQore/BaseClient.hpp>
|
||||
#include <QJsonArray>
|
||||
#include <QJsonDocument>
|
||||
#include <QUuid>
|
||||
|
||||
#include <context/DocumentContextReader.hpp>
|
||||
#include "refactor/ResponseCleaner.hpp"
|
||||
#include <context/DocumentReaderQtCreator.hpp>
|
||||
#include <context/Utils.hpp>
|
||||
#include "templates/PromptTemplateManager.hpp"
|
||||
#include "providers/ProvidersManager.hpp"
|
||||
#include "context/RulesLoader.hpp"
|
||||
#include <logger/Logger.hpp>
|
||||
#include <settings/ChatAssistantSettings.hpp>
|
||||
#include <settings/GeneralSettings.hpp>
|
||||
#include <settings/QuickRefactorSettings.hpp>
|
||||
#include <settings/ToolsSettings.hpp>
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
QuickRefactorHandler::QuickRefactorHandler(QObject *parent)
|
||||
: QObject(parent)
|
||||
, m_currentEditor(nullptr)
|
||||
, m_isRefactoringInProgress(false)
|
||||
, m_contextManager(this)
|
||||
{
|
||||
}
|
||||
|
||||
QuickRefactorHandler::~QuickRefactorHandler() {}
|
||||
|
||||
void QuickRefactorHandler::sendRefactorRequest(
|
||||
TextEditor::TextEditorWidget *editor, const QString &instructions)
|
||||
{
|
||||
if (m_isRefactoringInProgress) {
|
||||
cancelRequest();
|
||||
}
|
||||
|
||||
m_currentEditor = editor;
|
||||
|
||||
Utils::Text::Range range;
|
||||
if (editor->textCursor().hasSelection()) {
|
||||
QTextCursor cursor = editor->textCursor();
|
||||
int startPos = cursor.selectionStart();
|
||||
int endPos = cursor.selectionEnd();
|
||||
|
||||
QTextBlock startBlock = editor->document()->findBlock(startPos);
|
||||
int startLine = startBlock.blockNumber() + 1;
|
||||
int startColumn = startPos - startBlock.position();
|
||||
|
||||
QTextBlock endBlock = editor->document()->findBlock(endPos);
|
||||
int endLine = endBlock.blockNumber() + 1;
|
||||
int endColumn = endPos - endBlock.position();
|
||||
|
||||
Utils::Text::Position startPosition;
|
||||
startPosition.line = startLine;
|
||||
startPosition.column = startColumn;
|
||||
|
||||
Utils::Text::Position endPosition;
|
||||
endPosition.line = endLine;
|
||||
endPosition.column = endColumn;
|
||||
|
||||
range = Utils::Text::Range();
|
||||
range.begin = startPosition;
|
||||
range.end = endPosition;
|
||||
} else {
|
||||
QTextCursor cursor = editor->textCursor();
|
||||
int cursorPos = cursor.position();
|
||||
|
||||
QTextBlock block = editor->document()->findBlock(cursorPos);
|
||||
int line = block.blockNumber() + 1;
|
||||
int column = cursorPos - block.position();
|
||||
|
||||
Utils::Text::Position cursorPosition;
|
||||
cursorPosition.line = line;
|
||||
cursorPosition.column = column;
|
||||
range = Utils::Text::Range();
|
||||
range.begin = cursorPosition;
|
||||
range.end = cursorPosition;
|
||||
}
|
||||
|
||||
m_currentRange = range;
|
||||
prepareAndSendRequest(editor, instructions, range);
|
||||
}
|
||||
|
||||
void QuickRefactorHandler::prepareAndSendRequest(
|
||||
TextEditor::TextEditorWidget *editor,
|
||||
const QString &instructions,
|
||||
const Utils::Text::Range &range)
|
||||
{
|
||||
auto &settings = Settings::generalSettings();
|
||||
|
||||
auto &providerRegistry = Providers::ProvidersManager::instance();
|
||||
auto &promptManager = Templates::PromptTemplateManager::instance();
|
||||
|
||||
const auto providerName = settings.qrProvider();
|
||||
auto provider = providerRegistry.getProviderByName(providerName);
|
||||
|
||||
if (!provider) {
|
||||
QString error = QString("No provider found with name: %1").arg(providerName);
|
||||
LOG_MESSAGE(error);
|
||||
RefactorResult result;
|
||||
result.success = false;
|
||||
result.errorMessage = error;
|
||||
result.editor = editor;
|
||||
emit refactoringCompleted(result);
|
||||
return;
|
||||
}
|
||||
|
||||
const auto templateName = settings.qrTemplate();
|
||||
auto promptTemplate = promptManager.getChatTemplateByName(templateName);
|
||||
|
||||
if (!promptTemplate) {
|
||||
QString error = QString("No template found with name: %1").arg(templateName);
|
||||
LOG_MESSAGE(error);
|
||||
RefactorResult result;
|
||||
result.success = false;
|
||||
result.errorMessage = error;
|
||||
result.editor = editor;
|
||||
emit refactoringCompleted(result);
|
||||
return;
|
||||
}
|
||||
|
||||
QJsonObject payload{
|
||||
{"model", Settings::generalSettings().qrModel()}, {"stream", true}};
|
||||
|
||||
LLMCore::ContextData context = prepareContext(editor, range, instructions);
|
||||
|
||||
bool enableTools = Settings::quickRefactorSettings().useTools();
|
||||
bool enableThinking = Settings::quickRefactorSettings().useThinking();
|
||||
provider->prepareRequest(
|
||||
payload,
|
||||
promptTemplate,
|
||||
context,
|
||||
LLMCore::RequestType::QuickRefactoring,
|
||||
enableTools,
|
||||
enableThinking);
|
||||
|
||||
provider->client()->setMaxToolContinuations(
|
||||
Settings::toolsSettings().maxToolContinuations());
|
||||
|
||||
provider->client()->setTransferTimeout(
|
||||
static_cast<int>(Settings::generalSettings().requestTimeout() * 1000));
|
||||
|
||||
m_isRefactoringInProgress = true;
|
||||
|
||||
connect(
|
||||
provider->client(),
|
||||
&::LLMQore::BaseClient::requestCompleted,
|
||||
this,
|
||||
&QuickRefactorHandler::handleFullResponse,
|
||||
Qt::UniqueConnection);
|
||||
|
||||
connect(
|
||||
provider->client(),
|
||||
&::LLMQore::BaseClient::requestFinalized,
|
||||
this,
|
||||
&QuickRefactorHandler::handleRequestFinalized,
|
||||
Qt::UniqueConnection);
|
||||
|
||||
connect(
|
||||
provider->client(),
|
||||
&::LLMQore::BaseClient::requestFailed,
|
||||
this,
|
||||
&QuickRefactorHandler::handleRequestFailed,
|
||||
Qt::UniqueConnection);
|
||||
|
||||
const QString customEndpoint = Settings::generalSettings().qrCustomEndpoint();
|
||||
const QString endpoint = !customEndpoint.isEmpty() ? customEndpoint
|
||||
: promptTemplate->endpoint();
|
||||
auto requestId
|
||||
= provider->sendRequest(QUrl(Settings::generalSettings().qrUrl()), payload, endpoint);
|
||||
m_lastRequestId = requestId;
|
||||
QJsonObject request{{"id", requestId}};
|
||||
|
||||
m_activeRequests[requestId] = {request, provider};
|
||||
}
|
||||
|
||||
LLMCore::ContextData QuickRefactorHandler::prepareContext(
|
||||
TextEditor::TextEditorWidget *editor,
|
||||
const Utils::Text::Range &range,
|
||||
const QString &instructions)
|
||||
{
|
||||
LLMCore::ContextData context;
|
||||
|
||||
auto textDocument = editor->textDocument();
|
||||
Context::DocumentReaderQtCreator documentReader;
|
||||
auto documentInfo = documentReader.readDocument(textDocument->filePath().toUrlishString());
|
||||
|
||||
if (!documentInfo.document) {
|
||||
LOG_MESSAGE("Error: Document is not available");
|
||||
return context;
|
||||
}
|
||||
|
||||
QTextCursor cursor = editor->textCursor();
|
||||
int cursorPos = cursor.position();
|
||||
|
||||
Context::DocumentContextReader
|
||||
reader(documentInfo.document, documentInfo.mimeType, documentInfo.filePath);
|
||||
|
||||
QString taggedContent;
|
||||
bool readFullFile = Settings::quickRefactorSettings().readFullFile();
|
||||
|
||||
if (cursor.hasSelection()) {
|
||||
int selStart = cursor.selectionStart();
|
||||
int selEnd = cursor.selectionEnd();
|
||||
|
||||
QTextBlock startBlock = documentInfo.document->findBlock(selStart);
|
||||
int startLine = startBlock.blockNumber();
|
||||
int startColumn = selStart - startBlock.position();
|
||||
|
||||
QTextBlock endBlock = documentInfo.document->findBlock(selEnd);
|
||||
int endLine = endBlock.blockNumber();
|
||||
int endColumn = selEnd - endBlock.position();
|
||||
|
||||
QString contextBefore;
|
||||
if (readFullFile) {
|
||||
contextBefore = reader.readWholeFileBefore(startLine, startColumn);
|
||||
} else {
|
||||
contextBefore = reader.getContextBefore(
|
||||
startLine, startColumn, Settings::quickRefactorSettings().readStringsBeforeCursor() + 1);
|
||||
}
|
||||
|
||||
QString selectedText = cursor.selectedText();
|
||||
selectedText.replace(QChar(0x2029), "\n");
|
||||
|
||||
QString contextAfter;
|
||||
if (readFullFile) {
|
||||
contextAfter = reader.readWholeFileAfter(endLine, endColumn);
|
||||
} else {
|
||||
contextAfter = reader.getContextAfter(
|
||||
endLine, endColumn, Settings::quickRefactorSettings().readStringsAfterCursor() + 1);
|
||||
}
|
||||
|
||||
taggedContent = contextBefore;
|
||||
if (selStart == cursorPos) {
|
||||
taggedContent += "<cursor><selection_start>" + selectedText + "<selection_end>";
|
||||
} else {
|
||||
taggedContent += "<selection_start>" + selectedText + "<selection_end><cursor>";
|
||||
}
|
||||
taggedContent += contextAfter;
|
||||
} else {
|
||||
QTextBlock block = documentInfo.document->findBlock(cursorPos);
|
||||
int line = block.blockNumber();
|
||||
int column = cursorPos - block.position();
|
||||
|
||||
QString contextBefore;
|
||||
if (readFullFile) {
|
||||
contextBefore = reader.readWholeFileBefore(line, column);
|
||||
} else {
|
||||
contextBefore = reader.getContextBefore(
|
||||
line, column, Settings::quickRefactorSettings().readStringsBeforeCursor() + 1);
|
||||
}
|
||||
|
||||
QString contextAfter;
|
||||
if (readFullFile) {
|
||||
contextAfter = reader.readWholeFileAfter(line, column);
|
||||
} else {
|
||||
contextAfter = reader.getContextAfter(
|
||||
line, column, Settings::quickRefactorSettings().readStringsAfterCursor() + 1);
|
||||
}
|
||||
|
||||
taggedContent = contextBefore + "<cursor>" + contextAfter;
|
||||
}
|
||||
|
||||
QString systemPrompt = Settings::quickRefactorSettings().systemPrompt();
|
||||
|
||||
auto project = Context::RulesLoader::getActiveProject();
|
||||
if (project) {
|
||||
QString projectRules = Context::RulesLoader::loadRulesForProject(
|
||||
project, Context::RulesContext::QuickRefactor);
|
||||
|
||||
if (!projectRules.isEmpty()) {
|
||||
systemPrompt += "\n\n# Project Rules\n\n" + projectRules;
|
||||
LOG_MESSAGE("Loaded project rules for quick refactor");
|
||||
}
|
||||
}
|
||||
|
||||
systemPrompt += "\n\nFile information:";
|
||||
systemPrompt += "\nLanguage: " + documentInfo.mimeType;
|
||||
systemPrompt += "\nFile path: " + documentInfo.filePath;
|
||||
|
||||
systemPrompt += "\n\n# Code Context with Position Markers\n" + taggedContent;
|
||||
|
||||
systemPrompt += "\n\n# Output Requirements\n## What to Generate:";
|
||||
systemPrompt += cursor.hasSelection()
|
||||
? "\n- Generate ONLY the code that should REPLACE the selected text between "
|
||||
"<selection_start> and <selection_end> markers"
|
||||
"\n- Your output will completely replace the selected code"
|
||||
: "\n- Generate ONLY the code that should be INSERTED at the <cursor> position"
|
||||
"\n- Your output will be inserted at the cursor location";
|
||||
|
||||
systemPrompt += "\n\n## Formatting Rules:"
|
||||
"\n- Output ONLY the code itself, without ANY explanations or descriptions"
|
||||
"\n- Do NOT include markdown code blocks (no ```, no language tags)"
|
||||
"\n- Do NOT add comments explaining what you changed"
|
||||
"\n- Do NOT repeat existing code, be precise with context"
|
||||
"\n- Do NOT send in answer <cursor> or </cursor> and other tags"
|
||||
"\n- The output must be ready to insert directly into the editor as-is";
|
||||
|
||||
systemPrompt += "\n\n## Indentation and Whitespace:";
|
||||
|
||||
if (cursor.hasSelection()) {
|
||||
QTextBlock startBlock = documentInfo.document->findBlock(cursor.selectionStart());
|
||||
int leadingSpaces = 0;
|
||||
for (QChar c : startBlock.text()) {
|
||||
if (c == ' ') leadingSpaces++;
|
||||
else if (c == '\t') leadingSpaces += 4;
|
||||
else break;
|
||||
}
|
||||
if (leadingSpaces > 0) {
|
||||
systemPrompt += QString("\n- CRITICAL: The code to replace starts with %1 spaces of indentation"
|
||||
"\n- Your output MUST start with exactly %1 spaces (or equivalent tabs)"
|
||||
"\n- Each line in your output must maintain this base indentation")
|
||||
.arg(leadingSpaces);
|
||||
}
|
||||
systemPrompt += "\n- PRESERVE all indentation from the original code";
|
||||
} else {
|
||||
QTextBlock block = documentInfo.document->findBlock(cursorPos);
|
||||
QString lineText = block.text();
|
||||
int leadingSpaces = 0;
|
||||
for (QChar c : lineText) {
|
||||
if (c == ' ') leadingSpaces++;
|
||||
else if (c == '\t') leadingSpaces += 4;
|
||||
else break;
|
||||
}
|
||||
if (leadingSpaces > 0) {
|
||||
systemPrompt += QString("\n- CRITICAL: Current line has %1 spaces of indentation"
|
||||
"\n- If generating multiline code, EVERY line must start with at least %1 spaces"
|
||||
"\n- If generating single-line code, it will be inserted inline (no indentation needed)")
|
||||
.arg(leadingSpaces);
|
||||
}
|
||||
}
|
||||
|
||||
systemPrompt += "\n- Use the same indentation style (spaces or tabs) as the surrounding code"
|
||||
"\n- Maintain consistent indentation for nested blocks"
|
||||
"\n- Do NOT remove or reduce the base indentation level"
|
||||
"\n\n## Code Style:"
|
||||
"\n- Match the coding style of the surrounding code (naming, spacing, braces, etc.)"
|
||||
"\n- Preserve the original code structure when possible"
|
||||
"\n- Only change what is necessary to fulfill the user's request";
|
||||
|
||||
if (Settings::quickRefactorSettings().useOpenFilesInQuickRefactor()) {
|
||||
systemPrompt += "\n\n" + m_contextManager.openedFilesContext({documentInfo.filePath});
|
||||
}
|
||||
|
||||
context.systemPrompt = systemPrompt;
|
||||
|
||||
QVector<LLMCore::Message> messages;
|
||||
messages.append(
|
||||
{"user",
|
||||
instructions.isEmpty() ? "Refactor the code to improve its quality and maintainability."
|
||||
: instructions});
|
||||
context.history = messages;
|
||||
|
||||
return context;
|
||||
}
|
||||
|
||||
void QuickRefactorHandler::handleLLMResponse(
|
||||
const QString &response, const QJsonObject &request, bool isComplete)
|
||||
{
|
||||
if (request["id"].toString() != m_lastRequestId) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (isComplete) {
|
||||
m_isRefactoringInProgress = false;
|
||||
QString cleanedResponse = ResponseCleaner::clean(response);
|
||||
|
||||
RefactorResult result;
|
||||
result.newText = cleanedResponse;
|
||||
result.insertRange = m_currentRange;
|
||||
result.success = true;
|
||||
result.editor = m_currentEditor;
|
||||
|
||||
LOG_MESSAGE("Refactoring completed successfully. New code to insert: ");
|
||||
LOG_MESSAGE("---------- BEGIN REFACTORED CODE ----------");
|
||||
LOG_MESSAGE(cleanedResponse);
|
||||
LOG_MESSAGE("----------- END REFACTORED CODE -----------");
|
||||
|
||||
emit refactoringCompleted(result);
|
||||
}
|
||||
}
|
||||
|
||||
void QuickRefactorHandler::cancelRequest()
|
||||
{
|
||||
if (!m_isRefactoringInProgress)
|
||||
return;
|
||||
|
||||
const auto id = m_lastRequestId;
|
||||
m_isRefactoringInProgress = false;
|
||||
m_lastRequestId.clear();
|
||||
|
||||
auto it = m_activeRequests.find(id);
|
||||
if (it != m_activeRequests.end()) {
|
||||
auto provider = it.value().provider;
|
||||
m_activeRequests.erase(it);
|
||||
if (provider)
|
||||
provider->cancelRequest(id);
|
||||
}
|
||||
|
||||
RefactorResult result;
|
||||
result.success = false;
|
||||
result.errorMessage = "Refactoring request was cancelled";
|
||||
emit refactoringCompleted(result);
|
||||
}
|
||||
|
||||
void QuickRefactorHandler::handleFullResponse(const QString &requestId, const QString &fullText)
|
||||
{
|
||||
if (requestId == m_lastRequestId) {
|
||||
m_activeRequests.remove(requestId);
|
||||
QJsonObject request{{"id", requestId}};
|
||||
handleLLMResponse(fullText, request, true);
|
||||
}
|
||||
}
|
||||
|
||||
void QuickRefactorHandler::handleRequestFinalized(
|
||||
const ::LLMQore::RequestID &requestId, const ::LLMQore::CompletionInfo &info)
|
||||
{
|
||||
if (requestId != m_lastRequestId || !info.usage)
|
||||
return;
|
||||
|
||||
const auto &u = *info.usage;
|
||||
LOG_MESSAGE(
|
||||
QString("Quick refactor usage [%1]: prompt=%2 completion=%3 cached=%4 reasoning=%5")
|
||||
.arg(requestId)
|
||||
.arg(u.promptTokens)
|
||||
.arg(u.completionTokens)
|
||||
.arg(u.cachedPromptTokens)
|
||||
.arg(u.reasoningTokens));
|
||||
}
|
||||
|
||||
void QuickRefactorHandler::handleRequestFailed(const QString &requestId, const QString &error)
|
||||
{
|
||||
if (requestId == m_lastRequestId) {
|
||||
m_activeRequests.remove(requestId);
|
||||
m_isRefactoringInProgress = false;
|
||||
RefactorResult result;
|
||||
result.success = false;
|
||||
result.errorMessage = error;
|
||||
result.editor = m_currentEditor;
|
||||
emit refactoringCompleted(result);
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace QodeAssist
|
||||
78
sources/refactor/QuickRefactorHandler.hpp
Normal file
78
sources/refactor/QuickRefactorHandler.hpp
Normal file
@@ -0,0 +1,78 @@
|
||||
// Copyright (C) 2025-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <QJsonObject>
|
||||
#include <QObject>
|
||||
|
||||
#include <LLMQore/BaseClient.hpp>
|
||||
#include <texteditor/texteditor.h>
|
||||
#include <utils/textutils.h>
|
||||
|
||||
#include <context/ContextManager.hpp>
|
||||
#include <context/IDocumentReader.hpp>
|
||||
#include "llmcore/ContextData.hpp"
|
||||
#include "providers/Provider.hpp"
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
struct RefactorResult
|
||||
{
|
||||
QString newText;
|
||||
Utils::Text::Range insertRange;
|
||||
bool success;
|
||||
QString errorMessage;
|
||||
TextEditor::TextEditorWidget *editor{nullptr};
|
||||
};
|
||||
|
||||
class QuickRefactorHandler : public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
|
||||
public:
|
||||
explicit QuickRefactorHandler(QObject *parent = nullptr);
|
||||
~QuickRefactorHandler() override;
|
||||
|
||||
void sendRefactorRequest(TextEditor::TextEditorWidget *editor, const QString &instructions);
|
||||
|
||||
void cancelRequest();
|
||||
bool isProcessing() const { return m_isRefactoringInProgress; }
|
||||
|
||||
signals:
|
||||
void refactoringCompleted(const QodeAssist::RefactorResult &result);
|
||||
|
||||
private slots:
|
||||
void handleFullResponse(const QString &requestId, const QString &fullText);
|
||||
void handleRequestFinalized(
|
||||
const ::LLMQore::RequestID &requestId, const ::LLMQore::CompletionInfo &info);
|
||||
void handleRequestFailed(const QString &requestId, const QString &error);
|
||||
|
||||
private:
|
||||
void prepareAndSendRequest(
|
||||
TextEditor::TextEditorWidget *editor,
|
||||
const QString &instructions,
|
||||
const Utils::Text::Range &range);
|
||||
|
||||
void handleLLMResponse(const QString &response, const QJsonObject &request, bool isComplete);
|
||||
LLMCore::ContextData prepareContext(
|
||||
TextEditor::TextEditorWidget *editor,
|
||||
const Utils::Text::Range &range,
|
||||
const QString &instructions);
|
||||
|
||||
struct RequestContext
|
||||
{
|
||||
QJsonObject originalRequest;
|
||||
Providers::Provider *provider;
|
||||
};
|
||||
|
||||
QHash<QString, RequestContext> m_activeRequests;
|
||||
TextEditor::TextEditorWidget *m_currentEditor;
|
||||
Utils::Text::Range m_currentRange;
|
||||
bool m_isRefactoringInProgress;
|
||||
QString m_lastRequestId;
|
||||
Context::ContextManager m_contextManager;
|
||||
};
|
||||
|
||||
} // namespace QodeAssist
|
||||
100
sources/refactor/RefactorContextHelper.hpp
Normal file
100
sources/refactor/RefactorContextHelper.hpp
Normal file
@@ -0,0 +1,100 @@
|
||||
// Copyright (C) 2025-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <QString>
|
||||
#include <QTextCursor>
|
||||
#include <QTextBlock>
|
||||
|
||||
#include <texteditor/texteditor.h>
|
||||
#include <utils/textutils.h>
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
struct RefactorContext
|
||||
{
|
||||
QString originalText;
|
||||
QString textBeforeCursor;
|
||||
QString textAfterCursor;
|
||||
QString contextBefore;
|
||||
QString contextAfter;
|
||||
int startPos{0};
|
||||
int endPos{0};
|
||||
bool isInsertion{false};
|
||||
};
|
||||
|
||||
class RefactorContextHelper
|
||||
{
|
||||
public:
|
||||
static RefactorContext extractContext(TextEditor::TextEditorWidget *editor,
|
||||
const Utils::Text::Range &range,
|
||||
int contextLinesBefore = 3,
|
||||
int contextLinesAfter = 3)
|
||||
{
|
||||
RefactorContext ctx;
|
||||
if (!editor) {
|
||||
return ctx;
|
||||
}
|
||||
|
||||
QTextDocument *doc = editor->document();
|
||||
ctx.startPos = range.begin.toPositionInDocument(doc);
|
||||
ctx.endPos = range.end.toPositionInDocument(doc);
|
||||
ctx.isInsertion = (ctx.startPos == ctx.endPos);
|
||||
|
||||
if (!ctx.isInsertion) {
|
||||
QTextCursor cursor(doc);
|
||||
cursor.setPosition(ctx.startPos);
|
||||
cursor.setPosition(ctx.endPos, QTextCursor::KeepAnchor);
|
||||
ctx.originalText = cursor.selectedText();
|
||||
ctx.originalText.replace(QChar(0x2029), "\n");
|
||||
} else {
|
||||
QTextCursor cursor(doc);
|
||||
cursor.setPosition(ctx.startPos);
|
||||
|
||||
int posInBlock = cursor.positionInBlock();
|
||||
cursor.movePosition(QTextCursor::StartOfBlock);
|
||||
cursor.movePosition(QTextCursor::Right, QTextCursor::KeepAnchor, posInBlock);
|
||||
ctx.textBeforeCursor = cursor.selectedText();
|
||||
|
||||
cursor.setPosition(ctx.startPos);
|
||||
cursor.movePosition(QTextCursor::EndOfBlock, QTextCursor::KeepAnchor);
|
||||
ctx.textAfterCursor = cursor.selectedText();
|
||||
}
|
||||
|
||||
ctx.contextBefore = extractContextLines(doc, ctx.startPos, contextLinesBefore, true);
|
||||
ctx.contextAfter = extractContextLines(doc, ctx.endPos, contextLinesAfter, false);
|
||||
|
||||
return ctx;
|
||||
}
|
||||
|
||||
private:
|
||||
static QString extractContextLines(QTextDocument *doc, int position, int lineCount, bool before)
|
||||
{
|
||||
QTextCursor cursor(doc);
|
||||
cursor.setPosition(position);
|
||||
QTextBlock currentBlock = cursor.block();
|
||||
|
||||
QStringList lines;
|
||||
|
||||
if (before) {
|
||||
QTextBlock block = currentBlock.previous();
|
||||
for (int i = 0; i < lineCount && block.isValid(); ++i) {
|
||||
lines.prepend(block.text());
|
||||
block = block.previous();
|
||||
}
|
||||
} else {
|
||||
QTextBlock block = currentBlock.next();
|
||||
for (int i = 0; i < lineCount && block.isValid(); ++i) {
|
||||
lines.append(block.text());
|
||||
block = block.next();
|
||||
}
|
||||
}
|
||||
|
||||
return lines.join('\n');
|
||||
}
|
||||
};
|
||||
|
||||
} // namespace QodeAssist
|
||||
|
||||
187
sources/refactor/RefactorSuggestion.cpp
Normal file
187
sources/refactor/RefactorSuggestion.cpp
Normal file
@@ -0,0 +1,187 @@
|
||||
// Copyright (C) 2024-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#include "refactor/RefactorSuggestion.hpp"
|
||||
#include "completion/LLMSuggestion.hpp"
|
||||
|
||||
#include <QTextBlock>
|
||||
#include <QTextCursor>
|
||||
#include <QTextDocument>
|
||||
|
||||
#include <texteditor/texteditor.h>
|
||||
#include <logger/Logger.hpp>
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
namespace {
|
||||
QString extractLeadingWhitespace(const QString &text)
|
||||
{
|
||||
QString indent;
|
||||
int firstLineEnd = text.indexOf('\n');
|
||||
QString firstLine = (firstLineEnd != -1) ? text.left(firstLineEnd) : text;
|
||||
for (int i = 0; i < firstLine.length(); ++i) {
|
||||
if (firstLine[i].isSpace()) {
|
||||
indent += firstLine[i];
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
return indent;
|
||||
}
|
||||
} // anonymous namespace
|
||||
|
||||
RefactorSuggestion::RefactorSuggestion(const Data &suggestion, QTextDocument *sourceDocument)
|
||||
: TextEditor::TextSuggestion([&suggestion, sourceDocument]() {
|
||||
Data expandedData = suggestion;
|
||||
|
||||
int startPos = suggestion.range.begin.toPositionInDocument(sourceDocument);
|
||||
int endPos = suggestion.range.end.toPositionInDocument(sourceDocument);
|
||||
startPos = qBound(0, startPos, sourceDocument->characterCount());
|
||||
endPos = qBound(0, endPos, sourceDocument->characterCount());
|
||||
|
||||
if (startPos != endPos) {
|
||||
QTextCursor startCursor(sourceDocument);
|
||||
startCursor.setPosition(startPos);
|
||||
int startPosInBlock = startCursor.positionInBlock();
|
||||
|
||||
if (startPosInBlock > 0) {
|
||||
startCursor.movePosition(QTextCursor::StartOfBlock);
|
||||
}
|
||||
|
||||
QTextCursor endCursor(sourceDocument);
|
||||
endCursor.setPosition(endPos);
|
||||
int endPosInBlock = endCursor.positionInBlock();
|
||||
|
||||
if (endPosInBlock > 0) {
|
||||
endCursor.movePosition(QTextCursor::EndOfBlock);
|
||||
if (!endCursor.atEnd()) {
|
||||
endCursor.movePosition(QTextCursor::NextCharacter);
|
||||
}
|
||||
}
|
||||
|
||||
Utils::Text::Position expandedBegin = Utils::Text::Position::fromPositionInDocument(
|
||||
sourceDocument, startCursor.position());
|
||||
Utils::Text::Position expandedEnd = Utils::Text::Position::fromPositionInDocument(
|
||||
sourceDocument, endCursor.position());
|
||||
|
||||
expandedData.range = Utils::Text::Range(expandedBegin, expandedEnd);
|
||||
}
|
||||
|
||||
return expandedData;
|
||||
}(), sourceDocument)
|
||||
, m_suggestionData(suggestion)
|
||||
{
|
||||
const QString refactoredText = suggestion.text;
|
||||
|
||||
int startPos = suggestion.range.begin.toPositionInDocument(sourceDocument);
|
||||
int endPos = suggestion.range.end.toPositionInDocument(sourceDocument);
|
||||
startPos = qBound(0, startPos, sourceDocument->characterCount());
|
||||
endPos = qBound(0, endPos, sourceDocument->characterCount());
|
||||
|
||||
QTextCursor startCursor(sourceDocument);
|
||||
startCursor.setPosition(startPos);
|
||||
|
||||
if (startPos == endPos) {
|
||||
QTextBlock block = startCursor.block();
|
||||
QString blockText = block.text();
|
||||
int startPosInBlock = startCursor.positionInBlock();
|
||||
|
||||
QString leftText = blockText.left(startPosInBlock);
|
||||
QString rightText = blockText.mid(startPosInBlock);
|
||||
|
||||
QString displayText = leftText + refactoredText + rightText;
|
||||
replacementDocument()->setPlainText(displayText);
|
||||
|
||||
} else {
|
||||
QTextCursor fullLinesCursor(sourceDocument);
|
||||
fullLinesCursor.setPosition(startPos);
|
||||
fullLinesCursor.movePosition(QTextCursor::StartOfBlock);
|
||||
int fullLinesStart = fullLinesCursor.position();
|
||||
|
||||
fullLinesCursor.setPosition(endPos);
|
||||
fullLinesCursor.movePosition(QTextCursor::EndOfBlock);
|
||||
int fullLinesEnd = fullLinesCursor.position();
|
||||
|
||||
fullLinesCursor.setPosition(fullLinesStart);
|
||||
fullLinesCursor.setPosition(fullLinesEnd, QTextCursor::KeepAnchor);
|
||||
QString fullLinesText = fullLinesCursor.selectedText();
|
||||
fullLinesText.replace(QChar(0x2029), "\n");
|
||||
|
||||
QString oldIndent = extractLeadingWhitespace(fullLinesText);
|
||||
QString newIndent = extractLeadingWhitespace(refactoredText);
|
||||
|
||||
QString displayText = refactoredText;
|
||||
if (newIndent.length() < oldIndent.length()) {
|
||||
QString indentDiff = oldIndent.left(oldIndent.length() - newIndent.length());
|
||||
QStringList lines = refactoredText.split('\n');
|
||||
if (!lines.isEmpty() && !lines[0].trimmed().isEmpty()) {
|
||||
lines[0] = indentDiff + lines[0];
|
||||
displayText = lines.join('\n');
|
||||
}
|
||||
}
|
||||
|
||||
replacementDocument()->setPlainText(displayText);
|
||||
}
|
||||
}
|
||||
|
||||
bool RefactorSuggestion::apply()
|
||||
{
|
||||
const QString text = m_suggestionData.text;
|
||||
const Utils::Text::Range range = m_suggestionData.range;
|
||||
|
||||
const QTextCursor startCursor = range.begin.toTextCursor(sourceDocument());
|
||||
const QTextCursor endCursor = range.end.toTextCursor(sourceDocument());
|
||||
|
||||
const int startPos = startCursor.position();
|
||||
const int endPos = endCursor.position();
|
||||
|
||||
QTextCursor editCursor(sourceDocument());
|
||||
editCursor.beginEditBlock();
|
||||
|
||||
if (startPos == endPos) {
|
||||
editCursor.setPosition(startPos);
|
||||
editCursor.insertText(text);
|
||||
} else {
|
||||
editCursor.setPosition(startPos);
|
||||
editCursor.setPosition(endPos, QTextCursor::KeepAnchor);
|
||||
QString selectedText = editCursor.selectedText();
|
||||
selectedText.replace(QChar(0x2029), "\n");
|
||||
|
||||
QString oldIndent = extractLeadingWhitespace(selectedText);
|
||||
QString newIndent = extractLeadingWhitespace(text);
|
||||
|
||||
QString textToInsert = text;
|
||||
if (newIndent.length() < oldIndent.length()) {
|
||||
QString indentDiff = oldIndent.left(oldIndent.length() - newIndent.length());
|
||||
QStringList lines = text.split('\n');
|
||||
if (!lines.isEmpty() && !lines[0].trimmed().isEmpty()) {
|
||||
lines[0] = indentDiff + lines[0];
|
||||
textToInsert = lines.join('\n');
|
||||
}
|
||||
}
|
||||
|
||||
editCursor.setPosition(startPos);
|
||||
editCursor.setPosition(endPos, QTextCursor::KeepAnchor);
|
||||
editCursor.removeSelectedText();
|
||||
editCursor.insertText(textToInsert);
|
||||
}
|
||||
|
||||
editCursor.endEditBlock();
|
||||
return true;
|
||||
}
|
||||
|
||||
bool RefactorSuggestion::applyWord(TextEditor::TextEditorWidget *widget)
|
||||
{
|
||||
Q_UNUSED(widget)
|
||||
return apply();
|
||||
}
|
||||
|
||||
bool RefactorSuggestion::applyLine(TextEditor::TextEditorWidget *widget)
|
||||
{
|
||||
Q_UNUSED(widget)
|
||||
return apply();
|
||||
}
|
||||
|
||||
} // namespace QodeAssist
|
||||
|
||||
28
sources/refactor/RefactorSuggestion.hpp
Normal file
28
sources/refactor/RefactorSuggestion.hpp
Normal file
@@ -0,0 +1,28 @@
|
||||
// Copyright (C) 2024-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <texteditor/texteditor.h>
|
||||
#include <texteditor/textsuggestion.h>
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
class RefactorSuggestion : public TextEditor::TextSuggestion
|
||||
{
|
||||
public:
|
||||
RefactorSuggestion(const Data &suggestion, QTextDocument *sourceDocument);
|
||||
|
||||
bool apply() override;
|
||||
|
||||
bool applyWord(TextEditor::TextEditorWidget *widget) override;
|
||||
|
||||
bool applyLine(TextEditor::TextEditorWidget *widget) override;
|
||||
|
||||
private:
|
||||
Data m_suggestionData;
|
||||
};
|
||||
|
||||
} // namespace QodeAssist
|
||||
|
||||
195
sources/refactor/RefactorSuggestionHoverHandler.cpp
Normal file
195
sources/refactor/RefactorSuggestionHoverHandler.cpp
Normal file
@@ -0,0 +1,195 @@
|
||||
// Copyright (C) 2025-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#include "refactor/RefactorSuggestionHoverHandler.hpp"
|
||||
#include "refactor/RefactorSuggestion.hpp"
|
||||
|
||||
#include <QColor>
|
||||
#include <QHBoxLayout>
|
||||
#include <QPushButton>
|
||||
#include <QScopeGuard>
|
||||
#include <QTextBlock>
|
||||
#include <QTextCursor>
|
||||
#include <QWidget>
|
||||
|
||||
#include <texteditor/textdocumentlayout.h>
|
||||
#include <texteditor/texteditor.h>
|
||||
#include <utils/theme/theme.h>
|
||||
#include <utils/tooltip/tooltip.h>
|
||||
|
||||
#include <logger/Logger.hpp>
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
RefactorSuggestionHoverHandler::RefactorSuggestionHoverHandler()
|
||||
{
|
||||
setPriority(Priority_Suggestion);
|
||||
}
|
||||
|
||||
void RefactorSuggestionHoverHandler::setSuggestionRange(const Utils::Text::Range &range)
|
||||
{
|
||||
m_suggestionRange = range;
|
||||
m_hasSuggestion = true;
|
||||
}
|
||||
|
||||
void RefactorSuggestionHoverHandler::clearSuggestionRange()
|
||||
{
|
||||
m_hasSuggestion = false;
|
||||
}
|
||||
|
||||
void RefactorSuggestionHoverHandler::identifyMatch(
|
||||
TextEditor::TextEditorWidget *editorWidget,
|
||||
int pos,
|
||||
ReportPriority report)
|
||||
{
|
||||
|
||||
QScopeGuard cleanup([&] { report(Priority_None); });
|
||||
|
||||
if (!editorWidget->suggestionVisible()) {
|
||||
return;
|
||||
}
|
||||
|
||||
QTextCursor cursor(editorWidget->document());
|
||||
cursor.setPosition(pos);
|
||||
m_block = cursor.block();
|
||||
|
||||
#if QODEASSIST_QT_CREATOR_VERSION_MAJOR >= 17
|
||||
auto *suggestion = dynamic_cast<RefactorSuggestion *>(
|
||||
TextEditor::TextBlockUserData::suggestion(m_block));
|
||||
#else
|
||||
auto *userData = TextEditor::TextDocumentLayout::textUserData(m_block);
|
||||
if (!userData) {
|
||||
LOG_MESSAGE("RefactorSuggestionHoverHandler: No user data in block");
|
||||
return;
|
||||
}
|
||||
|
||||
auto *suggestion = dynamic_cast<RefactorSuggestion *>(userData->suggestion());
|
||||
#endif
|
||||
|
||||
if (!suggestion) {
|
||||
return;
|
||||
}
|
||||
|
||||
cleanup.dismiss();
|
||||
report(Priority_Suggestion);
|
||||
}
|
||||
|
||||
void RefactorSuggestionHoverHandler::operateTooltip(
|
||||
TextEditor::TextEditorWidget *editorWidget,
|
||||
const QPoint &point)
|
||||
{
|
||||
Q_UNUSED(point)
|
||||
|
||||
#if QODEASSIST_QT_CREATOR_VERSION_MAJOR >= 17
|
||||
auto *suggestion = dynamic_cast<RefactorSuggestion *>(
|
||||
TextEditor::TextBlockUserData::suggestion(m_block));
|
||||
#else
|
||||
auto *userData = TextEditor::TextDocumentLayout::textUserData(m_block);
|
||||
if (!userData) {
|
||||
LOG_MESSAGE("RefactorSuggestionHoverHandler::operateTooltip: No user data in block");
|
||||
return;
|
||||
}
|
||||
|
||||
auto *suggestion = dynamic_cast<RefactorSuggestion *>(userData->suggestion());
|
||||
#endif
|
||||
|
||||
if (!suggestion) {
|
||||
return;
|
||||
}
|
||||
|
||||
auto *widget = new QWidget();
|
||||
auto *layout = new QHBoxLayout(widget);
|
||||
layout->setContentsMargins(4, 3, 4, 3);
|
||||
layout->setSpacing(6);
|
||||
|
||||
const QColor normalBg = Utils::creatorColor(Utils::Theme::BackgroundColorNormal);
|
||||
const QColor hoverBg = Utils::creatorColor(Utils::Theme::BackgroundColorHover);
|
||||
const QColor selectedBg = Utils::creatorColor(Utils::Theme::BackgroundColorSelected);
|
||||
const QColor textColor = Utils::creatorColor(Utils::Theme::TextColorNormal);
|
||||
const QColor borderColor = Utils::creatorColor(Utils::Theme::SplitterColor);
|
||||
const QColor successColor = Utils::creatorColor(Utils::Theme::TextColorNormal);
|
||||
const QColor errorColor = Utils::creatorColor(Utils::Theme::TextColorError);
|
||||
|
||||
auto *applyButton = new QPushButton("✓ Apply", widget);
|
||||
applyButton->setFocusPolicy(Qt::NoFocus);
|
||||
applyButton->setToolTip("Apply refactoring (Tab)");
|
||||
applyButton->setCursor(Qt::PointingHandCursor);
|
||||
applyButton->setStyleSheet(QString(
|
||||
"QPushButton {"
|
||||
" background-color: %1;"
|
||||
" color: %2;"
|
||||
" border: 1px solid %3;"
|
||||
" border-radius: 3px;"
|
||||
" padding: 4px 12px;"
|
||||
" font-weight: bold;"
|
||||
" font-size: 11px;"
|
||||
" min-width: 60px;"
|
||||
"}"
|
||||
"QPushButton:hover {"
|
||||
" background-color: %4;"
|
||||
" border-color: %2;"
|
||||
"}"
|
||||
"QPushButton:pressed {"
|
||||
" background-color: %5;"
|
||||
"}")
|
||||
.arg(selectedBg.name())
|
||||
.arg(successColor.name())
|
||||
.arg(borderColor.name())
|
||||
.arg(selectedBg.lighter(110).name())
|
||||
.arg(selectedBg.darker(110).name()));
|
||||
QObject::connect(applyButton, &QPushButton::clicked, widget, [this]() {
|
||||
Utils::ToolTip::hide();
|
||||
if (m_applyCallback) {
|
||||
m_applyCallback();
|
||||
}
|
||||
});
|
||||
|
||||
auto *dismissButton = new QPushButton("✕ Dismiss", widget);
|
||||
dismissButton->setFocusPolicy(Qt::NoFocus);
|
||||
dismissButton->setToolTip("Dismiss refactoring (Esc)");
|
||||
dismissButton->setCursor(Qt::PointingHandCursor);
|
||||
dismissButton->setStyleSheet(QString(
|
||||
"QPushButton {"
|
||||
" background-color: %1;"
|
||||
" color: %2;"
|
||||
" border: 1px solid %3;"
|
||||
" border-radius: 3px;"
|
||||
" padding: 4px 12px;"
|
||||
" font-size: 11px;"
|
||||
" min-width: 60px;"
|
||||
"}"
|
||||
"QPushButton:hover {"
|
||||
" background-color: %4;"
|
||||
" color: %5;"
|
||||
" border-color: %5;"
|
||||
"}"
|
||||
"QPushButton:pressed {"
|
||||
" background-color: %6;"
|
||||
"}")
|
||||
.arg(normalBg.name())
|
||||
.arg(textColor.name())
|
||||
.arg(borderColor.name())
|
||||
.arg(hoverBg.name())
|
||||
.arg(errorColor.name())
|
||||
.arg(hoverBg.darker(110).name()));
|
||||
QObject::connect(dismissButton, &QPushButton::clicked, widget, [this]() {
|
||||
Utils::ToolTip::hide();
|
||||
if (m_dismissCallback) {
|
||||
m_dismissCallback();
|
||||
}
|
||||
});
|
||||
|
||||
layout->addWidget(applyButton);
|
||||
layout->addWidget(dismissButton);
|
||||
|
||||
const QRect cursorRect = editorWidget->cursorRect(editorWidget->textCursor());
|
||||
QPoint pos = editorWidget->viewport()->mapToGlobal(cursorRect.topLeft())
|
||||
- Utils::ToolTip::offsetFromPosition();
|
||||
pos.ry() -= widget->sizeHint().height();
|
||||
|
||||
Utils::ToolTip::show(pos, widget, editorWidget);
|
||||
}
|
||||
|
||||
} // namespace QodeAssist
|
||||
|
||||
57
sources/refactor/RefactorSuggestionHoverHandler.hpp
Normal file
57
sources/refactor/RefactorSuggestionHoverHandler.hpp
Normal file
@@ -0,0 +1,57 @@
|
||||
// Copyright (C) 2025-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <functional>
|
||||
#include <QTextBlock>
|
||||
|
||||
#include <texteditor/basehoverhandler.h>
|
||||
#include <utils/textutils.h>
|
||||
|
||||
namespace TextEditor {
|
||||
class TextEditorWidget;
|
||||
}
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
/**
|
||||
* @brief Hover handler for refactoring suggestions
|
||||
*
|
||||
* Shows interactive tooltip with Apply/Dismiss buttons when hovering over
|
||||
* a refactoring suggestion in the editor.
|
||||
*/
|
||||
class RefactorSuggestionHoverHandler : public TextEditor::BaseHoverHandler
|
||||
{
|
||||
public:
|
||||
using ApplyCallback = std::function<void()>;
|
||||
using DismissCallback = std::function<void()>;
|
||||
|
||||
RefactorSuggestionHoverHandler();
|
||||
|
||||
void setSuggestionRange(const Utils::Text::Range &range);
|
||||
void clearSuggestionRange();
|
||||
bool hasSuggestion() const { return m_hasSuggestion; }
|
||||
|
||||
void setApplyCallback(ApplyCallback callback) { m_applyCallback = std::move(callback); }
|
||||
void setDismissCallback(DismissCallback callback) { m_dismissCallback = std::move(callback); }
|
||||
|
||||
protected:
|
||||
void identifyMatch(TextEditor::TextEditorWidget *editorWidget,
|
||||
int pos,
|
||||
ReportPriority report) override;
|
||||
|
||||
void operateTooltip(TextEditor::TextEditorWidget *editorWidget,
|
||||
const QPoint &point) override;
|
||||
|
||||
private:
|
||||
Utils::Text::Range m_suggestionRange;
|
||||
bool m_hasSuggestion = false;
|
||||
ApplyCallback m_applyCallback;
|
||||
DismissCallback m_dismissCallback;
|
||||
QTextBlock m_block;
|
||||
};
|
||||
|
||||
} // namespace QodeAssist
|
||||
|
||||
104
sources/refactor/ResponseCleaner.hpp
Normal file
104
sources/refactor/ResponseCleaner.hpp
Normal file
@@ -0,0 +1,104 @@
|
||||
// Copyright (C) 2025-2026 Petr Mironychev
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <QString>
|
||||
#include <QStringList>
|
||||
#include <QRegularExpression>
|
||||
|
||||
namespace QodeAssist {
|
||||
|
||||
class ResponseCleaner
|
||||
{
|
||||
public:
|
||||
static QString clean(const QString &response)
|
||||
{
|
||||
QString cleaned = removeCodeBlocks(response);
|
||||
cleaned = trimWhitespace(cleaned);
|
||||
cleaned = removeExplanations(cleaned);
|
||||
return cleaned;
|
||||
}
|
||||
|
||||
private:
|
||||
static QString removeCodeBlocks(const QString &text)
|
||||
{
|
||||
if (!text.contains("```")) {
|
||||
return text;
|
||||
}
|
||||
|
||||
QRegularExpression codeBlockRegex("```\\w*\\n([\\s\\S]*?)```");
|
||||
QRegularExpressionMatch match = codeBlockRegex.match(text);
|
||||
if (match.hasMatch()) {
|
||||
return match.captured(1);
|
||||
}
|
||||
|
||||
int firstFence = text.indexOf("```");
|
||||
int lastFence = text.lastIndexOf("```");
|
||||
if (firstFence != -1 && lastFence > firstFence) {
|
||||
int firstNewLine = text.indexOf('\n', firstFence);
|
||||
if (firstNewLine != -1) {
|
||||
return text.mid(firstNewLine + 1, lastFence - firstNewLine - 1);
|
||||
}
|
||||
}
|
||||
|
||||
return text;
|
||||
}
|
||||
|
||||
static QString trimWhitespace(const QString &text)
|
||||
{
|
||||
QString result = text;
|
||||
while (result.startsWith('\n') || result.startsWith('\r')) {
|
||||
result = result.mid(1);
|
||||
}
|
||||
while (result.endsWith('\n') || result.endsWith('\r')) {
|
||||
result.chop(1);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
static QString removeExplanations(const QString &text)
|
||||
{
|
||||
static const QStringList explanationPrefixes = {
|
||||
"here's the", "here is the", "here's", "here is",
|
||||
"the refactored", "refactored code:", "code:",
|
||||
"i've refactored", "i refactored", "i've changed", "i changed"
|
||||
};
|
||||
|
||||
QStringList lines = text.split('\n');
|
||||
int startLine = 0;
|
||||
|
||||
for (int i = 0; i < qMin(3, lines.size()); ++i) {
|
||||
QString line = lines[i].trimmed().toLower();
|
||||
bool isExplanation = false;
|
||||
|
||||
for (const QString &prefix : explanationPrefixes) {
|
||||
if (line.startsWith(prefix) || line.contains(prefix + " code")) {
|
||||
isExplanation = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (line.length() < 50 && line.endsWith(':')) {
|
||||
isExplanation = true;
|
||||
}
|
||||
|
||||
if (isExplanation) {
|
||||
startLine = i + 1;
|
||||
} else if (!line.isEmpty()) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (startLine > 0 && startLine < lines.size()) {
|
||||
lines = lines.mid(startLine);
|
||||
return lines.join('\n');
|
||||
}
|
||||
|
||||
return text;
|
||||
}
|
||||
};
|
||||
|
||||
} // namespace QodeAssist
|
||||
|
||||
Reference in New Issue
Block a user