first commit

classes/helper/HelperTreeShops.php

@@ -0,0 +1,194 @@
+<?php
+/**
+ * Copyright since 2007 PrestaShop SA and Contributors
+ * PrestaShop is an International Registered Trademark & Property of PrestaShop SA
+ *
+ * NOTICE OF LICENSE
+ *
+ * This source file is subject to the Open Software License (OSL 3.0)
+ * that is bundled with this package in the file LICENSE.md.
+ * It is also available through the world-wide-web at this URL:
+ * https://opensource.org/licenses/OSL-3.0
+ * If you did not receive a copy of the license and are unable to
+ * obtain it through the world-wide-web, please send an email
+ * to license@prestashop.com so we can send you a copy immediately.
+ *
+ * DISCLAIMER
+ *
+ * Do not edit or add to this file if you wish to upgrade PrestaShop to newer
+ * versions in the future. If you wish to customize PrestaShop for your
+ * needs please refer to https://devdocs.prestashop.com/ for more information.
+ *
+ * @author    PrestaShop SA and Contributors <contact@prestashop.com>
+ * @copyright Since 2007 PrestaShop SA and Contributors
+ * @license   https://opensource.org/licenses/OSL-3.0 Open Software License (OSL 3.0)
+ */
+class HelperTreeShopsCore extends TreeCore
+{
+    const DEFAULT_TEMPLATE = 'tree_shops.tpl';
+    const DEFAULT_NODE_FOLDER_TEMPLATE = 'tree_node_folder_checkbox_shops.tpl';
+    const DEFAULT_NODE_ITEM_TEMPLATE = 'tree_node_item_checkbox_shops.tpl';
+
+    private $_lang;
+    private $_selected_shops;
+
+    public function __construct($id, $title = null, $lang = null)
+    {
+        parent::__construct($id);
+
+        if (isset($title)) {
+            $this->setTitle($title);
+        }
+
+        $this->setLang($lang);
+    }
+
+    public function getData()
+    {
+        if (!isset($this->_data)) {
+            $this->setData(Shop::getTree());
+        }
+
+        return $this->_data;
+    }
+
+    public function setLang($value)
+    {
+        $this->_lang = $value;
+
+        return $this;
+    }
+
+    public function getLang()
+    {
+        if (!isset($this->_lang)) {
+            $this->setLang($this->getContext()->employee->id_lang);
+        }
+
+        return $this->_lang;
+    }
+
+    public function getNodeFolderTemplate()
+    {
+        if (!isset($this->_node_folder_template)) {
+            $this->setNodeFolderTemplate(self::DEFAULT_NODE_FOLDER_TEMPLATE);
+        }
+
+        return $this->_node_folder_template;
+    }
+
+    public function getNodeItemTemplate()
+    {
+        if (!isset($this->_node_item_template)) {
+            $this->setNodeItemTemplate(self::DEFAULT_NODE_ITEM_TEMPLATE);
+        }
+
+        return $this->_node_item_template;
+    }
+
+    public function setSelectedShops($value)
+    {
+        if (!is_array($value)) {
+            throw new PrestaShopException('Selected shops value must be an array');
+        }
+
+        $this->_selected_shops = $value;
+
+        return $this;
+    }
+
+    public function getSelectedShops()
+    {
+        if (!isset($this->_selected_shops)) {
+            $this->_selected_shops = [];
+        }
+
+        return $this->_selected_shops;
+    }
+
+    public function getTemplate()
+    {
+        if (!isset($this->_template)) {
+            $this->setTemplate(self::DEFAULT_TEMPLATE);
+        }
+
+        return $this->_template;
+    }
+
+    public function render($data = null, $use_default_actions = true, $use_selected_shop = true)
+    {
+        if (!isset($data)) {
+            $data = $this->getData();
+        }
+
+        if ($use_default_actions) {
+            $this->setActions([
+                new TreeToolbarLink(
+                    $this->translator->trans('Collapse all', [], 'Admin.Actions'),
+                    '#',
+                    '$(\'#' . $this->getId() . '\').tree(\'collapseAll\'); return false;',
+                    'icon-collapse-alt'
+                ),
+                new TreeToolbarLink(
+                    $this->translator->trans('Expand all', [], 'Admin.Actions'),
+                    '#',
+                    '$(\'#' . $this->getId() . '\').tree(\'expandAll\'); return false;',
+                    'icon-expand-alt'
+                ),
+                new TreeToolbarLink(
+                    $this->translator->trans('Check all', [], 'Admin.Actions'),
+                    '#',
+                    'checkAllAssociatedShops($(\'#' . $this->getId() . '\')); return false;',
+                    'icon-check-sign'
+                ),
+                new TreeToolbarLink(
+                    $this->translator->trans('Uncheck all', [], 'Admin.Actions'),
+                    '#',
+                    'uncheckAllAssociatedShops($(\'#' . $this->getId() . '\')); return false;',
+                    'icon-check-empty'
+                ),
+            ]);
+        }
+
+        if ($use_selected_shop) {
+            $this->setAttribute('selected_shops', $this->getSelectedShops());
+        }
+
+        return parent::render($data);
+    }
+
+    public function renderNodes($data = null)
+    {
+        if (!isset($data)) {
+            $data = $this->getData();
+        }
+
+        if (!is_array($data) && !$data instanceof Traversable) {
+            throw new PrestaShopException('Data value must be an traversable array');
+        }
+
+        $html = '';
+
+        foreach ($data as $item) {
+            if (array_key_exists('shops', $item)
+                && !empty($item['shops'])) {
+                $html .= $this->getContext()->smarty->createTemplate(
+                    $this->getTemplateFile($this->getNodeFolderTemplate()),
+                    $this->getContext()->smarty
+                )->assign($this->getAttributes())->assign([
+                    'children' => $this->renderNodes($item['shops']),
+                    'node' => $item,
+                ])->fetch();
+            } else {
+                $html .= $this->getContext()->smarty->createTemplate(
+                    $this->getTemplateFile($this->getNodeItemTemplate()),
+                    $this->getContext()->smarty
+                )->assign($this->getAttributes())->assign([
+                    'node' => $item,
+                ])->fetch();
+            }
+        }
+
+        return $html;
+    }
+}