forked from OSchip/llvm-project
90 lines
2.9 KiB
C++
90 lines
2.9 KiB
C++
//===--- 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 "Logger.h"
|
|
#include "ParsedAST.h"
|
|
#include "SourceCode.h"
|
|
#include "refactor/Tweak.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;
|
|
Intent intent() const override { return Intent::Refactor; }
|
|
|
|
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 llvm::createStringError(llvm::inconvertibleErrorCode(),
|
|
"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
|