/**
* Returns a tree data as array
*
* @param bool|Model|NestedSetBehavior $root Parent item for a tree data (false, if from root item)
* @param bool|integer $maxLevel Max level for a tree data (false for no limit)
* @param string $keyAttribute Attribute of model for array key
* @param string[] $extraAttributes Additional attribute names for include to tree
* @return array
*/
public function tree($root = false, $maxLevel = false, $keyAttribute = 'id', array $extraAttributes = [])
{
/** @var NestedSetBehavior|Model $items */
$tree = [];
if ($root === false) {
$ownerClass = $this->owner->modelClass;
$items = $ownerClass::find()->roots()->all();
} else {
if (!$maxLevel || $root->level <= $maxLevel) {
$items = $root->children()->all();
} else {
return $tree;
}
}
foreach ($items as $item) {
$tree[$item->{$keyAttribute}] = ['id' => $item->id, 'name' => $item->{$item->titleAttribute}, 'children' => !$maxLevel || $item->level < $maxLevel ? $this->tree($item, $maxLevel, $keyAttribute, $extraAttributes) : null];
if (count($extraAttributes) > 0) {
$tree[$item->{$keyAttribute}] = ArrayHelper::merge($tree[$item->{$keyAttribute}], $item->toArray($extraAttributes));
}
}
return $tree;
}
