Add Dependencies in project group

bhargavg · bhargavg · commit 5cd0b68e7f15 · 2016-03-19T02:08:06.000+05:30
diff --git a/Sources/Transmute/transmute().swift b/Sources/Transmute/transmute().swift
@@ -12,7 +12,7 @@ import PackageType
 import Utility
 import func libc.exit
 
-public func transmute(packages: [Package], rootdir: String) throws -> ([Module], [Product]) {
+public func transmute(packages: [Package], rootdir: String) throws -> (modules: [Module], externalModules: [Module], products: [Product]) {
 
     var products: [Product] = []
     var map: [Package: [Module]] = [:]
@@ -66,18 +66,10 @@ public func transmute(packages: [Package], rootdir: String) throws -> ([Module],
     // ensure modules depend on the modules of any dependent packages
     fillModuleGraph(packages, modulesForPackage: { map[$0]! })
 
-    var set = Set<Module>()
-    var stack = packages.flatMap{ map[$0] ?? [] }
-    var modules = [Module]()
+    let depPackages = packages.filter{ $0.path != rootdir }
 
-    while !stack.isEmpty {
-        let module = stack.removeFirst()
-        if !set.contains(module) {
-            set.insert(module)
-            stack += module.dependencies
-            modules.append(module)
-        }
-    }
+    let modules = recursiveDependencies(packages.flatMap{ map[$0] ?? [] })
+    let extModules = recursiveDependencies(depPackages.flatMap{ map[$0] ?? [] })
 
-    return (modules, products)
-}
+    return (modules, extModules, products)
+}
diff --git a/Sources/Xcodeproj/Module+PBXProj.swift b/Sources/Xcodeproj/Module+PBXProj.swift
@@ -35,6 +35,7 @@ let rootGroupReference =                            "___RootGroup_"
 let productsGroupReference =                        "____Products_"
 let testProductsGroupReference =                    "TestProducts_"
 let sourcesGroupReference =                         "_____Sources_"
+let dependenciesGroupReference =                    "Dependencies_"
 let testsGroupReference =                           "_______Tests_"
 let linkPhaseFileRefPrefix =                        "_LinkFileRef_"
 let sourceGroupFileRefPrefix =                      "__PBXFileRef_"
diff --git a/Sources/Xcodeproj/generate().swift b/Sources/Xcodeproj/generate().swift
@@ -16,7 +16,7 @@ import POSIX
  Generates an xcodeproj at the specified path.
  - Returns: the path to the generated project
 */
-public func generate(dstdir dstdir: String, projectName: String, srcroot: String, modules: [SwiftModule], products: [Product]) throws -> String {
+public func generate(dstdir dstdir: String, projectName: String, srcroot: String, modules: [SwiftModule], externalModules: [SwiftModule], products: [Product]) throws -> String {
 
     let xcodeprojName = "\(projectName).xcodeproj"
     let xcodeprojPath = try mkdir(dstdir, xcodeprojName)
@@ -25,7 +25,7 @@ public func generate(dstdir dstdir: String, projectName: String, srcroot: String
 
 ////// the pbxproj file describes the project and its targets
     try open(xcodeprojPath, "project.pbxproj") { fwrite in
-        pbxproj(srcroot: srcroot, projectRoot: dstdir, modules: modules, products: products, printer: fwrite)
+        pbxproj(srcroot: srcroot, projectRoot: dstdir, modules: modules, externalModules: externalModules, products: products, printer: fwrite)
     }
 
 ////// the scheme acts like an aggregate target for all our targets
diff --git a/Sources/Xcodeproj/pbxproj().swift b/Sources/Xcodeproj/pbxproj().swift
@@ -15,9 +15,11 @@
 import PackageType
 import Utility
 
-public func pbxproj(srcroot srcroot: String, projectRoot: String, modules: [SwiftModule], products _: [Product], printer print: (String) -> Void) {
-    let nontests = modules.filter{ !($0 is TestModule) }
-    let tests = modules.filter{ $0 is TestModule }
+public func pbxproj(srcroot srcroot: String, projectRoot: String, modules: [SwiftModule], externalModules: [SwiftModule], products _: [Product], printer print: (String) -> Void) {
+
+    let rootModulesSet = Set(modules).subtract(Set(externalModules))
+    let nonTestRootModules = rootModulesSet.filter{ !($0 is TestModule) }
+    let (tests, nonTests) = modules.partition{ $0 is TestModule }
 
     print("// !$*UTF8*$!")
     print("{")
@@ -47,7 +49,7 @@ public func pbxproj(srcroot srcroot: String, projectRoot: String, modules: [Swif
 ////// root group
     print("        \(rootGroupReference) = {")
     print("            isa = PBXGroup;")
-    print("            children = (\(sourcesGroupReference), \(testsGroupReference), \(productsGroupReference));")
+    print("            children = (\(sourcesGroupReference), \(dependenciesGroupReference), \(testsGroupReference), \(productsGroupReference));")
     print("            sourceTree = '<group>';")
     print("        };")
 
@@ -147,11 +149,19 @@ public func pbxproj(srcroot srcroot: String, projectRoot: String, modules: [Swif
 ////// “Sources” group
     print("        \(sourcesGroupReference) = {")
     print("            isa = PBXGroup;")
-    print("            children = (" + nontests.map{ $0.groupReference }.joined(separator: ", ") + ");")
+    print("            children = (" + nonTestRootModules.map{ $0.groupReference }.joined(separator: ", ") + ");")
     print("            name = Sources;")
     print("            sourceTree = '<group>';")
     print("        };")
 
+    ////// “Dependencies” group
+    print("        \(dependenciesGroupReference) = {")
+    print("            isa = PBXGroup;")
+    print("            children = (" + externalModules.map{ $0.groupReference }.joined(separator: ", ") + ");")
+    print("            name = Dependencies;")
+    print("            sourceTree = '<group>';")
+    print("        };")
+
 ////// “Tests” group
     print("        \(testsGroupReference) = {")
     print("            isa = PBXGroup;")
@@ -184,7 +194,7 @@ public func pbxproj(srcroot srcroot: String, projectRoot: String, modules: [Swif
     }
 
 ////// “Products” group
-    productReferences += nontests.map { $0.productReference }
+    productReferences += nonTests.map { $0.productReference }
 
     print("        \(productsGroupReference) = {")
     print("            isa = PBXGroup;")
diff --git a/Sources/swift-build/main.swift b/Sources/swift-build/main.swift
@@ -67,7 +67,7 @@ do {
         case .Build(let conf):
             let dirs = try directories()
             let packages = try fetch(dirs.root)
-            let (modules, products) = try transmute(packages, rootdir: dirs.root)
+            let (modules, externalModules, products) = try transmute(packages, rootdir: dirs.root)
             let yaml = try describe(dirs.build, conf, modules, products, Xcc: opts.Xcc, Xld: opts.Xld, Xswiftc: opts.Xswiftc)
             try build(YAMLPath: yaml, target: "default")
 
@@ -93,8 +93,10 @@ do {
         case .GenerateXcodeproj(let outpath):
             let dirs = try directories()
             let packages = try fetch(dirs.root)
-            let (modules, products) = try transmute(packages, rootdir: dirs.root)
+            let (modules, externalModules, products) = try transmute(packages, rootdir: dirs.root)
+            
             let swiftModules = modules.flatMap{ $0 as? SwiftModule }
+            let externalSwiftModules = externalModules.flatMap{ $0 as? SwiftModule }
 
             let projectName: String
             let dstdir: String
@@ -114,7 +116,7 @@ do {
                 projectName = packageName
             }
 
-            let outpath = try Xcodeproj.generate(dstdir: dstdir, projectName: projectName, srcroot: dirs.root, modules: swiftModules, products: products)
+            let outpath = try Xcodeproj.generate(dstdir: dstdir, projectName: projectName, srcroot: dirs.root, modules: swiftModules, externalModules: externalSwiftModules, products: products)
 
             print("generated:", outpath.prettied)
     }
diff --git a/Tests/Transmute/ModuleTests.swift b/Tests/Transmute/ModuleTests.swift
@@ -201,8 +201,8 @@ extension ModuleTests {
             let prefix = Path.join(prefix, "App")
             let manifest = try Manifest(path: prefix)
             let packages = try get(manifest, manifestParser: { try Manifest(path: $0, baseURL: $1) })
-            let (modules, _) = try transmute(packages, rootdir: prefix)
-
+            let (modules, _,  _) = try transmute(packages, rootdir: prefix)
+            
             XCTAssertEqual(modules.count, 3)
             XCTAssertEqual(recursiveDependencies(modules).count, 3)
             XCTAssertTrue(modules.dropFirst().first is CModule)
@@ -212,7 +212,7 @@ extension ModuleTests {
             let prefix = Path.join(prefix, "App")
             let manifest = try Manifest(path: prefix)
             let packages = try get(manifest, manifestParser: { try Manifest(path: $0, baseURL: $1) })
-            let (modules, _) = try transmute(packages, rootdir: prefix)
+            let (modules, _, _) = try transmute(packages, rootdir: prefix)
 
             XCTAssertEqual(modules.count, 2)
             XCTAssertTrue(modules.first is CModule)
diff --git a/Tests/Xcodeproj/TestGeneration.swift b/Tests/Xcodeproj/TestGeneration.swift
@@ -51,7 +51,7 @@ class TestGeneration: XCTestCase {
             let modules = dummy()
             let products: [Product] = []
 
-            let outpath = try Xcodeproj.generate(dstdir: dstdir, projectName: projectName, srcroot: srcroot, modules: modules, products: products)
+            let outpath = try Xcodeproj.generate(dstdir: dstdir, projectName: projectName, srcroot: srcroot, modules: modules, externalModules: [], products: products)
 
             XCTAssertDirectoryExists(outpath)
             XCTAssertEqual(outpath, Path.join(dstdir, "\(projectName).xcodeproj"))
