Made saving state optional

Author: arogachev

src/behaviors/NestedSetsManagementBehavior.php

@@ -2,6 +2,7 @@
 
 namespace arogachev\tree\behaviors;
 
+use arogachev\tree\helpers\ArrayHelper;
 use yii\base\Behavior;
 
 class NestedSetsManagementBehavior extends Behavior
@@ -10,4 +11,64 @@ class NestedSetsManagementBehavior extends Behavior
      * @var string
      */
     public $isOpenedAttribute = 'is_opened';
+
+    /**
+     * @var boolean
+     */
+    public $saveState = false;
+
+
+    /**
+     * @return array
+     */
+    public function getHierarchicalArray()
+    {
+        /* @var $modelClass \yii\db\ActiveRecord */
+        $modelClass = $this->owner->className();
+        /* @var $baseModel \creocoder\nestedsets\NestedSetsBehavior */
+        $baseModel = new $this->owner;
+        $models = $modelClass::find()
+            ->orderBy($baseModel->leftAttribute)
+            ->all();
+
+        $depth = 0;
+        $c = 0;
+        $result = [];
+        $path = [];
+
+        foreach ($models as $model) {
+            $depthValue = $model->{$baseModel->depthAttribute};
+
+            if ($depthValue > $depth) {
+                $c = 0;
+            } elseif ($depthValue < $depth) {
+                for ($i = $depth - $depthValue; $i; $i--) {
+                    unset($path[count($path) - 1]);
+                }
+
+                $c = $path[count($path) - 1] + 1;
+            }
+
+            $path[$depthValue] = $c;
+
+            $value = [
+                'id' => $model->primaryKey,
+                'text' => $model->name,
+            ];
+
+            if ($this->saveState) {
+                $value['state'] = [
+                    'opened' => $model->{$model->isOpenedAttribute},
+                ];
+            }
+
+            ArrayHelper::saveByPath($result, $path, $value);
+
+            $depth = $depthValue;
+
+            $c++;
+        }
+
+        return $result;
+    }
 }

src/controllers/TreeController.php

@@ -2,7 +2,8 @@
 
 namespace arogachev\tree\controllers;
 
-use arogachev\tree\helpers\NestedSetsHelper;
+use arogachev\tree\behaviors\NestedSetsManagementBehavior;
+use creocoder\nestedsets\NestedSetsBehavior;
 use Yii;
 use yii\db\ActiveRecord;
 use yii\filters\ContentNegotiator;
@@ -50,7 +51,7 @@ public function beforeAction($action)
      */
     public function actionGetTree()
     {
-        return NestedSetsHelper::getHierarchicalArray(Yii::$app->request->get('modelClass'));
+        return $this->getModel(null, true)->getHierarchicalArray();
     }
 
     /**
@@ -149,7 +150,7 @@ public function actionInsertAfter()
     /**
      * @param null|string $paramName
      * @param boolean $createIfNotFound
-     * @return \yii\db\ActiveRecord|\creocoder\nestedsets\NestedSetsBehavior
+     * @return \yii\db\ActiveRecord|NestedSetsBehavior|NestedSetsManagementBehavior
      * @throws BadRequestHttpException
      * @throws NotFoundHttpException
      */

src/helpers/NestedSetsHelper.php

@@ -45,6 +45,21 @@ public function init()
             $this->options['id'] = $this->getId();
         }
 
+        $clientEvents = [
+            'create_node' => 'yii.tree.createNode',
+            'move_node' => 'yii.tree.moveNode',
+            'rename_node' => 'yii.tree.renameNode',
+            'delete_node' => 'yii.tree.deleteNode',
+        ];
+
+        $model = new $this->modelClass;
+        if ($model->saveState) {
+            $clientEvents = array_merge($clientEvents, [
+                'open_node' => 'yii.tree.openNode',
+                'close_node' => 'yii.tree.closeNode',
+            ]);
+        }
+
         $this->jsTreeOptions = ArrayHelper::merge([
             'clientOptions' => [
                 'core' => [
@@ -65,14 +80,7 @@ public function init()
                     ],
                 ],
             ],
-            'clientEvents' => [
-                'open_node' => 'yii.tree.openNode',
-                'close_node' => 'yii.tree.closeNode',
-                'create_node' => 'yii.tree.createNode',
-                'move_node' => 'yii.tree.moveNode',
-                'rename_node' => 'yii.tree.renameNode',
-                'delete_node' => 'yii.tree.deleteNode',
-            ],
+            'clientEvents' => $clientEvents,
         ], $this->jsTreeOptions);
     }