require "rubygems"
require "webrick"
require "launch"

##
# Launch::WEBrickHTTPServer adds launchd support to WEBrick.
#
# To use, replace WEBrick::HTTPServer with Launch::WEBrickHTTPServer.
#
# By default Launch::WEBrickHTTPServer expects to find the socket list under
# <tt>'WEBrickSockets'</tt> but this may be overridden through the
# +:LaunchdSockets+ option.
#
# The server will automatically shut down when a TERM signal is sent.  If you
# wish to perform other shutdown actions override TERM but be sure to shut
# down webrick.
#
# An example WEBrick server using Launch::WEBrickHTTPServer would be:
#
#   require 'launch/webrick'
#
#   Launch::WEBrickHTTPServer.new(:DocumentRoot => ARGV.shift).start
#
# Here is an example plist for this server which listens on port 8000:
#
#   <?xml version="1.0" encoding="UTF-8"?>
#   <!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
#   <plist version="1.0">
#   <dict>
#     <key>Label</key>
#     <string>net.segment7.launch.webrick</string>
#     <key>ProgramArguments</key>
#     <array>
#       <string>/path/to/ruby</string>
#       <string>/path/to/webrick</string>
#       <string>/Users/your_user/Sites</string>
#     </array>
#     <key>ServiceIPC</key>
#     <true/>
#     <key>Sockets</key>
#     <dict>
#       <key>WEBrickSockets</key>
#       <dict>
#         <key>SockServiceName</key>
#         <string>8000</string>
#       </dict>
#     </dict>
#   </dict>
#   </plist>
module Launch
  class WEBrickHTTPServer < WEBrick::HTTPServer
    ##
    # Initializes an HTTP server with +options+ and set the server's listeners
    # using Launch.  A TERM handler to shut down the server is automatically
    # set.
    #
    # +:LaunchdSockets+ may be set to change the socket key from the default of
    # <tt>'WEBrickSockets'</tt>
    def initialize(options = {})
      options[:DoNotListen] = true
      sockets_key = options.delete(:LaunchdSockets) || 'WEBrickSockets'

      super

      @job = Launch::Job.checkin
      sockets = @job.sockets[sockets_key]
      listeners.replace(socket)

      trap("TERM") { shutdown }
    end
  end
end