Store the current user in the CurrentUserDesignAction

1 files changed, 22 insertions, 4 deletions

diff --git a/lib/currentuserdesignaction.php b/lib/currentuserdesignaction.php
index 490f87d13..7cd892022 100644
--- a/lib/currentuserdesignaction.php
+++ b/lib/currentuserdesignaction.php
@@ -22,7 +22,7 @@
  * @category  Action
  * @package   StatusNet
  * @author    Evan Prodromou <evan@status.net>
- * @copyright 2009 StatusNet, Inc.
+ * @copyright 2009-2010 StatusNet, Inc.
  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  * @link      http://status.net/
  */
@@ -40,12 +40,31 @@ if (!defined('STATUSNET') && !defined('LACONICA')) {
  * @category Action
  * @package  StatusNet
  * @author   Evan Prodromou <evan@status.net>
+ * @author   Zach Copley    <zach@status.net>
  * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  * @link     http://status.net/
  *
  */
 class CurrentUserDesignAction extends Action
 {
+
+    protected $cur = null; // The current user
+
+    /**
+     * For initializing members of the class. Set a the
+     * current user here.
+     *
+     * @param array $argarray misc. arguments
+     *
+     * @return boolean true
+     */
+    function prepare($argarray)
+    {
+        parent::prepare($argarray);
+
+        $this->cur = common_current_user();
+    }
+
     /**
      * A design for this action
      *
@@ -55,9 +74,7 @@ class CurrentUserDesignAction extends Action
      */
     function getDesign()
     {
-        $cur = common_current_user();
-
-        if (!empty($cur)) {
+        if (!empty($this->cur)) {
 
             $design = $cur->getDesign();
 
@@ -68,4 +85,5 @@ class CurrentUserDesignAction extends Action
 
         return parent::getDesign();
     }
+
 }