AnnotateHighlightings.cpp
3 KB
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
//===--- AnnotateHighlightings.cpp -------------------------------*- C++-*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "SemanticHighlighting.h"
#include "refactor/Tweak.h"
#include "llvm/ADT/StringRef.h"
namespace clang {
namespace clangd {
namespace {
/// Annotate all highlighting tokens in the current file. This is a hidden tweak
/// which is used to debug semantic highlightings.
/// Before:
/// void f() { int abc; }
/// ^^^^^^^^^^^^^^^^^^^^^
/// After:
/// void /* entity.name.function.cpp */ f() { int /* variable.cpp */ abc; }
class AnnotateHighlightings : public Tweak {
public:
const char *id() const override final;
bool prepare(const Selection &Inputs) override { return true; }
Expected<Effect> apply(const Selection &Inputs) override;
std::string title() const override { return "Annotate highlighting tokens"; }
Intent intent() const override { return Refactor; }
bool hidden() const override { return true; }
};
REGISTER_TWEAK(AnnotateHighlightings)
Expected<Tweak::Effect> AnnotateHighlightings::apply(const Selection &Inputs) {
const Decl *CommonDecl = nullptr;
for (auto N = Inputs.ASTSelection.commonAncestor(); N && !CommonDecl;
N = N->Parent)
CommonDecl = N->ASTNode.get<Decl>();
std::vector<HighlightingToken> HighlightingTokens;
if (!CommonDecl) {
// Now we hit the TUDecl case where commonAncestor() returns null
// intendedly. We only annotate tokens in the main file, so use the default
// traversal scope (which is the top level decls of the main file).
HighlightingTokens = getSemanticHighlightings(*Inputs.AST);
} else {
// Store the existing scopes.
const auto &BackupScopes = Inputs.AST->getASTContext().getTraversalScope();
// Narrow the traversal scope to the selected node.
Inputs.AST->getASTContext().setTraversalScope(
{const_cast<Decl *>(CommonDecl)});
HighlightingTokens = getSemanticHighlightings(*Inputs.AST);
// Restore the traversal scope.
Inputs.AST->getASTContext().setTraversalScope(BackupScopes);
}
auto &SM = Inputs.AST->getSourceManager();
tooling::Replacements Result;
llvm::StringRef FilePath = SM.getFilename(Inputs.Cursor);
for (const auto &Token : HighlightingTokens) {
assert(Token.R.start.line == Token.R.end.line &&
"Token must be at the same line");
auto InsertOffset = positionToOffset(Inputs.Code, Token.R.start);
if (!InsertOffset)
return InsertOffset.takeError();
auto InsertReplacement = tooling::Replacement(
FilePath, *InsertOffset, 0,
("/* " + toTextMateScope(Token.Kind) + " */").str());
if (auto Err = Result.add(InsertReplacement))
return std::move(Err);
}
return Effect::mainFileEdit(SM, std::move(Result));
}
} // namespace
} // namespace clangd
} // namespace clang