instiki/vendor/rails/actionpack/lib/action_controller/flash.rb

214 lines
6.5 KiB
Ruby
Raw Normal View History

2007-01-22 14:43:50 +01:00
module ActionController #:nodoc:
# The flash provides a way to pass temporary objects between actions. Anything you place in the flash will be exposed
# to the very next action and then cleared out. This is a great way of doing notices and alerts, such as a create
# action that sets <tt>flash[:notice] = "Successfully created"</tt> before redirecting to a display action that can
# then expose the flash to its template. Actually, that exposure is automatically done. Example:
2007-01-22 14:43:50 +01:00
#
# class PostsController < ActionController::Base
2007-01-22 14:43:50 +01:00
# def create
# # save post
# flash[:notice] = "Successfully created post"
# redirect_to posts_path(@post)
2007-01-22 14:43:50 +01:00
# end
#
# def show
2007-01-22 14:43:50 +01:00
# # doesn't need to assign the flash notice to the template, that's done automatically
# end
# end
#
# show.html.erb
# <% if flash[:notice] %>
# <div class="notice"><%= flash[:notice] %></div>
# <% end %>
2007-01-22 14:43:50 +01:00
#
# This example just places a string in the flash, but you can put any object in there. And of course, you can put as
# many as you like at a time too. Just remember: They'll be gone by the time the next action has been performed.
2007-01-22 14:43:50 +01:00
#
# See docs on the FlashHash class for more details about the flash.
module Flash
def self.included(base)
base.class_eval do
include InstanceMethods
2010-05-25 19:45:45 +02:00
alias_method_chain :perform_action, :flash
alias_method_chain :reset_session, :flash
2010-05-25 19:45:45 +02:00
alias_method_chain :redirect_to, :flash
helper_method :alert
helper_method :notice
2007-01-22 14:43:50 +01:00
end
end
2007-01-22 14:43:50 +01:00
class FlashNow #:nodoc:
def initialize(flash)
@flash = flash
end
2007-01-22 14:43:50 +01:00
def []=(k, v)
@flash[k] = v
@flash.discard(k)
v
end
2007-01-22 14:43:50 +01:00
def [](k)
@flash[k]
end
end
2007-01-22 14:43:50 +01:00
class FlashHash < Hash
def initialize #:nodoc:
super
@used = {}
end
2007-01-22 14:43:50 +01:00
def []=(k, v) #:nodoc:
keep(k)
super
end
2007-01-22 14:43:50 +01:00
def update(h) #:nodoc:
h.keys.each { |k| keep(k) }
2007-01-22 14:43:50 +01:00
super
end
2007-01-22 14:43:50 +01:00
alias :merge! :update
2007-01-22 14:43:50 +01:00
def replace(h) #:nodoc:
@used = {}
super
end
2007-01-22 14:43:50 +01:00
# Sets a flash that will not be available to the next action, only to the current.
#
# flash.now[:message] = "Hello current action"
#
2007-01-22 14:43:50 +01:00
# This method enables you to use the flash as a central messaging system in your app.
# When you need to pass an object to the next action, you use the standard flash assign (<tt>[]=</tt>).
# When you need to pass an object to the current action, you use <tt>now</tt>, and your object will
# vanish when the current action is done.
#
# Entries set via <tt>now</tt> are accessed the same way as standard entries: <tt>flash['my-key']</tt>.
def now
FlashNow.new(self)
2007-01-22 14:43:50 +01:00
end
2007-01-22 14:43:50 +01:00
# Keeps either the entire current flash or a specific flash entry available for the next action:
#
# flash.keep # keeps the entire flash
# flash.keep(:notice) # keeps only the "notice" entry, the rest of the flash is discarded
def keep(k = nil)
2007-01-22 14:43:50 +01:00
use(k, false)
end
# Marks the entire flash or a single flash entry to be discarded by the end of the current action:
2007-01-22 14:43:50 +01:00
#
# flash.discard # discard the entire flash at the end of the current action
# flash.discard(:warning) # discard only the "warning" entry at the end of the current action
def discard(k = nil)
2007-01-22 14:43:50 +01:00
use(k)
end
2007-01-22 14:43:50 +01:00
# Mark for removal entries that were kept, and delete unkept ones.
#
# This method is called automatically by filters, so you generally don't need to care about it.
def sweep #:nodoc:
keys.each do |k|
2007-01-22 14:43:50 +01:00
unless @used[k]
use(k)
else
delete(k)
@used.delete(k)
end
end
# clean up after keys that could have been left over by calling reject! or shift on the flash
(@used.keys - keys).each{ |k| @used.delete(k) }
2007-01-22 14:43:50 +01:00
end
def store(session, key = "flash")
return if self.empty?
session[key] = self
end
2007-01-22 14:43:50 +01:00
private
# Used internally by the <tt>keep</tt> and <tt>discard</tt> methods
# use() # marks the entire flash as used
# use('msg') # marks the "msg" entry as used
# use(nil, false) # marks the entire flash as unused (keeps it around for one more action)
# use('msg', false) # marks the "msg" entry as unused (keeps it around for one more action)
def use(k=nil, v=true)
unless k.nil?
@used[k] = v
else
keys.each{ |key| use(key, v) }
2007-01-22 14:43:50 +01:00
end
end
end
module InstanceMethods #:nodoc:
protected
def perform_action_with_flash
perform_action_without_flash
if defined? @_flash
@_flash.store(session)
remove_instance_variable(:@_flash)
end
end
def reset_session_with_flash
reset_session_without_flash
remove_instance_variable(:@_flash) if defined? @_flash
end
2010-05-25 19:45:45 +02:00
def redirect_to_with_flash(options = {}, response_status_and_flash = {}) #:doc:
if alert = response_status_and_flash.delete(:alert)
flash[:alert] = alert
end
if notice = response_status_and_flash.delete(:notice)
flash[:notice] = notice
end
if other_flashes = response_status_and_flash.delete(:flash)
flash.update(other_flashes)
end
redirect_to_without_flash(options, response_status_and_flash)
end
# Access the contents of the flash. Use <tt>flash["notice"]</tt> to
# read a notice you put there or <tt>flash["notice"] = "hello"</tt>
# to put a new one.
def flash #:doc:
if !defined?(@_flash)
@_flash = session["flash"] || FlashHash.new
@_flash.sweep
2007-01-22 14:43:50 +01:00
end
@_flash
2007-01-22 14:43:50 +01:00
end
2010-05-25 19:45:45 +02:00
# Convenience accessor for flash[:alert]
def alert
flash[:alert]
end
# Convenience accessor for flash[:alert]=
def alert=(message)
flash[:alert] = message
end
# Convenience accessor for flash[:notice]
def notice
flash[:notice]
end
# Convenience accessor for flash[:notice]=
def notice=(message)
flash[:notice] = message
end
2007-01-22 14:43:50 +01:00
end
end
end