2006-02-09 21:38:18 -05:00
|
|
|
require 'rubygems'
|
2006-02-09 20:06:55 -05:00
|
|
|
require 'mongrel'
|
|
|
|
require 'cgi'
|
2006-02-11 20:30:33 -05:00
|
|
|
require 'daemons/daemonize'
|
|
|
|
require 'mongrel/command'
|
2006-02-09 20:06:55 -05:00
|
|
|
|
|
|
|
|
|
|
|
class CGIFixed < ::CGI
|
|
|
|
public :env_table
|
2006-02-10 01:45:50 -05:00
|
|
|
attr_reader :options
|
2006-02-09 20:06:55 -05:00
|
|
|
|
|
|
|
def initialize(params, data, out, *args)
|
|
|
|
@env_table = params
|
|
|
|
@args = *args
|
|
|
|
@input = StringIO.new(data)
|
|
|
|
@out = out
|
2006-02-10 01:45:50 -05:00
|
|
|
@options = {}
|
2006-02-09 20:06:55 -05:00
|
|
|
super(*args)
|
|
|
|
end
|
|
|
|
|
2006-02-10 01:45:50 -05:00
|
|
|
def header(options = "text/html")
|
|
|
|
if options.class == Hash
|
|
|
|
# passing in a header so need to keep the status around and other options
|
|
|
|
@options = options
|
|
|
|
end
|
|
|
|
|
|
|
|
super(options)
|
|
|
|
end
|
|
|
|
|
|
|
|
def status
|
|
|
|
s = @options["Status"] || @options["status"]
|
|
|
|
s[0 .. s.index(' ')] || "200"
|
|
|
|
end
|
|
|
|
|
2006-02-09 20:06:55 -05:00
|
|
|
def args
|
|
|
|
@args
|
|
|
|
end
|
|
|
|
|
|
|
|
def env_table
|
|
|
|
@env_table
|
|
|
|
end
|
|
|
|
|
|
|
|
def stdinput
|
|
|
|
@input
|
|
|
|
end
|
|
|
|
|
|
|
|
def stdoutput
|
|
|
|
@out
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
class RailsHandler < Mongrel::HttpHandler
|
2006-02-14 07:19:02 -05:00
|
|
|
|
|
|
|
def initialize(dir, mime_map = {})
|
2006-02-09 20:06:55 -05:00
|
|
|
@files = Mongrel::DirHandler.new(dir,false)
|
|
|
|
@guard = Mutex.new
|
2006-02-14 07:19:02 -05:00
|
|
|
|
|
|
|
# register the requested mime types
|
|
|
|
mime_map.each {|k,v| Mongrel::DirHandler::add_mime_type(k,v) }
|
2006-02-09 20:06:55 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def process(request, response)
|
|
|
|
# not static, need to talk to rails
|
|
|
|
return if response.socket.closed?
|
|
|
|
|
|
|
|
if @files.can_serve(request.params["PATH_INFO"])
|
|
|
|
@files.process(request,response)
|
|
|
|
else
|
|
|
|
cgi = CGIFixed.new(request.params, request.body, response.socket)
|
|
|
|
begin
|
|
|
|
|
|
|
|
@guard.synchronize do
|
|
|
|
# Rails is not thread safe so must be run entirely within synchronize
|
|
|
|
Dispatcher.dispatch(cgi, ActionController::CgiRequest::DEFAULT_SESSION_OPTIONS, response.body)
|
|
|
|
end
|
2006-02-10 01:45:50 -05:00
|
|
|
|
|
|
|
response.status = cgi.status
|
2006-02-09 20:06:55 -05:00
|
|
|
response.send_status
|
|
|
|
response.send_body
|
|
|
|
rescue Object => rails_error
|
2006-02-10 01:45:50 -05:00
|
|
|
STDERR.puts "calling Dispatcher.dispatch #{rails_error}"
|
|
|
|
STDERR.puts rails_error.backtrace.join("\n")
|
2006-02-09 20:06:55 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
2006-02-11 22:37:38 -05:00
|
|
|
|
2006-02-11 20:30:33 -05:00
|
|
|
class StartCommand < Mongrel::Command::Command
|
2006-02-09 20:06:55 -05:00
|
|
|
|
2006-02-11 20:30:33 -05:00
|
|
|
def configure
|
|
|
|
options [
|
2006-02-14 07:19:02 -05:00
|
|
|
["-e", "--environment ENV", "Rails environment to run as", :@environment, ENV['RAILS_ENV'] || "development"],
|
|
|
|
["-d", "--daemonize", "Whether to run in the background or not", :@daemon, false],
|
|
|
|
['-p', '--port PORT', "Which port to bind to", :@port, 3000],
|
|
|
|
['-a', '--address ADDR', "Address to bind to", :@address, "0.0.0.0"],
|
|
|
|
['-l', '--log FILE', "Where to write log messages", :@log_file, "log/mongrel.log"],
|
|
|
|
['-P', '--pid FILE', "Where to write the PID", :@pid_file, "log/mongrel.pid"],
|
|
|
|
['-n', '--num-procs INT', "Number of processor threads to use", :@num_procs, 20],
|
|
|
|
['-t', '--timeout SECONDS', "Timeout all requests after SECONDS time", :@timeout, 120],
|
|
|
|
['-m', '--mime PATH', "A YAML file that lists additional MIME types", :@mime_map, nil],
|
|
|
|
['-c', '--chdir PATH', "Change to dir before starting (will be expanded)", :@cwd, Dir.pwd],
|
|
|
|
['-r', '--root PATH', "Set the document root (default 'public')", :@docroot, "public"],
|
2006-02-11 20:30:33 -05:00
|
|
|
]
|
|
|
|
end
|
|
|
|
|
|
|
|
def validate
|
2006-02-14 07:19:02 -05:00
|
|
|
@cwd = File.expand_path(@cwd)
|
|
|
|
valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
|
|
|
|
|
|
|
|
# change there to start, then we'll have to come back after daemonize
|
|
|
|
Dir.chdir(@cwd)
|
|
|
|
|
2006-02-11 20:30:33 -05:00
|
|
|
valid_dir? File.dirname(@log_file), "Path to log file not valid: #@log_file"
|
|
|
|
valid_dir? File.dirname(@pid_file), "Path to pid file not valid: #@pid_file"
|
2006-02-14 07:19:02 -05:00
|
|
|
valid_dir? @docroot, "Path to docroot not valid: #@docroot"
|
|
|
|
valid_exists? @mime_map, "MIME mapping file does not exist: #@mime_map" if @mime_map
|
|
|
|
|
2006-02-11 20:30:33 -05:00
|
|
|
return @valid
|
|
|
|
end
|
2006-02-10 01:45:50 -05:00
|
|
|
|
2006-02-14 07:19:02 -05:00
|
|
|
def daemonize
|
|
|
|
# save this for later since daemonize will hose it
|
2006-02-11 20:30:33 -05:00
|
|
|
if @daemon
|
2006-02-14 07:19:02 -05:00
|
|
|
puts "Started Mongrel server in #@environment mode at #@address:#@port"
|
|
|
|
Daemonize.daemonize(log_file=File.join(@cwd, @log_file))
|
|
|
|
|
2006-02-11 20:30:33 -05:00
|
|
|
# change back to the original starting directory
|
2006-02-14 07:19:02 -05:00
|
|
|
Dir.chdir(@cwd)
|
|
|
|
|
|
|
|
open(@pid_file,"w") {|f| f.write(Process.pid) }
|
2006-02-11 20:30:33 -05:00
|
|
|
else
|
2006-02-14 07:19:02 -05:00
|
|
|
puts "Running Mongrel server in #@environment mode at #@address:#@port"
|
2006-02-11 20:30:33 -05:00
|
|
|
end
|
2006-02-14 07:19:02 -05:00
|
|
|
end
|
2006-02-11 20:30:33 -05:00
|
|
|
|
2006-02-14 07:19:02 -05:00
|
|
|
def load_mime_map
|
|
|
|
mime = {}
|
|
|
|
|
|
|
|
# configure any requested mime map
|
|
|
|
if @mime_map
|
|
|
|
puts "Loading additional MIME types from #@mime_map"
|
|
|
|
mime.merge!(YAML.load_file(@mime_map))
|
2006-02-11 20:30:33 -05:00
|
|
|
|
2006-02-14 07:19:02 -05:00
|
|
|
# check all the mime types to make sure they are the right format
|
|
|
|
mime.each {|k,v| puts "WARNING: MIME type #{k} must start with '.'" if k.index(".") != 0 }
|
|
|
|
end
|
|
|
|
|
|
|
|
return mime
|
|
|
|
end
|
|
|
|
|
|
|
|
def configure_rails
|
|
|
|
ENV['RAILS_ENV'] = @environment
|
2006-02-11 20:30:33 -05:00
|
|
|
require 'config/environment'
|
2006-02-14 07:19:02 -05:00
|
|
|
|
|
|
|
# configure the rails handler
|
|
|
|
rails = RailsHandler.new(@docroot, load_mime_map)
|
|
|
|
|
|
|
|
return rails
|
|
|
|
end
|
|
|
|
|
|
|
|
def start_mongrel(rails)
|
|
|
|
# start up mongrel with the right configurations
|
|
|
|
server = Mongrel::HttpServer.new(@address, @port, @num_procs.to_i, @timeout.to_i)
|
|
|
|
server.register("/", rails)
|
|
|
|
server.run
|
2006-02-11 20:30:33 -05:00
|
|
|
|
|
|
|
begin
|
2006-02-14 07:19:02 -05:00
|
|
|
puts "Server ready."
|
|
|
|
server.acceptor.join
|
2006-02-11 20:30:33 -05:00
|
|
|
rescue Interrupt
|
2006-02-11 22:37:38 -05:00
|
|
|
puts "Interrupted."
|
|
|
|
end
|
|
|
|
end
|
2006-02-14 07:19:02 -05:00
|
|
|
|
|
|
|
def run
|
|
|
|
daemonize
|
|
|
|
rails = configure_rails
|
|
|
|
start_mongrel(rails)
|
|
|
|
end
|
2006-02-11 22:37:38 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class StopCommand < Mongrel::Command::Command
|
|
|
|
|
|
|
|
def configure
|
|
|
|
options [
|
2006-02-14 07:19:02 -05:00
|
|
|
['-c', '--chdir PATH', "Change to dir before starting (will be expanded)", :@cwd, Dir.pwd],
|
|
|
|
['-P', '--pid FILE', "Where to write the PID", :@pid_file, "log/mongrel.pid"]
|
2006-02-11 22:37:38 -05:00
|
|
|
]
|
|
|
|
end
|
|
|
|
|
|
|
|
def validate
|
2006-02-14 07:19:02 -05:00
|
|
|
@cwd = File.expand_path(@cwd)
|
|
|
|
valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
|
|
|
|
|
|
|
|
@pid_file = File.join(@cwd,@pid_file)
|
2006-02-11 22:37:38 -05:00
|
|
|
valid_exists? @pid_file, "PID file #@pid_file does not exist. Not running?"
|
2006-02-14 07:19:02 -05:00
|
|
|
|
2006-02-11 22:37:38 -05:00
|
|
|
return @valid
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
def run
|
|
|
|
pid = open(@pid_file).read.to_i
|
|
|
|
print "Stopping Mongrel at PID #{pid}..."
|
|
|
|
begin
|
|
|
|
Process.kill("INT", pid)
|
|
|
|
rescue Errno::ESRCH
|
|
|
|
puts "Process does not exist. Not running."
|
2006-02-11 20:30:33 -05:00
|
|
|
end
|
2006-02-11 22:37:38 -05:00
|
|
|
|
|
|
|
File.unlink(@pid_file)
|
|
|
|
puts "Done."
|
2006-02-11 20:30:33 -05:00
|
|
|
end
|
2006-02-09 21:38:18 -05:00
|
|
|
end
|
2006-02-11 20:30:33 -05:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mongrel::Command::Registry.instance.run ARGV
|