119 lines
4.3 KiB
Ruby
119 lines
4.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'roda/proxy/version'
|
|
require 'net_http_unix'
|
|
|
|
# :nodoc:
|
|
class Roda
|
|
# :nodoc:
|
|
module RodaPlugins
|
|
# Roda plugin for simple API proxying
|
|
module Proxy
|
|
|
|
# Respond to the configure method to set the destination when proxying
|
|
# Expects the following options:
|
|
# [to] Required. The scheme and host of the proxy. Should not end with a slash.
|
|
# [path_prefix] Optional. The path to append to the above for proxying.
|
|
# The current request path will be prefixed on to this value.
|
|
# Should begin and end with a +/+. Defaults to +/+.
|
|
# For example, if the path prefix is +/foo/+ and the request received
|
|
# by Roda is +GET /postcode/lookup+, The proxied request will be dispatched
|
|
# to +GET /home/postcode/lookup+
|
|
# Example:
|
|
# plugin :proxy, to: 'https://foo.bar', path: '/my/api'
|
|
def self.configure(app, opts = {})
|
|
app.opts[:proxy_to] = opts.fetch(:to, nil)
|
|
app.opts[:proxy_path] = opts.fetch(:path_prefix, '/')
|
|
|
|
raise 'Proxy host not set, use "plugin :proxy, to: http://example.com"' unless app.opts[:proxy_to]
|
|
end
|
|
|
|
# :nodoc:
|
|
module RequestMethods
|
|
|
|
# Proxies the request, forwarding all headers except +Host+ which is
|
|
# rewritten to be the destination host. The response headers, body and
|
|
# status are returned to the client.
|
|
def proxy
|
|
client = NetX::HTTPUnix.new(_sock_url)
|
|
request_class = Net::HTTP.const_get(:"#{env['REQUEST_METHOD'].to_s.downcase.capitalize}")
|
|
req = request_class.new(_proxy_url.to_s, _proxy_headers)
|
|
env_body = env["rack.input"].read
|
|
env_content_type = env['CONTENT_TYPE']
|
|
unless env_body.nil?
|
|
req.body = env_body
|
|
end
|
|
unless env_content_type.nil?
|
|
req.content_type = env_content_type
|
|
end
|
|
f_response = client.request(req)
|
|
f_response.content_length = nil #prevent duplicate header Content-Length and content-length
|
|
_respond(f_response)
|
|
end
|
|
|
|
# Conditionally proxies when +condition+ is true and with selective probability.
|
|
# For instance, to proxy 50% of the time:
|
|
# r.proxy_when(probability: 0.5)
|
|
# Condition can be a truthy value or a block / lambda, in which case
|
|
# the result from the +#call+ is expected to be truthy.
|
|
# r.proxy_when( r.env['HTTP_PROXY_ME'] == 'true' )
|
|
# The two parameters can be combined, the probability is evaluated first.
|
|
# r.proxy_when( r.env['HTTP_PROXY_ME'] == 'true', probability: 0.8 )
|
|
# If and only if this method choses not to proxy is the block evaluated.
|
|
# The block is then expected to return a meaningful response to Roda.
|
|
def proxy_when(condition = true, probability: 1.0)
|
|
shall_proxy = Random.rand(0.0..1.0) <= probability
|
|
|
|
if shall_proxy && ( condition.respond_to?(:call) ? condition.call : condition )
|
|
proxy
|
|
else
|
|
yield(self)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def _sock_url
|
|
roda_class.opts[:proxy_to]
|
|
end
|
|
def _proxy_url
|
|
uri = URI("#{roda_class.opts[:proxy_path]}#{env['PATH_INFO'][1..-1]}")
|
|
if !env['QUERY_STRING'].empty?
|
|
uri.query = env['QUERY_STRING']
|
|
end
|
|
uri
|
|
end
|
|
|
|
def _proxy_headers
|
|
env_host = env['HTTP_HOST'].to_s
|
|
env_new = env
|
|
.select { |k, _v| k.start_with? 'HTTP_' }
|
|
.reject { |k, _v| k == 'HTTP_HOST' }
|
|
.transform_keys do |k|
|
|
k.sub(/^HTTP_/, '')
|
|
.split('_')
|
|
.map(&:capitalize)
|
|
.join('-')
|
|
end
|
|
if env_host
|
|
env_new.merge!({'Host' => env_host})
|
|
end
|
|
env_new
|
|
end
|
|
|
|
def _respond(proxied_response)
|
|
response.status = proxied_response.code.to_i
|
|
proxied_response
|
|
.each_header.to_h
|
|
.reject { |k, v| k.downcase == 'transfer-encoding' }
|
|
.each { |k, v| response[k] = v }
|
|
response.write(proxied_response.body)
|
|
end
|
|
|
|
end
|
|
end
|
|
|
|
register_plugin :proxy, Proxy
|
|
end
|
|
end
|