Sha256: 311eff5c20564c7c86ce91b7e4aee9d1fa82c39cbcbc30cdd6c06693478133d8
Contents?: true
Size: 1.98 KB
Versions: 43
Compression:
Stored size: 1.98 KB
Contents
# frozen-string-literal: true
#
class Roda
module RodaPlugins
# The default_headers plugin accepts a hash of headers,
# and overrides the default_headers method in the
# response class to be a copy of the headers.
#
# Note that when using this module, you should not
# attempt to mutate of the values set in the default
# headers hash.
#
# Example:
#
# plugin :default_headers, 'Content-Type'=>'text/csv'
#
# You can modify the default headers later by loading the
# plugin again:
#
# plugin :default_headers, 'Foo'=>'bar'
# plugin :default_headers, 'Bar'=>'baz'
module DefaultHeaders
# Merge the given headers into the existing default headers, if any.
def self.configure(app, headers={})
app.opts[:default_headers] = (app.default_headers || app::RodaResponse::DEFAULT_HEADERS).merge(headers).freeze
end
module ClassMethods
# The default response headers to use for the current class.
def default_headers
opts[:default_headers]
end
# Optimize the response class set_default_headers method if it hasn't been
# overridden and all default headers are strings.
def freeze
if (headers = opts[:default_headers]).all?{|k, v| k.is_a?(String) && v.is_a?(String)} &&
(self::RodaResponse.instance_method(:set_default_headers).owner == Base::ResponseMethods)
self::RodaResponse.class_eval(<<-END, __FILE__, __LINE__+1)
private
def set_default_headers
h = @headers
#{headers.map{|k,v| "h[#{k.inspect}] ||= #{v.inspect}"}.join('; ')}
end
END
end
super
end
end
module ResponseMethods
# Get the default headers from the related roda class.
def default_headers
roda_class.default_headers
end
end
end
register_plugin(:default_headers, DefaultHeaders)
end
end
Version data entries
43 entries across 43 versions & 1 rubygems