SwiftEditorDiagConsumer.h

//===--- SwiftEditorDiagConsumer.h - ----------------------------*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_SOURCEKIT_LIB_SWIFTLANG_SWIFTEDITORDIAGCONSUMER_H
#defineLLVM_SOURCEKIT_LIB_SWIFTLANG_SWIFTEDITORDIAGCONSUMER_H

#include"SourceKit/Core/LangSupport.h"
#include"swift/AST/DiagnosticConsumer.h"
#include"llvm/ADT/DenseMap.h"
#include"llvm/ADT/StringMap.h"

namespaceSourceKit {

classEditorDiagConsumer : publicswift::DiagnosticConsumer {
typedef std::vector<DiagnosticEntryInfo> DiagnosticsTy;
/// Maps from a BufferID to the diagnostics that were emitted inside that
/// buffer.
 llvm::DenseMap<unsigned, DiagnosticsTy> BufferDiagnostics;
 DiagnosticsTy InvalidLocDiagnostics;

 llvm::StringMap<BufferInfoSharedPtr> BufferInfos;

int LastDiagBufferID = -1;
unsigned LastDiagIndex = 0;

boolhaveLastDiag() {
return LastDiagBufferID >= 0;
 }
voidclearLastDiag() {
 LastDiagBufferID = -1;
 }
 DiagnosticEntryInfo &getLastDiag() {
assert(haveLastDiag());
return BufferDiagnostics[LastDiagBufferID][LastDiagIndex];
 }

bool HadAnyError = false;

 BufferInfoSharedPtr getBufferInfo(StringRef FileName,
 std::optional<unsigned> BufferID,
 swift::SourceManager &SM);

public:
/// The diagnostics are returned in source-order.
 ArrayRef<DiagnosticEntryInfo> getDiagnosticsForBuffer(unsigned BufferID) const {
 ArrayRef<DiagnosticEntryInfo> Diags;
auto DiagFound = BufferDiagnostics.find(BufferID);
if (DiagFound != BufferDiagnostics.end())
 Diags = DiagFound->second;
return Diags;
 }

voidgetAllDiagnostics(SmallVectorImpl<DiagnosticEntryInfo> &Result);

boolhadAnyError() const { return HadAnyError; }

voidhandleDiagnostic(swift::SourceManager &SM,
const swift::DiagnosticInfo &Info) override;
};

} // namespace SourceKit

#endif