1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
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_const = symbol.upcase
raise NoMethodError, "To respond to a custom format, register it as a MIME type first:" +
"http://guides.rubyonrails.org/action_controller_overview.html#restful-downloads." +
"If you meant to respond to a variant like :tablet or :phone, not a custom format," +
"be sure to nest your variant response within a format response: format.html" +
"{ |html| html.tablet { ..." unless Mime.const_defined?(mime_const)
mime_constant = Mime.const_get(mime_const)
if Mime::SET.include?(mime_constant)
AbstractController::Collector.generate_method_for_mime(mime_constant)
send(symbol, &block)
else
super
end
end
end
end
|
