#!/usr/bin/env ruby

class Object
  def trapr_wrap(signal, newproc, function, *args)
    #puts "Trapping #{signal} with #{newproc.inspect}"
    oldproc = trap "#{signal}", newproc
    self.send("#{function}", *args)
  ensure
    #puts "Reseting trap. #{signal} -> #{oldproc.inspect}"
    trap "#{signal}", oldproc
  end
end

class Eggplant
  def initialize
  end
  
  def foo
    trap "SIGINT", Proc.new {
      puts "Trapped in foo!"
    }
    puts "I'm in foo for 5 seconds!"
    sleep(5);
    puts "Leaving foo."
  end

  def bar
    puts "I'm bar'n it up for 5 seconds!"
    sleep(5)
    puts "Done bar'n"
  end

end

trap "SIGINT", Proc.new { puts "Base trap." }
t = Eggplant.new
p = Proc.new { puts "Trapr'd!" }
# Trap bindings will be released when :foo exits
t.trapr_wrap "SIGINT", p, :foo
# Trap bindings will be released when :bar exits
t.trapr_wrap "SIGINT", p, :bar
puts "Back to base. T-minus 5."
sleep(5)


#  [1:hinmanm@Xanadu:~/src/ruby/trapr]% ./trapr.rb
#  I'm in foo for 5 seconds!
#  ^CTrapped in foo!
#  Leaving foo.
#  I'm bar'n it up for 5 seconds!
#  ^CTrapr'd!
#  Done bar'n
#  Back to base. T-minus 5.
#  ^CBase trap.