111 lines
2.7 KiB
Ruby
111 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'uri'
|
|
require 'net/http'
|
|
require 'rack'
|
|
require 'capybara/server/middleware'
|
|
require 'capybara/server/animation_disabler'
|
|
require 'capybara/server/checker'
|
|
|
|
module Capybara
|
|
# @api private
|
|
class Server
|
|
class << self
|
|
def ports
|
|
@ports ||= {}
|
|
end
|
|
end
|
|
|
|
attr_reader :app, :port, :host
|
|
|
|
def initialize(app,
|
|
*deprecated_options,
|
|
port: Capybara.server_port,
|
|
host: Capybara.server_host,
|
|
reportable_errors: Capybara.server_errors,
|
|
extra_middleware: [])
|
|
warn 'Positional arguments, other than the application, to Server#new are deprecated, please use keyword arguments' unless deprecated_options.empty?
|
|
@app = app
|
|
@extra_middleware = extra_middleware
|
|
@server_thread = nil # suppress warnings
|
|
@host = deprecated_options[1] || host
|
|
@reportable_errors = deprecated_options[2] || reportable_errors
|
|
@port = deprecated_options[0] || port
|
|
@port ||= Capybara::Server.ports[port_key]
|
|
@port ||= find_available_port(host)
|
|
@checker = Checker.new(@host, @port)
|
|
end
|
|
|
|
def reset_error!
|
|
middleware.clear_error
|
|
end
|
|
|
|
def error
|
|
middleware.error
|
|
end
|
|
|
|
def using_ssl?
|
|
@checker.ssl?
|
|
end
|
|
|
|
def responsive?
|
|
return false if @server_thread&.join(0)
|
|
|
|
res = @checker.request { |http| http.get('/__identify__') }
|
|
|
|
return res.body == app.object_id.to_s if res.is_a?(Net::HTTPSuccess) || res.is_a?(Net::HTTPRedirection)
|
|
rescue SystemCallError, Net::ReadTimeout, OpenSSL::SSL::SSLError
|
|
false
|
|
end
|
|
|
|
def wait_for_pending_requests
|
|
timer = Capybara::Helpers.timer(expire_in: 60)
|
|
while pending_requests?
|
|
raise 'Requests did not finish in 60 seconds' if timer.expired?
|
|
|
|
sleep 0.01
|
|
end
|
|
end
|
|
|
|
def boot
|
|
unless responsive?
|
|
Capybara::Server.ports[port_key] = port
|
|
|
|
@server_thread = Thread.new do
|
|
Capybara.server.call(middleware, port, host)
|
|
end
|
|
|
|
timer = Capybara::Helpers.timer(expire_in: 60)
|
|
until responsive?
|
|
raise 'Rack application timed out during boot' if timer.expired?
|
|
|
|
@server_thread.join(0.1)
|
|
end
|
|
end
|
|
|
|
self
|
|
end
|
|
|
|
private
|
|
|
|
def middleware
|
|
@middleware ||= Middleware.new(app, @reportable_errors, @extra_middleware)
|
|
end
|
|
|
|
def port_key
|
|
Capybara.reuse_server ? app.object_id : middleware.object_id
|
|
end
|
|
|
|
def pending_requests?
|
|
middleware.pending_requests?
|
|
end
|
|
|
|
def find_available_port(host)
|
|
server = TCPServer.new(host, 0)
|
|
server.addr[1]
|
|
ensure
|
|
server&.close
|
|
end
|
|
end
|
|
end
|