-
Notifications
You must be signed in to change notification settings - Fork 1.7k
/
Copy pathSwiftDiagnosticsConsumer.h
34 lines (26 loc) · 1.13 KB
/
SwiftDiagnosticsConsumer.h
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
#pragma once
#include <swift/AST/DiagnosticConsumer.h>
#include "swift/extractor/infra/SwiftLocationExtractor.h"
#include "swift/logging/SwiftLogging.h"
namespace codeql {
class TrapDomain;
class SwiftDiagnosticsConsumer : public swift::DiagnosticConsumer {
public:
explicit SwiftDiagnosticsConsumer(TrapDomain& targetFile)
: trap(targetFile), locationExtractor(targetFile) {}
void handleDiagnostic(swift::SourceManager& sourceManager,
const swift::DiagnosticInfo& diagInfo) override;
private:
static std::string getDiagMessage(swift::SourceManager& sourceManager,
const swift::DiagnosticInfo& diagInfo);
void forwardToLog(swift::SourceManager& sourceManager,
const swift::DiagnosticInfo& diagInfo,
const std::string& message);
void forwardToLog(swift::SourceManager& sourceManager, const swift::DiagnosticInfo& diagInfo) {
forwardToLog(sourceManager, diagInfo, getDiagMessage(sourceManager, diagInfo));
}
TrapDomain& trap;
SwiftLocationExtractor locationExtractor;
Logger logger{"compiler"};
};
} // namespace codeql