Allow passing extra environment variables to PuppetCommand.format_com…

…mand
theforeman · Oct 18, 2024 · 343d39f · 343d39f
1 parent 05ecf82
commit 343d39f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/kafo/puppet_command.rb b/lib/kafo/puppet_command.rb
@@ -50,11 +50,11 @@ def self.aio_puppet?
       false
     end
 
-    def self.format_command(command)
+    def self.format_command(command, extra_env = {})
       if aio_puppet?
-        [clean_env_vars, command, :unsetenv_others => true]
+        [clean_env_vars.merge(extra_env), command, :unsetenv_others => true]
       else
-        [::ENV, command, :unsetenv_others => false]
+        [::ENV.to_h.merge(extra_env), command, :unsetenv_others => false]
       end
     end