rack-cname_request 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f2c45ae4f0970d558960667f9730829e21f6d9ca
-  data.tar.gz: 4aba8bc690a5af39cde5187487d3fbf94f3f9ad2
+  metadata.gz: 76b6ee695dfd46b74650f95e38362e587c5cd7c9
+  data.tar.gz: a4d4b43f7ef474e871a27c0e7574a2cdbbbcb72c
 SHA512:
-  metadata.gz: 6a2bae2aa4de9d71a356ab547ee30764e99fe4de0c9cc854b5deb7b5598b7fe79d492bcdfb8cae55b5716ba64c9da384965c7f91a1c1d4f340cf38220634b3b6
-  data.tar.gz: 8501f8ce43de9f3e404b00e98086dd5ac8bd78ffd48bc959ddd0a8df4d05ac7ca3220ebb9779b09e00810986aee8eb1cc0942f1c936f08dc2df8e075b9b36a3d
+  metadata.gz: 8d4f73aa1cb54d1704f5f6e28c0895bd0afa05d2895b8bfeb864d0a4509a070578820901fc4623fd4299aa106c61020eea27a5fc946fed33b1a03052e68c94bc
+  data.tar.gz: f86d4a2af0575e7d229a9339b63a047b116530291de5ad636348e4649a25ba9511e77c7f91079d30dfa737f7e70789f8b4f76daeb34f1c4329e40ae3c3b5a63c

data/lib/rack/cname_request.rb

@@ -3,26 +3,55 @@ require 'rack/cname_request/version'
 module Rack
   class CnameRequest
 
-    attr_reader :http_cname_header_key
+    attr_reader :http_cname_header_key, :host_whitelist
 
-    def initialize(app, cname_header_key: nil)
+    def initialize(app, cname_header_key: nil, only: [])
       @app = app
       @http_cname_header_key = "HTTP_#{cname_header_key}"
+      @host_whitelist = only || []
     end
 
     def call(env)
       status, headers, body = @app.call env
-
-      if headers['Location'] && env[http_cname_header_key]
-        uri = URI(headers['Location'])
-        uri.host = env[http_cname_header_key]
-        headers['Location'] = uri.to_s
+      original_location = headers['Location']
+      if original_location
+        modifier = LocationModifier.new(original_location, env[http_cname_header_key], host_whitelist: host_whitelist)
+        headers['Location'] = modifier.modified_location if modifier.should_modify?
       end
 
       [status, headers, body]
     end
 
+  end
+
+  class LocationModifier
+
+    attr_reader :location_uri, :cname_header_value, :host_whitelist
+
+    def initialize(location, cname_header_value, host_whitelist: [])
+      @location_uri = URI(location) if location
+      @cname_header_value = cname_header_value
+      @host_whitelist = host_whitelist || []
+    end
+
+    def should_modify?
+       comes_from_cname_proxy? && location_host_in_white_list?
+    end
+
+    def modified_location
+      location_uri.host = cname_header_value
+      location_uri.to_s
+    end
+
     private
 
+    def comes_from_cname_proxy?
+      cname_header_value
+    end
+
+    def location_host_in_white_list?
+      host_whitelist.include?(location_uri.host)
+    end
+
   end
 end

data/lib/rack/cname_request/version.rb

@@ -1,5 +1,5 @@
 module Rack
   class CnameRequest
-    VERSION = '0.1.1'
+    VERSION = '0.1.2'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rack-cname_request
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - FENG Zhichao
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-01-03 00:00:00.000000000 Z
+date: 2018-01-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler