Class: Rackup::Handler::WEBrick

# File 'lib/rackup/handler/webrick.rb', line 66

def initialize(server, app)
  super server
  @app = app
end

# File 'lib/rackup/handler/webrick.rb', line 32

def self.run(app, **options)
  environment  = ENV['RACK_ENV'] || 'development'
  default_host = environment == 'development' ? 'localhost' : nil

  if !options[:BindAddress] || options[:Host]
    options[:BindAddress] = options.delete(:Host) || default_host
  end
  options[:Port] ||= 8080
  if options[:SSLEnable]
    require 'webrick/https'
  end

  @server = Server.new(app, options)
  yield @server if block_given?
  @server.start
end

# File 'lib/rackup/handler/webrick.rb', line 103

def service(req, res)
  env = req.meta_vars
  env.delete_if { |k, v| v.nil? }

  input = Input.new(req)

  env.update(
    ::Rack::RACK_INPUT => input,
    ::Rack::RACK_ERRORS => $stderr,
    ::Rack::RACK_URL_SCHEME => ["yes", "on", "1"].include?(env[::Rack::HTTPS]) ? "https" : "http",
    ::Rack::RACK_IS_HIJACK => true,
  )

  env[::Rack::QUERY_STRING] ||= ""

  # Handle OPTIONS * requests which have no path
  if req.unparsed_uri == "*"
    env[::Rack::PATH_INFO] = "*"
    env[::Rack::REQUEST_PATH] = "*"

    # Ensure SERVER_NAME and SERVER_PORT are set from server config.
    # (WEBrick allows these to be nil for OPTIONS * requests)
    # See https://github.com/ruby/webrick/pull/182 for a proper fix.
    server_name = env[::Rack::SERVER_NAME]
    if server_name.nil? || server_name == ""
      env[::Rack::SERVER_NAME] = @server[:ServerName] || @server[:BindAddress] || "localhost"
    end

    # Legacy versions of WEBrick can set server_port to "" in some cases:
    server_port = env[::Rack::SERVER_PORT]
    if server_port.nil? || server_port == ""
      env[::Rack::SERVER_PORT] = (@server[:Port] || 80).to_s
    end
  else
    unless env[::Rack::PATH_INFO] == ""
      # Strip the script name prefix from the path to get path info
      script_name_length = env[::Rack::SCRIPT_NAME].length
      env[::Rack::PATH_INFO] = req.request_uri.path[script_name_length..-1] || ""
    end
    env[::Rack::REQUEST_PATH] ||= env[::Rack::SCRIPT_NAME] + env[::Rack::PATH_INFO]
  end

  status, headers, body = @app.call(env)
  begin
    res.status = status

    if value = headers[::Rack::RACK_HIJACK]
      io_lambda = value
      body = nil
    elsif !body.respond_to?(:to_path) && !body.respond_to?(:each)
      io_lambda = body
      body = nil
    end

    if value = headers.delete('set-cookie')
      res.cookies.concat(Array(value))
    end

    headers.each do |key, value|
      # Skip keys starting with rack., per Rack SPEC
      next if key.start_with?('rack.')

      # Since WEBrick won't accept repeated headers,
      # merge the values per RFC 1945 section 4.2.
      value = value.join(", ") if Array === value
      res[key] = value
    end

    if io_lambda
      protocol = headers['rack.protocol'] || headers['upgrade']

      if protocol
        # Set all the headers correctly for an upgrade response:
        res.upgrade!(protocol)
      end
      res.body = io_lambda
    elsif body.respond_to?(:to_path)
      res.body = ::File.open(body.to_path, 'rb')
    else
      buffer = String.new
      body.each do |part|
        buffer << part
      end
      res.body = buffer
    end
  ensure
    body.close if body.respond_to?(:close)
  end
end