ts/src/utils/tree.js

export const useTree = () => {
  return {
    filterTreeNodeByField,
    getAllKeys,
  }
}

function filterTreeNodeByField({ treeData, params, paramName, icon = '' }) {
  return treeData.reduce((acc, node) => {
    if (params.includes(node[paramName]) && !node.children) {
      acc.push({ ...node, icon })
    }
    if (node.children) {
      acc = acc.concat(
        filterTreeNodeByField({
          treeData: node.children,
          params,
          paramName,
          icon: node.icon,
        })
      )
    }
    return acc
  }, [])
}

function getAllKeys(treeData, key) {
  const data = 'value' in treeData ? treeData.value : treeData

  return data.reduce((acc, node) => {
    // console.log(node, '---')
    acc.push(node[key])
    if (node.children) {
      acc = acc.concat(getAllKeys(node.children, key))
    }
    return acc
  }, [])
}