mirror of
https://github.com/github/codeql.git
synced 2026-02-16 15:03:41 +01:00
45 lines
1.5 KiB
C++
45 lines
1.5 KiB
C++
#include "swift/extractor/invocation/SwiftDiagnosticsConsumer.h"
|
|
#include "swift/extractor/trap/generated/TrapEntries.h"
|
|
#include "swift/extractor/trap/TrapDomain.h"
|
|
|
|
#include <swift/AST/DiagnosticEngine.h>
|
|
#include <swift/Basic/SourceManager.h>
|
|
#include <llvm/ADT/SmallString.h>
|
|
#include <llvm/Support/raw_ostream.h>
|
|
#include <string>
|
|
|
|
using namespace codeql;
|
|
|
|
static int diagnosticsKind(const swift::DiagnosticInfo& diagInfo) {
|
|
switch (diagInfo.Kind) {
|
|
case swift::DiagnosticKind::Error:
|
|
return 1;
|
|
case swift::DiagnosticKind::Warning:
|
|
return 2;
|
|
case swift::DiagnosticKind::Note:
|
|
return 3;
|
|
case swift::DiagnosticKind::Remark:
|
|
return 4;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void SwiftDiagnosticsConsumer::handleDiagnostic(swift::SourceManager& sourceManager,
|
|
const swift::DiagnosticInfo& diagInfo) {
|
|
auto message = getDiagMessage(sourceManager, diagInfo);
|
|
DiagnosticsTrap diag{};
|
|
diag.id = trap.createLabel<DiagnosticsTag>();
|
|
diag.kind = diagnosticsKind(diagInfo);
|
|
diag.text = message;
|
|
trap.emit(diag);
|
|
locationExtractor.attachLocation(sourceManager, diagInfo.Loc, diagInfo.Loc, diag.id);
|
|
}
|
|
|
|
std::string SwiftDiagnosticsConsumer::getDiagMessage(swift::SourceManager& sourceManager,
|
|
const swift::DiagnosticInfo& diagInfo) {
|
|
llvm::SmallString<256> text;
|
|
llvm::raw_svector_ostream out(text);
|
|
swift::DiagnosticEngine::formatDiagnosticText(out, diagInfo.FormatString, diagInfo.FormatArgs);
|
|
return text.str().str();
|
|
}
|