llvm-project/clang/tools/clang-refactor/ToolRefactoringResultConsumer.h
Alex Lorenz f5ca27cc37 [refactor] allow the use of refactoring diagnostics
This commit allows the refactoring library to use its own set of
refactoring-specific diagnostics to reports things like initiation errors.

Differential Revision: https://reviews.llvm.org/D38772

llvm-svn: 315924
2017-10-16 18:28:26 +00:00

49 lines
1.4 KiB
C++

//===--- ToolRefactoringResultConsumer.h - ----------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
#define LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
#include "clang/AST/ASTContext.h"
#include "clang/Tooling/Refactoring/RefactoringResultConsumer.h"
namespace clang {
namespace refactor {
/// An interface that subclasses the \c RefactoringResultConsumer interface
/// that stores the reference to the TU-specific diagnostics engine.
class ClangRefactorToolConsumerInterface
: public tooling::RefactoringResultConsumer {
public:
/// Called when a TU is entered.
void beginTU(ASTContext &Context) {
assert(!Diags && "Diags has been set");
Diags = &Context.getDiagnostics();
}
/// Called when the tool is done with a TU.
void endTU() {
assert(Diags && "Diags unset");
Diags = nullptr;
}
DiagnosticsEngine &getDiags() const {
assert(Diags && "no diags");
return *Diags;
}
private:
DiagnosticsEngine *Diags = nullptr;
};
} // end namespace refactor
} // end namespace clang
#endif // LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H