class ActionController::MimeResponds::Collector

def any(*args, &block)
  if args.any?
    args.each { |type| send(type, &block) }
  else
    custom(Mime::ALL, &block)
  end
end

def custom(mime_type, &block)
  mime_type = mime_type.is_a?(Mime::Type) ? mime_type : Mime::Type.lookup(mime_type.to_s)
  @order << mime_type
  @responses[mime_type] ||= block
end

def initialize(&block)
  @order, @responses, @default_response = [], {}, block
end

def response_for(mime)
  @responses[mime] || @responses[Mime::ALL] || @default_response
end