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.
53 lines
1.7 KiB
53 lines
1.7 KiB
5 months ago
|
var helperCreateTreeFunc = require('./helperCreateTreeFunc')
|
||
|
|
|
||
|
|
var arrayEach = require('./arrayEach')
|
||
|
|
|
||
|
|
var assign = require('./assign')
|
||
|
|
|
||
|
|
function searchTreeItem (matchParent, parent, obj, iterate, context, path, node, parseChildren, opts) {
|
||
|
|
var paths, nodes, rest, isMatch, hasChild
|
||
|
|
var rests = []
|
||
|
|
var hasOriginal = opts.original
|
||
|
|
var sourceData = opts.data
|
||
|
|
var mapChildren = opts.mapChildren || parseChildren
|
||
|
|
var isEvery = opts.isEvery
|
||
|
|
arrayEach(obj, function (item, index) {
|
||
|
|
paths = path.concat(['' + index])
|
||
|
|
nodes = node.concat([item])
|
||
|
|
isMatch = (matchParent && !isEvery) || iterate.call(context, item, index, obj, paths, parent, nodes)
|
||
|
|
hasChild = parseChildren && item[parseChildren]
|
||
|
|
if (isMatch || hasChild) {
|
||
|
|
if (hasOriginal) {
|
||
|
|
rest = item
|
||
|
|
} else {
|
||
|
|
rest = assign({}, item)
|
||
|
|
if (sourceData) {
|
||
|
|
rest[sourceData] = item
|
||
|
|
}
|
||
|
|
}
|
||
|
|
rest[mapChildren] = searchTreeItem(isMatch, item, item[parseChildren], iterate, context, paths, nodes, parseChildren, opts)
|
||
|
|
if (isMatch || rest[mapChildren].length) {
|
||
|
|
rests.push(rest)
|
||
|
|
}
|
||
|
|
} else if (isMatch) {
|
||
|
|
rests.push(rest)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
return rests
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 从树结构中根据回调查找数据
|
||
|
|
*
|
||
|
|
* @param {Object} obj 对象/数组
|
||
|
|
* @param {Function} iterate(item, index, items, path, parent, nodes) 回调
|
||
|
|
* @param {Object} options {children: 'children'}
|
||
|
|
* @param {Object} context 上下文
|
||
|
|
* @return {Array}
|
||
|
|
*/
|
||
|
|
var searchTree = helperCreateTreeFunc(function (parent, obj, iterate, context, path, nodes, parseChildren, opts) {
|
||
|
|
return searchTreeItem(0, parent, obj, iterate, context, path, nodes, parseChildren, opts)
|
||
|
|
})
|
||
|
|
|
||
|
|
module.exports = searchTree
|