From f1b5f5168b8a6b5b39752e704c344fd91d86b846 Mon Sep 17 00:00:00 2001
From: Paul van Tilburg <paul@mozcode.nl>
Date: Mon, 30 Jan 2012 13:57:51 +0100
Subject: [PATCH] Port the Markaby method override support to Mab

---
 stoptime.rb | 31 +++++++++++++++++--------------
 1 file changed, 17 insertions(+), 14 deletions(-)

diff --git a/stoptime.rb b/stoptime.rb
index 4eac2b2..e3a0708 100644
--- a/stoptime.rb
+++ b/stoptime.rb
@@ -77,21 +77,24 @@ end
 
 # = The Stop… Camping Time! Markaby extensions
 # FIXME: update for Mab!
-module StopTime::Helpers
-  #SUPPORTED = [:get, :post]
+module StopTime::Mab
+  SUPPORTED = [:get, :post]
 
-  # # Adds a method override field in form tags for (usually) unsupported
-  # # methods by browsers (i.e.  PUT and DELETE) by injecting a hidden field.
-  # def tag!(tag, *attrs)
-  #   return super unless tag == :form && block_given?
-  #   attrs = attrs.last.is_a?(Hash) ? attrs.last : {}
-  #   meth = attrs[:method]
-  #   attrs[:method] = 'post' if override = !SUPPORTED.include?(meth)
-  #   super(tag, attrs) do
-  #     input :type => 'hidden', :name => '_method', :value => meth if override
-  #     yield
-  #   end
-  # end
+  def mab_done(tag)
+    return super unless tag.name == :form
+
+    meth = tag.attributes[:method]
+    tag.attributes[:method] = 'post' if override = !SUPPORTED.include?(meth)
+    # Inject a hidden input element with the proper method to the tag block
+    # if the form method is unsupported.
+    orig_blk = tag.block
+    tag.block = proc do
+      input :type => 'hidden', :name => '_method', :value => meth
+      orig_blk.call
+    end if override
+  end
+
+  include Mab::Indentation
 
 end