Class: Rackup::Handler::WEBrick

Inherits:
WEBrick::HTTPServlet::AbstractServlet
  • Object
show all
Defined in:
lib/rackup/handler/webrick.rb

Defined Under Namespace

Classes: Input, Server

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(server, app) ⇒ WEBrick

Returns a new instance of WEBrick.



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

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

Class Method Details

.run(app, **options) {|@server| ... } ⇒ Object

Yields:

  • (@server) 


32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# 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

.shutdown ⇒ Object 



59
60
61
62
63
64
 
# File 'lib/rackup/handler/webrick.rb', line 59

def self.shutdown
  if @server
    @server.shutdown
    @server = nil
  end
end

.valid_options ⇒ Object 



49
50
51
52
53
54
55
56
57
 
# File 'lib/rackup/handler/webrick.rb', line 49

def self.valid_options
  environment  = ENV['RACK_ENV'] || 'development'
  default_host = environment == 'development' ? 'localhost' : '0.0.0.0'

  {
    "Host=HOST" => "Hostname to listen on (default: #{default_host})",
    "Port=PORT" => "Port to listen on (default: 8080)",
  }
end

Instance Method Details

#service(req, res) ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
 
# 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