Add C++ support in ClangModule

swiftlang · May 21, 2016 · 5ec54ab · 5ec54ab
1 parent 8a4372b
commit 5ec54ab
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 2 deletions.
diff --git a/Sources/Build/Command.compile(ClangModule).swift b/Sources/Build/Command.compile(ClangModule).swift
@@ -86,7 +86,10 @@ extension Command {
             var args = basicArgs
             args += ["-MD", "-MT", "dependencies", "-MF", path.deps]
             args += ["-c", path.source, "-o", path.object]
-
+            // Append -std flag if file is cpp.
+            if path.source.isCpp {
+               args += ["-std=c++14"]
+            }
             let clang = ClangTool(desc: "Compile \(module.name) \(path.filename)",
                                   inputs: dependencies + [path.source],
                                   outputs: [path.object],
@@ -105,6 +108,7 @@ extension Command {
         var args = module.basicArgs
         args += module.optimizationFlags(conf)
         args += ["-L\(prefix)"]
+        args += module.languageArgs
         args += module.linkFlags
         args += module.sources.compilePathsForBuildDir(wd).map{$0.object}
         args += Xld

diff --git a/Sources/Build/misc.swift b/Sources/Build/misc.swift
@@ -43,6 +43,42 @@ extension CModule {
     }
 }
 
+extension String {
+    var isCpp: Bool {
+        guard let ext = self.fileExt else {
+            return false
+        }
+        return ext == "cpp"
+    }
+}
+
+extension ClangModule {
+    // Returns language specific arguments for a ClangModule.
+    var languageArgs: [String] {
+        var args = [String]() 
+        // Check if this module contains any cpp file.
+        var linkCpp = self.containsCppFiles
+        // Otherwise check if any of its dependencies contains a cpp file.
+        if !linkCpp {
+            for case let dep as ClangModule in recursiveDependencies {
+                if dep.containsCppFiles {
+                    linkCpp = true
+                    break
+                }
+            }
+        }
+        // Link C++ if found any cpp source. 
+        if linkCpp {
+            args += ["-lc++"]
+        }
+        return args
+    }
+
+    var containsCppFiles: Bool {
+        return sources.paths.contains { $0.isCpp } 
+    }
+}
+
 extension ClangModule {
 
     public enum ModuleMapError: ErrorProtocol {

diff --git a/Sources/PackageModel/Sources.swift b/Sources/PackageModel/Sources.swift
@@ -29,7 +29,7 @@ public struct Sources {
 
     static public var validSwiftExtensions = Set<String>(["swift"])
 
-    static public var validCExtensions = Set<String>(["c", "m"])
+    static public var validCExtensions = Set<String>(["c", "m", "cc", "cpp", "cxx"])
 
     static public var validExtensions = { validSwiftExtensions.union(validCExtensions) }()
 }