forked from OSchip/llvm-project
94 lines
3.4 KiB
C++
94 lines
3.4 KiB
C++
//===--- tools/extra/clang-rename/RenamingAction.cpp - Clang rename tool --===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// \brief Provides an action to rename every symbol at a point.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "RenamingAction.h"
|
|
#include "USRLocFinder.h"
|
|
#include "clang/AST/ASTConsumer.h"
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/Basic/FileManager.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/FrontendAction.h"
|
|
#include "clang/Lex/Lexer.h"
|
|
#include "clang/Lex/Preprocessor.h"
|
|
#include "clang/Tooling/CommonOptionsParser.h"
|
|
#include "clang/Tooling/Refactoring.h"
|
|
#include "clang/Tooling/Tooling.h"
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
using namespace llvm;
|
|
|
|
namespace clang {
|
|
namespace rename {
|
|
|
|
class RenamingASTConsumer : public ASTConsumer {
|
|
public:
|
|
RenamingASTConsumer(
|
|
const std::vector<std::string> &NewNames,
|
|
const std::vector<std::string> &PrevNames,
|
|
const std::vector<std::vector<std::string>> &USRList,
|
|
std::map<std::string, tooling::Replacements> &FileToReplaces,
|
|
bool PrintLocations)
|
|
: NewNames(NewNames), PrevNames(PrevNames), USRList(USRList),
|
|
FileToReplaces(FileToReplaces), PrintLocations(PrintLocations) {}
|
|
|
|
void HandleTranslationUnit(ASTContext &Context) override {
|
|
for (unsigned I = 0; I < NewNames.size(); ++I)
|
|
HandleOneRename(Context, NewNames[I], PrevNames[I], USRList[I]);
|
|
}
|
|
|
|
void HandleOneRename(ASTContext &Context, const std::string &NewName,
|
|
const std::string &PrevName,
|
|
const std::vector<std::string> &USRs) {
|
|
const SourceManager &SourceMgr = Context.getSourceManager();
|
|
std::vector<SourceLocation> RenamingCandidates;
|
|
std::vector<SourceLocation> NewCandidates;
|
|
|
|
NewCandidates =
|
|
getLocationsOfUSRs(USRs, PrevName, Context.getTranslationUnitDecl());
|
|
RenamingCandidates.insert(RenamingCandidates.end(), NewCandidates.begin(),
|
|
NewCandidates.end());
|
|
|
|
unsigned PrevNameLen = PrevName.length();
|
|
for (const auto &Loc : RenamingCandidates) {
|
|
if (PrintLocations) {
|
|
FullSourceLoc FullLoc(Loc, SourceMgr);
|
|
errs() << "clang-rename: renamed at: " << SourceMgr.getFilename(Loc)
|
|
<< ":" << FullLoc.getSpellingLineNumber() << ":"
|
|
<< FullLoc.getSpellingColumnNumber() << "\n";
|
|
}
|
|
// FIXME: better error handling.
|
|
tooling::Replacement Replace(SourceMgr, Loc, PrevNameLen, NewName);
|
|
llvm::Error Err = FileToReplaces[Replace.getFilePath()].add(Replace);
|
|
if (Err)
|
|
llvm::errs() << "Renaming failed in " << Replace.getFilePath() << "! "
|
|
<< llvm::toString(std::move(Err)) << "\n";
|
|
}
|
|
}
|
|
|
|
private:
|
|
const std::vector<std::string> &NewNames, &PrevNames;
|
|
const std::vector<std::vector<std::string>> &USRList;
|
|
std::map<std::string, tooling::Replacements> &FileToReplaces;
|
|
bool PrintLocations;
|
|
};
|
|
|
|
std::unique_ptr<ASTConsumer> RenamingAction::newASTConsumer() {
|
|
return llvm::make_unique<RenamingASTConsumer>(NewNames, PrevNames, USRList,
|
|
FileToReplaces, PrintLocations);
|
|
}
|
|
|
|
} // namespace rename
|
|
} // namespace clang
|