lib/sitehub/forward_proxies.rb in sitehub-0.4.2 vs lib/sitehub/forward_proxies.rb in sitehub-0.4.3

@@ -5,46 +5,45 @@
 require 'rack/utils'
 require 'em-http'
 
 class SiteHub
   class ForwardProxies
-
     NOT_FOUND = Rack::Response.new(['page not found'], 404, {})
-    def call env
+    def call(env)
       source_request = Rack::Request.new(env)
 
       forward_proxy = mapped_route(path: source_request.path, request: source_request)
 
       return forward_proxy.call(env) if forward_proxy
       NOT_FOUND
     end
 
     def init
-      forward_proxies.values.each { |proxy| proxy.build }
+      forward_proxies.values.each(&:build)
       self
     end
 
-    def << route
+    def <<(route)
       forward_proxies[route.mapped_path] = route
     end
 
     def mapped_route(path:, request:)
+      fwd_proxy_builder = forward_proxies[mapping(path)]
+      fwd_proxy_builder ? fwd_proxy_builder.resolve(id: request.cookies[RECORDED_ROUTES_COOKIE], env: request.env) : nil
+    end
 
-      key = forward_proxies.keys.find do |key|
+    def mapping(path)
+      forward_proxies.keys.find do |key|
         case key
-          when Regexp
-            key.match(path)
-          else
-            path == key
+        when Regexp
+          key.match(path)
+        else
+          path == key
         end
       end
-
-      forward_proxy_builder = forward_proxies[key]
-      forward_proxy_builder ? forward_proxy_builder.resolve(id: request.cookies[RECORDED_ROUTES_COOKIE], env: request.env) : nil
     end
 
     def forward_proxies
       @forward_proxies ||= {}
     end
-
   end
 end