Module: AeReverseProxy::ControllerCallbackMethod

Defined in:
lib/ae_reverse_proxy/controller_callback_method.rb

Instance Method Summary collapse

Instance Method Details

#reverse_proxy(uri) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/ae_reverse_proxy/controller_callback_method.rb', line 7

def reverse_proxy(uri)
  proxy_uri = Addressable::URI.parse(uri)

  client = AeReverseProxy::Client.new(proxy_uri) do |config|
    config.on_response do |_code, response|
      blacklist = [
        'Connection',           # Always close connection
        'Transfer-Encoding',    # Let Rails calculate this
        'Content-Length', # Let Rails calculate this
      ]

      response.each_capitalized do |key, value|
        next if blacklist.include?(key)

        headers[key] = value
      end
    end

    config.on_set_cookies do |_code, _response, set_cookies|
      set_cookies.each do |key, attributes|
        cookies[key] = attributes
      end
    end

    config.on_redirect do |code, _response, redirect_url|
      request_uri = Addressable::URI.parse(request.url)
      redirect_uri = Addressable::URI.parse(redirect_url)

      # Make redirect uri absolute if it's relative by
      # joining it with the request url
      redirect_uri = request_uri.join(redirect_url) if redirect_uri.host.nil?

      if !redirect_uri.port.nil? && (redirect_uri.port == proxy_uri.port)
        # Make sure it's consistent with our request port
        redirect_uri.port = request.port
      end

      redirect_to redirect_uri.to_s, status: code
      return
    end

    config.on_complete do |code, response|
      content_type = response['Content-Type']
      body = response.body.to_s

      if content_type&.match(/image/)
        send_data body, content_type: content_type, disposition: 'inline', status: code
      else
        render body: body, content_type: content_type, status: code
      end
    end
  end

  client.forward_request(request.env)
end