diff -r 000000000000 -r 4eba9c11703f web/Zend/Layout/Controller/Plugin/Layout.php --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/web/Zend/Layout/Controller/Plugin/Layout.php Mon Dec 13 18:29:26 2010 +0100 @@ -0,0 +1,156 @@ +setLayout($layout); + } + } + + /** + * Retrieve layout object + * + * @return Zend_Layout + */ + public function getLayout() + { + return $this->_layout; + } + + /** + * Set layout object + * + * @param Zend_Layout $layout + * @return Zend_Layout_Controller_Plugin_Layout + */ + public function setLayout(Zend_Layout $layout) + { + $this->_layout = $layout; + return $this; + } + + /** + * Set layout action helper + * + * @param Zend_Layout_Controller_Action_Helper_Layout $layoutActionHelper + * @return Zend_Layout_Controller_Plugin_Layout + */ + public function setLayoutActionHelper(Zend_Layout_Controller_Action_Helper_Layout $layoutActionHelper) + { + $this->_layoutActionHelper = $layoutActionHelper; + return $this; + } + + /** + * Retrieve layout action helper + * + * @return Zend_Layout_Controller_Action_Helper_Layout + */ + public function getLayoutActionHelper() + { + return $this->_layoutActionHelper; + } + + /** + * postDispatch() plugin hook -- render layout + * + * @param Zend_Controller_Request_Abstract $request + * @return void + */ + public function postDispatch(Zend_Controller_Request_Abstract $request) + { + $layout = $this->getLayout(); + $helper = $this->getLayoutActionHelper(); + + // Return early if forward detected + if (!$request->isDispatched() + || $this->getResponse()->isRedirect() + || ($layout->getMvcSuccessfulActionOnly() + && (!empty($helper) && !$helper->isActionControllerSuccessful()))) + { + return; + } + + // Return early if layout has been disabled + if (!$layout->isEnabled()) { + return; + } + + $response = $this->getResponse(); + $content = $response->getBody(true); + $contentKey = $layout->getContentKey(); + + if (isset($content['default'])) { + $content[$contentKey] = $content['default']; + } + if ('default' != $contentKey) { + unset($content['default']); + } + + $layout->assign($content); + + $fullContent = null; + $obStartLevel = ob_get_level(); + try { + $fullContent = $layout->render(); + $response->setBody($fullContent); + } catch (Exception $e) { + while (ob_get_level() > $obStartLevel) { + $fullContent .= ob_get_clean(); + } + $request->setParam('layoutFullContent', $fullContent); + $request->setParam('layoutContent', $layout->content); + $response->setBody(null); + throw $e; + } + + } +}