util-helpers

import { __values } from "tslib";
import { isArray, isObject } from 'ut2';
/**
 * 内部实现
 *
 * @private
 * @param {Object[]} tree 树结构数据
 * @param {function} predicate 遍历每一项执行的函数，参数是当前遍历到的节点数据，如果返回 Truthy 将返回包含该节点的所有父级节点
 * @param {string} [childrenField] 子级字段名
 * @param {Object[]} [path=[]] 当前遍历路径
 * @returns {Object[]}
 */
function internalFindTreeSelect(tree, predicate, childrenField, path) {
    var e_1, _a;
    if (path === void 0) { path = []; }
    if (!isArray(tree)) {
        return [];
    }
    try {
        for (var tree_1 = __values(tree), tree_1_1 = tree_1.next(); !tree_1_1.done; tree_1_1 = tree_1.next()) {
            var item = tree_1_1.value;
            path.push(item);
            if (predicate(item)) {
                return path;
            }
            if (isObject(item)) {
                // @ts-ignore
                var childs = item[childrenField];
                if (isArray(childs) && childs.length > 0) {
                    var findChildren = internalFindTreeSelect(childs, predicate, childrenField, path);
                    if (findChildren.length > 0) {
                        return findChildren;
                    }
                }
            }
            path.pop();
        }
    }
    catch (e_1_1) { e_1 = { error: e_1_1 }; }
    finally {
        try {
            if (tree_1_1 && !tree_1_1.done && (_a = tree_1.return)) _a.call(tree_1);
        }
        finally { if (e_1) throw e_1.error; }
    }
    return [];
}
/**
 * 查找包含当前节点的所有父级节点
 *
 * @static
 * @alias module:Tree.findTreeSelect
 * @since 4.14.0
 * @param {Object[]} tree 树结构数据
 * @param {function} predicate 遍历每一项执行的函数，参数是当前遍历到的节点数据，如果返回 Truthy 将返回包含该节点的所有父级节点
 * @param {string} [childrenField='children'] 子级字段名
 * @returns {Object[]}
 * @example
 * const menus = [{ id: '1', name: '首页', code: 'trade', pid: null }, { id: '2', name: '交易管理', code: 'trade', pid: null, children: [{ id: '3', name: '交易查询', code: 'trade-1', pid: '2', children: [{ id: '4', name: '交易查询-查询操作', code: 'trade-1-1', pid: '3' }]}]}, { id: '5', name: '权限管理', code: 'authorization', pid: null, children: [{ id: '6', name: '角色管理', code: 'authorization-1', pid: '5' }, { id: '7', name: '用户管理', code: 'authorization-2', pid: '5' }]}];
 *
 * findTreeSelect(menus, item => item.id === '2');
 * // [{id:'2',name:'交易管理',code:'trade',pid:null,children:[{id:'3',name:'交易查询',code:'trade-1',pid:'2',children:[{id:'4',name:'交易查询-查询操作',code:'trade-1-1',pid:'3'}]}]}]
 *
 * findTreeSelect(menus, item => item.id === '7');
 * // [{id:'5',name:'权限管理',code:'authorization',pid:null,children:[{id:'6',name:'角色管理',code:'authorization-1',pid:'5'},{id:'7',name:'用户管理',code:'authorization-2',pid:'5'}]},{id:'7',name:'用户管理',code:'authorization-2',pid:'5'}]
 *
 * findTreeSelect(menus, item => item.id === 'not found');
 * // []
 */
function findTreeSelect(tree, predicate, childrenField) {
    if (childrenField === void 0) { childrenField = 'children'; }
    return internalFindTreeSelect(tree, predicate, childrenField);
}
export default findTreeSelect;
findTreeSelect.js
