Add module when loading a controller method

jenssegers · Jan 18, 2013 · beb2554 · beb2554
1 parent 0edaa5e
commit beb2554
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/third_party/HMVC/Loader.php b/third_party/HMVC/Loader.php
@@ -80,8 +80,17 @@ public function controller($uri, $params = array(), $return = FALSE) {
                 $uri = $module . '/' . $uri;
             }
         }
+
+        // Add module
+        $this->add_module($module);
 
-        return $this->_load_controller($uri, $params, $return);
+        // Execute the controller method and capture output
+        $void = $this->_load_controller($uri, $params, $return);
+
+        // Remove module
+        $this->remove_module();
+
+        return $void;
     }
 
     /**