ant-design-vue/components/vc-tree-select/demo/util.js

/* eslint no-loop-func: 0, no-console: 0 */

export function generateData (x = 3, y = 2, z = 1, gData = []) {
  // x：每一级下的节点总数。y：每级节点里有y个节点、存在子节点。z：树的level层级数（0表示一级）
  function _loop (_level, _preKey, _tns) {
    const preKey = _preKey || '0'
    const tns = _tns || gData

    const children = []
    for (let i = 0; i < x; i++) {
      const key = `${preKey}-${i}`
      tns.push({
        label: `${key}-label`,
        value: `${key}-value`,
        key,
        disabled: key === '0-0-0-1' || false,
      })
      if (i < y) {
        children.push(key)
      }
    }
    if (_level < 0) {
      return tns
    }
    const __level = _level - 1
    children.forEach((key, index) => {
      tns[index].children = []
      return _loop(__level, key, tns[index].children)
    })
  }
  _loop(z)
  return gData
}
export function calcTotal (x = 3, y = 2, z = 1) {
  /* eslint no-param-reassign:0*/
  const rec = (n) => n >= 0 ? x * Math.pow(y, n--) + rec(n) : 0
  return rec(z + 1)
}
console.log('总节点数（单个tree）：', calcTotal())
export const gData = generateData()

export function generateTreeNodes (treeNode) {
  const arr = []
  const key = treeNode.eventKey
  for (let i = 0; i < 3; i++) {
    arr.push({ label: `${key}-${i}-label`, value: `${key}-${i}-value`, key: `${key}-${i}` })
  }
  return arr
}

function setLeaf (treeData, curKey, level) {
  const loopLeaf = (data, lev) => {
    const l = lev - 1
    data.forEach((item) => {
      if ((item.key.length > curKey.length) ? item.key.indexOf(curKey) !== 0
        : curKey.indexOf(item.key) !== 0) {
        return
      }
      if (item.children) {
        loopLeaf(item.children, l)
      } else if (l < 1) {
        item.isLeaf = true
      }
    })
  }
  loopLeaf(treeData, level + 1)
}

export function getNewTreeData (treeData, curKey, child, level) {
  const loop = (data) => {
    if (level < 1 || curKey.length - 3 > level * 2) return
    data.forEach((item) => {
      if (curKey.indexOf(item.key) === 0) {
        if (item.children) {
          loop(item.children)
        } else {
          item.children = child
        }
      }
    })
  }
  loop(treeData)
  setLeaf(treeData, curKey, level)
}

function loopData (data, callback) {
  const loop = (d, level = 0) => {
    d.forEach((item, index) => {
      const pos = `${level}-${index}`
      if (item.children) {
        loop(item.children, pos)
      }
      callback(item, index, pos)
    })
  }
  loop(data)
}

function isPositionPrefix (smallPos, bigPos) {
  if (bigPos.length < smallPos.length) {
    return false
  }
  // attention: "0-0-1" "0-0-10"
  if ((bigPos.length > smallPos.length) && (bigPos.charAt(smallPos.length) !== '-')) {
    return false
  }
  return bigPos.substr(0, smallPos.length) === smallPos
}
// console.log(isPositionPrefix("0-1", "0-10-1"));

export function getFilterValue (val, sVal, delVal) {
  const allPos = []
  const delPos = []
  loopData(gData, (item, index, pos) => {
    if (sVal.indexOf(item.value) > -1) {
      allPos.push(pos)
    }
    if (delVal.indexOf(item.value) > -1) {
      delPos.push(pos)
    }
  })
  const newPos = []
  delPos.forEach((item) => {
    allPos.forEach((i) => {
      if (isPositionPrefix(item, i) || isPositionPrefix(i, item)) {
        // 过滤掉 父级节点 和 所有子节点。
        // 因为 node节点 不选时，其 父级节点 和 所有子节点 都不选。
        return
      }
      newPos.push(i)
    })
  })
  const newVal = []
  if (newPos.length) {
    loopData(gData, (item, index, pos) => {
      if (newPos.indexOf(pos) > -1) {
        newVal.push(item.value)
      }
    })
  }
  return newVal
}