SourceFile::getDisplayDecls also walks @_exported imports

Author: QuietMisdreavus

include/swift/AST/SourceFile.h

@@ -393,6 +393,8 @@ class SourceFile final : public FileUnit {
 public:
   virtual void getTopLevelDecls(SmallVectorImpl<Decl*> &results) const override;
 
+  virtual void getDisplayDecls(SmallVectorImpl<Decl*> &results) const override;
+
   virtual void
   getOperatorDecls(SmallVectorImpl<OperatorDecl *> &results) const override;
 

lib/AST/Module.cpp

@@ -822,6 +822,17 @@ void SourceFile::getTopLevelDecls(SmallVectorImpl<Decl*> &Results) const {
   Results.append(decls.begin(), decls.end());
 }
 
+void SourceFile::getDisplayDecls(SmallVectorImpl<Decl*> &Results) const {
+  if (Imports.hasValue()) {
+    for (auto import : *Imports) {
+      if (import.options.contains(ImportFlags::Exported)) {
+        import.module.importedModule->getDisplayDecls(Results);
+      }
+    }
+  }
+  getTopLevelDecls(Results);
+}
+
 void ModuleDecl::getOperatorDecls(
     SmallVectorImpl<OperatorDecl *> &results) const {
   // For a parsed module, we can check the source cache on the module rather

test/SymbolGraph/ClangImporter/EmitWhileBuilding.swift

@@ -3,13 +3,17 @@
 // RUN: %target-swift-frontend(mock-sdk: %clang-importer-sdk) -enable-objc-interop -emit-module-path %t/EmitWhileBuilding.framework/Modules/EmitWhileBuilding.swiftmodule/%target-swiftmodule-name -import-underlying-module -F %t -module-name EmitWhileBuilding -disable-objc-attr-requires-foundation-module %s -emit-symbol-graph -emit-symbol-graph-dir %t
 // RUN: %{python} -m json.tool %t/EmitWhileBuilding.symbols.json %t/EmitWhileBuilding.formatted.symbols.json
 // RUN: %FileCheck %s --input-file %t/EmitWhileBuilding.formatted.symbols.json
+// RUN: %FileCheck %s --input-file %t/EmitWhileBuilding.formatted.symbols.json --check-prefix HEADER
 
 // REQUIRES: objc_interop
 
 import Foundation
 
 public enum SwiftEnum {}
 
+// HEADER:       "precise": "c:@testVariable"
+
+// CHECK:        "precise": "s:17EmitWhileBuilding9SwiftEnumO",
 // CHECK:        "declarationFragments": [
 // CHECK-NEXT:       {
 // CHECK-NEXT:           "kind": "keyword",