require 'optparse' require 'uri' require 'puma/server' require 'puma/const' module Puma class CLI DefaultTCPHost = "0.0.0.0" DefaultTCPPort = 9292 def initialize(argv, stdout=STDOUT, stderr=STDERR) @argv = argv @stdout = stdout @stderr = stderr @events = Events.new @stdout, @stderr setup_options end def log(str) @stdout.puts str end def error(str) @stderr.puts "ERROR: #{str}" exit 1 end def setup_options @options = { :min_threads => 0, :max_threads => 16 } @binds = [] @parser = OptionParser.new do |o| o.on '-t', '--threads INT', "min:max threads to use (default 0:16)" do |arg| min, max = arg.split(":") if max @options[:min_threads] = min.to_i @options[:max_threads] = max.to_i else @options[:min_threads] = 0 @options[:max_threads] = arg.to_i end end o.on "-b", "--bind URI", "URI to bind to (tcp:// and unix:// only)" do |arg| @binds << arg end end @parser.banner = "puma " @parser.on_tail "-h", "--help", "Show help" do log @parser exit 1 end end def load_rackup @app, options = Rack::Builder.parse_file @rackup @options.merge! options options.each do |key,val| if key.to_s[0,4] == "bind" @binds << val end end end def run @parser.parse! @argv @rackup = ARGV.shift || "config.ru" unless File.exists?(@rackup) raise "Missing rackup file '#{@rackup}'" end load_rackup if @binds.empty? @options[:Host] ||= DefaultTCPHost @options[:Port] ||= DefaultTCPPort end min_t = @options[:min_threads] max_t = @options[:max_threads] server = Puma::Server.new @app, @events server.min_threads = min_t server.max_threads = max_t log "Puma #{Puma::Const::PUMA_VERSION} starting..." log "* Min threads: #{min_t}, max threads: #{max_t}" if @options[:Host] log "* Listening on tcp://#{@options[:Host]}:#{@options[:Port]}" server.add_tcp_listener @options[:Host], @options[:Port] end @binds.each do |str| uri = URI.parse str case uri.scheme when "tcp" log "* Listening on #{str}" server.add_tcp_listener uri.host, uri.port when "unix" log "* Listening on #{str}" path = "#{uri.host}#{uri.path}" server.add_unix_listener path else error "Invalid URI: #{str}" end end if server.attempt_bonjour(@rackup) log "* Announced services via bonjour" end log "Use Ctrl-C to stop" server.run.join end end end