aboutsummaryrefslogtreecommitdiffstats
path: root/tests/unit/mockup/texteditor/codeassist/assistinterface.h
blob: e9f5dd35b2e93f89e6c1921847e0fe2a09f4a23f (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
80
81
82
83
84
85
86
87
88
// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0+ OR GPL-3.0 WITH Qt-GPL-exception-1.0

#pragma once

#include <QTextCursor>

#include <QTextDocument>

#include "../assistenums.h"

namespace TextEditor {

class AssistInterface
{
public:
    AssistInterface(const QByteArray &text,
                    int position)
        : textDocument_(QString::fromUtf8(text)),
          position_(position)
    {}

    AssistInterface(QTextDocument *textDocument,
                    int position,
                    const QString &fileName,
                    AssistReason reason)
        : textDocument_(textDocument),
          fileName_(fileName),
          position_(position),
          reason_(reason)
    {}

    QTextDocument *textDocument() const;
    virtual int position() const;
    virtual QChar characterAt(int position) const;
    virtual QString textAt(int position, int length) const;
    virtual QString fileName() const;
    virtual AssistReason reason() const;


private:
    mutable QTextDocument textDocument_;
    QString fileName_;
    int position_;
    AssistReason reason_ = IdleEditor;
};

inline QTextDocument *AssistInterface::textDocument() const
{
    return &textDocument_;
}

inline int AssistInterface::position() const
{
    return position_;
}

inline QChar AssistInterface::characterAt(int position) const
{
    return textDocument_.characterAt(position);
}

inline QString AssistInterface::textAt(int position, int length) const
{
    QTextCursor textCursor(&textDocument_);
    if (position < 0)
        position = 0;
    textCursor.movePosition(QTextCursor::End);
    if (position + length > textCursor.position())
        length = textCursor.position() - position;

    textCursor.setPosition(position);
    textCursor.setPosition(position + length, QTextCursor::KeepAnchor);

    // selectedText() returns U+2029 (PARAGRAPH SEPARATOR) instead of newline
    return textCursor.selectedText().replace(QChar::ParagraphSeparator, QLatin1Char('\n'));
}

inline AssistReason AssistInterface::reason() const
{
    return reason_;
}

inline QString AssistInterface::fileName() const
{
    return fileName_;
}
}