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
|
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
#include <qqmlformatting_p.h>
#include <qqmlcodemodel_p.h>
#include <qqmllsutils_p.h>
#include <QtQmlDom/private/qqmldomitem_p.h>
#include <QtQmlDom/private/qqmldomindentinglinewriter_p.h>
#include <QtQmlDom/private/qqmldomoutwriter_p.h>
QT_BEGIN_NAMESPACE
Q_LOGGING_CATEGORY(formatLog, "qt.languageserver.formatting")
QQmlDocumentFormatting::QQmlDocumentFormatting(QmlLsp::QQmlCodeModel *codeModel)
: QQmlBaseModule(codeModel)
{
}
QString QQmlDocumentFormatting::name() const
{
return u"QQmlDocumentFormatting"_s;
}
void QQmlDocumentFormatting::registerHandlers(QLanguageServer *, QLanguageServerProtocol *protocol)
{
protocol->registerDocumentFormattingRequestHandler(getRequestHandler());
}
void QQmlDocumentFormatting::setupCapabilities(
const QLspSpecification::InitializeParams &,
QLspSpecification::InitializeResult &serverCapabilities)
{
// TODO: Allow customized formatting in future
serverCapabilities.capabilities.documentFormattingProvider = true;
}
void QQmlDocumentFormatting::process(RequestPointerArgument request)
{
QList<QLspSpecification::TextEdit> result;
ResponseScopeGuard guard(result, request->m_response);
using namespace QQmlJS::Dom;
QmlLsp::OpenDocument doc = m_codeModel->openDocumentByUrl(
QQmlLSUtils::lspUriToQmlUrl(request->m_parameters.textDocument.uri));
DomItem file = doc.snapshot.doc.fileObject(GoTo::MostLikely);
if (!file) {
guard.setError(QQmlLSUtilsErrorMessage{
0, u"Could not find the file %1"_s.arg(doc.snapshot.doc.canonicalFilePath()) });
return;
}
if (!file.field(Fields::isValid).value().toBool(false)) {
guard.setError(QQmlLSUtilsErrorMessage{ 0, u"Cannot format invalid documents!"_s });
return;
}
if (auto envPtr = file.environment().ownerAs<DomEnvironment>())
envPtr->clearReferenceCache();
auto qmlFile = file.ownerAs<QmlFile>();
if (!qmlFile || !qmlFile->isValid()) {
file.iterateErrors(
[](const DomItem &, const ErrorMessage &msg) {
errorToQDebug(msg);
return true;
},
true);
guard.setError(QQmlLSUtilsErrorMessage{
0, u"Failed to parse %1"_s.arg(file.canonicalFilePath()) });
return;
}
// TODO: implement formatting options
// For now, qmlformat's default options.
LineWriterOptions options;
options.updateOptions = LineWriterOptions::Update::None;
options.attributesSequence = LineWriterOptions::AttributesSequence::Preserve;
QLspSpecification::TextEdit formattedText;
LineWriter lw([&formattedText](QStringView s) {formattedText.newText += s.toUtf8(); }, QString(), options);
OutWriter ow(lw);
file.writeOutForFile(ow, WriteOutCheck::None);
ow.flush();
const auto &code = qmlFile->code();
const auto [endLine, endColumn] = QQmlLSUtils::textRowAndColumnFrom(code, code.length());
Q_UNUSED(endColumn);
formattedText.range = QLspSpecification::Range{ QLspSpecification::Position{ 0, 0 },
QLspSpecification::Position{ endLine + 1, 0 } };
result.append(formattedText);
}
QT_END_NAMESPACE
|
