Controller plugin that will help with PRG (see wikipedia link below)

http://en.wikipedia.org/wiki/Post/Redirect/Get
mnapoli · Jul 13, 2012 · 7f3029b · 7f3029b
1 parent 95011ba
commit 7f3029b
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 6 deletions.
diff --git a/library/Zend/Mvc/Controller/Plugin/PostRedirectGet.php b/library/Zend/Mvc/Controller/Plugin/PostRedirectGet.php
@@ -0,0 +1,32 @@
+<?php
+
+namespace Zend\Mvc\Controller\Plugin;
+
+use Zend\Mvc\Router\Exception\RuntimeException;
+
+class PostRedirectGet extends AbstractPlugin
+{
+    public function __invoke($redirect)
+    {
+        $controller = $this->getController();
+        $request = $controller->getRequest();
+        $flashMessenger = $controller->flashMessenger()->setNamespace('prg-post');
+
+        if ($request->isPost()) {
+            $flashMessenger->addMessage($request->getPost()->toArray());
+            try {
+                return $controller->redirect()->toRoute($redirect);
+            } catch (RuntimeException $e) {
+                return $controller->redirect()->toUrl($redirect);
+            }
+        } else {
+            $messages = $flashMessenger->getMessages();
+            if (count($messages)) {
+                return $messages[0];
+            } else {
+                return false;
+            }
+        }
+    }
+}
+
diff --git a/library/Zend/Mvc/Controller/PluginManager.php b/library/Zend/Mvc/Controller/PluginManager.php
@@ -33,12 +33,14 @@ class PluginManager extends AbstractPluginManager
      * @var array
      */
     protected $invokableClasses = array(
-        'flashmessenger' => 'Zend\Mvc\Controller\Plugin\FlashMessenger',
-        'forward'        => 'Zend\Mvc\Controller\Plugin\Forward',
-        'layout'         => 'Zend\Mvc\Controller\Plugin\Layout',
-        'params'         => 'Zend\Mvc\Controller\Plugin\Params',
-        'redirect'       => 'Zend\Mvc\Controller\Plugin\Redirect',
-        'url'            => 'Zend\Mvc\Controller\Plugin\Url',
+        'flashmessenger'  => 'Zend\Mvc\Controller\Plugin\FlashMessenger',
+        'forward'         => 'Zend\Mvc\Controller\Plugin\Forward',
+        'layout'          => 'Zend\Mvc\Controller\Plugin\Layout',
+        'params'          => 'Zend\Mvc\Controller\Plugin\Params',
+        'postredirectget' => 'Zend\Mvc\Controller\Plugin\PostRedirectGet',
+        'prg'             => 'Zend\Mvc\Controller\Plugin\PostRedirectGet',
+        'redirect'        => 'Zend\Mvc\Controller\Plugin\Redirect',
+        'url'             => 'Zend\Mvc\Controller\Plugin\Url',
     );
 
     /**