57 lines
1.8 KiB
Ruby
57 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Sinatra
|
|
PARAMS_CONFIG = {}
|
|
|
|
if ARGV.any?
|
|
require 'optparse'
|
|
parser = OptionParser.new do |op|
|
|
op.on('-p port', 'set the port (default is 4567)') { |val| PARAMS_CONFIG[:port] = Integer(val) }
|
|
op.on('-s server', 'specify rack server/handler') { |val| PARAMS_CONFIG[:server] = val }
|
|
op.on('-q', 'turn on quiet mode (default is off)') { PARAMS_CONFIG[:quiet] = true }
|
|
op.on('-x', 'turn on the mutex lock (default is off)') { PARAMS_CONFIG[:lock] = true }
|
|
op.on('-e env', 'set the environment (default is development)') do |val|
|
|
ENV['RACK_ENV'] = val
|
|
PARAMS_CONFIG[:environment] = val.to_sym
|
|
end
|
|
op.on('-o addr', "set the host (default is (env == 'development' ? 'localhost' : '0.0.0.0'))") do |val|
|
|
PARAMS_CONFIG[:bind] = val
|
|
end
|
|
end
|
|
begin
|
|
parser.parse!(ARGV.dup)
|
|
rescue StandardError => e
|
|
PARAMS_CONFIG[:optparse_error] = e
|
|
end
|
|
end
|
|
|
|
require 'sinatra/base'
|
|
|
|
class Application < Base
|
|
# we assume that the first file that requires 'sinatra' is the
|
|
# app_file. all other path related options are calculated based
|
|
# on this path by default.
|
|
set :app_file, caller_files.first || $0
|
|
|
|
set :run, proc { File.expand_path($0) == File.expand_path(app_file) }
|
|
|
|
if run? && ARGV.any?
|
|
error = PARAMS_CONFIG.delete(:optparse_error)
|
|
raise error if error
|
|
|
|
PARAMS_CONFIG.each { |k, v| set k, v }
|
|
end
|
|
end
|
|
|
|
remove_const(:PARAMS_CONFIG)
|
|
at_exit { Application.run! if $!.nil? && Application.run? }
|
|
end
|
|
|
|
# include would include the module in Object
|
|
# extend only extends the `main` object
|
|
extend Sinatra::Delegator
|
|
|
|
class Rack::Builder
|
|
include Sinatra::Delegator
|
|
end
|