aboutsummaryrefslogtreecommitdiffstats
path: root/sources/shiboken6/ApiExtractor/codesnip.cpp
blob: 255b9a392faaf3c92e76188892d22bb738da1c2a (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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
// Copyright (C) 2022 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0

#include "codesnip.h"

#include "qtcompat.h"
#include "exception.h"
#include "typedatabase.h"

#include <QtCore/QDebug>

using namespace Qt::StringLiterals;

QString TemplateInstance::expandCode() const
{
    const auto templateEntry = TypeDatabase::instance()->findTemplate(m_name);
    if (!templateEntry) {
        const QString m = u"<insert-template> referring to non-existing template '"_s
                          + m_name + u"'."_s;
        throw Exception(m);
    }

    QString code = templateEntry->code();
    for (auto it = replaceRules.cbegin(), end = replaceRules.cend(); it != end; ++it)
        code.replace(it.key(), it.value());
    while (!code.isEmpty() && code.at(code.size() - 1).isSpace())
        code.chop(1);
    QString result = u"// TEMPLATE - "_s + m_name + u" - START"_s;
    if (!code.startsWith(u'\n'))
        result += u'\n';
    result += code;
    result += u"\n// TEMPLATE - "_s + m_name + u" - END\n"_s;
    return result;
}

// ---------------------- CodeSnipFragment

static QString fragmentToCodeHelper(const QString &c)
{
    return c;
}

static QString fragmentToCodeHelper(const TemplateInstance &p)
{
    return p.expandCode();
}

static QString fragmentToCode(const CodeSnipFragment &codeFrag)
{
    return std::visit([](auto f) { return fragmentToCodeHelper(f); }, codeFrag);
}

static bool isEmptyFragmentHelper(const QString &c)
{
    return c.isEmpty();
}

static bool isEmptyFragmentHelper(const TemplateInstance &)
{
    return false;
}

static bool isEmptyFragment(const CodeSnipFragment &codeFrag)
{
    return std::visit([](auto f) { return isEmptyFragmentHelper(f); }, codeFrag);
}

static void formatDebugHelper(QDebug &d, const QString &code)
{
    const auto lines = QStringView{code}.split(u'\n');
    for (qsizetype i = 0, size = lines.size(); i < size; ++i) {
        if (i)
            d << "\\n";
        d << lines.at(i).trimmed();
    }
}

static void formatDebugHelper(QDebug &d, const TemplateInstance &t)
{
     d << "template=\"" << t.name() << '"';
}

QDebug operator<<(QDebug d, const CodeSnipFragment &codeFrag)
{
    QDebugStateSaver saver(d);
    d.noquote();
    d.nospace();
    std::visit([&d](auto f) { formatDebugHelper(d, f); }, codeFrag);
    return d;
}

// ---------------------- CodeSnipAbstract

void CodeSnipAbstract::addCode(const QString &code)
{
    m_codeList.emplace_back(CodeSnipFragment(fixSpaces(code)));
}

QString CodeSnipAbstract::code() const
{
    QString res;
    for (const auto &codeFrag : m_codeList)
        res.append(fragmentToCode(codeFrag));
    return res;
}

void CodeSnipAbstract::purgeEmptyFragments()
{
    auto end = std::remove_if(m_codeList.begin(), m_codeList.end(), isEmptyFragment);
    m_codeList.erase(end, m_codeList.end());
}

QRegularExpression CodeSnipAbstract::placeHolderRegex(int index)
{
    return QRegularExpression(u'%' + QString::number(index) + "\\b"_L1);
}

void purgeEmptyCodeSnips(QList<CodeSnip> *list)
{
    for (auto it = list->begin(); it != list->end(); ) {
        it->purgeEmptyFragments();
        if (it->isEmpty())
            it = list->erase(it);
        else
            ++it;
    }
}