021f3d24f3
Assuming the type ":touch", Collector.new was calling send(:touch), which instead of triggering method_missing and generating a new collector method, actually invoked the private method `touch` inherited from Object. By generating the method for each mime type as it is registered, the private methods on Object can never be reached by `send`, because the `Collector` will have them before `send` is called on it. To do this, a callback mechanism was added to Mime::Type This allows someone to add a callback for whenever a new mime type is registered. The callback then gets called with the new mime as a parameter. This is then used in AbstractController::Collector to generate new collector methods after each mime is registered.
37 lines
1017 B
Ruby
37 lines
1017 B
Ruby
require "action_dispatch/http/mime_type"
|
|
|
|
module AbstractController
|
|
module Collector
|
|
def self.generate_method_for_mime(mime)
|
|
sym = mime.is_a?(Symbol) ? mime : mime.to_sym
|
|
const = sym.upcase
|
|
class_eval <<-RUBY, __FILE__, __LINE__ + 1
|
|
def #{sym}(*args, &block) # def html(*args, &block)
|
|
custom(Mime::#{const}, *args, &block) # custom(Mime::HTML, *args, &block)
|
|
end # end
|
|
RUBY
|
|
end
|
|
|
|
Mime::SET.each do |mime|
|
|
generate_method_for_mime(mime)
|
|
end
|
|
|
|
Mime::Type.register_callback do |mime|
|
|
generate_method_for_mime(mime) unless self.instance_methods.include?(mime.to_sym)
|
|
end
|
|
|
|
protected
|
|
|
|
def method_missing(symbol, &block)
|
|
mime_constant = Mime.const_get(symbol.upcase)
|
|
|
|
if Mime::SET.include?(mime_constant)
|
|
AbstractController::Collector.generate_method_for_mime(mime_constant)
|
|
send(symbol, &block)
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|
|
end
|