remove proxy feature into its own extension
This commit is contained in:
Thomas Reynolds
parent
12a00e3811
commit
bb2005388a
|
|
@ -140,9 +140,6 @@ module Middleman
|
||||||
|
|
||||||
# Treat project as a blog
|
# Treat project as a blog
|
||||||
autoload :Blog, "middleman/features/blog"
|
autoload :Blog, "middleman/features/blog"
|
||||||
|
|
||||||
# Proxy web services requests in dev mode only
|
|
||||||
autoload :Proxy, "middleman/features/proxy"
|
|
||||||
|
|
||||||
# guard-livereload
|
# guard-livereload
|
||||||
autoload :LiveReload, "middleman/features/live_reload"
|
autoload :LiveReload, "middleman/features/live_reload"
|
||||||
|
|
|
||||||
|
|
@ -1,194 +0,0 @@
|
||||||
# ===========================================================================
|
|
||||||
# Original Project: Abbot - SproutCore Build Tools
|
|
||||||
# Copyright: ©2009 Apple Inc.
|
|
||||||
# portions copyright @2006-2011 Strobe Inc.
|
|
||||||
# and contributors
|
|
||||||
# ===========================================================================
|
|
||||||
|
|
||||||
begin
|
|
||||||
require 'net/https'
|
|
||||||
Middleman::HTTPS_ENABLED = true
|
|
||||||
rescue LoadError => e
|
|
||||||
require 'net/http'
|
|
||||||
Middleman::HTTPS_ENABLED = false
|
|
||||||
end
|
|
||||||
|
|
||||||
module Middleman::Features::Proxy
|
|
||||||
class << self
|
|
||||||
def registered(app)
|
|
||||||
app.extend ClassMethods
|
|
||||||
app.use Middleman::Features::Proxy::Rack
|
|
||||||
end
|
|
||||||
alias :included :registered
|
|
||||||
end
|
|
||||||
|
|
||||||
class Collection
|
|
||||||
def initialize(app)
|
|
||||||
@app = app
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.proxies
|
|
||||||
@@proxies ||= {}
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.add(path, options={})
|
|
||||||
@@proxies ||= {}
|
|
||||||
@@proxies[path] = options[:to]
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
module ClassMethods
|
|
||||||
# Proxies requests to the path
|
|
||||||
#
|
|
||||||
# proxy '/twitter', "http://twitter/web/service"
|
|
||||||
def proxy(path, options={})
|
|
||||||
Middleman::Features::Proxy::Collection.add(path, options)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
# Rack application proxies requests as needed for the given project.
|
|
||||||
module Rack
|
|
||||||
|
|
||||||
def initialize(app)
|
|
||||||
@app = app
|
|
||||||
end
|
|
||||||
|
|
||||||
def call(env)
|
|
||||||
url = env['PATH_INFO']
|
|
||||||
|
|
||||||
@proxies = Middleman::Features::Proxy::Collection.proxies
|
|
||||||
@proxies.each do |proxy, value|
|
|
||||||
if url.match(/^#{Regexp.escape(proxy.to_s)}/)
|
|
||||||
return handle_proxy(value, proxy.to_s, env)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
return [404, {}, "not found"]
|
|
||||||
end
|
|
||||||
|
|
||||||
def handle_proxy(proxy, proxy_url, env)
|
|
||||||
if proxy[:secure] && !Middleman::HTTPS_ENABLED
|
|
||||||
$stderr.puts "~ WARNING: HTTPS is not supported on your system, using HTTP instead.\n"
|
|
||||||
$stderr.puts" If you are using Ubuntu, you can run `apt-get install libopenssl-ruby`\n"
|
|
||||||
proxy[:secure] = false
|
|
||||||
end
|
|
||||||
|
|
||||||
origin_host = env['SERVER_NAME'] # capture the origin host for cookies
|
|
||||||
http_method = env['REQUEST_METHOD'].to_s.downcase
|
|
||||||
url = env['PATH_INFO']
|
|
||||||
params = env['QUERY_STRING']
|
|
||||||
|
|
||||||
# collect headers...
|
|
||||||
headers = {}
|
|
||||||
env.each do |key, value|
|
|
||||||
next unless key =~ /^HTTP_/
|
|
||||||
key = key.gsub(/^HTTP_/,'').downcase.sub(/^\w/){|l| l.upcase}.gsub(/_(\w)/){|l| "-#{$1.upcase}"} # remove HTTP_, dasherize and titleize
|
|
||||||
if !key.eql? "Version"
|
|
||||||
headers[key] = value
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
# Rack documentation says CONTENT_TYPE and CONTENT_LENGTH aren't prefixed by HTTP_
|
|
||||||
headers['Content-Type'] = env['CONTENT_TYPE'] if env['CONTENT_TYPE']
|
|
||||||
|
|
||||||
length = env['CONTENT_LENGTH']
|
|
||||||
headers['Content-Length'] = length if length
|
|
||||||
|
|
||||||
http_host, http_port = proxy[:to].split(':')
|
|
||||||
http_port = proxy[:secure] ? '443' : '80' if http_port.nil?
|
|
||||||
|
|
||||||
# added 4/23/09 per Charles Jolley, corrects problem
|
|
||||||
# when making requests to virtual hosts
|
|
||||||
headers['Host'] = "#{http_host}:#{http_port}"
|
|
||||||
|
|
||||||
if proxy[:url]
|
|
||||||
url = url.sub(/^#{Regexp.escape proxy_url}/, proxy[:url])
|
|
||||||
end
|
|
||||||
|
|
||||||
http_path = [url]
|
|
||||||
http_path << params if params && params.size>0
|
|
||||||
http_path = http_path.join('?')
|
|
||||||
|
|
||||||
response = nil
|
|
||||||
no_body_method = %w(get copy head move options trace)
|
|
||||||
|
|
||||||
done = false
|
|
||||||
tries = 0
|
|
||||||
until done
|
|
||||||
http = ::Net::HTTP.new(http_host, http_port)
|
|
||||||
|
|
||||||
if proxy[:secure]
|
|
||||||
http.use_ssl = true
|
|
||||||
http.verify_mode = OpenSSL::SSL::VERIFY_NONE
|
|
||||||
end
|
|
||||||
|
|
||||||
http.start do |web|
|
|
||||||
if no_body_method.include?(http_method)
|
|
||||||
response = web.send(http_method, http_path, headers)
|
|
||||||
else
|
|
||||||
http_body = env['rack.input']
|
|
||||||
http_body.rewind # May not be necessary but can't hurt
|
|
||||||
|
|
||||||
req = Net::HTTPGenericRequest.new(http_method.upcase,
|
|
||||||
true, true, http_path, headers)
|
|
||||||
req.body_stream = http_body if length.to_i > 0
|
|
||||||
response = web.request(req)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
status = response.code # http status code
|
|
||||||
protocol = proxy[:secure] ? 'https' : 'http'
|
|
||||||
|
|
||||||
$stderr.puts "~ PROXY: #{http_method.upcase} #{status} #{url} -> #{protocol}://#{http_host}:#{http_port}#{http_path}\n"
|
|
||||||
|
|
||||||
# display and construct specific response headers
|
|
||||||
response_headers = {}
|
|
||||||
ignore_headers = ['transfer-encoding', 'keep-alive', 'connection']
|
|
||||||
response.each do |key, value|
|
|
||||||
next if ignore_headers.include?(key.downcase)
|
|
||||||
# If this is a cookie, strip out the domain. This technically may
|
|
||||||
# break certain scenarios where services try to set cross-domain
|
|
||||||
# cookies, but those services should not be doing that anyway...
|
|
||||||
value.gsub!(/domain=[^\;]+\;? ?/,'') if key.downcase == 'set-cookie'
|
|
||||||
# Location headers should rewrite the hostname if it is included.
|
|
||||||
value.gsub!(/^http:\/\/#{http_host}(:[0-9]+)?\//, "http://#{http_host}/") if key.downcase == 'location'
|
|
||||||
# content-length is returning char count not bytesize
|
|
||||||
if key.downcase == 'content-length'
|
|
||||||
if response.body.respond_to?(:bytesize)
|
|
||||||
value = response.body.bytesize.to_s
|
|
||||||
elsif response.body.respond_to?(:size)
|
|
||||||
value = response.body.size.to_s
|
|
||||||
else
|
|
||||||
value = '0'
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
$stderr.puts << " #{key}: #{value}\n"
|
|
||||||
response_headers[key] = value
|
|
||||||
end
|
|
||||||
|
|
||||||
if [301, 302, 303, 307].include?(status.to_i) && proxy[:redirect] != false
|
|
||||||
$stderr.puts '~ REDIRECTING: '+response_headers['location']+"\n"
|
|
||||||
|
|
||||||
uri = URI.parse(response_headers['location']);
|
|
||||||
http_host = uri.host
|
|
||||||
http_port = uri.port
|
|
||||||
http_path = uri.path
|
|
||||||
http_path += '?'+uri.query if uri.query
|
|
||||||
|
|
||||||
tries += 1
|
|
||||||
if tries > 10
|
|
||||||
raise "Too many redirects!"
|
|
||||||
end
|
|
||||||
else
|
|
||||||
done = true
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
# Thin doesn't like null bodies
|
|
||||||
response_body = response.body || ''
|
|
||||||
|
|
||||||
return [status, ::Rack::Utils::HeaderHash.new(response_headers), [response_body]]
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
Loading…
Reference in a new issue