{"version":3,"file":"tree.mjs","sources":["../../../../../../../packages/components/table/src/store/tree.ts"],"sourcesContent":["// @ts-nocheck\nimport { computed, getCurrentInstance, ref, unref, watch } from 'vue'\nimport { getRowIdentity, walkTreeNode } from '../util'\n\nimport type { WatcherPropsData } from '.'\nimport type { Table, TableProps } from '../table/defaults'\n\nfunction useTree<T>(watcherData: WatcherPropsData<T>) {\n  const expandRowKeys = ref<string[]>([])\n  const treeData = ref<unknown>({})\n  const indent = ref(16)\n  const lazy = ref(false)\n  const lazyTreeNodeMap = ref({})\n  const lazyColumnIdentifier = ref('hasChildren')\n  const childrenColumnName = ref('children')\n  const instance = getCurrentInstance() as Table<T>\n  const normalizedData = computed(() => {\n    if (!watcherData.rowKey.value) return {}\n    const data = watcherData.data.value || []\n    return normalize(data)\n  })\n  const normalizedLazyNode = computed(() => {\n    const rowKey = watcherData.rowKey.value\n    const keys = Object.keys(lazyTreeNodeMap.value)\n    const res = {}\n    if (!keys.length) return res\n    keys.forEach((key) => {\n      if (lazyTreeNodeMap.value[key].length) {\n        const item = { children: [] }\n        lazyTreeNodeMap.value[key].forEach((row) => {\n          const currentRowKey = getRowIdentity(row, rowKey)\n          item.children.push(currentRowKey)\n          if (row[lazyColumnIdentifier.value] && !res[currentRowKey]) {\n            res[currentRowKey] = { children: [] }\n          }\n        })\n        res[key] = item\n      }\n    })\n    return res\n  })\n\n  const normalize = (data) => {\n    const rowKey = watcherData.rowKey.value\n    const res = {}\n    walkTreeNode(\n      data,\n      (parent, children, level) => {\n        const parentId = getRowIdentity(parent, rowKey)\n        if (Array.isArray(children)) {\n          res[parentId] = {\n            children: children.map((row) => getRowIdentity(row, rowKey)),\n            level,\n          }\n        } else if (lazy.value) {\n          res[parentId] = {\n            children: [],\n            lazy: true,\n            level,\n          }\n        }\n      },\n      childrenColumnName.value,\n      lazyColumnIdentifier.value\n    )\n    return res\n  }\n\n  const updateTreeData = (\n    ifChangeExpandRowKeys = false,\n    ifExpandAll = instance.store?.states.defaultExpandAll.value\n  ) => {\n    const nested = normalizedData.value\n    const normalizedLazyNode_ = normalizedLazyNode.value\n    const keys = Object.keys(nested)\n    const newTreeData = {}\n    if (keys.length) {\n      const oldTreeData = unref(treeData)\n      const rootLazyRowKeys = []\n      const getExpanded = (oldValue, key) => {\n        if (ifChangeExpandRowKeys) {\n          if (expandRowKeys.value) {\n            return ifExpandAll || expandRowKeys.value.includes(key)\n          } else {\n            return !!(ifExpandAll || oldValue?.expanded)\n          }\n        } else {\n          const included =\n            ifExpandAll ||\n            (expandRowKeys.value && expandRowKeys.value.includes(key))\n          return !!(oldValue?.expanded || included)\n        }\n      }\n      keys.forEach((key) => {\n        const oldValue = oldTreeData[key]\n        const newValue = { ...nested[key] }\n        newValue.expanded = getExpanded(oldValue, key)\n        if (newValue.lazy) {\n          const { loaded = false, loading = false } = oldValue || {}\n          newValue.loaded = !!loaded\n          newValue.loading = !!loading\n          rootLazyRowKeys.push(key)\n        }\n        newTreeData[key] = newValue\n      })\n      const lazyKeys = Object.keys(normalizedLazyNode_)\n      if (lazy.value && lazyKeys.length && rootLazyRowKeys.length) {\n        lazyKeys.forEach((key) => {\n          const oldValue = oldTreeData[key]\n          const lazyNodeChildren = normalizedLazyNode_[key].children\n          if (rootLazyRowKeys.includes(key)) {\n            if (newTreeData[key].children.length !== 0) {\n              throw new Error('[BTable]children must be an empty array.')\n            }\n            newTreeData[key].children = lazyNodeChildren\n          } else {\n            const { loaded = false, loading = false } = oldValue || {}\n            newTreeData[key] = {\n              lazy: true,\n              loaded: !!loaded,\n              loading: !!loading,\n              expanded: getExpanded(oldValue, key),\n              children: lazyNodeChildren,\n              level: '',\n            }\n          }\n        })\n      }\n    }\n    treeData.value = newTreeData\n    instance.store?.updateTableScrollY()\n  }\n\n  watch(\n    () => expandRowKeys.value,\n    () => {\n      updateTreeData(true)\n    }\n  )\n\n  watch(\n    () => normalizedData.value,\n    () => {\n      updateTreeData()\n    }\n  )\n  watch(\n    () => normalizedLazyNode.value,\n    () => {\n      updateTreeData()\n    }\n  )\n\n  const updateTreeExpandKeys = (value: string[]) => {\n    expandRowKeys.value = value\n    updateTreeData()\n  }\n\n  const toggleTreeExpansion = (row: T, expanded?: boolean) => {\n    instance.store.assertRowKey()\n\n    const rowKey = watcherData.rowKey.value\n    const id = getRowIdentity(row, rowKey)\n    const data = id && treeData.value[id]\n    if (id && data && 'expanded' in data) {\n      const oldExpanded = data.expanded\n      expanded = typeof expanded === 'undefined' ? !data.expanded : expanded\n      treeData.value[id].expanded = expanded\n      if (oldExpanded !== expanded) {\n        instance.emit('expand-change', row, expanded)\n      }\n      instance.store.updateTableScrollY()\n    }\n  }\n\n  const loadOrToggle = (row) => {\n    instance.store.assertRowKey()\n    const rowKey = watcherData.rowKey.value\n    const id = getRowIdentity(row, rowKey)\n    const data = treeData.value[id]\n    if (lazy.value && data && 'loaded' in data && !data.loaded) {\n      loadData(row, id, data)\n    } else {\n      toggleTreeExpansion(row, undefined)\n    }\n  }\n\n  const loadData = (row: T, key: string, treeNode) => {\n    const { load } = instance.props as unknown as TableProps<T>\n    if (load && !treeData.value[key].loaded) {\n      treeData.value[key].loading = true\n      load(row, treeNode, (data) => {\n        if (!Array.isArray(data)) {\n          throw new TypeError('[BTable] data must be an array')\n        }\n        treeData.value[key].loading = false\n        treeData.value[key].loaded = true\n        treeData.value[key].expanded = true\n        if (data.length) {\n          lazyTreeNodeMap.value[key] = data\n        }\n        instance.emit('expand-change', row, true)\n      })\n    }\n  }\n\n  return {\n    loadData,\n    loadOrToggle,\n    toggleTreeExpansion,\n    updateTreeExpandKeys,\n    updateTreeData,\n    normalize,\n    states: {\n      expandRowKeys,\n      treeData,\n      indent,\n      lazy,\n      lazyTreeNodeMap,\n      lazyColumnIdentifier,\n      childrenColumnName,\n    },\n  }\n}\n\nexport default useTree\n"],"names":[],"mappings":";;;AAOA,SAAS,QAAW,WAAkC,EAAA;AACpD,EAAM,MAAA,aAAA,GAAgB,GAAc,CAAA,EAAE,CAAA,CAAA;AACtC,EAAM,MAAA,QAAA,GAAW,GAAa,CAAA,EAAE,CAAA,CAAA;AAChC,EAAM,MAAA,MAAA,GAAS,IAAI,EAAE,CAAA,CAAA;AACrB,EAAM,MAAA,IAAA,GAAO,IAAI,KAAK,CAAA,CAAA;AACtB,EAAM,MAAA,eAAA,GAAkB,GAAI,CAAA,EAAE,CAAA,CAAA;AAC9B,EAAM,MAAA,oBAAA,GAAuB,IAAI,aAAa,CAAA,CAAA;AAC9C,EAAM,MAAA,kBAAA,GAAqB,IAAI,UAAU,CAAA,CAAA;AACzC,EAAA,MAAM,WAAW,kBAAmB,EAAA,CAAA;AACpC,EAAM,MAAA,cAAA,GAAiB,SAAS,MAAM;AACpC,IAAI,IAAA,CAAC,YAAY,MAAO,CAAA,KAAA;AAAO,MAAA,OAAO,EAAC,CAAA;AACvC,IAAA,MAAM,IAAO,GAAA,WAAA,CAAY,IAAK,CAAA,KAAA,IAAS,EAAC,CAAA;AACxC,IAAA,OAAO,UAAU,IAAI,CAAA,CAAA;AAAA,GACtB,CAAA,CAAA;AACD,EAAM,MAAA,kBAAA,GAAqB,SAAS,MAAM;AACxC,IAAM,MAAA,MAAA,GAAS,YAAY,MAAO,CAAA,KAAA,CAAA;AAClC,IAAA,MAAM,IAAO,GAAA,MAAA,CAAO,IAAK,CAAA,eAAA,CAAgB,KAAK,CAAA,CAAA;AAC9C,IAAA,MAAM,MAAM,EAAC,CAAA;AACb,IAAA,IAAI,CAAC,IAAK,CAAA,MAAA;AAAQ,MAAO,OAAA,GAAA,CAAA;AACzB,IAAK,IAAA,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACpB,MAAI,IAAA,eAAA,CAAgB,KAAM,CAAA,GAAA,CAAA,CAAK,MAAQ,EAAA;AACrC,QAAA,MAAM,IAAO,GAAA,EAAE,QAAU,EAAA,EAAG,EAAA,CAAA;AAC5B,QAAA,eAAA,CAAgB,KAAM,CAAA,GAAA,CAAA,CAAK,OAAQ,CAAA,CAAC,GAAQ,KAAA;AAC1C,UAAM,MAAA,aAAA,GAAgB,cAAe,CAAA,GAAA,EAAK,MAAM,CAAA,CAAA;AAChD,UAAK,IAAA,CAAA,QAAA,CAAS,KAAK,aAAa,CAAA,CAAA;AAChC,UAAA,IAAI,GAAI,CAAA,oBAAA,CAAqB,KAAU,CAAA,IAAA,CAAC,IAAI,aAAgB,CAAA,EAAA;AAC1D,YAAA,GAAA,CAAI,aAAiB,CAAA,GAAA,EAAE,QAAU,EAAA,EAAG,EAAA,CAAA;AAAA,WACtC;AAAA,SACD,CAAA,CAAA;AACD,QAAA,GAAA,CAAI,GAAO,CAAA,GAAA,IAAA,CAAA;AAAA,OACb;AAAA,KACD,CAAA,CAAA;AACD,IAAO,OAAA,GAAA,CAAA;AAAA,GACR,CAAA,CAAA;AAED,EAAM,MAAA,SAAA,GAAY,CAAC,IAAS,KAAA;AAC1B,IAAM,MAAA,MAAA,GAAS,YAAY,MAAO,CAAA,KAAA,CAAA;AAClC,IAAA,MAAM,MAAM,EAAC,CAAA;AACb,IAAA,YAAA;AAAA,MACE,IAAA;AAAA,MACA,CAAC,MAAQ,EAAA,QAAA,EAAU,KAAU,KAAA;AAC3B,QAAM,MAAA,QAAA,GAAW,cAAe,CAAA,MAAA,EAAQ,MAAM,CAAA,CAAA;AAC9C,QAAI,IAAA,KAAA,CAAM,OAAQ,CAAA,QAAQ,CAAG,EAAA;AAC3B,UAAA,GAAA,CAAI,QAAY,CAAA,GAAA;AAAA,YACd,QAAA,EAAU,SAAS,GAAI,CAAA,CAAC,QAAQ,cAAe,CAAA,GAAA,EAAK,MAAM,CAAC,CAAA;AAAA,YAC3D,KAAA;AAAA,WACF,CAAA;AAAA,SACF,MAAA,IAAW,KAAK,KAAO,EAAA;AACrB,UAAA,GAAA,CAAI,QAAY,CAAA,GAAA;AAAA,YACd,UAAU,EAAC;AAAA,YACX,IAAM,EAAA,IAAA;AAAA,YACN,KAAA;AAAA,WACF,CAAA;AAAA,SACF;AAAA,OACF;AAAA,MACA,kBAAmB,CAAA,KAAA;AAAA,MACnB,oBAAqB,CAAA,KAAA;AAAA,KACvB,CAAA;AACA,IAAO,OAAA,GAAA,CAAA;AAAA,GACT,CAAA;AAEA,EAAM,MAAA,cAAA,GAAiB,CACrB,qBAAwB,GAAA,KAAA,EACxB,cAAc,QAAS,CAAA,KAAA,EAAO,MAAO,CAAA,gBAAA,CAAiB,KACnD,KAAA;AACH,IAAA,MAAM,SAAS,cAAe,CAAA,KAAA,CAAA;AAC9B,IAAA,MAAM,sBAAsB,kBAAmB,CAAA,KAAA,CAAA;AAC/C,IAAM,MAAA,IAAA,GAAO,MAAO,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAC/B,IAAA,MAAM,cAAc,EAAC,CAAA;AACrB,IAAA,IAAI,KAAK,MAAQ,EAAA;AACf,MAAM,MAAA,WAAA,GAAc,MAAM,QAAQ,CAAA,CAAA;AAClC,MAAA,MAAM,kBAAkB,EAAC,CAAA;AACzB,MAAM,MAAA,WAAA,GAAc,CAAC,QAAA,EAAU,GAAQ,KAAA;AACrC,QAAA,IAAI,qBAAuB,EAAA;AACzB,UAAA,IAAI,cAAc,KAAO,EAAA;AACvB,YAAA,OAAO,WAAe,IAAA,aAAA,CAAc,KAAM,CAAA,QAAA,CAAS,GAAG,CAAA,CAAA;AAAA,WACjD,MAAA;AACL,YAAO,OAAA,CAAC,EAAE,WAAA,IAAe,QAAU,EAAA,QAAA,CAAA,CAAA;AAAA,WACrC;AAAA,SACK,MAAA;AACL,UAAA,MAAM,WACJ,WACC,IAAA,aAAA,CAAc,SAAS,aAAc,CAAA,KAAA,CAAM,SAAS,GAAG,CAAA,CAAA;AAC1D,UAAO,OAAA,CAAC,EAAE,QAAA,EAAU,QAAY,IAAA,QAAA,CAAA,CAAA;AAAA,SAClC;AAAA,OACF,CAAA;AACA,MAAK,IAAA,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACpB,QAAA,MAAM,WAAW,WAAY,CAAA,GAAA,CAAA,CAAA;AAC7B,QAAA,MAAM,QAAW,GAAA,EAAE,GAAG,MAAA,CAAO,GAAK,CAAA,EAAA,CAAA;AAClC,QAAS,QAAA,CAAA,QAAA,GAAW,WAAY,CAAA,QAAA,EAAU,GAAG,CAAA,CAAA;AAC7C,QAAA,IAAI,SAAS,IAAM,EAAA;AACjB,UAAA,MAAM,EAAE,MAAS,GAAA,KAAA,EAAO,UAAU,KAAM,EAAA,GAAI,YAAY,EAAC,CAAA;AACzD,UAAS,QAAA,CAAA,MAAA,GAAS,CAAC,CAAC,MAAA,CAAA;AACpB,UAAS,QAAA,CAAA,OAAA,GAAU,CAAC,CAAC,OAAA,CAAA;AACrB,UAAA,eAAA,CAAgB,KAAK,GAAG,CAAA,CAAA;AAAA,SAC1B;AACA,QAAA,WAAA,CAAY,GAAO,CAAA,GAAA,QAAA,CAAA;AAAA,OACpB,CAAA,CAAA;AACD,MAAM,MAAA,QAAA,GAAW,MAAO,CAAA,IAAA,CAAK,mBAAmB,CAAA,CAAA;AAChD,MAAA,IAAI,IAAK,CAAA,KAAA,IAAS,QAAS,CAAA,MAAA,IAAU,gBAAgB,MAAQ,EAAA;AAC3D,QAAS,QAAA,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACxB,UAAA,MAAM,WAAW,WAAY,CAAA,GAAA,CAAA,CAAA;AAC7B,UAAM,MAAA,gBAAA,GAAmB,oBAAoB,GAAK,CAAA,CAAA,QAAA,CAAA;AAClD,UAAI,IAAA,eAAA,CAAgB,QAAS,CAAA,GAAG,CAAG,EAAA;AACjC,YAAA,IAAI,WAAY,CAAA,GAAA,CAAA,CAAK,QAAS,CAAA,MAAA,KAAW,CAAG,EAAA;AAC1C,cAAM,MAAA,IAAI,MAAM,0CAA0C,CAAA,CAAA;AAAA,aAC5D;AACA,YAAA,WAAA,CAAY,KAAK,QAAW,GAAA,gBAAA,CAAA;AAAA,WACvB,MAAA;AACL,YAAA,MAAM,EAAE,MAAS,GAAA,KAAA,EAAO,UAAU,KAAM,EAAA,GAAI,YAAY,EAAC,CAAA;AACzD,YAAA,WAAA,CAAY,GAAO,CAAA,GAAA;AAAA,cACjB,IAAM,EAAA,IAAA;AAAA,cACN,MAAA,EAAQ,CAAC,CAAC,MAAA;AAAA,cACV,OAAA,EAAS,CAAC,CAAC,OAAA;AAAA,cACX,QAAA,EAAU,WAAY,CAAA,QAAA,EAAU,GAAG,CAAA;AAAA,cACnC,QAAU,EAAA,gBAAA;AAAA,cACV,KAAO,EAAA,EAAA;AAAA,aACT,CAAA;AAAA,WACF;AAAA,SACD,CAAA,CAAA;AAAA,OACH;AAAA,KACF;AACA,IAAA,QAAA,CAAS,KAAQ,GAAA,WAAA,CAAA;AACjB,IAAA,QAAA,CAAS,OAAO,kBAAmB,EAAA,CAAA;AAAA,GACrC,CAAA;AAEA,EAAA,KAAA;AAAA,IACE,MAAM,aAAc,CAAA,KAAA;AAAA,IACpB,MAAM;AACJ,MAAA,cAAA,CAAe,IAAI,CAAA,CAAA;AAAA,KACrB;AAAA,GACF,CAAA;AAEA,EAAA,KAAA;AAAA,IACE,MAAM,cAAe,CAAA,KAAA;AAAA,IACrB,MAAM;AACJ,MAAe,cAAA,EAAA,CAAA;AAAA,KACjB;AAAA,GACF,CAAA;AACA,EAAA,KAAA;AAAA,IACE,MAAM,kBAAmB,CAAA,KAAA;AAAA,IACzB,MAAM;AACJ,MAAe,cAAA,EAAA,CAAA;AAAA,KACjB;AAAA,GACF,CAAA;AAEA,EAAM,MAAA,oBAAA,GAAuB,CAAC,KAAoB,KAAA;AAChD,IAAA,aAAA,CAAc,KAAQ,GAAA,KAAA,CAAA;AACtB,IAAe,cAAA,EAAA,CAAA;AAAA,GACjB,CAAA;AAEA,EAAM,MAAA,mBAAA,GAAsB,CAAC,GAAA,EAAQ,QAAuB,KAAA;AAC1D,IAAA,QAAA,CAAS,MAAM,YAAa,EAAA,CAAA;AAE5B,IAAM,MAAA,MAAA,GAAS,YAAY,MAAO,CAAA,KAAA,CAAA;AAClC,IAAM,MAAA,EAAA,GAAK,cAAe,CAAA,GAAA,EAAK,MAAM,CAAA,CAAA;AACrC,IAAM,MAAA,IAAA,GAAO,EAAM,IAAA,QAAA,CAAS,KAAM,CAAA,EAAA,CAAA,CAAA;AAClC,IAAI,IAAA,EAAA,IAAM,IAAQ,IAAA,UAAA,IAAc,IAAM,EAAA;AACpC,MAAA,MAAM,cAAc,IAAK,CAAA,QAAA,CAAA;AACzB,MAAA,QAAA,GAAW,OAAO,QAAA,KAAa,WAAc,GAAA,CAAC,KAAK,QAAW,GAAA,QAAA,CAAA;AAC9D,MAAS,QAAA,CAAA,KAAA,CAAM,IAAI,QAAW,GAAA,QAAA,CAAA;AAC9B,MAAA,IAAI,gBAAgB,QAAU,EAAA;AAC5B,QAAS,QAAA,CAAA,IAAA,CAAK,eAAiB,EAAA,GAAA,EAAK,QAAQ,CAAA,CAAA;AAAA,OAC9C;AACA,MAAA,QAAA,CAAS,MAAM,kBAAmB,EAAA,CAAA;AAAA,KACpC;AAAA,GACF,CAAA;AAEA,EAAM,MAAA,YAAA,GAAe,CAAC,GAAQ,KAAA;AAC5B,IAAA,QAAA,CAAS,MAAM,YAAa,EAAA,CAAA;AAC5B,IAAM,MAAA,MAAA,GAAS,YAAY,MAAO,CAAA,KAAA,CAAA;AAClC,IAAM,MAAA,EAAA,GAAK,cAAe,CAAA,GAAA,EAAK,MAAM,CAAA,CAAA;AACrC,IAAM,MAAA,IAAA,GAAO,SAAS,KAAM,CAAA,EAAA,CAAA,CAAA;AAC5B,IAAA,IAAI,KAAK,KAAS,IAAA,IAAA,IAAQ,YAAY,IAAQ,IAAA,CAAC,KAAK,MAAQ,EAAA;AAC1D,MAAS,QAAA,CAAA,GAAA,EAAK,IAAI,IAAI,CAAA,CAAA;AAAA,KACjB,MAAA;AACL,MAAA,mBAAA,CAAoB,KAAK,KAAS,CAAA,CAAA,CAAA;AAAA,KACpC;AAAA,GACF,CAAA;AAEA,EAAA,MAAM,QAAW,GAAA,CAAC,GAAQ,EAAA,GAAA,EAAa,QAAa,KAAA;AAClD,IAAM,MAAA,EAAE,IAAK,EAAA,GAAI,QAAS,CAAA,KAAA,CAAA;AAC1B,IAAA,IAAI,IAAQ,IAAA,CAAC,QAAS,CAAA,KAAA,CAAM,KAAK,MAAQ,EAAA;AACvC,MAAS,QAAA,CAAA,KAAA,CAAM,KAAK,OAAU,GAAA,IAAA,CAAA;AAC9B,MAAK,IAAA,CAAA,GAAA,EAAK,QAAU,EAAA,CAAC,IAAS,KAAA;AAC5B,QAAA,IAAI,CAAC,KAAA,CAAM,OAAQ,CAAA,IAAI,CAAG,EAAA;AACxB,UAAM,MAAA,IAAI,UAAU,gCAAgC,CAAA,CAAA;AAAA,SACtD;AACA,QAAS,QAAA,CAAA,KAAA,CAAM,KAAK,OAAU,GAAA,KAAA,CAAA;AAC9B,QAAS,QAAA,CAAA,KAAA,CAAM,KAAK,MAAS,GAAA,IAAA,CAAA;AAC7B,QAAS,QAAA,CAAA,KAAA,CAAM,KAAK,QAAW,GAAA,IAAA,CAAA;AAC/B,QAAA,IAAI,KAAK,MAAQ,EAAA;AACf,UAAA,eAAA,CAAgB,MAAM,GAAO,CAAA,GAAA,IAAA,CAAA;AAAA,SAC/B;AACA,QAAS,QAAA,CAAA,IAAA,CAAK,eAAiB,EAAA,GAAA,EAAK,IAAI,CAAA,CAAA;AAAA,OACzC,CAAA,CAAA;AAAA,KACH;AAAA,GACF,CAAA;AAEA,EAAO,OAAA;AAAA,IACL,QAAA;AAAA,IACA,YAAA;AAAA,IACA,mBAAA;AAAA,IACA,oBAAA;AAAA,IACA,cAAA;AAAA,IACA,SAAA;AAAA,IACA,MAAQ,EAAA;AAAA,MACN,aAAA;AAAA,MACA,QAAA;AAAA,MACA,MAAA;AAAA,MACA,IAAA;AAAA,MACA,eAAA;AAAA,MACA,oBAAA;AAAA,MACA,kBAAA;AAAA,KACF;AAAA,GACF,CAAA;AACF;;;;"}