You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

55 lines
1.7 KiB

/*
* Copyright [2022] [YangLiu of copyright http://www.maxkey.top]
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { NzFormatEmitEvent, NzTreeNode, NzTreeNodeOptions } from 'ng-zorro-antd/tree';
export class TreeNodes {
activated!: NzTreeNode;
request!: any[];
nodes!: any[];
checkable!: boolean;
checkedKeys!: any[];
selectedKeys!: any[];
_rootNode!: any;
constructor(checkable: boolean) {
this.checkable = checkable;
this.checkedKeys = [];
this.selectedKeys = [];
}
init(treeAttrs: any) {
this._rootNode = { title: treeAttrs.rootNode.title, key: treeAttrs.rootNode.key, expanded: true, isLeaf: false };
this.request = treeAttrs.nodes;
}
build(): any[] {
return this.buildTree(this._rootNode);
}
buildTree(rootNode: any): any[] {
let treeNodes: any[] = [];
for (let node of this.request) {
if (node.key != rootNode.key && node.parentKey == rootNode.key) {
let treeNode = { title: node.title, key: node.key, expanded: false, isLeaf: true };
this.buildTree(treeNode);
treeNodes.push(treeNode);
rootNode.isLeaf = false;
}
}
rootNode.children = treeNodes;
return [rootNode];
}
}