Backport addenv (#721)

Author: tkf

Project.toml

@@ -1,6 +1,6 @@
 name = "Compat"
 uuid = "34da2185-b29b-5c13-b0c7-acf172513d20"
-version = "3.15.0"
+version = "3.16.0"
 
 [deps]
 Base64 = "2a0f44e3-6c83-55bd-87e4-b1978d98bd5f"

README.md

@@ -55,6 +55,8 @@ changes in `julia`.
 
 ## Supported features
 
+* New function `addenv` for adding environment mappings into a `Cmd` object, returning the new `Cmd` object ([#37244]). (since Compat 3.16)
+
 * `contains(haystack, needle)` and its one argument partially applied form `contains(haystack)` have been added, it acts like `occursin(needle, haystack)` ([#35132]). (since Compat 3.15)
 
 * `startswith(x)` and `endswith(x)`, returning partially-applied versions of the functions, similar to existing methods like `isequal(x)` ([#35052]). (since Compat 3.15)
@@ -197,3 +199,4 @@ Note that you should specify the correct minimum version for `Compat` in the
 [#34352]: https://github.com/JuliaLang/julia/pull/34352
 [#35132]: https://github.com/JuliaLang/julia/pull/35132
 [#35052]: https://github.com/JuliaLang/julia/pull/35052
+[#37244]: https://github.com/JuliaLang/julia/pull/37244

src/Compat.jl

@@ -607,6 +607,31 @@ if VERSION < v"1.5.0-DEV.438" # 0a43c0f1d21ce9c647c49111d93927369cd20f85
     Base.startswith(s) = Base.Fix2(startswith, s)
 end
 
+# https://github.com/JuliaLang/julia/pull/37244
+if VERSION < v"1.6.0-DEV.873" # 18198b1bf85125de6cec266eac404d31ccc2e65c
+    export addenv
+    function addenv(cmd::Cmd, env::Dict)
+        new_env = Dict{String,String}()
+        if cmd.env !== nothing
+            for (k, v) in split.(cmd.env, "=")
+                new_env[string(k)::String] = string(v)::String
+            end
+        end
+        for (k, v) in env
+            new_env[string(k)::String] = string(v)::String
+        end
+        return setenv(cmd, new_env)
+    end
+
+    function addenv(cmd::Cmd, pairs::Pair{<:AbstractString}...)
+        return addenv(cmd, Dict(k => v for (k, v) in pairs))
+    end
+
+    function addenv(cmd::Cmd, env::Vector{<:AbstractString})
+        return addenv(cmd, Dict(k => v for (k, v) in split.(env, "=")))
+    end
+end
+
 include("iterators.jl")
 include("deprecated.jl")
 

test/runtests.jl

@@ -556,6 +556,34 @@ end
     @test endswith("d")("abcd")
 end
 
+# From spawn.jl
+shcmd = `sh`
+havebb = false
+if Sys.iswindows()
+    busybox = download("https://cache.julialang.org/https://frippery.org/files/busybox/busybox.exe", joinpath(tempdir(), "busybox.exe"))
+    havebb = try # use busybox-w32 on windows, if available
+        success(`$busybox`)
+        true
+    catch
+        false
+    end
+    if havebb
+        shcmd = `$busybox sh`
+    end
+end
+
+# https://github.com/JuliaLang/julia/pull/37244
+@testset "addenv()" begin
+    cmd = Cmd(`$shcmd -c "echo \$FOO \$BAR"`, env=Dict("FOO" => "foo"))
+    @test strip(String(read(cmd))) == "foo"
+    cmd = addenv(cmd, "BAR" => "bar")
+    @test strip(String(read(cmd))) == "foo bar"
+    cmd = addenv(cmd, Dict("FOO" => "bar"))
+    @test strip(String(read(cmd))) == "bar bar"
+    cmd = addenv(cmd, ["FOO=baz"])
+    @test strip(String(read(cmd))) == "baz bar"
+end
+
 include("iterators.jl")
 
 nothing