ModelTree.vue 2.54 KB
<template>
  <div class="m-4 mr-0 overflow-hidden bg-white">
    <BasicTree
      ref="treeRef"
      title="知识库"
      search
      toolbar
      treeWrapperClassName="h-[calc(100%-35px)] overflow-auto"
      :clickRowToExpand="false"
      :defaultExpandAll="true"
      :treeData="treeData"
      :fieldNames="{ key: 'businessId', title: 'modelName' }"
      @select="handleSelect"
    />
    <ModelDetailModal @register="registerModal" />
  </div>
</template>
<script lang="ts" setup>
  import { nextTick, onMounted, ref, unref } from 'vue';
  import { BasicTree, TreeActionType, TreeItem } from '@/components/Tree';
  import { Nullable } from '@vben/types';
  import { TreeData } from '@/views/metaModel/knowledgeModel/modelData';
  import ModelDetailModal from './modelDetailModal.vue';
  import { useModal } from '@/components/Modal';

  defineOptions({ name: 'DeptTree' });

  const emit = defineEmits(['select']);

  const treeData = ref<TreeItem[]>([]);
  const treeRef = ref<Nullable<TreeActionType>>(null);
  function getTree() {
    const tree = unref(treeRef);
    if (!tree) {
      throw new Error('tree is null!');
    }
    return tree;
  }
  const [registerModal, { openModal }] = useModal();
  async function fetch() {
    const data = TreeData;
    treeData.value = handleTree(data, 'businessId', undefined, undefined, undefined);
    await nextTick(() => {
      getTree().expandAll(true);
    });
  }

  function handleTree(data, id, parentId, children, rootId) {
    id = id || 'id';
    parentId = parentId || 'parentId';
    children = children || 'children';
    rootId =
      rootId ||
      Math.min.apply(
        Math,
        data.map((item) => {
          return item[parentId];
        }),
      ) ||
      0;
    // 对源数据深度克隆
    const cloneData = JSON.parse(JSON.stringify(data));
    // 循环所有项
    const treeData = cloneData.filter((father) => {
      const branchArr = cloneData.filter((child) => {
        // 返回每一项的子级数组
        return father[id] === child[parentId];
      });
      branchArr.length > 0 ? (father.children = branchArr) : '';
      // 返回第一层
      return father[parentId] === rootId;
    });
    return treeData !== '' ? treeData : data;
  }

  function handleSelect(keys) {
    emit('select', keys[0]);
    const selectedNode = getTree().getSelectedNode(keys[0]);
    if (keys[0] >= 200) {
      openModal(true, {
        isUpdate: false,
        modelName: selectedNode.modelName,
        type: selectedNode.type
      });
    }
  }

  onMounted(() => {
    fetch();
  });
</script>