Merge pull request #17 from banjun/fix-llvm-cache-dir-error

Fix LLVM ERROR: cannot get default cache directory
banjun · Feb 5, 2024 · f56cc5b · f56cc5b
2 parents a33514b + 3549dc7
commit f56cc5b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Sources/Core/NSTaskCommand.swift b/Sources/Core/NSTaskCommand.swift
@@ -11,14 +11,20 @@ struct NSTaskCommand {
     }
 
     @discardableResult
-    func run(clearEnvironments: Bool = true) throws -> (stdout: String?, stderr: String?) {
+    func run(clearEnvironments: Bool = true, setEnvHome: String? = NSHomeDirectory()) throws -> (stdout: String?, stderr: String?) {
         let NSTask: AnyClass? = NSClassFromString("NSTask")
         let task = NSTask?.value(forKey: "new") as? NSObject
         guard let task else { throw Error.nsTaskUnavailable }
 
+        var environment = task.value(forKey: "environment") as? [AnyHashable: Any] ?? [:]
         if clearEnvironments {
-            task.setValue([:], forKey: "environment")
+            environment.removeAll()
         }
+        if let setEnvHome {
+            // preserve or add HOME to avoid error: `LLVM ERROR: cannot get default cache directory`
+            environment["HOME"] = setEnvHome
+        }
+        task.setValue(environment, forKey: "environment")
 
         task.setValue(launchPath, forKey: "launchPath")
         task.setValue(args, forKey: "arguments")