#
# FastCGI Wrapper
#   trick to wrap a cgi in fastcgi easily
#
# Author: Tatsuki Sugiura <sugi@nemui.org>
# Licence: Ruby's
#
#
# Usage:
#  write wrapper script like following;
#    require 'fcgiwrap'
#    FCGIWrap.each {
#      load '/path/to/original.cgi'
#    }
#

require 'cgi'
require 'fcgi'
require 'singleton'

# for ruby 1.8 blow
if RUBY_VERSION.tr(".", "0").to_i < 10801
  alias $stdout $defout
  def ENV.clear
    ENV.each_key {|k| ENV.delete(k) }
  end
  def ENV.update(from)
    from.each {|k,v| ENV[k] = v}
  end
end

class FCGIWrap
  include Singleton

  VERSION = "0.1.6"

  def initialize
    cgi = nil
    shutdown_p = false
  end
  attr_accessor :cgi, :shutdown_p

  class << self
    def cgi
      FCGIWrap.instance.cgi
    end

    def shutdown?
      FCGIWrap.instance.shutdown_p
    end

    def shutdown=(f)
      FCGIWrap.instance.shutdown_p = f
    end

    def each_request
      trap(:PIPE){ exit } 
      trap(:TERM){ self.cgi ? (shutdown = true) : exit } 
      trap(:INT){ self.cgi ? (shutdown = true) : exit } 
      FCGI.each_cgi { |cgi|
        FCGIWrap.instance.cgi = cgi
	ENV.clear
	ENV.update(self.cgi.env_table)
        begin
          yield
	rescue SystemExit
	  shutdown? && raise
        ensure
          FCGIWrap.instance.cgi = nil
          #Thread.list.each { |t|
          #  Thread.current == t and next
          #  t.kill
          #}
        end
	shutdown? && exit
      }
    end
    alias each each_request
    alias each_cgi each_request
    alias loop each_request
  end
end

module Kernel
  def print(*args)
    if FCGIWrap.cgi
      FCGIWrap.cgi.print(*args)
    else
      # IGNORE: 
      # STDERR.puts "WARN: Useless print"
    end
  end
  def puts(*args)
    Kernel.print(*args.map{|s| t = s.dup.to_s; t !~ /\n$/ and t += "\n"; t })
  end
end

class CGI
  class << self
    def new(*args)
      FCGIWrap.cgi ? FCGIWrap.cgi : super(*args)
    end
  end
end

# vim: set sts=2 sw=2 expandtab: