Add the ability to control the environment passed to the manifest and plugin compilation

Author: abertelrud

Sources/PackageLoading/ManifestLoader.swift

@@ -44,6 +44,9 @@ public protocol ManifestResourceProvider {
 
     /// Extra flags to pass the Swift compiler.
     var swiftCompilerFlags: [String] { get }
+    
+    /// Additional environment entries to pass to the Swift compiler and linker.
+    var swiftCompilerEnvironment: [String: String] { get }
 
     /// XCTest Location
     var xctestLocation: AbsolutePath? { get }
@@ -63,6 +66,10 @@ public extension ManifestResourceProvider {
     var swiftCompilerFlags: [String] {
         return []
     }
+    
+    var swiftCompilerEnvironment: [String: String] {
+        return [:]
+    }
 
     var xctestLocation: AbsolutePath? {
         return nil
@@ -786,8 +793,10 @@ public final class ManifestLoader: ManifestLoaderProtocol {
             let compiledManifestFile = tmpDir.appending(component: "\(packageIdentity)-manifest\(executableSuffix)")
             cmd += ["-o", compiledManifestFile.pathString]
 
+            let compilerEnv = ProcessEnv.vars.merging(resources.swiftCompilerEnvironment, uniquingKeysWith: { $1 })
+
             // Compile the manifest.
-            let compilerResult = try Process.popen(arguments: cmd)
+            let compilerResult = try Process.popen(arguments: cmd, environment: compilerEnv)
             let compilerOutput = try (compilerResult.utf8Output() + compilerResult.utf8stderrOutput()).spm_chuzzle()
             result.compilerOutput = compilerOutput
 

Sources/Workspace/DefaultPluginScriptRunner.swift

@@ -114,8 +114,10 @@ public struct DefaultPluginScriptRunner: PluginScriptRunner {
         command += sources.paths.map { $0.pathString }
         let compiledExec = cacheDir.appending(component: "compiled-plugin")
         command += ["-o", compiledExec.pathString]
+        
+        let environment = ProcessEnv.vars.merging(resources.swiftCompilerEnvironment, uniquingKeysWith: { $1 })
 
-        let result = try Process.popen(arguments: command)
+        let result = try Process.popen(arguments: command, environment: environment)
         let output = try (result.utf8Output() + result.utf8stderrOutput()).spm_chuzzle() ?? ""
         if result.exitStatus != .terminated(code: 0) {
             // TODO: Make this a proper error.

Tests/PackageLoadingTests/PD_5_0_LoadingTests.swift

@@ -547,4 +547,66 @@ class PackageDescription5_0LoadingTests: PackageDescriptionLoadingTests {
             XCTAssertEqual(manifest.dependencies, [])
         }
     }
+
+    func testManifestLoaderEnvironment() throws {
+        try testWithTemporaryDirectory { path in
+            let fs = localFileSystem
+
+            struct CustomManifestResources: ManifestResourceProvider {
+                var swiftCompiler: AbsolutePath
+                var libDir: AbsolutePath
+                var binDir: AbsolutePath?
+                var sdkRoot: AbsolutePath?
+                var swiftCompilerEnvironment: [String: String]
+            }
+
+            let packagePath = path.appending(component: "pkg")
+            let manifestPath = packagePath.appending(component: "Package.swift")
+            try fs.writeFileContents(manifestPath) { stream in
+                stream <<< """
+                // swift-tools-version:5
+                import PackageDescription
+                
+                let package = Package(
+                    name: "Trivial",
+                    targets: [
+                        .target(
+                            name: "foo",
+                            dependencies: []),
+                    ]
+                )
+                """
+            }
+
+            let moduleTraceFilePath = path.appending(component: "swift-module-trace")
+            let customResources = CustomManifestResources(
+                swiftCompiler: Resources.default.swiftCompiler,
+                libDir: Resources.default.libDir,
+                binDir: Resources.default.binDir,
+                sdkRoot: Resources.default.sdkRoot,
+                swiftCompilerEnvironment: ["SWIFT_LOADED_MODULE_TRACE_FILE": moduleTraceFilePath.pathString])
+            let manifestLoader = ManifestLoader(
+                manifestResources: customResources,
+                serializedDiagnostics: true,
+                isManifestSandboxEnabled: false,
+                cacheDir: nil)
+
+            let diagnostics = DiagnosticsEngine()
+            let manifest = try manifestLoader.load(
+                at: manifestPath.parentDirectory,
+                packageKind: .local,
+                packageLocation: manifestPath.pathString,
+                toolsVersion: .v5,
+                fileSystem: fs,
+                diagnostics: diagnostics
+            )
+
+            XCTAssertTrue(diagnostics.diagnostics.isEmpty)
+            XCTAssertEqual(manifest.name, "Trivial")
+            
+            if let moduleTraceJSON = try? localFileSystem.readFileContents(moduleTraceFilePath).validDescription {
+                XCTAssert(moduleTraceJSON.contains("PackageDescription"))
+            }
+        }
+    }
 }