Contents

require 'childprocess'
require 'tempfile'
require 'shellwords'

module ::Cuken
  module Api
    module Aruba
      class Process
        include Shellwords

        def remove_ansi_codes(str)
          str.gsub(/\e\[(\d+)m/, '')
        end

        def initialize(cmd, exit_timeout, io_wait)
          @exit_timeout = exit_timeout
          @io_wait = io_wait

          @out = ::Tempfile.new("aruba-out")
          @err = ::Tempfile.new("aruba-err")
          @process = ChildProcess.build(*shellwords(cmd))
          @process.io.stdout = @out
          @process.io.stderr = @err
          @process.duplex = true
        end

        def run!(&block)
          @process.start
          yield self if block_given?
        end

        def stdin
          wait_for_io do
            @process.io.stdin
          end
        end

        def output
          stdout + stderr
        end

        def stdout
          wait_for_io do
            @out.rewind
            remove_ansi_codes(@out.read)
          end
        end

        def stderr
          wait_for_io do
            @err.rewind
            remove_ansi_codes(@err.read)
          end
        end

        def stop
          if @process
            stdout && stderr # flush output
            @process.poll_for_exit(@exit_timeout)
            @process.exit_code
          end
        end

        private

        def wait_for_io(&block)
          sleep @io_wait if @process.alive?
          yield
        end
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
cuken-0.1.22	lib/cuken/api/aruba/process.rb
cuken-0.1.21	lib/cuken/api/aruba/process.rb
cuken-0.1.20	lib/cuken/api/aruba/process.rb
cuken-0.1.19	lib/cuken/api/aruba/process.rb
cuken-0.1.18	lib/cuken/api/aruba/process.rb
cuken-0.1.17	lib/cuken/api/aruba/process.rb
cuken-0.1.16	lib/cuken/api/aruba/process.rb
cuken-0.1.15	lib/cuken/api/aruba/process.rb