mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
33 lines
689 B
Ruby
33 lines
689 B
Ruby
require 'active_support/memoizable'
|
|
|
|
module ActionDispatch
|
|
module Http
|
|
class Headers < ::Hash
|
|
@@env_cache = Hash.new { |h,k| h[k] = "HTTP_#{k.upcase.gsub(/-/, '_')}" }
|
|
|
|
def initialize(*args)
|
|
|
|
if args.size == 1 && args[0].is_a?(Hash)
|
|
super()
|
|
update(args[0])
|
|
else
|
|
super
|
|
end
|
|
end
|
|
|
|
def [](header_name)
|
|
if include?(header_name)
|
|
super
|
|
else
|
|
super(env_name(header_name))
|
|
end
|
|
end
|
|
|
|
private
|
|
# Converts a HTTP header name to an environment variable name.
|
|
def env_name(header_name)
|
|
@@env_cache[header_name]
|
|
end
|
|
end
|
|
end
|
|
end
|