1
0
Fork 0
mirror of https://github.com/puma/puma.git synced 2022-11-09 13:48:40 -05:00
puma--puma/lib/puma/cli.rb

235 lines
6.5 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
2011-09-27 12:23:03 -04:00
require 'optparse'
2011-09-27 13:53:45 -04:00
require 'uri'
require 'puma'
require 'puma/configuration'
require 'puma/launcher'
require 'puma/const'
require 'puma/events'
2011-09-27 12:23:03 -04:00
module Puma
2015-03-13 18:07:08 -04:00
class << self
# The CLI exports its Puma::Configuration object here to allow
# apps to pick it up. An app needs to use it conditionally though
# since it is not set if the app is launched via another
# mechanism than the CLI class.
attr_accessor :cli_config
end
2011-12-01 18:23:14 -05:00
# Handles invoke a Puma::Server in a command line style.
#
2011-09-27 12:23:03 -04:00
class CLI
2016-02-04 10:49:01 -05:00
KEYS_NOT_TO_PERSIST_IN_STATE = Launcher::KEYS_NOT_TO_PERSIST_IN_STATE
2011-12-01 18:23:14 -05:00
# Create a new CLI object using +argv+ as the command line
# arguments.
#
# +stdout+ and +stderr+ can be set to IO-like objects which
# this object will report status on.
#
def initialize(argv, events=Events.stdio)
2012-08-01 13:11:27 -04:00
@debug = false
@argv = argv.dup
2011-09-27 12:23:03 -04:00
@events = events
2011-09-27 17:33:17 -04:00
@conf = nil
@stdout = nil
@stderr = nil
@append = false
@control_url = nil
@control_options = {}
2011-09-27 12:23:03 -04:00
setup_options
begin
@parser.parse! @argv
if file = @argv.shift
@conf.configure do |user_config, file_config|
file_config.rackup file
end
end
rescue UnsupportedOption
exit 1
end
@conf.configure do |user_config, file_config|
if @stdout || @stderr
user_config.stdout_redirect @stdout, @stderr, @append
end
if @control_url
user_config.activate_control_app @control_url, @control_options
end
end
@launcher = Puma::Launcher.new(@conf, :events => @events, :argv => argv)
end
attr_reader :launcher
2013-07-06 00:13:29 -04:00
2011-12-01 18:23:14 -05:00
# Parse the options, load the rackup, start the server and wait
# for it to finish.
#
2011-11-22 00:15:40 -05:00
def run
@launcher.run
2013-02-05 22:09:40 -05:00
end
2014-02-19 12:22:11 -05:00
private
2015-03-13 21:23:09 -04:00
def unsupported(str)
@events.error(str)
raise UnsupportedOption
end
def configure_control_url(command_line_arg)
if command_line_arg
@control_url = command_line_arg
elsif Puma.jruby?
unsupported "No default url available on JRuby"
end
end
2015-03-13 19:08:07 -04:00
# Build the OptionParser object to handle the available options.
#
2015-03-13 19:08:07 -04:00
def setup_options
@conf = Configuration.new do |user_config, file_config|
@parser = OptionParser.new do |o|
o.on "-b", "--bind URI", "URI to bind to (tcp://, unix://, ssl://)" do |arg|
user_config.bind arg
end
2015-03-13 19:08:07 -04:00
o.on "-C", "--config PATH", "Load PATH as a config file" do |arg|
file_config.load arg
end
2015-03-13 19:08:07 -04:00
o.on "--control-url URL", "The bind url to use for the control server. Use 'auto' to use temp unix server" do |arg|
configure_control_url(arg)
end
o.on "--control-token TOKEN",
"The token to use as authentication for the control server" do |arg|
@control_options[:auth_token] = arg
end
2015-03-13 19:08:07 -04:00
o.on "-d", "--daemon", "Daemonize the server into the background" do
user_config.daemonize
user_config.quiet
end
2015-03-13 19:08:07 -04:00
o.on "--debug", "Log lowlevel debugging information" do
user_config.debug
end
2015-03-13 19:08:07 -04:00
o.on "--dir DIR", "Change to DIR before starting" do |d|
user_config.directory d
end
2015-03-13 19:08:07 -04:00
o.on "-e", "--environment ENVIRONMENT",
"The environment to run the Rack app on (default development)" do |arg|
user_config.environment arg
end
2015-03-13 19:08:07 -04:00
o.on "-I", "--include PATH", "Specify $LOAD_PATH directories" do |arg|
$LOAD_PATH.unshift(*arg.split(':'))
end
2015-03-13 19:08:07 -04:00
o.on "-p", "--port PORT", "Define the TCP port to bind to",
"Use -b for more advanced options" do |arg|
user_config.bind "tcp://#{Configuration::DefaultTCPHost}:#{arg}"
end
2015-03-13 19:08:07 -04:00
o.on "--pidfile PATH", "Use PATH as a pidfile" do |arg|
user_config.pidfile arg
end
2015-03-13 19:08:07 -04:00
o.on "--preload", "Preload the app. Cluster mode only" do
user_config.preload_app!
end
2015-03-13 19:08:07 -04:00
o.on "--prune-bundler", "Prune out the bundler env if possible" do
user_config.prune_bundler
end
2015-03-13 19:08:07 -04:00
o.on "--extra-runtime-dependencies GEM1,GEM2", "Defines any extra needed gems when using --prune-bundler" do |arg|
user_config.extra_runtime_dependencies arg.split(',')
end
o.on "-q", "--quiet", "Do not log requests internally (default true)" do
user_config.quiet
end
2015-03-13 19:08:07 -04:00
o.on "-v", "--log-requests", "Log requests as they occur" do
user_config.log_requests
end
o.on "-R", "--restart-cmd CMD",
"The puma command to run during a hot restart",
"Default: inferred" do |cmd|
user_config.restart_command cmd
end
2015-03-13 19:08:07 -04:00
o.on "-S", "--state PATH", "Where to store the state details" do |arg|
user_config.state_path arg
end
2015-03-13 19:08:07 -04:00
o.on '-t', '--threads INT', "min:max threads to use (default 0:16)" do |arg|
min, max = arg.split(":")
if max
user_config.threads min, max
else
user_config.threads min, min
end
2015-03-13 19:08:07 -04:00
end
o.on "--tcp-mode", "Run the app in raw TCP mode instead of HTTP mode" do
user_config.tcp_mode!
end
2015-03-13 19:08:07 -04:00
o.on "--early-hints", "Enable early hints support" do
user_config.early_hints
end
o.on "-V", "--version", "Print the version information" do
puts "puma version #{Puma::Const::VERSION}"
exit 0
end
2015-03-13 19:08:07 -04:00
o.on "-w", "--workers COUNT",
"Activate cluster mode: How many worker processes to create" do |arg|
user_config.workers arg
end
2015-03-13 19:08:07 -04:00
o.on "--tag NAME", "Additional text to display in process listing" do |arg|
user_config.tag arg
end
2015-03-13 19:08:07 -04:00
o.on "--redirect-stdout FILE", "Redirect STDOUT to a specific file" do |arg|
@stdout = arg.to_s
end
o.on "--redirect-stderr FILE", "Redirect STDERR to a specific file" do |arg|
@stderr = arg.to_s
end
o.on "--[no-]redirect-append", "Append to redirected files" do |val|
@append = val
end
o.banner = "puma <options> <rackup file>"
2015-03-13 19:08:07 -04:00
o.on_tail "-h", "--help", "Show help" do
$stdout.puts o
exit 0
end
2015-03-13 19:08:07 -04:00
end
end
end
2011-09-27 12:23:03 -04:00
end
end