class Falcon::Proxy

def call(request)
	if endpoint = lookup(request)
		@count += 1
		
		request = self.prepare_request(request)
		
		client = connect(endpoint)
		
		client.call(request)
	else
		super
	end
rescue
	Async.logger.error(self) {$!}
	return Async::HTTP::Response[502, {'content-type' => 'text/plain'}, ["#{$!.inspect}: #{$!.backtrace.join("\n")}"]]
end

def close
	@clients.each_value(&:close)
	
	super
end

def connect(endpoint)
	@clients[endpoint] ||= Async::HTTP::Client.new(endpoint)
end

def initialize(app, hosts)
	super(app)
	
	@server_context = nil
	
	@hosts = hosts
	@clients = {}
	
	@count = 0
end

def lookup(request)
	# Trailing dot and port is ignored/normalized.
	if authority = request.authority.sub(/(\.)?(:\d+)?$/, '')
		return @hosts[authority]
	end
end

def prepare_headers(headers)
	if connection = headers[CONNECTION]
		headers.slice!(connection)
	end
	
	headers.slice!(HOP_HEADERS)
end

def prepare_request(request)
	forwarded = []
	
	if address = request.remote_address
		request.headers.add(X_FORWARDED_FOR, address.ip_address)
		forwarded << "for=#{address.ip_address}"
	end
	
	if scheme = request.scheme
		request.headers.add(X_FORWARDED_PROTO, scheme)
		forwarded << "proto=#{scheme}"
	end
	
	unless forwarded.empty?
		request.headers.add(FORWARDED, forwarded.join(';'))
	end
	
	request.headers.add(VIA, "#{request.version} #{self.class}")
	
	self.prepare_headers(request.headers)
	
	return request
end