ObjCLocalizeStringLiteral.cpp
2.81 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
//===--- ObjcLocalizeStringLiteral.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 "ParsedAST.h"
#include "SourceCode.h"
#include "refactor/Tweak.h"
#include "support/Logger.h"
#include "clang/AST/ExprObjC.h"
#include "clang/Basic/LangOptions.h"
#include "clang/Basic/SourceLocation.h"
#include "clang/Basic/SourceManager.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 {
/// Wraps an Objective-C string literal with the NSLocalizedString macro.
/// Before:
/// @"description"
/// ^^^
/// After:
/// NSLocalizedString(@"description", @"")
class ObjCLocalizeStringLiteral : public Tweak {
public:
const char *id() const override final;
llvm::StringLiteral kind() const override {
return CodeAction::REFACTOR_KIND;
}
bool prepare(const Selection &Inputs) override;
Expected<Tweak::Effect> apply(const Selection &Inputs) override;
std::string title() const override;
private:
const clang::ObjCStringLiteral *Str = nullptr;
};
REGISTER_TWEAK(ObjCLocalizeStringLiteral)
bool ObjCLocalizeStringLiteral::prepare(const Selection &Inputs) {
const SelectionTree::Node *N = Inputs.ASTSelection.commonAncestor();
if (!N)
return false;
// Allow the refactoring even if the user selected only the C string part
// of the expression.
if (N->ASTNode.get<StringLiteral>()) {
if (N->Parent)
N = N->Parent;
}
Str = dyn_cast_or_null<ObjCStringLiteral>(N->ASTNode.get<Stmt>());
return Str;
}
Expected<Tweak::Effect>
ObjCLocalizeStringLiteral::apply(const Selection &Inputs) {
auto *AST = Inputs.AST;
auto &SM = AST->getSourceManager();
const auto &TB = AST->getTokens();
auto Toks = TB.spelledForExpanded(TB.expandedTokens(Str->getSourceRange()));
if (!Toks || Toks->empty())
return error("Failed to find tokens to replace.");
// Insert `NSLocalizedString(` before the literal.
auto Reps = tooling::Replacements(tooling::Replacement(
SM, Toks->front().location(), 0, "NSLocalizedString("));
// Insert `, @"")` after the literal.
if (auto Err = Reps.add(
tooling::Replacement(SM, Toks->back().endLocation(), 0, ", @\"\")")))
return std::move(Err);
return Effect::mainFileEdit(SM, std::move(Reps));
}
std::string ObjCLocalizeStringLiteral::title() const {
return "Wrap in NSLocalizedString";
}
} // namespace
} // namespace clangd
} // namespace clang