3adfb6a3ee
Change the contract of GetStyle so that it returns an error when an error occurs (i.e. when it writes to stderr), and only returns the fallback style when it can't find a configuration file. Differential Revision: https://reviews.llvm.org/D28081 llvm-svn: 292174
105 lines
3.5 KiB
C++
105 lines
3.5 KiB
C++
//===--- Refactoring.cpp - Framework for clang refactoring tools ----------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Implements tools to support refactorings.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Tooling/Refactoring.h"
|
|
#include "clang/Basic/DiagnosticOptions.h"
|
|
#include "clang/Basic/FileManager.h"
|
|
#include "clang/Basic/SourceManager.h"
|
|
#include "clang/Format/Format.h"
|
|
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
#include "clang/Lex/Lexer.h"
|
|
#include "clang/Rewrite/Core/Rewriter.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/raw_os_ostream.h"
|
|
|
|
namespace clang {
|
|
namespace tooling {
|
|
|
|
RefactoringTool::RefactoringTool(
|
|
const CompilationDatabase &Compilations, ArrayRef<std::string> SourcePaths,
|
|
std::shared_ptr<PCHContainerOperations> PCHContainerOps)
|
|
: ClangTool(Compilations, SourcePaths, PCHContainerOps) {}
|
|
|
|
std::map<std::string, Replacements> &RefactoringTool::getReplacements() {
|
|
return FileToReplaces;
|
|
}
|
|
|
|
int RefactoringTool::runAndSave(FrontendActionFactory *ActionFactory) {
|
|
if (int Result = run(ActionFactory)) {
|
|
return Result;
|
|
}
|
|
|
|
LangOptions DefaultLangOptions;
|
|
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
TextDiagnosticPrinter DiagnosticPrinter(llvm::errs(), &*DiagOpts);
|
|
DiagnosticsEngine Diagnostics(
|
|
IntrusiveRefCntPtr<DiagnosticIDs>(new DiagnosticIDs()),
|
|
&*DiagOpts, &DiagnosticPrinter, false);
|
|
SourceManager Sources(Diagnostics, getFiles());
|
|
Rewriter Rewrite(Sources, DefaultLangOptions);
|
|
|
|
if (!applyAllReplacements(Rewrite)) {
|
|
llvm::errs() << "Skipped some replacements.\n";
|
|
}
|
|
|
|
return saveRewrittenFiles(Rewrite);
|
|
}
|
|
|
|
bool RefactoringTool::applyAllReplacements(Rewriter &Rewrite) {
|
|
bool Result = true;
|
|
for (const auto &Entry : groupReplacementsByFile(
|
|
Rewrite.getSourceMgr().getFileManager(), FileToReplaces))
|
|
Result = tooling::applyAllReplacements(Entry.second, Rewrite) && Result;
|
|
return Result;
|
|
}
|
|
|
|
int RefactoringTool::saveRewrittenFiles(Rewriter &Rewrite) {
|
|
return Rewrite.overwriteChangedFiles() ? 1 : 0;
|
|
}
|
|
|
|
bool formatAndApplyAllReplacements(
|
|
const std::map<std::string, Replacements> &FileToReplaces,
|
|
Rewriter &Rewrite, StringRef Style) {
|
|
SourceManager &SM = Rewrite.getSourceMgr();
|
|
FileManager &Files = SM.getFileManager();
|
|
|
|
bool Result = true;
|
|
for (const auto &FileAndReplaces : groupReplacementsByFile(
|
|
Rewrite.getSourceMgr().getFileManager(), FileToReplaces)) {
|
|
const std::string &FilePath = FileAndReplaces.first;
|
|
auto &CurReplaces = FileAndReplaces.second;
|
|
|
|
const FileEntry *Entry = Files.getFile(FilePath);
|
|
FileID ID = SM.getOrCreateFileID(Entry, SrcMgr::C_User);
|
|
StringRef Code = SM.getBufferData(ID);
|
|
|
|
auto CurStyle = format::getStyle(Style, FilePath, "LLVM");
|
|
if (!CurStyle) {
|
|
llvm::errs() << llvm::toString(CurStyle.takeError()) << "\n";
|
|
return false;
|
|
}
|
|
|
|
auto NewReplacements =
|
|
format::formatReplacements(Code, CurReplaces, *CurStyle);
|
|
if (!NewReplacements) {
|
|
llvm::errs() << llvm::toString(NewReplacements.takeError()) << "\n";
|
|
return false;
|
|
}
|
|
Result = applyAllReplacements(*NewReplacements, Rewrite) && Result;
|
|
}
|
|
return Result;
|
|
}
|
|
|
|
} // end namespace tooling
|
|
} // end namespace clang
|