buff-shell_out 0.1.1 → 0.2.0

data/lib/buff/shell_out.rb

@@ -14,10 +14,13 @@ module Buff
     #
     # @param [String] command
     #   The command to execute
+    # @param [Hash] env
+    #   A hash of enviroment variables to set during execution
     #
     # @return [ShellOut::Response]
-    def shell_out(command)
-      process_status, out, err = jruby? ? jruby_out(command) : mri_out(command)
+    def shell_out(command, env = {})
+      env = process_env_vars(env)
+      process_status, out, err = jruby? ? jruby_out(command, env) : mri_out(command, env)
       Response.new(process_status, out, err)
     end
 
@@ -25,11 +28,11 @@ module Buff
 
       # @param [String] command
       #   The command to execute
-      def mri_out(command)
+      def mri_out(command, env = {})
         out, err = Tempfile.new('mixin.shell_out.stdout'), Tempfile.new('mixin.shell_out.stderr')
 
         begin
-          pid         = Process.spawn(command, out: out.to_i, err: err.to_i)
+          pid         = Process.spawn(env, command, out: out.to_i, err: err.to_i)
           pid, status = Process.waitpid2(pid)
           # Check if we're getting back a process status because win32-process 6.x was a fucking MURDERER.
           # https://github.com/djberg96/win32-process/blob/master/lib/win32/process.rb#L494-L519
@@ -48,10 +51,10 @@ module Buff
 
       # @param [String] command
       #   The command to execute
-      def jruby_out(command)
+      def jruby_out(command, env = {})
         out, err = StringIO.new, StringIO.new
         $stdout, $stderr = out, err
-        system(command)
+        system(env, command)
 
         exitstatus = $?.exitstatus
         out.rewind
@@ -61,5 +64,17 @@ module Buff
       ensure
         $stdout, $stderr = STDOUT, STDERR
       end
+
+      # Builds a new hash from the given vars hash. The new hash's keys are all uppercase
+      # strings representing environment variables.
+      #
+      # @param [Hash] vars
+      #   A hash of environment variables
+      def process_env_vars(vars)
+        vars.inject({}) do |acc, (key, val)|
+          acc[key.to_s.upcase] = val
+          acc
+        end
+      end
   end
 end

data/lib/buff/shell_out/version.rb

@@ -1,5 +1,5 @@
 module Buff
   module ShellOut
-    VERSION = "0.1.1"
+    VERSION = "0.2.0"
   end
 end

data/spec/buff/shell_out_spec.rb

@@ -20,7 +20,7 @@ describe Buff::ShellOut do
       before { described_class.stub(jruby?: false) }
 
       it "delegates to #mri_out" do
-        described_class.should_receive(:mri_out).with(command)
+        expect(described_class).to receive(:mri_out).with(command, {})
         result
       end
 
@@ -37,7 +37,7 @@ describe Buff::ShellOut do
       before { described_class.stub(jruby?: true) }
 
       it "delegates to #jruby_out" do
-        described_class.should_receive(:jruby_out).with(command)
+        expect(described_class).to receive(:jruby_out).with(command, {})
         result
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: buff-shell_out
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-10-22 00:00:00.000000000 Z
+date: 2014-08-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: buff-ruby_engine
@@ -216,7 +216,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3294574992538716380
+      hash: -1099429043491521833
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.23