97 lines
3.5 KiB
C++
97 lines
3.5 KiB
C++
|
// Copyright (c) 2013 The Chromium Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style license that can be
|
||
|
|
// found in the LICENSE file.
|
||
|
|
//
|
||
|
|
// This implements a Clang tool to rewrite all instances of scoped_array<T> to
|
||
|
|
// scoped_ptr<T[]>. The former is being deprecated in favor of the latter, to
|
||
|
|
// allow for an eventual transition from scoped_ptr to unique_ptr.
|
||
|
|
|
||
|
|
#include "clang/AST/ASTContext.h"
|
||
|
|
#include "clang/ASTMatchers/ASTMatchers.h"
|
||
|
|
#include "clang/ASTMatchers/ASTMatchFinder.h"
|
||
|
|
#include "clang/Basic/SourceManager.h"
|
||
|
|
#include "clang/Frontend/FrontendActions.h"
|
||
|
|
#include "clang/Lex/Lexer.h"
|
||
|
|
#include "clang/Tooling/CommonOptionsParser.h"
|
||
|
|
#include "clang/Tooling/Refactoring.h"
|
||
|
|
#include "clang/Tooling/Tooling.h"
|
||
|
|
#include "llvm/Support/CommandLine.h"
|
||
|
|
|
||
|
|
using clang::ast_matchers::MatchFinder;
|
||
|
|
using clang::ast_matchers::hasDeclaration;
|
||
|
|
using clang::ast_matchers::hasName;
|
||
|
|
using clang::ast_matchers::id;
|
||
|
|
using clang::ast_matchers::loc;
|
||
|
|
using clang::ast_matchers::qualType;
|
||
|
|
using clang::ast_matchers::recordDecl;
|
||
|
|
using clang::tooling::CommonOptionsParser;
|
||
|
|
using clang::tooling::Replacement;
|
||
|
|
using clang::tooling::Replacements;
|
||
|
|
using llvm::StringRef;
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
class RewriterCallback : public MatchFinder::MatchCallback {
|
||
|
|
public:
|
||
|
|
RewriterCallback(Replacements* replacements) : replacements_(replacements) {}
|
||
|
|
virtual void run(const MatchFinder::MatchResult& result) LLVM_OVERRIDE;
|
||
|
|
|
||
|
|
private:
|
||
|
|
Replacements* const replacements_;
|
||
|
|
};
|
||
|
|
|
||
|
|
void RewriterCallback::run(const MatchFinder::MatchResult& result) {
|
||
|
|
const clang::TypeLoc type_location =
|
||
|
|
*result.Nodes.getNodeAs<clang::TypeLoc>("loc");
|
||
|
|
clang::CharSourceRange range = clang::CharSourceRange::getTokenRange(
|
||
|
|
result.SourceManager->getSpellingLoc(type_location.getLocStart()),
|
||
|
|
result.SourceManager->getSpellingLoc(type_location.getLocEnd()));
|
||
|
|
// TODO(dcheng): Log an error?
|
||
|
|
if (!range.isValid())
|
||
|
|
return;
|
||
|
|
std::string replacement_text = clang::Lexer::getSourceText(
|
||
|
|
range, *result.SourceManager, result.Context->getLangOpts());
|
||
|
|
// TODO(dcheng): Log errors?
|
||
|
|
if (!StringRef(replacement_text).startswith("scoped_array<") ||
|
||
|
|
!StringRef(replacement_text).endswith(">"))
|
||
|
|
return;
|
||
|
|
replacement_text.replace(strlen("scoped_"), strlen("array"), "ptr");
|
||
|
|
replacement_text.insert(replacement_text.size() - 1, "[]");
|
||
|
|
replacements_->insert(
|
||
|
|
Replacement(*result.SourceManager, range, replacement_text));
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
static llvm::cl::extrahelp common_help(CommonOptionsParser::HelpMessage);
|
||
|
|
|
||
|
|
int main(int argc, const char* argv[]) {
|
||
|
|
CommonOptionsParser options(argc, argv);
|
||
|
|
clang::tooling::ClangTool tool(options.getCompilations(),
|
||
|
|
options.getSourcePathList());
|
||
|
|
|
||
|
|
Replacements replacements;
|
||
|
|
RewriterCallback callback(&replacements);
|
||
|
|
MatchFinder match_finder;
|
||
|
|
match_finder.addMatcher(
|
||
|
|
id("loc",
|
||
|
|
loc(qualType(hasDeclaration(recordDecl(hasName("::scoped_array")))))),
|
||
|
|
&callback);
|
||
|
|
|
||
|
|
int result =
|
||
|
|
tool.run(clang::tooling::newFrontendActionFactory(&match_finder));
|
||
|
|
if (result != 0)
|
||
|
|
return result;
|
||
|
|
|
||
|
|
// Serialization format is documented in tools/clang/scripts/run_tool.py
|
||
|
|
llvm::outs() << "==== BEGIN EDITS ====\n";
|
||
|
|
for (Replacements::const_iterator it = replacements.begin();
|
||
|
|
it != replacements.end(); ++it) {
|
||
|
|
llvm::outs() << "r:" << it->getFilePath() << ":" << it->getOffset() << ":"
|
||
|
|
<< it->getLength() << ":" << it->getReplacementText() << "\n";
|
||
|
|
}
|
||
|
|
llvm::outs() << "==== END EDITS ====\n";
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|