From afaeb15b7ddbe131fa1d21af826fde92d559a326 Mon Sep 17 00:00:00 2001
From: Jerome Nadaud <jerome.nadaud@prestashop.com>
Date: Wed, 13 Nov 2013 20:10:13 +0100
Subject: [PATCH] // Define default template dir on tree class

---
 classes/helper/HelperForm.php      |  4 ++-
 classes/tree/Tree.php              |  6 ++--
 classes/tree/TreeToolbar.php       | 10 +++++--
 classes/tree/TreeToolbarButton.php | 44 +++++++++++++++++++++++++++---
 4 files changed, 53 insertions(+), 11 deletions(-)

diff --git a/classes/helper/HelperForm.php b/classes/helper/HelperForm.php
index ec1c69178..ab97b0893 100644
--- a/classes/helper/HelperForm.php
+++ b/classes/helper/HelperForm.php
@@ -245,7 +245,7 @@ class HelperFormCore extends Helper
 	 *
 	 * @return string
 	 */
-	public function renderAssoShop($params, $disable_shared = false)
+	public function renderAssoShop($disable_shared = false, $template_directory = null)
 	{
 		if (!Shop::isFeatureActive())
 			return;
@@ -288,6 +288,8 @@ class HelperFormCore extends Helper
 		}*/
 
 		$tree = new HelperTreeShops('shop-tree', 'Shops');
+		if (isset($template_directory))
+			$tree->setTemplateDirectory($template_directory);
 		$tree->setSelectedShops($assos);
 		$tree->setAttribute('table', $this->table);
 		return $tree->render();
diff --git a/classes/tree/Tree.php b/classes/tree/Tree.php
index 95b82f13c..8db9f8f47 100644
--- a/classes/tree/Tree.php
+++ b/classes/tree/Tree.php
@@ -61,7 +61,7 @@ class TreeCore
 		if (!isset($this->_toolbar))
 			$this->setToolbar(new TreeToolbarCore());
 
-		$this->getToolbar()->setActions($value);
+		$this->getToolbar()->setTemplateDirectory($this->getTemplateDirectory())->setActions($value);
 		return $this;
 	}
 
@@ -70,7 +70,7 @@ class TreeCore
 		if (!isset($this->_toolbar))
 			$this->setToolbar(new TreeToolbarCore());
 
-		return $this->getToolbar()->getActions();
+		return $this->getToolbar()->setTemplateDirectory($this->getTemplateDirectory())->getActions();
 	}
 
 	public function setAttribute($name, $value)
@@ -293,7 +293,7 @@ class TreeCore
 		if (!isset($this->_toolbar))
 			$this->setToolbar(new TreeToolbarCore());
 
-		$this->getToolbar()->addAction($action);
+		$this->getToolbar()->setTemplateDirectory($this->getTemplateDirectory())->addAction($action);
 		return $this;
 	}
 
diff --git a/classes/tree/TreeToolbar.php b/classes/tree/TreeToolbar.php
index 195333560..5ed6239cd 100644
--- a/classes/tree/TreeToolbar.php
+++ b/classes/tree/TreeToolbar.php
@@ -40,12 +40,13 @@ class TreeToolbarCore implements ITreeToolbarCore
 		return $this->render();
 	}
 
-	public function setActions($value)
+	public function setActions($actions)
 	{
-		if (!is_array($value) && !$value instanceof Traversable)
+		if (!is_array($actions) && !$actions instanceof Traversable)
 			throw new PrestaShopException('Action value must be an traversable array');
 
-		$this->_actions = $value;
+		foreach($actions as $action)
+			$this->addAction($action);
 	}
 
 	public function getActions()
@@ -165,6 +166,9 @@ class TreeToolbarCore implements ITreeToolbarCore
 		if (!isset($this->_actions))
 			$this->_actions = array();
 
+		if (isset($this->_template_directory))
+			$action->setTemplateDirectory($this->getTemplateDirectory());
+
 		$this->_actions[] = $action;
 		return $this;
 	}
diff --git a/classes/tree/TreeToolbarButton.php b/classes/tree/TreeToolbarButton.php
index 942642027..3ce66d638 100644
--- a/classes/tree/TreeToolbarButton.php
+++ b/classes/tree/TreeToolbarButton.php
@@ -155,13 +155,49 @@ abstract class TreeToolbarButtonCore
 	public function getTemplateDirectory()
 	{
 		if (!isset($this->_template_directory))
-			$this->_template_directory = $this->_normalizeDirectory(
-				$this->getContext()->smarty->getTemplateDir(0)
-				.self::DEFAULT_TEMPLATE_DIRECTORY);
+			$this->_template_directory = $this->_normalizeDirectory(self::DEFAULT_TEMPLATE_DIRECTORY);
 
 		return $this->_template_directory;
 	}
 
+	public function getTemplateFile($template)
+	{
+		if (preg_match_all('/((?:^|[A-Z])[a-z]+)/', get_class($this->getContext()->controller), $matches) !== FALSE)
+			$controllerName = strtolower($matches[0][1]);
+
+		if ($this->getContext()->controller instanceof ModuleAdminController)
+			return $this->_normalizeDirectory(
+				$this->getContext()->controller->getTemplatePath())
+				.$this->getTemplateDirectory().$template;
+		else if ($this->getContext()->controller instanceof AdminController
+			&& isset($controllerName) && file_exists($this->_normalizeDirectory(
+				$this->getContext()->smarty->getTemplateDir(0)).'controllers'
+				.DIRECTORY_SEPARATOR
+				.$controllerName
+				.DIRECTORY_SEPARATOR
+				.$this->getTemplateDirectory().$template))
+			return $this->_normalizeDirectory(
+				$this->getContext()->smarty->getTemplateDir(0)).'controllers'
+				.DIRECTORY_SEPARATOR
+				.$controllerName
+				.DIRECTORY_SEPARATOR
+				.$this->getTemplateDirectory().$template;
+		else if (file_exists($this->_normalizeDirectory(
+				$this->getContext()->smarty->getTemplateDir(1))
+				.$this->getTemplateDirectory().$template))
+				return $this->_normalizeDirectory(
+					$this->getContext()->smarty->getTemplateDir(1))
+					.$this->getTemplateDirectory().$template;
+		else if (file_exists($this->_normalizeDirectory(
+				$this->getContext()->smarty->getTemplateDir(0))
+				.$this->getTemplateDirectory().$template))
+				return $this->_normalizeDirectory(
+				$this->getContext()->smarty->getTemplateDir(0))
+				.$this->getTemplateDirectory().$template;
+		else
+			return $this->getTemplateDirectory().$template;
+	}
+
 	public function hasAttribute($name)
 	{
 		return (isset($this->_attributes)
@@ -171,7 +207,7 @@ abstract class TreeToolbarButtonCore
 	public function render()
 	{
 		return $this->getContext()->smarty->createTemplate(
-			$this->getTemplateDirectory().$this->getTemplate(),
+			$this->getTemplateFile($this->getTemplate()),
 			$this->getContext()->smarty
 		)->assign($this->getAttributes())->fetch();
 	}