Add restore blog method

jgarib · Jun 5, 2017 · 9765505 · 9765505
1 parent 2165662
commit 9765505
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/Twig/Extension/WordpressExtension.php b/Twig/Extension/WordpressExtension.php
@@ -99,6 +99,12 @@ public function switchBlog($id)
         $this->manager = $this->managerRegistry->getManager();
     }
 
+    public function restoreBlog()
+    {
+        $this->managerRegistry->restorePreviousBlog();
+        $this->manager = $this->managerRegistry->getManager();
+    }
+
     public function findOneOptionBy($criteria)
     {
         if (is_string($criteria)) {

diff --git a/Wordpress/ManagerRegistry.php b/Wordpress/ManagerRegistry.php
@@ -35,6 +35,7 @@ class ManagerRegistry implements ManagerRegistryInterface
     protected $rootDir;
     protected $environment;
     protected $currentBlogId = 1;
+    protected $previousBlogId = 1;
     protected $managers = [];
 
     public function __construct(
@@ -60,7 +61,7 @@ public function __construct(
     public function getManager($blogId = null)
     {
         if ($blogId !== null && $blogId !== $this->currentBlogId) {
-            $this->currentBlogId = $blogId;
+            $this->setCurrentBlogId($blogId);
         }
 
         if (!isset($this->managers[$this->currentBlogId])) {
@@ -85,13 +86,28 @@ public function getManager($blogId = null)
     }
 
     /**
+     * Switches the active blog until the user calls the restorePreviousBlog() method.
+     *
      * @param $blogId
      */
     public function setCurrentBlogId($blogId)
     {
+        if ($this->currentBlogId === $blogId) {
+            return;
+        }
+
+        $this->previousBlogId = $this->currentBlogId;
         $this->currentBlogId = $blogId;
     }
 
+    /**
+     * Switches active blog back after user calls the setCurrentBlogId() method.
+     */
+    public function restorePreviousBlog()
+    {
+        $this->setCurrentBlogId($this->previousBlogId);
+    }
+
     /**
      * Loads a configured object manager metadata, query or result cache driver.
      *