74 lines
2.1 KiB
Ruby
74 lines
2.1 KiB
Ruby
require 'webrick'
|
|
require 'stringio'
|
|
require 'rack/content_length'
|
|
|
|
module Rack
|
|
module Handler
|
|
class WEBrick < ::WEBrick::HTTPServlet::AbstractServlet
|
|
def self.run(app, options={})
|
|
options[:BindAddress] = options.delete(:Host) if options[:Host]
|
|
@server = ::WEBrick::HTTPServer.new(options)
|
|
@server.mount "/", Rack::Handler::WEBrick, app
|
|
yield @server if block_given?
|
|
@server.start
|
|
end
|
|
|
|
def self.shutdown
|
|
@server.shutdown
|
|
@server = nil
|
|
end
|
|
|
|
def initialize(server, app)
|
|
super server
|
|
@app = Rack::ContentLength.new(app)
|
|
end
|
|
|
|
def service(req, res)
|
|
env = req.meta_vars
|
|
env.delete_if { |k, v| v.nil? }
|
|
|
|
rack_input = StringIO.new(req.body.to_s)
|
|
rack_input.set_encoding(Encoding::BINARY) if rack_input.respond_to?(:set_encoding)
|
|
|
|
env.update({"rack.version" => Rack::VERSION,
|
|
"rack.input" => rack_input,
|
|
"rack.errors" => $stderr,
|
|
|
|
"rack.multithread" => true,
|
|
"rack.multiprocess" => false,
|
|
"rack.run_once" => false,
|
|
|
|
"rack.url_scheme" => ["yes", "on", "1"].include?(ENV["HTTPS"]) ? "https" : "http"
|
|
})
|
|
|
|
env["HTTP_VERSION"] ||= env["SERVER_PROTOCOL"]
|
|
env["QUERY_STRING"] ||= ""
|
|
env["REQUEST_PATH"] ||= "/"
|
|
unless env["PATH_INFO"] == ""
|
|
path, n = req.request_uri.path, env["SCRIPT_NAME"].length
|
|
env["PATH_INFO"] = path[n, path.length-n]
|
|
end
|
|
|
|
status, headers, body = @app.call(env)
|
|
begin
|
|
res.status = status.to_i
|
|
headers.each { |k, vs|
|
|
if k.downcase == "set-cookie"
|
|
res.cookies.concat vs.split("\n")
|
|
else
|
|
vs.split("\n").each { |v|
|
|
res[k] = v
|
|
}
|
|
end
|
|
}
|
|
body.each { |part|
|
|
res.body << part
|
|
}
|
|
ensure
|
|
body.close if body.respond_to? :close
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|