60 lines
1.7 KiB
Ruby
60 lines
1.7 KiB
Ruby
module ActionController #:nodoc:
|
|
module Flash
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
class_attribute :_flash_types, instance_accessor: false
|
|
self._flash_types = []
|
|
|
|
delegate :flash, to: :request
|
|
add_flash_types(:alert, :notice)
|
|
end
|
|
|
|
module ClassMethods
|
|
# Creates new flash types. You can pass as many types as you want to create
|
|
# flash types other than the default <tt>alert</tt> and <tt>notice</tt> in
|
|
# your controllers and views. For instance:
|
|
#
|
|
# # in application_controller.rb
|
|
# class ApplicationController < ActionController::Base
|
|
# add_flash_types :warning
|
|
# end
|
|
#
|
|
# # in your controller
|
|
# redirect_to user_path(@user), warning: "Incomplete profile"
|
|
#
|
|
# # in your view
|
|
# <%= warning %>
|
|
#
|
|
# This method will automatically define a new method for each of the given
|
|
# names, and it will be available in your views.
|
|
def add_flash_types(*types)
|
|
types.each do |type|
|
|
next if _flash_types.include?(type)
|
|
|
|
define_method(type) do
|
|
request.flash[type]
|
|
end
|
|
helper_method type
|
|
|
|
self._flash_types += [type]
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
def redirect_to(options = {}, response_status_and_flash = {}) #:doc:
|
|
self.class._flash_types.each do |flash_type|
|
|
if type = response_status_and_flash.delete(flash_type)
|
|
flash[flash_type] = type
|
|
end
|
|
end
|
|
|
|
if other_flashes = response_status_and_flash.delete(:flash)
|
|
flash.update(other_flashes)
|
|
end
|
|
|
|
super(options, response_status_and_flash)
|
|
end
|
|
end
|
|
end
|