Files
codeql/swift/extractor/invocation/SwiftDiagnosticsConsumer.h
2022-11-08 14:56:50 +01:00

25 lines
735 B
C++

#pragma once
#include <swift/AST/DiagnosticConsumer.h>
#include "swift/extractor/infra/SwiftLocationExtractor.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);
TrapDomain& trap;
SwiftLocationExtractor locationExtractor;
};
} // namespace codeql