Class: Falcon::Proxy

Inherits:
Async::HTTP::Middleware
  • Object
show all
Defined in:
lib/falcon/proxy.rb

Constant Summary collapse

FORWARDED = 
'forwarded'.freeze
X_FORWARDED_FOR = 
'x-forwarded-for'.freeze
X_FORWARDED_PROTO = 
'x-forwarded-proto'.freeze
VIA = 
'via'.freeze
CONNECTION = 
::HTTP::Protocol::CONNECTION
HOP_HEADERS = 
[
  'connection',
  'keep-alive',
  'public',
  'proxy-authenticate',
  'transfer-encoding',
  'upgrade',
]

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(app, hosts) ⇒ Proxy

Returns a new instance of Proxy.



51
52
53
54
55
56
57
58
59
60
 
# File 'lib/falcon/proxy.rb', line 51

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

Instance Attribute Details

#count ⇒ Object (readonly)

Returns the value of attribute count.



62
63
64
 
# File 'lib/falcon/proxy.rb', line 62

def count
  @count
end

Instance Method Details

#call(request) ⇒ Object 



113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/falcon/proxy.rb', line 113

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

#close ⇒ Object 



64
65
66
67
68
 
# File 'lib/falcon/proxy.rb', line 64

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

#connect(endpoint) ⇒ Object 



70
71
72
 
# File 'lib/falcon/proxy.rb', line 70

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

#lookup(request) ⇒ Object 



74
75
76
77
78
79
 
# File 'lib/falcon/proxy.rb', line 74

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

#prepare_headers(headers) ⇒ Object 



81
82
83
84
85
86
87
 
# File 'lib/falcon/proxy.rb', line 81

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

#prepare_request(request) ⇒ Object 



89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
 
# File 'lib/falcon/proxy.rb', line 89

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