Add module when loading a controller method

Author: jenssegers

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;
     }
 
     /**