Use factory proc to construct transform headers

cyu · Mar 17, 2023 · a53de88 · a53de88
1 parent 8e9701c
commit a53de88
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/lib/rack/cors/resource.rb b/lib/rack/cors/resource.rb
@@ -66,7 +66,7 @@ def to_headers(env)
           'access-control-max-age' => max_age.to_s
         }
         h['access-control-allow-credentials'] = 'true' if credentials
-        defined?(Rack::Utils::HeaderHash) ? Rack::Utils::HeaderHash.new(h) : h
+        header_proc.call(h)
       end
 
       protected
@@ -127,6 +127,16 @@ def compile(path)
           raise TypeError, path
         end
       end
+
+      def header_proc
+        @header_proc ||= begin
+          if defined?(Rack::Headers)
+            ->(h) { h }
+          else
+            ->(h) { Rack::Utils::HeaderHash.new(h) }
+          end
+        end
+      end
     end
   end
 end