| //===--- SwiftEditorDiagConsumer.h - ----------------------------*- C++ -*-===// |
| // |
| // This source file is part of the Swift.org open source project |
| // |
| // Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors |
| // Licensed under Apache License v2.0 with Runtime Library Exception |
| // |
| // See http://swift.org/LICENSE.txt for license information |
| // See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef LLVM_SOURCEKIT_LIB_SWIFTLANG_SWIFTEDITORDIAGCONSUMER_H |
| #define LLVM_SOURCEKIT_LIB_SWIFTLANG_SWIFTEDITORDIAGCONSUMER_H |
| |
| #include "SourceKit/Core/LangSupport.h" |
| #include "swift/Basic/DiagnosticConsumer.h" |
| #include "llvm/ADT/DenseMap.h" |
| |
| namespace SourceKit { |
| |
| class EditorDiagConsumer : public swift::DiagnosticConsumer { |
| typedef std::vector<DiagnosticEntryInfo> DiagnosticsTy; |
| /// Maps from a BufferID to the diagnostics that were emitted inside that |
| /// buffer. |
| llvm::DenseMap<unsigned, DiagnosticsTy> BufferDiagnostics; |
| |
| SmallVector<unsigned, 8> InputBufIDs; |
| int LastDiagBufferID = -1; |
| unsigned LastDiagIndex = 0; |
| |
| bool haveLastDiag() { |
| return LastDiagBufferID >= 0; |
| } |
| void clearLastDiag() { |
| LastDiagBufferID = -1; |
| } |
| DiagnosticEntryInfo &getLastDiag() { |
| assert(haveLastDiag()); |
| return BufferDiagnostics[LastDiagBufferID][LastDiagIndex]; |
| } |
| |
| bool HadInvalidLocError = false; |
| bool HadAnyError = false; |
| |
| public: |
| void setInputBufferIDs(ArrayRef<unsigned> BufferIDs) { |
| InputBufIDs.append(BufferIDs.begin(), BufferIDs.end()); |
| std::sort(InputBufIDs.begin(), InputBufIDs.end()); |
| } |
| |
| bool isInputBufferID(unsigned BufferID) const { |
| return std::binary_search(InputBufIDs.begin(), InputBufIDs.end(), BufferID); |
| } |
| |
| /// 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; |
| } |
| |
| bool hadErrorWithInvalidLoc() const { return HadInvalidLocError; } |
| |
| bool hadAnyError() const { return HadAnyError; } |
| |
| void handleDiagnostic(swift::SourceManager &SM, swift::SourceLoc Loc, |
| swift::DiagnosticKind Kind, StringRef Text, |
| const swift::DiagnosticInfo &Info) override; |
| }; |
| |
| } // namespace SourceKit. |
| |
| #endif |