mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2025-11-16 06:52:48 -05:00
refactor: Full rework quick refactor (#257)
This commit is contained in:
Petr Mironychev
GitHub
202
RefactorSuggestion.cpp
Normal file
202
RefactorSuggestion.cpp
Normal file
@ -0,0 +1,202 @@
|
||||
/*
|
||||
* 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 "RefactorSuggestion.hpp"
|
||||
#include "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
|
||||
|
||||
Reference in New Issue
Block a user