├── .gitignore
├── src
    ├── Exceptions
    │   └── TreeException.php
    ├── Tree.php
    ├── Node.php
    └── Generator.php
├── composer.json
└── README.md


/.gitignore:
--------------------------------------------------------------------------------
1 | /vendor
2 | /.idea


--------------------------------------------------------------------------------
/src/Exceptions/TreeException.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace Betterde\Tree\Exceptions;
 4 | 
 5 | use Exception;
 6 | 
 7 | class TreeException extends Exception
 8 | {
 9 | 
10 | }


--------------------------------------------------------------------------------
/composer.json:
--------------------------------------------------------------------------------
 1 | {
 2 |     "name": "betterde/tree",
 3 |     "description": "Tree generator",
 4 |     "keywords": ["Tree", "Node", "generator"],
 5 |     "type": "library",
 6 |     "license": "MIT",
 7 |     "authors": [
 8 |         {
 9 |             "name": "betterde",
10 |             "email": "george@betterde.com"
11 |         }
12 |     ],
13 |     "require": {
14 |         "illuminate/support": "^5|^6|^7",
15 |         "illuminate/database": "^5|^6|^7"
16 |     },
17 |     "autoload": {
18 |         "psr-4": {
19 |             "Betterde\\Tree\\": "src/"
20 |         }
21 |     },
22 |     "minimum-stability": "dev"
23 | }
24 | 


--------------------------------------------------------------------------------
/src/Tree.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace Betterde\Tree;
 4 | 
 5 | use Illuminate\Database\Eloquent\Model;
 6 | 
 7 | class Tree extends Model
 8 | {
 9 |     /**
10 |      * 生成的树
11 |      *
12 |      * @var
13 |      * Date: 22/03/2018
14 |      * @author George
15 |      */
16 |     protected $tree;
17 | 
18 |     /**
19 |      * Tree constructor.
20 |      * @param array $attributes
21 |      */
22 |     public function __construct(array $attributes = [])
23 |     {
24 |         parent::__construct($attributes);
25 |         $this->tree = collect([]);
26 |     }
27 | 
28 |     /**
29 |      * 添加节点
30 |      *
31 |      * Date: 13/04/2018
32 |      * @author George
33 |      * @param Node $node
34 |      */
35 |     public function addNode(Node $node)
36 |     {
37 |         $this->tree->push($node);
38 |     }
39 | 
40 |     public function get()
41 |     {
42 |         return $this->tree;
43 |     }
44 | }
45 | 


--------------------------------------------------------------------------------
/src/Node.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace Betterde\Tree;
 4 | 
 5 | use Illuminate\Database\Eloquent\Model;
 6 | 
 7 | use Arr;
 8 | 
 9 | /**
10 |  * Date: 13/04/2018
11 |  * @author George
12 |  * @property *
13 |  * @package Betterde\Tree
14 |  */
15 | class Node extends Model
16 | {
17 |     protected $guarded = [];
18 | 
19 |     /**
20 |      * Node constructor.
21 |      * @param array $attributes
22 |      */
23 |     public function __construct($attributes = [])
24 |     {
25 |         parent::__construct($attributes);
26 |         $this->fill($attributes);
27 |     }
28 | 
29 |     /**
30 |      * 添加子节点
31 |      *
32 |      * Date: 13/04/2018
33 |      * @author George
34 |      * @param Node $node
35 |      * @param string $childrenKey
36 |      * 11/06/2020 修改 array_has 函数为 Arr::has
37 |      */
38 |     public function addChildren(Node $node, string $childrenKey = 'children')
39 |     {
40 |         if (Arr::has($this->attributes, $childrenKey)) {
41 |             $this->attributes[$childrenKey]->push($node);
42 |         } else {
43 |             $this->attributes[$childrenKey] = collect([]);
44 |             $this->attributes[$childrenKey]->push($node);
45 |         }
46 |     }
47 | }
48 | 


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | # Use guide
 2 | 
 3 | ```
 4 | use Betterde\Tree\Generator;
 5 | 
 6 | $menus = [
 7 |     [
 8 |         'id' => 1,
 9 |         'parent_id' => 0,
10 |         'label' => 'Dashboard',
11 |         'link' => '/dashboard',
12 |         'icon' => null,
13 |         'roles' => 'Admin,'
14 |     ],
15 |     [
16 |         'id' => 2,
17 |         'parent_id' => 1,
18 |         'label' => 'Fiance',
19 |         'link' => '/dashboard/finace',
20 |         'icon' => null,
21 |         'roles' => 'Admin,Fiance'
22 |     ],
23 |     [
24 |         'id' => 3,
25 |         'parent_id' => 1,
26 |         'label' => 'Operation',
27 |         'link' => '/dashboard/operation',
28 |         'icon' => null,
29 |         'roles' => 'Admin,Operation'
30 |     ],
31 | ];
32 | 
33 | $generator = new Generator();
34 | $tree = $generator->make($menus, 'id', 'parent_id', 'sub_menus', 0);
35 | ```
36 | # Generated result
37 | ```php
38 | [
39 |     [
40 |         'id' => 1,
41 |         'parent_id' => 0,
42 |         'label' => 'Dashboard',
43 |         'link' => '/dashboard',
44 |         'icon' => null,
45 |         'roles' => 'Admin,',
46 |         'sub_menus' => [
47 |             [
48 |                 'id' => 2,
49 |                 'parent_id' => 1,
50 |                 'label' => 'Fiance',
51 |                 'link' => '/dashboard/finace',
52 |                 'icon' => null,
53 |                 'roles' => 'Admin,Fiance'
54 |             ],
55 |             [
56 |                 'id' => 3,
57 |                 'parent_id' => 1,
58 |                 'label' => 'Operation',
59 |                 'link' => '/dashboard/operation',
60 |                 'icon' => null,
61 |                 'roles' => 'Admin,Operation'
62 |             ]
63 |         ]
64 |     ]
65 | ];
66 | ```
67 | 


--------------------------------------------------------------------------------
/src/Generator.php:
--------------------------------------------------------------------------------
  1 | <?php
  2 | 
  3 | namespace Betterde\Tree;
  4 | 
  5 | use Illuminate\Support\Collection;
  6 | use Illuminate\Database\Eloquent\Model;
  7 | use Betterde\Tree\Exceptions\TreeException;
  8 | 
  9 | /**
 10 |  * 树生成器
 11 |  *
 12 |  * Date: 13/04/2018
 13 |  * @author George
 14 |  * @package Betterde\Tree
 15 |  */
 16 | class Generator
 17 | {
 18 |     /**
 19 |      * 生成的树
 20 |      *
 21 |      * @var
 22 |      * Date: 22/03/2018
 23 |      * @author George
 24 |      */
 25 |     protected $tree;
 26 | 
 27 |     /**
 28 |      * 暂存区
 29 |      *
 30 |      * @var Collection
 31 |      * Date: 22/03/2018
 32 |      * @author George
 33 |      */
 34 |     protected $collection;
 35 | 
 36 |     /**
 37 |      * 定义主键名称
 38 |      *
 39 |      * @var string
 40 |      * Date: 22/03/2018
 41 |      * @author George
 42 |      */
 43 |     private $primaryKey = 'id';
 44 | 
 45 |     /**
 46 |      * 定义子键名称
 47 |      *
 48 |      * @var string
 49 |      * Date: 22/03/2018
 50 |      * @author George
 51 |      */
 52 |     private $childrenKey = 'children';
 53 | 
 54 |     /**
 55 |      * Generator constructor.
 56 |      */
 57 |     public function __construct()
 58 |     {
 59 |         $this->tree = new Tree();
 60 |     }
 61 | 
 62 | 
 63 |     /**
 64 |      * 生成树结构
 65 |      *
 66 |      * Date: 09/04/2018
 67 |      * @param $collection
 68 |      * @param string $primaryKey
 69 |      * @param string $parentKey
 70 |      * @param string $childrenKey
 71 |      * @param string $topvalue
 72 |      * @return Collection|mixed
 73 |      * @throws TreeException
 74 |      * @author George
 75 |      */
 76 |     public function make($collection, string $primaryKey = 'id', string $parentKey = 'parent_id', string $childrenKey = 'children', string $topvalue = "")
 77 |     {
 78 |         $collection = $this->transform($collection);
 79 | 
 80 |         $this->primaryKey = $primaryKey;
 81 |         $this->childrenKey = $childrenKey;
 82 | 
 83 |         $this->collection = $collection->groupBy($parentKey);
 84 | 
 85 |         if ($this->addTopNode($topvalue)) {
 86 |             return $collection;
 87 |         }
 88 | 
 89 |         if (count($this->collection) > 0) {
 90 |             foreach ($this->tree->get() as $key => $item) {
 91 |                 $this->recursion($item);
 92 |             }
 93 |         }
 94 | 
 95 |         return $this->tree->get();
 96 |     }
 97 | 
 98 |     /**
 99 |      * 添加顶级节点
100 |      *
101 |      * Date: 13/04/2018
102 |      * @author George
103 |      * @param $topvalue
104 |      * @return bool
105 |      * @throws TreeException
106 |      */
107 |     public function addTopNode($topvalue)
108 |     {
109 |         $nodes = $this->collection->pull($topvalue);
110 | 
111 |         if (empty($nodes)) {
112 |             return true;
113 |         }
114 |         foreach ($nodes as $item) {
115 |             $node = new Node($this->isArray($item));
116 |             $this->tree->addNode($node);
117 |         }
118 |         return false;
119 |     }
120 | 
121 |     /**
122 |      * 递归生成树结构
123 |      *
124 |      * Date: 22/03/2018
125 |      * @author George
126 |      * @param $item
127 |      */
128 |     private function recursion(Node $item)
129 |     {
130 |         if ($this->collection->has($item->{$this->primaryKey})) {
131 |             $this->collection->pull($item->{$this->primaryKey})->map(function ($temp) use ($item) {
132 |                 $node = new Node($this->isArray($temp));
133 |                 $item->addChildren($node, $this->childrenKey);
134 |                 $this->recursion($node);
135 |             });
136 |         }
137 |     }
138 | 
139 |     /**
140 |      * 转换数据类型
141 |      *
142 |      * Date: 09/04/2018
143 |      * @author George
144 |      * @param $collection
145 |      * @return Collection
146 |      * @throws TreeException
147 |      */
148 |     private function transform($collection)
149 |     {
150 |         if ($collection instanceof Collection) {
151 |             return $collection;
152 |         }
153 | 
154 |         if (is_array($collection)) {
155 |             return collect($collection);
156 |         }
157 | 
158 |         throw new TreeException('数据类型错误');
159 |     }
160 | 
161 |     /**
162 |      * Date: 13/04/2018
163 |      * @author George
164 |      * @param $item
165 |      * @return array
166 |      * @throws TreeException
167 |      */
168 |     public function isArray($item)
169 |     {
170 |         if (is_array($item)) {
171 |             return $item;
172 |         }
173 | 
174 |         if ($item instanceof Model) {
175 |             return $item->toArray();
176 |         }
177 |         throw new TreeException('数据类型错误');
178 |     }
179 | }
180 | 


--------------------------------------------------------------------------------