Add multiple domain support for rack

technicalpickles · Jun 5, 2009 · bfe769c · bfe769c
1 parent 7e8baf2
commit bfe769c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,3 +1,6 @@
+2009-06-05 - Sean Huber (shuber@huberry.com)
+  * Add multiple domain support for rack
+
 2009-06-04 - Sean Huber (shuber@huberry.com)
   * Update test related rake tasks
   * Remove huberry namespace

diff --git a/lib/proxy.rb b/lib/proxy.rb
@@ -15,7 +15,11 @@ def self.replace_host_with(&block)
   private
 
     def self.before_dispatch(dispatcher)
-      request = dispatcher.instance_variable_get('@request')
+      request = dispatcher.instance_variable_get('@request') || dispatcher.instance_variable_get('@env')
+      if request.is_a?(Hash)
+        request.each(&:inspect) # TODO: we can't access the 'rack.request' key without doing this first...wtf
+        request = request['rack.request']
+      end
       new_host = replace_host_with_proc.call(request)
       request.env['HTTP_X_FORWARDED_HOST'] = [request.host, new_host].join(', ') unless new_host.blank?
     end