You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

48 lines
1.4 KiB

//===--- ToolRefactoringResultConsumer.h - ----------------------*- 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
//
//===----------------------------------------------------------------------===//
#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