update Diagnostic.

2024-08-31 20:56:19 +08:00
parent e8f72639fe
commit 91bc14c129
5 changed files with 38 additions and 10 deletions
--- a/include/AST/Diagnostic.h
+++ b/include/AST/Diagnostic.h
@@ -8,4 +8,17 @@
 #include <clang/Sema/Sema.h>
 #include <clang/Tooling/CompilationDatabase.h>
 #include <clang/Tooling/Syntax/Tokens.h>
-#include <clang/Lex/PPCallbacks.h>
+#include <clang/Lex/PPCallbacks.h>
+
+namespace clice {
+
+class Diagnostic : public clang::DiagnosticConsumer {
+public:
+    void BeginSourceFile(const clang::LangOptions& Opts, const clang::Preprocessor* PP) override;
+
+    void HandleDiagnostic(clang::DiagnosticsEngine::Level DiagLevel, const clang::Diagnostic& Info) override;
+
+    void EndSourceFile() override;
+};
+
+}  // namespace clice