gaoluyang
2026-06-24 712aa51536236d43e87273e4ce45ac5691dffad8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
<script lang="ts" setup>
import type { DataNode } from 'ant-design-vue/es/tree';
 
import type { SystemDeptApi } from '#/api/system/dept';
 
import { ref } from 'vue';
 
import { useVbenModal } from '..\..\..\..\packages\effects\common-ui\src';
import { handleTree } from '..\..\..\..\packages\utils\src';
 
import { Card, Col, Row, Tree } from 'ant-design-vue';
 
import { getSimpleDeptList } from '#/api/system/dept';
 
defineOptions({ name: 'DeptSelectModal' });
 
const props = withDefaults(
  defineProps<{
    cancelText?: string; // 取消按钮文本
    checkStrictly?: boolean; // checkable 状态下节点选择完全受控
    confirmText?: string; // 确认按钮文本
    multiple?: boolean; // 是否支持多选
    title?: string; // 标题
  }>(),
  {
    cancelText: '取消',
    checkStrictly: false,
    confirmText: '确认',
    multiple: true,
    title: '部门选择',
  },
);
 
const emit = defineEmits<{
  confirm: [deptList: SystemDeptApi.Dept[]];
}>();
 
type checkedKeys = number[] | { checked: number[]; halfChecked: number[] };
const deptTree = ref<DataNode[]>([]); // 部门树形结构
const selectedDeptIds = ref<checkedKeys>([]); // 选中的部门 ID 列表
const deptData = ref<SystemDeptApi.Dept[]>([]); // 部门数据
 
const [Modal, modalApi] = useVbenModal({
  async onConfirm() {
    // 获取选中的部门 ID
    const selectedIds: number[] = Array.isArray(selectedDeptIds.value)
      ? selectedDeptIds.value
      : selectedDeptIds.value.checked || [];
    const deptArray = deptData.value.filter((dept) =>
      selectedIds.includes(dept.id!),
    );
    emit('confirm', deptArray);
    // 关闭并提示
    await modalApi.close();
  },
  async onOpenChange(isOpen: boolean) {
    if (!isOpen) {
      deptTree.value = [];
      selectedDeptIds.value = [];
      return;
    }
    // 加载数据
    const data = modalApi.getData();
    if (!data) {
      return;
    }
    modalApi.lock();
    try {
      deptData.value = await getSimpleDeptList();
      deptTree.value = handleTree(deptData.value) as DataNode[];
      // // 设置已选择的部门
      if (data.selectedList?.length) {
        const selectedIds = data.selectedList
          .map((dept: SystemDeptApi.Dept) => dept.id)
          .filter((id: number) => id !== undefined);
        selectedDeptIds.value = props.checkStrictly
          ? {
              checked: selectedIds,
              halfChecked: [],
            }
          : selectedIds;
      }
    } finally {
      modalApi.unlock();
    }
  },
  destroyOnClose: true,
});
 
/** 处理选中状态变化 */
function handleCheck() {
  if (!props.multiple) {
    // 单选模式下,只保留最后选择的节点
    if (Array.isArray(selectedDeptIds.value)) {
      const lastSelectedId =
        selectedDeptIds.value[selectedDeptIds.value.length - 1];
      if (lastSelectedId) {
        selectedDeptIds.value = [lastSelectedId];
      }
    } else {
      // checkStrictly 为 true 时,selectedDeptIds 是一个对象
      const checked = selectedDeptIds.value.checked || [];
      if (checked.length > 0) {
        const lastSelectedId = checked[checked.length - 1];
        selectedDeptIds.value = {
          checked: [lastSelectedId!],
          halfChecked: [],
        };
      }
    }
  }
}
</script>
<template>
  <Modal :title="title" key="dept-select-modal" class="w-2/5">
    <Row class="h-full">
      <Col :span="24">
        <Card class="h-full">
          <Tree
            :tree-data="deptTree"
            v-if="deptTree.length > 0"
            v-model:checked-keys="selectedDeptIds"
            :checkable="true"
            :check-strictly="checkStrictly"
            :field-names="{ title: 'name', key: 'id' }"
            :default-expand-all="true"
            @check="handleCheck"
          />
        </Card>
      </Col>
    </Row>
  </Modal>
</template>