/
TreeWidget.php
127 lines (104 loc) · 2.6 KB
/
TreeWidget.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
<?php
namespace pendalf89\tree;
use yii\helpers\Html;
/**
* Class TreeWidget
*
* Displays models in tree view.
*
* @author Zabolotskikh Boris <zabolotskich@bk.ru>
* @package pendalf89\blog\widgets
*/
class TreeWidget extends \yii\base\Widget
{
/**
* @var string primary key
*/
public $id = 'id';
/**
* @var string parent id
*/
public $parent_id = 'parent_id';
/**
* @var array models
*/
public $models = [];
/**
* @var string callback function for disply content of each item. Have $model argument.
*/
public $value = '';
/**
* @var array container html options
*/
public $containerOptions = [];
/**
* @var bool container options already set or not
*/
private $containerOptionsIsSet = false;
/**
* Run the widget
* @return string tree
*/
public function run()
{
return $this->buildTree($this->createTreeArray());
}
/**
* Create tree array from models.
* @return array tree array
*/
private function createTreeArray()
{
$tree = [];
foreach($this->models as $model) {
$tree[] = [
'id' => $model->{$this->id},
'parent_id' => $model->{$this->parent_id},
'model' => $model,
];
}
return $tree;
}
/**
* Check node for child.
* @param array $rows
* @param int $id parent id
* @return bool
*/
private function hasChild($rows, $id)
{
foreach ($rows as $row) {
if ($row['parent_id'] == $id) {
return true;
}
}
return false;
}
/**
* Create tree.
* @param array $rows
* @param int $parent_id
* @return string tree
*/
private function buildTree($rows, $parent_id = 0)
{
$containerAttributes = '';
if (!$this->containerOptionsIsSet) {
$containerAttributes = Html::renderTagAttributes($this->containerOptions);
$this->containerOptionsIsSet = true;
}
$result = "<ul$containerAttributes>";
foreach ($rows as $row) {
if ($row['parent_id'] == $parent_id) {
$value = call_user_func_array($this->value, ['model' => $row['model']]);
$result .= "<li>$value";
if ($this->hasChild($rows, $row['id'])) {
$result.= $this->buildTree($rows, $row['id']);
}
$result .= '</li>';
}
}
$result .= '</ul>';
return $result;
}
}