1//===--- ToolRefactoringResultConsumer.h - ----------------------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
10#define LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
11
12#include "clang/AST/ASTContext.h"
13#include "clang/Tooling/Refactoring/RefactoringResultConsumer.h"
14
15namespace clang {
16namespace refactor {
17
18/// An interface that subclasses the \c RefactoringResultConsumer interface
19/// that stores the reference to the TU-specific diagnostics engine.
20class ClangRefactorToolConsumerInterface
21    : public tooling::RefactoringResultConsumer {
22public:
23  /// Called when a TU is entered.
24  void beginTU(ASTContext &Context) {
25    assert(!Diags && "Diags has been set");
26    Diags = &Context.getDiagnostics();
27  }
28
29  /// Called when the tool is done with a TU.
30  void endTU() {
31    assert(Diags && "Diags unset");
32    Diags = nullptr;
33  }
34
35  DiagnosticsEngine &getDiags() const {
36    assert(Diags && "no diags");
37    return *Diags;
38  }
39
40private:
41  DiagnosticsEngine *Diags = nullptr;
42};
43
44} // end namespace refactor
45} // end namespace clang
46
47#endif // LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
48