collector.rb 886 B

1234567891011121314151617181920212223242526272829303132
  1. require "action_dispatch/http/mime_type"
  2. module AbstractController
  3. module Collector
  4. def self.generate_method_for_mime(mime)
  5. sym = mime.is_a?(Symbol) ? mime : mime.to_sym
  6. const = sym.to_s.upcase
  7. class_eval <<-RUBY, __FILE__, __LINE__ + 1
  8. def #{sym}(*args, &block) # def html(*args, &block)
  9. custom(Mime::#{const}, *args, &block) # custom(Mime::HTML, *args, &block)
  10. end # end
  11. RUBY
  12. end
  13. Mime::SET.each do |mime|
  14. generate_method_for_mime(mime)
  15. end
  16. protected
  17. def method_missing(symbol, &block)
  18. mime_constant = Mime.const_get(symbol.to_s.upcase)
  19. if Mime::SET.include?(mime_constant)
  20. AbstractController::Collector.generate_method_for_mime(mime_constant)
  21. send(symbol, &block)
  22. else
  23. super
  24. end
  25. end
  26. end
  27. end