AST printer: print declarations from a module in alphabetical order

Also, move module interface printing to libAST to make it reuseable.


Swift SVN r10932

Author: gribozavr

include/swift/AST/ModuleInterfacePrinting.h

@@ -0,0 +1,28 @@
+//===--- ModuleInterfacePrinting.h - Routines to print module interface ---===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2015 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 SWIFT_AST_MODULE_INTERFACE_PRINTING_H
+#define SWIFT_AST_MODULE_INTERFACE_PRINTING_H
+
+#include "swift/Basic/LLVM.h"
+
+namespace swift {
+class Module;
+struct PrintOptions;
+
+void printModuleInterface(Module *M, raw_ostream &OS,
+                          const PrintOptions &Options);
+
+} // namespace swift
+
+#endif // SWIFT_AST_MODULE_INTERFACE_PRINTING_H
+

lib/AST/ASTPrinter.cpp

@@ -20,6 +20,8 @@
 #include "swift/AST/Decl.h"
 #include "swift/AST/Expr.h"
 #include "swift/AST/Module.h"
+#include "swift/AST/ModuleInterfacePrinting.h"
+#include "swift/AST/NameLookup.h"
 #include "swift/AST/PrintOptions.h"
 #include "swift/AST/Stmt.h"
 #include "swift/AST/TypeVisitor.h"
@@ -1602,3 +1604,39 @@ void TypeBase::print(raw_ostream &OS, const PrintOptions &PO) const {
   Type(const_cast<TypeBase *>(this)).print(OS, PO);
 }
 
+void swift::printModuleInterface(Module *M, raw_ostream &OS,
+                                 const PrintOptions &Options) {
+  auto AdjustedOptions = Options;
+  // Don't print empty curly braces while printing the module interface.
+  AdjustedOptions.FunctionDefinitions = false;
+
+  SmallVector<ValueDecl *, 32> Decls;
+  VectorDeclConsumer Consumer(Decls);
+
+  M->lookupVisibleDecls(Module::AccessPathTy(), Consumer,
+                        NLKind::UnqualifiedLookup);
+
+  // Sort the declarations so that we print them in a consistent order.
+  std::sort(Decls.begin(), Decls.end(),
+            [](ValueDecl *LHS, ValueDecl *RHS) {
+    StringRef LHSName = LHS->getName().str();
+    StringRef RHSName = RHS->getName().str();
+    if (LHSName.compare(RHSName) < 0)
+      return true;
+    // FIXME: this is not sufficient to establish a total order for overloaded
+    // decls.
+    return RHS->getKind() < RHS->getKind();
+  });
+
+  for (auto *VD : Decls) {
+    VD->print(OS, AdjustedOptions);
+    OS << '\n';
+    if (auto NTD = dyn_cast<NominalTypeDecl>(VD)) {
+      for (auto Ext : NTD->getExtensions()) {
+        Ext->print(OS, AdjustedOptions);
+        OS << '\n';
+      }
+    }
+  }
+}
+