ASTPrinter: Move the logic for collecting all module groups from SourceKit to an IDE API, NFC.

Author: nkcsgexi

include/swift/IDE/ModuleInterfacePrinting.h

@@ -16,6 +16,8 @@
 #include "swift/Basic/LLVM.h"
 #include "swift/Basic/OptionSet.h"
 
+#include <vector>
+
 namespace swift {
 class ASTContext;
 class ASTPrinter;
@@ -38,6 +40,9 @@ enum class ModuleTraversal : unsigned {
 /// Options used to describe the traversal of a module for printing.
 typedef OptionSet<ModuleTraversal> ModuleTraversalOptions;
 
+ArrayRef<StringRef> collectModuleGroups(ModuleDecl *M,
+                                        std::vector<StringRef> &Scratch);
+
 void printModuleInterface(ModuleDecl *M,
                           ModuleTraversalOptions TraversalOptions,
                           ASTPrinter &Printer, const PrintOptions &Options,

lib/IDE/ModuleInterfacePrinting.cpp

@@ -176,6 +176,17 @@ void findExtensionsFromConformingProtocols(Decl *D,
   }
 }
 
+ArrayRef<StringRef>
+swift::ide::collectModuleGroups(Module *M, std::vector<StringRef> &Scratch) {
+  for (auto File : M->getFiles()) {
+    File->collectAllGroups(Scratch);
+  }
+  std::sort(Scratch.begin(), Scratch.end(), [](StringRef L, StringRef R) {
+    return L.compare_lower(R) < 0;
+  });
+  return llvm::makeArrayRef(Scratch);
+}
+
 void swift::ide::printSubmoduleInterface(
        Module *M,
        ArrayRef<StringRef> FullModuleName,

tools/SourceKit/lib/SwiftLang/SwiftDocSupport.cpp

@@ -901,11 +901,6 @@ void SwiftLangSupport::findModuleGroups(StringRef ModuleName,
     Receiver(Groups, Error);
     return;
   }
-  for (auto File : M->getFiles()) {
-    File->collectAllGroups(Groups);
-  }
-  std::sort(Groups.begin(), Groups.end(), [](StringRef L, StringRef R) {
-    return L.compare_lower(R) < 0;
-  });
-  Receiver(Groups, Error);
+  std::vector<StringRef> Scratch;
+  Receiver(collectModuleGroups(M, Scratch), Error);
 }