lib/ramaze/helper/flash.rb in ramaze-0.3.5 vs lib/ramaze/helper/flash.rb in ramaze-0.3.9

@@ -20,15 +20,40 @@
   #
   #   do_stuff if flash[:error]
   #
   # On the request after this, flash[:error] is gone.
 
-  module FlashHelper
-    private
-
+  module Helper::Flash
+    trait :tag => "<div class='flash' id='flash_%key'>%value</div>"
     # answers with Session.current.flash
 
     def flash
       Session.current.flash
+    end
+
+    # Use in your template to display all flash messages that may be stored.
+    # For example, given you stored:
+    #
+    #   flash # => { :error => 'Pleae enter your name'
+    #                :info => 'Do you see the fnords?' }
+    #
+    # Then a flashbox would display:
+    #
+    #   <div class='flash' id='flash_error'>Please enter your name</div>
+    #   <div class='flash' id='flash_info'>Do you see the fnords?</div>
+    #
+    # This is designed to be customized permanently or per usage:
+    #
+    #   flashbox("<div class='flash_%key'>%value</div>")
+    #
+    # Where any occurrence of %key and %value will be replaced by the actual
+    # contents of each element of flash
+
+    def flashbox(tag = Helper::Flash.trait[:tag])
+      flash.map{|key, value|
+        [key.to_s, value.to_s]
+      }.sort.map{|(key, value)|
+        tag.gsub(/%key/, key.to_s).gsub(/%value/, value.to_s)
+      }.join("\n")
     end
   end
 end