also add sourceOrigin for local protocols without docs

Author: QuietMisdreavus

lib/SymbolGraphGen/Edge.cpp

@@ -36,6 +36,15 @@ const ValueDecl *getForeignProtocolRequirement(const ValueDecl *VD, const Module
     requirements.pop();
   }
 }
+
+const ValueDecl *getProtocolRequirement(const ValueDecl *VD) {
+  auto reqs = VD->getSatisfiedProtocolRequirements();
+
+  if (!reqs.empty())
+    return reqs.front();
+  else
+    return nullptr;
+}
 } // end anonymous namespace
 
 void Edge::serialize(llvm::json::OStream &OS) const {
@@ -82,15 +91,19 @@ void Edge::serialize(llvm::json::OStream &OS) const {
     if (const auto *ID = Source.getDeclInheritingDocs())
       InheritingDecl = ID;
 
-
     if (!InheritingDecl && Source.getSynthesizedBaseTypeDecl())
       InheritingDecl = Source.getSymbolDecl();
 
     if (!InheritingDecl) {
       if (const auto *ID = getForeignProtocolRequirement(Source.getSymbolDecl(), &Graph->M))
         InheritingDecl = ID;
     }
-    
+
+    if (!InheritingDecl) {
+      if (const auto *ID = getProtocolRequirement(Source.getSymbolDecl()))
+        InheritingDecl = ID;
+    }
+
     // If our source symbol is a inheriting decl, write in information about
     // where it's inheriting docs from.
     if (InheritingDecl) {

test/SymbolGraph/Relationships/Synthesized/InheritedDocs.swift

@@ -8,6 +8,7 @@
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes BONUS-DOCS
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes EXTRA
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes LOCAL
+// RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes SUPER
 
 // RUN: %target-swift-symbolgraph-extract -module-name InheritedDocs -I %t -pretty-print -output-dir %t -skip-inherited-docs
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes CHECK,SKIP
@@ -16,6 +17,7 @@
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes BONUS-SKIP
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes EXTRA
 // RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes LOCAL
+// RUN: %FileCheck %s --input-file %t/InheritedDocs.symbols.json --check-prefixes SUPER
 
 // RUN: %empty-directory(%t)
 // RUN: %target-build-swift %s -module-name InheritedDocs -emit-module -emit-module-path %t/InheritedDocs.swiftmodule -emit-symbol-graph -emit-symbol-graph-dir %t/ -skip-inherited-docs
@@ -60,19 +62,29 @@
 
 // local implementations of a local protocol still need to a relation to that protocol
 
-// LOCAL:      "source": "s:13InheritedDocs1SV9localFuncyyF"
+// LOCAL:           "source": "s:13InheritedDocs1SV9localFuncyyF"
 // LOCAL-NEXT:      "target": "s:13InheritedDocs1SV"
 // LOCAL-NEXT:      "sourceOrigin"
 // LOCAL-NEXT:        "identifier": "s:13InheritedDocs1PP9localFuncyyF"
 // LOCAL-NEXT:        "displayName": "P.localFunc()"
 
+// ...both with and without docs
+
+// SUPER:           "source": "s:13InheritedDocs1SV9superFuncyyF"
+// SUPER-NEXT:      "target": "s:13InheritedDocs1SV"
+// SUPER-NEXT:      "sourceOrigin"
+// SUPER-NEXT:        "identifier": "s:13InheritedDocs1PP9superFuncyyF"
+// SUPER-NEXT:        "displayName": "P.superFunc()"
+
 /// Protocol P
 public protocol P {
     /// Some Function
     func someFunc()
 
     /// It's a local function!
     func localFunc()
+
+    func superFunc()
 }
 
 public extension P {
@@ -86,4 +98,6 @@ public extension P {
 
 public struct S: P {
     public func localFunc() {}
+
+    public func superFunc() {}
 }