RawStringLiteral.cpp
3.29 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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
//===--- RawStringLiteral.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 "Logger.h"
#include "ParsedAST.h"
#include "SourceCode.h"
#include "refactor/Tweak.h"
#include "clang/AST/ASTContext.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/AST/Stmt.h"
#include "clang/Basic/LangOptions.h"
#include "clang/Basic/SourceLocation.h"
#include "clang/Basic/SourceManager.h"
#include "clang/Lex/Lexer.h"
#include "clang/Tooling/Core/Replacement.h"
#include "llvm/ADT/None.h"
#include "llvm/ADT/Optional.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/iterator_range.h"
#include "llvm/Support/Casting.h"
#include "llvm/Support/Error.h"
namespace clang {
namespace clangd {
namespace {
/// Converts a string literal to a raw string.
/// Before:
/// printf("\"a\"\nb");
/// ^^^^^^^^^
/// After:
/// printf(R"("a"
/// b)");
class RawStringLiteral : public Tweak {
public:
const char *id() const override final;
bool prepare(const Selection &Inputs) override;
Expected<Effect> apply(const Selection &Inputs) override;
std::string title() const override { return "Convert to raw string"; }
Intent intent() const override { return Refactor; }
private:
const clang::StringLiteral *Str = nullptr;
};
REGISTER_TWEAK(RawStringLiteral)
static bool isNormalString(const StringLiteral &Str, SourceLocation Cursor,
SourceManager &SM) {
// All chunks must be normal ASCII strings, not u8"..." etc.
if (!Str.isAscii())
return false;
SourceLocation LastTokenBeforeCursor;
for (auto I = Str.tokloc_begin(), E = Str.tokloc_end(); I != E; ++I) {
if (I->isMacroID()) // No tokens in the string may be macro expansions.
return false;
if (SM.isBeforeInTranslationUnit(*I, Cursor) || *I == Cursor)
LastTokenBeforeCursor = *I;
}
// Token we care about must be a normal "string": not raw, u8, etc.
const char* Data = SM.getCharacterData(LastTokenBeforeCursor);
return Data && *Data == '"';
}
static bool needsRaw(llvm::StringRef Content) {
return Content.find_first_of("\"\n\t") != StringRef::npos;
}
static bool canBeRaw(llvm::StringRef Content) {
for (char C : Content)
if (!llvm::isPrint(C) && C != '\n' && C != '\t')
return false;
return !Content.contains(")\"");
}
bool RawStringLiteral::prepare(const Selection &Inputs) {
const SelectionTree::Node *N = Inputs.ASTSelection.commonAncestor();
if (!N)
return false;
Str = dyn_cast_or_null<StringLiteral>(N->ASTNode.get<Stmt>());
return Str &&
isNormalString(*Str, Inputs.Cursor, Inputs.AST->getSourceManager()) &&
needsRaw(Str->getBytes()) && canBeRaw(Str->getBytes());
}
Expected<Tweak::Effect> RawStringLiteral::apply(const Selection &Inputs) {
auto &SM = Inputs.AST->getSourceManager();
auto Reps = tooling::Replacements(
tooling::Replacement(SM, Str, ("R\"(" + Str->getBytes() + ")\"").str(),
Inputs.AST->getLangOpts()));
return Effect::mainFileEdit(SM, std::move(Reps));
}
} // namespace
} // namespace clangd
} // namespace clang