// Improve flexibility of dispatcher + fix controllers access in BO (only admin controllers are loaded by the dispatcher)

git-svn-id: http://dev.prestashop.com/svn/v1/branches/1.5.x@8764 b9a71923-0436-4b27-9f14-aed3839534dd

admin-dev/functions.php

@@ -233,6 +233,11 @@ function translate($string)
  */
 function checkingTab($tab)
 {
+	static $controllers;
+
+	if (is_null($controllers))
+		$controllers = Dispatcher::getControllers(_PS_ADMIN_DIR_.'/tabs/');
+
 	$tab = trim($tab);
 	if (!Validate::isTabName($tab))
 		return false;
@@ -244,10 +249,10 @@ function checkingTab($tab)
 		echo sprintf(Tools::displayError('Tab %s cannot be found.'),$tab);
 		return false;
 	}
-	if ($row['module'] AND file_exists(_PS_MODULE_DIR_.'/'.$row['module'].'/'.Dispatcher::$controllers[$tab].'.php'))
+	if ($row['module'] AND file_exists(_PS_MODULE_DIR_.'/'.$row['module'].'/'.$controllers[$tab].'.php'))
-		include_once(_PS_MODULE_DIR_.'/'.$row['module'].'/'.Dispatcher::$controllers[$tab].'.php');
+		include_once(_PS_MODULE_DIR_.'/'.$row['module'].'/'.$controllers[$tab].'.php');
-	elseif (file_exists(_PS_ADMIN_DIR_.'/tabs/'.Dispatcher::$controllers[$tab].'.php'))
+	elseif (file_exists(_PS_ADMIN_DIR_.'/tabs/'.$controllers[$tab].'.php'))
-		include_once(_PS_ADMIN_DIR_.'/tabs/'.Dispatcher::$controllers[$tab].'.php');
+		include_once(_PS_ADMIN_DIR_.'/tabs/'.$controllers[$tab].'.php');
 	if (!class_exists($tab, false) OR !$row['id_tab'])
 	{
 		echo sprintf(Tools::displayError('Tab file %s cannot be found.'),$tab);

admin-dev/header.inc.php

@@ -278,7 +278,8 @@ echo '			</select>
 if (empty($tab))
 	echo '<div class="mainsubtablist" style="display:none"></div>';
 // This is made to display the subtab list
-$id_current_tab = (int)(Tab::getIdFromClassName(Dispatcher::$controllers[$tab]));
+$controllers = Dispatcher::getControllers(_PS_ADMIN_DIR_.'/tabs/');
+$id_current_tab = (int)Tab::getIdFromClassName($controllers[$tab]);
 $myCurrentTab = new Tab($id_current_tab);
 $tabs = Tab::getTabs(Context::getContext()->language->id, 0);
 $echoLis = '';

admin-dev/index.php

@@ -28,5 +28,8 @@
 define('_PS_ADMIN_DIR_', getcwd());
 require(dirname(__FILE__).'/../config/config.inc.php');
 require(dirname(__FILE__).'/functions.php');
-Dispatcher::getInstance()->dispatch();
 
+Dispatcher::getInstance()->setDefaultController('adminhome');
+Dispatcher::getInstance()->setControllerNotFound('adminhome');
+Dispatcher::getInstance()->setControllerDirectories(array(_PS_ADMIN_DIR_.'/tabs/', _PS_ADMIN_CONTROLLER_DIR_));
+Dispatcher::getInstance()->dispatch();

admin-dev/tabs/AdminModulesPositions.php

@@ -496,7 +496,7 @@ EOF;
 		echo '<br /><select id="em_list_'.$shopID.'">';
 
 		// @todo do something better with controllers
-		$controllers = Dispatcher::getControllers();
+		$controllers = Dispatcher::getControllers(_PS_FRONT_CONTROLLER_DIR_);
 		ksort($controllers);
 		foreach ($controllers as $k => $v)
 		{

classes/Dispatcher.php

@@ -36,14 +36,7 @@ class DispatcherCore
 	public static $instance = null;
 
 	/**
-	 * @var array list of available controllers
+	 * @var array List of default routes
-	 */
-	public static $controllers;
-
-	/**
-	 * List of default routes
-	 *
-	 * @var array
 	 */
 	public $default_routes = array(
 		'product_rule' => array(
@@ -124,40 +117,45 @@ class DispatcherCore
 	);
 
 	/**
-	 * If true, use routes to build URL (mod rewrite must be activated)
+	 * @var bool If true, use routes to build URL (mod rewrite must be activated)
-	 *
-	 * @var bool
 	 */
 	protected $use_routes = false;
 
 	/**
-	 * List of loaded routes
+	 * @var array List of loaded routes
-	 *
-	 * @var array
 	 */
 	protected $routes = array();
 
 	/**
-	 * Current controller name
+	 * @var string Current controller name
-	 *
-	 * @var string
 	 */
 	protected $controller;
 
 	/**
-	 * Current request uri
+	 * @var string Current request uri
-	 *
-	 * @var string
 	 */
 	protected $request_uri;
 
 	/**
-	 * Store empty route (a route with an empty rule)
+	 * @var array Store empty route (a route with an empty rule)
-	 *
-	 * @var array
 	 */
 	protected $empty_route;
 
+	/**
+	 * @var string Set default controller, which will be used if http parameter 'controller' is empty
+	 */
+	protected $default_controller = 'index';
+
+	/**
+	 * @var string Controller to use if found controller doesn't exist
+	 */
+	protected $controller_not_found = 'pagenotfound';
+
+	/**
+	 * @var array List of controllers where are stored controllers
+	 */
+	protected $controller_directories = array();
+
 	/**
 	 * Get current instance of dispatcher (singleton)
 	 *
@@ -175,6 +173,8 @@ class DispatcherCore
 	 */
 	protected function __construct()
 	{
+		$this->setDefaultController('index');
+		$this->setControllerDirectories(_PS_FRONT_CONTROLLER_DIR_);
 		$this->use_routes = (bool)Configuration::get('PS_REWRITING_SETTINGS');
 		$this->loadRoutes();
 
@@ -185,6 +185,38 @@ class DispatcherCore
 			$this->request_uri = $_SERVER['HTTP_X_REWRITE_URL'];
 	}
 
+	/**
+	 * Set default controller, which will be used if http parameter 'controller' is empty
+	 *
+	 * @param string $controller
+	 */
+	public function setDefaultController($controller)
+	{
+		$this->default_controller = $controller;
+	}
+
+	/**
+	 * Controller to use if found controller doesn't exist
+	 *
+	 * @var string $controller
+	 */
+	public function setControllerNotFound($controller)
+	{
+		$this->controller_not_found = $controller;
+	}
+
+	/**
+	 * Set list of controllers where are stored controllers
+	 *
+	 * @param mixed A directory, or an array of directory
+	 */
+	public function setControllerDirectories($dir)
+	{
+		if (!is_array($dir))
+			$dir = array($dir);
+		$this->controller_directories = $dir;
+	}
+
 	/**
 	 * Find the controller and instantiate it
 	 */
@@ -202,20 +234,20 @@ class DispatcherCore
 
 		// Get current controller and list of controllers
 		$this->getController();
-		$controllers = Dispatcher::getControllers();
+		$controllers = Dispatcher::getControllers($this->controller_directories);
 
 		if (!$this->controller)
 			$this->controller = (defined('_PS_ADMIN_DIR_')) ? 'adminhome' : 'index';
 
 		// For retrocompatibility with admin/tabs/ old system
-		if (defined('_PS_ADMIN_DIR_') && file_exists(_PS_ADMIN_DIR_.'/tabs/'.$controllers[$this->controller].'.php'))
+		if (isset($controllers[$this->controller]) && defined('_PS_ADMIN_DIR_') && file_exists(_PS_ADMIN_DIR_.'/tabs/'.$controllers[$this->controller].'.php'))
 		{
 			require_once(_PS_ADMIN_DIR_.'/functions.php');
 			runAdminTab();
 			return;
 		}
 		else if (!isset($controllers[$this->controller]))
-			$this->controller = 'pagenotfound';
+			$this->controller = $this->controller_not_found;
 
 		// Instantiate controller
 		Controller::getController($controllers[$this->controller])->run();
@@ -446,17 +478,17 @@ class DispatcherCore
 	/**
 	 * Get list of all available controllers
 	 *
+	 * @var mixed $dirs
 	 * @return array
 	 */
-	public static function getControllers()
+	public static function getControllers($dirs)
 	{
-		if (self::$controllers)
+		if (!is_array($dirs))
-			return self::$controllers;
+			$dirs = array($dirs);
 
 		$controllers = array();
-		if (defined('_PS_ADMIN_DIR_'))
+		foreach ($dirs as $dir)
-			$controllers = array_merge($controllers, Dispatcher::getControllersInDirectory(_PS_ADMIN_DIR_.'/tabs/', ''));
+			$controllers = array_merge($controllers, Dispatcher::getControllersInDirectory($dir));
-		$controllers = array_merge($controllers, Dispatcher::getControllersInDirectory(_PS_CONTROLLER_DIR_));
 
 		// Add default controllers
 		$controllers['index'] = 'IndexController';
@@ -465,18 +497,16 @@ class DispatcherCore
 		if (isset($controllers['compare']))
 			$controllers['productscomparison'] = $controllers['compare'];
 
-		self::$controllers = $controllers;
+		return $controllers;
-		return self::$controllers;
 	}
 
 	/**
 	 * Get list of available controllers from the specified dir
 	 *
 	 * @param string dir directory to scan (recursively)
-	 * @param filename suffix (without .php extension). Others files will be ignored.
 	 * @return array
 	 */
-	public static function getControllersInDirectory($dir, $suffix = 'Controller')
+	public static function getControllersInDirectory($dir)
 	{
 		$controllers = array();
 		$controller_files = scandir($dir);
@@ -486,10 +516,10 @@ class DispatcherCore
 			{
 				if (is_dir($dir.$controller_filename))
 					$controllers += Dispatcher::getControllersInDirectory($dir.$controller_filename.DIRECTORY_SEPARATOR);
-				else if (substr($controller_filename, - (strlen($suffix) + 4), - 4) == $suffix)
+				else if ($controller_filename != 'index.php')
 				{
-					$subdir = str_replace(_PS_CONTROLLER_DIR_, '', $dir);
+					$key = str_replace(array('controller.php', '.php'), array('', ''), strtolower($controller_filename));
-					$controllers[strtolower(substr($controller_filename, 0, - (strlen($suffix) + 4)))] = basename($controller_filename, '.php');
+					$controllers[$key] = basename($controller_filename, '.php');
 				}
 			}
 		}

config/defines.inc.php

@@ -71,6 +71,8 @@ define('_MODULE_DIR_',        	__PS_BASE_URI__.'modules/');
 define('_PS_ROOT_DIR_',             realpath($currentDir.'/..'));
 define('_PS_CLASS_DIR_',            _PS_ROOT_DIR_.'/classes/');
 define('_PS_CONTROLLER_DIR_',       _PS_ROOT_DIR_.'/controllers/');
+define('_PS_FRONT_CONTROLLER_DIR_', _PS_ROOT_DIR_.'/controllers/front/');
+define('_PS_ADMIN_CONTROLLER_DIR_', _PS_ROOT_DIR_.'/controllers/admin/');
 define('_PS_TRANSLATIONS_DIR_',     _PS_ROOT_DIR_.'/translations/');
 define('_PS_DOWNLOAD_DIR_',         _PS_ROOT_DIR_.'/download/');
 define('_PS_MAIL_DIR_',             _PS_ROOT_DIR_.'/mails/');