---

yaml
---
r: 210441
b: refs/heads/swift-5.0-branch
c: b78987a
h: refs/heads/master
i:
  210439: 1fbfdd9

Author: benlangmuir

[refs]

@@ -644,7 +644,7 @@ refs/tags/swift-DEVELOPMENT-SNAPSHOT-2017-12-23-a: b7c074342459a645779f106c42bf4
 refs/heads/master-llvm-swift5-transition: 8ace18c8953afb3d7d94cf04cacc0b51a7e5f1e3
 "refs/heads/revert-12883-disable_modelio_test_ios": a77ae373b809a0d8cb460cf3d1585d618510d242
 refs/heads/revert-13597-master: cccee1df039d072215f9bddc2cbc1e32a8d5d5ee
-refs/heads/swift-5.0-branch: 8c78533e57f6c67d0fbc921f55d291bfaef10480
+refs/heads/swift-5.0-branch: b78987a8c53597e853804452885f9c1f86620cdf
 refs/tags/swift-4.1-DEVELOPMENT-SNAPSHOT-2017-12-23-a: b32214f7e04339dfada623b6b76dbebfb41e4541
 refs/tags/swift-4.1-DEVELOPMENT-SNAPSHOT-2017-12-24-a: 1eb0be506c0744c7eff0550a10240286046e181d
 refs/tags/swift-4.1-DEVELOPMENT-SNAPSHOT-2017-12-25-a: f35a91502bad0065c83d8760407c23be7b899f48

branches/swift-5.0-branch/test/SourceKit/CompileNotifications/Inputs/parse-error.swift

@@ -0,0 +1 @@
+func 1() {}

branches/swift-5.0-branch/test/SourceKit/CompileNotifications/Inputs/sema-error.swift

@@ -0,0 +1 @@
+_ = unknown

branches/swift-5.0-branch/test/SourceKit/CompileNotifications/diagnostics.swift

@@ -0,0 +1,40 @@
+// RUN: %sourcekitd-test -req=track-compiles == -req=sema %s -- %s | %FileCheck %s -check-prefix=NODIAGS
+// NODIAGS: key.notification: source.notification.compile-did-finish
+// NODIAGS-NEXT: key.diagnostics: [
+// NODIAGS-NEXT: ]
+
+// RUN: %sourcekitd-test -req=track-compiles == -req=sema %S/Inputs/parse-error.swift -- %S/Inputs/parse-error.swift | %FileCheck %s -check-prefix=PARSE
+// PARSE: key.notification: source.notification.compile-did-finish
+// PARSE-NEXT: key.diagnostics: [
+// PARSE-NEXT:   {
+// PARSE-NEXT:     key.line: 1
+// PARSE-NEXT:     key.column: 6
+// PARSE-NEXT:     key.filepath: "{{.*}}parse-error.swift"
+// PARSE-NEXT:     key.severity: source.diagnostic.severity.error
+// PARSE-NEXT:     key.description: "function name
+// PARSE-NEXT:   }
+// PARSE-NEXT: ]
+
+// Diagnostic from other file.
+// RUN: %sourcekitd-test -req=track-compiles == -req=sema %s -- %s %S/Inputs/parse-error.swift | %FileCheck %s -check-prefix=PARSE
+
+// RUN: %sourcekitd-test -req=track-compiles == -req=sema %S/Inputs/sema-error.swift -- %S/Inputs/sema-error.swift | %FileCheck %s -check-prefix=SEMA
+// SEMA: key.notification: source.notification.compile-did-finish
+// SEMA-NEXT: key.diagnostics: [
+// SEMA-NEXT:   {
+// SEMA-NEXT:     key.line: 1
+// SEMA-NEXT:     key.column: 5
+// SEMA-NEXT:     key.filepath: "{{.*}}sema-error.swift"
+// SEMA-NEXT:     key.severity: source.diagnostic.severity.error
+// SEMA-NEXT:     key.description: "use of
+// SEMA-NEXT:     key.ranges: [
+
+// RUN: %sourcekitd-test -req=track-compiles == -req=sema %s -- %s -Xcc -include -Xcc /doesnotexist | %FileCheck %s -check-prefix=CLANG_IMPORTER
+// CLANG_IMPORTER: key.notification: source.notification.compile-did-finish,
+// CLANG_IMPORTER-NEXT: key.diagnostics: [
+// CLANG_IMPORTER-NEXT:   {
+// CLANG_IMPORTER-NEXT:     key.line:
+// CLANG_IMPORTER-NEXT:     key.column:
+// CLANG_IMPORTER-NEXT:     key.filepath: "<{{.*}}>"
+// CLANG_IMPORTER-NEXT:     key.severity: source.diagnostic.severity.error,
+// CLANG_IMPORTER-NEXT:     key.description: {{.*}}not found

branches/swift-5.0-branch/tools/SourceKit/include/SourceKit/Support/Tracing.h

@@ -13,13 +13,17 @@
 #ifndef LLVM_SOURCEKIT_SUPPORT_TRACING_H
 #define LLVM_SOURCEKIT_SUPPORT_TRACING_H
 
+#include "SourceKit/Core/LLVM.h"
 #include "SourceKit/Support/UIdent.h"
 #include "swift/Basic/OptionSet.h"
+#include "llvm/ADT/ArrayRef.h"
 #include "llvm/ADT/Optional.h"
 
 #include <vector>
 
 namespace SourceKit {
+  struct DiagnosticEntryInfo;
+
 namespace trace {
 
 struct SwiftArguments {
@@ -76,7 +80,8 @@ class TraceConsumer {
                                 const StringPairs &OpArgs) = 0;
 
   // Operation previously started with startXXX has finished
-  virtual void operationFinished(uint64_t OpId, OperationKind OpKind) = 0;
+  virtual void operationFinished(uint64_t OpId, OperationKind OpKind,
+                                 ArrayRef<DiagnosticEntryInfo> Diagnostics) = 0;
 
   /// Returns the set of operations this consumer is interested in.
   ///
@@ -99,7 +104,8 @@ uint64_t startOperation(OperationKind OpKind,
                         const StringPairs &OpArgs = StringPairs());
 
 // Operation previously started with startXXX has finished
-void operationFinished(uint64_t OpId, OperationKind OpKind);
+void operationFinished(uint64_t OpId, OperationKind OpKind,
+                       ArrayRef<DiagnosticEntryInfo> Diagnostics);
 
 // Register trace consumer.
 void registerConsumer(TraceConsumer *Consumer);
@@ -134,9 +140,9 @@ class TracedOperation final {
     OpId = startOperation(OpKind, Inv, OpArgs);
   }
 
-  void finish() {
+  void finish(ArrayRef<DiagnosticEntryInfo> Diagnostics = llvm::None) {
     if (OpId.hasValue()) {
-      operationFinished(OpId.getValue(), OpKind);
+      operationFinished(OpId.getValue(), OpKind, Diagnostics);
       OpId.reset();
     }
   }

branches/swift-5.0-branch/tools/SourceKit/lib/Support/Tracing.cpp

@@ -54,11 +54,12 @@ uint64_t trace::startOperation(trace::OperationKind OpKind,
 }
 
 // Operation previously started with startXXX has finished
-void trace::operationFinished(uint64_t OpId, trace::OperationKind OpKind) {
+void trace::operationFinished(uint64_t OpId, trace::OperationKind OpKind,
+                              ArrayRef<DiagnosticEntryInfo> Diagnostics) {
   if (trace::anyEnabled()) {
     llvm::sys::ScopedLock L(consumersLock);
     for (auto *consumer : consumers) {
-      consumer->operationFinished(OpId, OpKind);
+      consumer->operationFinished(OpId, OpKind, Diagnostics);
     }
   }
 }

branches/swift-5.0-branch/tools/SourceKit/lib/SwiftLang/SwiftASTManager.cpp

@@ -878,6 +878,12 @@ ASTUnitRef ASTProducer::createASTUnit(SwiftASTManager::Implementation &MgrImpl,
   // TypeResolver is set before.
   ASTRef->Impl.TypeResolver = createLazyResolver(CompIns.getASTContext());
 
+  if (TracedOp.enabled()) {
+    SmallVector<DiagnosticEntryInfo, 8> Diagnostics;
+    Consumer.getAllDiagnostics(Diagnostics);
+    TracedOp.finish(Diagnostics);
+  }
+
   return ASTRef;
 }
 

branches/swift-5.0-branch/tools/SourceKit/lib/SwiftLang/SwiftEditor.cpp

@@ -46,6 +46,28 @@ using namespace SourceKit;
 using namespace swift;
 using namespace ide;
 
+static std::vector<unsigned> getSortedBufferIDs(
+    const llvm::DenseMap<unsigned, std::vector<DiagnosticEntryInfo>> &Map) {
+  std::vector<unsigned> bufferIDs;
+  bufferIDs.reserve(Map.size());
+  for (auto I = Map.begin(), E = Map.end(); I != E; ++I) {
+    bufferIDs.push_back(I->getFirst());
+  }
+  llvm::array_pod_sort(bufferIDs.begin(), bufferIDs.end());
+  return bufferIDs;
+}
+
+void EditorDiagConsumer::getAllDiagnostics(
+    SmallVectorImpl<DiagnosticEntryInfo> &Result) {
+  // Note: we cannot reuse InputBufIds because there may be diagnostics outside
+  // the inputs.  Instead, sort the extant buffers.
+  auto bufferIDs = getSortedBufferIDs(BufferDiagnostics);
+  for (unsigned bufferID : bufferIDs) {
+    const auto &diags = BufferDiagnostics[bufferID];
+    Result.append(diags.begin(), diags.end());
+  }
+}
+
 void EditorDiagConsumer::handleDiagnostic(
     SourceManager &SM, SourceLoc Loc, DiagnosticKind Kind,
     StringRef FormatString, ArrayRef<DiagnosticArgument> FormatArgs,

branches/swift-5.0-branch/tools/SourceKit/lib/SwiftLang/SwiftEditorDiagConsumer.h

@@ -62,6 +62,8 @@ class EditorDiagConsumer : public swift::DiagnosticConsumer {
     return Diags;
   }
 
+  void getAllDiagnostics(SmallVectorImpl<DiagnosticEntryInfo> &Result);
+
   bool hadErrorWithInvalidLoc() const { return HadInvalidLocError; }
 
   bool hadAnyError() const { return HadAnyError; }

branches/swift-5.0-branch/tools/SourceKit/tools/sourcekitd/bin/XPC/Service/XpcTracing.cpp

@@ -98,7 +98,9 @@ class XpcTraceConsumer : public SourceKit::trace::TraceConsumer {
                         const StringPairs &OpArgs) override;
 
   // Operation previously started with startXXX has finished
-  void operationFinished(uint64_t OpId, OperationKind OpKind) override;
+  void operationFinished(
+      uint64_t OpId, OperationKind OpKind,
+      ArrayRef<SourceKit::DiagnosticEntryInfo> Diagnostics) override;
 };
 
 // Trace start of SourceKit operation
@@ -117,7 +119,9 @@ void XpcTraceConsumer::operationStarted(uint64_t OpId,
 }
 
 // Operation previously started with startXXX has finished
-void XpcTraceConsumer::operationFinished(uint64_t OpId, OperationKind OpKind) {
+void XpcTraceConsumer::operationFinished(
+    uint64_t OpId, OperationKind OpKind,
+    ArrayRef<SourceKit::DiagnosticEntryInfo> Diagnostics) {
   xpc_object_t Contents = xpc_array_create(nullptr, 0);
   append(Contents, trace::ActionKind::OperationFinished);
   append(Contents, OpId);

branches/swift-5.0-branch/tools/SourceKit/tools/sourcekitd/lib/API/Requests.cpp

@@ -2695,7 +2695,8 @@ class CompileTrackingConsumer final : public trace::TraceConsumer {
   void operationStarted(uint64_t OpId, trace::OperationKind OpKind,
                         const trace::SwiftInvocation &Inv,
                         const trace::StringPairs &OpArgs) override;
-  void operationFinished(uint64_t OpId, trace::OperationKind OpKind) override;
+  void operationFinished(uint64_t OpId, trace::OperationKind OpKind,
+                         ArrayRef<DiagnosticEntryInfo> Diagnostics) override;
   swift::OptionSet<trace::OperationKind> desiredOperations() override {
     return swift::OptionSet<trace::OperationKind>() |
            trace::OperationKind::PerformSema |
@@ -2721,8 +2722,9 @@ void CompileTrackingConsumer::operationStarted(
   sourcekitd::postNotification(RespBuilder.createResponse());
 }
 
-void CompileTrackingConsumer::operationFinished(uint64_t OpId,
-                                                trace::OperationKind OpKind) {
+void CompileTrackingConsumer::operationFinished(
+    uint64_t OpId, trace::OperationKind OpKind,
+    ArrayRef<DiagnosticEntryInfo> Diagnostics) {
   if (!desiredOperations().contains(OpKind))
     return;
 
@@ -2731,7 +2733,11 @@ void CompileTrackingConsumer::operationFinished(uint64_t OpId,
   auto Dict = RespBuilder.getDictionary();
   Dict.set(KeyNotification, CompileDidFinishUID);
   Dict.set(KeyCompileID, std::to_string(OpId));
-  // Diagnostics
+  auto DiagArray = Dict.setArray(KeyDiagnostics);
+  for (const auto &DiagInfo : Diagnostics) {
+    fillDictionaryForDiagnosticInfo(DiagArray.appendDictionary(), DiagInfo);
+  }
+
   sourcekitd::postNotification(RespBuilder.createResponse());
 }