blob: b69f9d673fb6b69c38640718895146e550d1b7b2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
//===--- 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
|