├── .gitignore
├── test
    ├── slow_script.rb
    └── rtimeout_test.rb
├── README
├── Manifest
├── bin
    └── rtimeout
├── Rakefile
└── lib
    └── rtimeout.rb


/.gitignore:
--------------------------------------------------------------------------------
1 | rtimeout.gemspec
2 | pkg
3 | 


--------------------------------------------------------------------------------
/test/slow_script.rb:
--------------------------------------------------------------------------------
1 | sleep 1.5
2 | puts "It worked!"
3 | 


--------------------------------------------------------------------------------
/README:
--------------------------------------------------------------------------------
1 | $ rtimeout <timeout> <script to run>
2 | 
3 | For: Github 
4 | By : chris@ozmm.org
5 | 


--------------------------------------------------------------------------------
/Manifest:
--------------------------------------------------------------------------------
1 | bin/rtimeout
2 | lib/rtimeout.rb
3 | README
4 | test/rtimeout_test.rb
5 | test/slow_script.rb
6 | Manifest
7 | 


--------------------------------------------------------------------------------
/bin/rtimeout:
--------------------------------------------------------------------------------
 1 | #!/usr/bin/env ruby
 2 | $LOAD_PATH.unshift File.dirname(__FILE__) + '/../lib'
 3 | require 'rtimeout'
 4 | 
 5 | if ARGV.size < 2
 6 |   STDERR.puts "Usage: $ rtimeout <timeout> <command to run>" 
 7 |   exit! 1
 8 | end
 9 | 
10 | Rtimeout.read(ARGV.shift, ARGV.join(' '))
11 | 


--------------------------------------------------------------------------------
/Rakefile:
--------------------------------------------------------------------------------
 1 | require 'rubygems'
 2 | require 'rake'
 3 | 
 4 | version = '1.0'
 5 | 
 6 | begin
 7 |   require 'echoe'
 8 | 
 9 |   Echoe.new('rtimeout', version) do |p|
10 |     p.rubyforge_name = 'github'
11 |     p.summary        = "For timing out system commands."
12 |     p.url            = "http://github.com/defunkt/rtimeout"
13 |     p.author         = 'Chris Wanstrath'
14 |     p.email          = "chris@ozmm.org"
15 |     p.dependencies   = ['open4 >=0.9.6']
16 |   end
17 | 
18 | rescue LoadError => boom
19 |   puts "You are missing a dependency required for meta-operations on this gem."
20 |   puts "#{boom.to_s.capitalize}."
21 | end
22 | 


--------------------------------------------------------------------------------
/lib/rtimeout.rb:
--------------------------------------------------------------------------------
 1 | require 'rubygems'
 2 | require 'open4'
 3 | 
 4 | class Rtimeout
 5 |   def initialize(timeout, command)
 6 |     @timeout = timeout.to_i
 7 |     @command = command
 8 |     @pid, _, @io, _ = Open4.popen4(command)
 9 |   end
10 | 
11 |   def read
12 |     while result = IO.select([@io], nil, nil, @timeout)
13 |       out = @io.read(1)
14 |       break if out.nil?
15 |       $stdout.print out
16 |     end
17 | 
18 |     if result.nil?
19 |       system "kill -9 #{@pid}"
20 |       $stderr.puts "rtimeout: Command `#{@command}` timed out." 
21 |     end
22 |   end
23 | 
24 |   def self.read(timeout, command)
25 |     new(timeout, command).read
26 |   end
27 | end
28 | 


--------------------------------------------------------------------------------
/test/rtimeout_test.rb:
--------------------------------------------------------------------------------
 1 | require 'rubygems'
 2 | require 'test/unit'
 3 | require 'mocha'
 4 | 
 5 | $LOAD_PATH.unshift File.dirname(__FILE__) + '/../lib'
 6 | require 'rtimeout'
 7 | 
 8 | def context(*args, &block)
 9 |   return super unless (name = args.first) && block
10 |   require 'test/unit'
11 |   klass = Class.new(Test::Unit::TestCase) do
12 |     def self.specify(name, &block) 
13 |       define_method("test_#{name.gsub(/\W/,'_')}", &block) 
14 |     end
15 |     def self.xspecify(*args) end
16 |     def self.setup(&block) define_method(:setup, &block) end
17 |     def self.teardown(&block) define_method(:teardown, &block) end
18 |   end
19 |   klass.class_eval &block
20 | end
21 | 
22 | module FakeIO
23 |   def fake_stdout
24 |     old_io = $stdout.dup
25 |     tempio = StringIO.new
26 | 
27 |     $stdout = tempio
28 |     yield
29 |     tempio.string
30 |   ensure
31 |     $stdout = old_io
32 |   end
33 | 
34 |   def fake_stderr
35 |     old_io = $stderr.dup
36 |     tempio = StringIO.new
37 | 
38 |     $stderr = tempio
39 |     yield
40 |     tempio.string
41 |   ensure
42 |     $stderr = old_io
43 |   end
44 | end
45 | 
46 | context "rtimeout" do
47 |   include FakeIO
48 | 
49 |   setup do
50 |     @test_script = 'ruby ' + File.dirname(__FILE__) + '/slow_script.rb'
51 |   end
52 | 
53 |   specify "prints to stdout on success" do
54 |     out = fake_stdout do
55 |       Rtimeout.read(3, @test_script)
56 |     end
57 | 
58 |     assert_equal "It worked!\n", out
59 |   end
60 | 
61 |   specify "prints to stderr on timeout" do
62 |     out = fake_stderr do
63 |       Rtimeout.read(1, @test_script)
64 |     end
65 | 
66 |     assert_equal "rtimeout: Command `#{@test_script}` timed out.\n", out
67 |   end
68 | end
69 | 


--------------------------------------------------------------------------------