blob: 28da286f1f7074c2cd826336f4cd03d1b9062522 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
#include "copilotsuggestion.h"
#include <texteditor/texteditor.h>
#include <utils/stringutils.h>
using namespace Utils;
using namespace TextEditor;
using namespace LanguageServerProtocol;
namespace Copilot::Internal {
CopilotSuggestion::CopilotSuggestion(const QList<Completion> &completions,
QTextDocument *origin,
int currentCompletion)
: m_completions(completions)
, m_currentCompletion(currentCompletion)
{
const Completion completion = completions.value(currentCompletion);
const Position start = completion.range().start();
const Position end = completion.range().end();
QString text = start.toTextCursor(origin).block().text();
int length = text.length() - start.character();
if (start.line() == end.line())
length = end.character() - start.character();
text.replace(start.character(), length, completion.text());
document()->setPlainText(text);
m_start = completion.position().toTextCursor(origin);
m_start.setKeepPositionOnInsert(true);
setCurrentPosition(m_start.position());
}
bool CopilotSuggestion::apply()
{
reset();
const Completion completion = m_completions.value(m_currentCompletion);
QTextCursor cursor = completion.range().toSelection(m_start.document());
cursor.insertText(completion.text());
return true;
}
bool CopilotSuggestion::applyWord(TextEditorWidget *widget)
{
const Completion completion = m_completions.value(m_currentCompletion);
const QTextCursor cursor = completion.range().toSelection(m_start.document());
QTextCursor currentCursor = widget->textCursor();
const QString text = completion.text();
const int startPos = currentCursor.positionInBlock() - cursor.positionInBlock()
+ (cursor.selectionEnd() - cursor.selectionStart());
const int next = endOfNextWord(text, startPos);
if (next == -1)
return apply();
// TODO: Allow adding more than one line
QString subText = text.mid(startPos, next - startPos);
subText = subText.left(subText.indexOf('\n'));
if (subText.isEmpty())
return false;
currentCursor.insertText(subText);
return false;
}
void CopilotSuggestion::reset()
{
m_start.removeSelectedText();
}
int CopilotSuggestion::position()
{
return m_start.position();
}
} // namespace Copilot::Internal
|
