From 3672ab9c3051dd9911d84275b4c685b3e8540264 Mon Sep 17 00:00:00 2001
From: shizhongming <zhongming4762@hotmail.com>
Date: Tue, 23 Apr 2024 22:31:23 +0800
Subject: [PATCH] =?UTF-8?q?perf(=E7=B3=BB=E7=BB=9F=E6=A8=A1=E5=9D=97-?=
 =?UTF-8?q?=E8=8F=9C=E5=8D=95=E7=AE=A1=E7=90=86):=201=E3=80=81=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E8=8F=9C=E5=8D=95=E7=AE=A1=E7=90=86=E5=8A=A0=E8=BD=BD?=
 =?UTF-8?q?=202=E3=80=81=E8=8F=9C=E5=8D=95=E5=8F=AF=E4=BF=AE=E6=94=B9?=
 =?UTF-8?q?=E7=88=B6=E7=BA=A7=E8=8F=9C=E5=8D=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../views/function/FunctionListView.config.ts |  16 ++-
 .../views/function/FunctionListView.vue       | 115 ++++++++++++++----
 2 files changed, 97 insertions(+), 34 deletions(-)

diff --git a/src/modules/smart-system/views/function/FunctionListView.config.ts b/src/modules/smart-system/views/function/FunctionListView.config.ts
index 8516383c3..907c86edb 100644
--- a/src/modules/smart-system/views/function/FunctionListView.config.ts
+++ b/src/modules/smart-system/views/function/FunctionListView.config.ts
@@ -98,18 +98,16 @@ export const getAddEditForm = (t: Function): FormSchema[] => {
       show: false,
     },
     {
-      field: 'parentName',
-      label: '上级',
-      component: 'Input',
-      componentProps: {
-        disabled: true,
-      },
+      field: 'isTopAdd',
+      label: '',
+      component: 'Switch',
+      defaultValue: false,
+      show: false,
     },
     {
       field: 'parentId',
-      label: '',
-      component: 'Input',
-      show: false,
+      label: '上级',
+      slot: 'addEdit-parentId',
     },
     {
       field: 'functionName',
diff --git a/src/modules/smart-system/views/function/FunctionListView.vue b/src/modules/smart-system/views/function/FunctionListView.vue
index 5a2787dd9..e0d1d262a 100644
--- a/src/modules/smart-system/views/function/FunctionListView.vue
+++ b/src/modules/smart-system/views/function/FunctionListView.vue
@@ -25,12 +25,27 @@
           </a-radio>
         </a-radio-group>
       </template>
+      <template #addEdit-parentId="{ model, field }">
+        <TreeSelect
+          v-model:value="model[field]"
+          :fieldNames="{ label: 'functionName', value: 'functionId', children: 'children' }"
+          :tree-data="getTreeData(model)"
+        >
+          <template #title="{ functionType, functionName }">
+            <a-tag :color="getTagData(functionType).color">
+              {{ getTagData(functionType).text }}
+            </a-tag>
+            {{ functionName }}
+          </template>
+        </TreeSelect>
+      </template>
     </SmartTable>
   </div>
 </template>
 
 <script lang="ts" setup>
-  import { reactive, ref, unref } from 'vue';
+  import { onMounted, reactive, ref, unref } from 'vue';
+  import { TreeSelect } from 'ant-design-vue';
 
   import {
     ActionItem,
@@ -46,6 +61,7 @@
   import StringUtils from '@/utils/StringUtils';
   import { SystemPermissions } from '@/modules/smart-system/constants/SystemConstants';
   import { hasPermission } from '@/utils/auth';
+  import TreeUtils from '@/utils/TreeUtils';
 
   const permissions = SystemPermissions.function;
   const { t } = useI18n();
@@ -107,10 +123,56 @@
     ];
   };
 
-  const [
-    registerTable,
-    { showAddModal, editByRowModal, getSearchForm, getTableInstance, deleteByRow },
-  ] = useSmartTable({
+  const functionTreeData = ref<Recordable[]>([]);
+  onMounted(async () => {
+    functionTreeData.value = await listApi({
+      parameter: {
+        'functionType@in': ['10', '20'],
+      },
+    });
+  });
+  const getTreeData = (model: Recordable) => {
+    const { functionType, isTopAdd } = model;
+    console.log(isTopAdd);
+    let treeData: Recordable[] = [];
+    if (isTopAdd !== true) {
+      let dataList: Recordable[] = [];
+      if (functionType === 'CATALOG' || functionType === 'MENU') {
+        dataList = unref(functionTreeData)
+          .filter((item) => item.functionType === 'CATALOG')
+          .map((item) => {
+            return {
+              ...item,
+            };
+          });
+      } else {
+        dataList = unref(functionTreeData).map((item) => {
+          return {
+            ...item,
+          };
+        });
+      }
+      if (dataList.length > 0) {
+        treeData =
+          TreeUtils.convertList2Tree(
+            dataList,
+            (row) => row.functionId,
+            (row) => row.parentId,
+            0,
+          ) || [];
+      }
+    }
+    return [
+      {
+        functionId: 0,
+        functionName: '根目录',
+        children: treeData,
+        functionType: 'CATALOG',
+      },
+    ];
+  };
+
+  const [registerTable, { showAddModal, editByRowModal, deleteByRow }] = useSmartTable({
     id: 'FunctionListView',
     columns: tableColumns,
     resizableConfig: {},
@@ -139,17 +201,19 @@
       },
     },
     treeConfig: {
-      lazy: true,
-      loadMethod: ({ row }) => {
-        const { searchSymbolForm } = getSearchForm().getSearchFormParameter();
-        const parameter = {
-          parameter: {
-            ...searchSymbolForm,
-            'parentId@=': row.functionId,
-          },
-        };
-        return listApi(parameter);
-      },
+      reserve: true,
+      // lazy: true,
+      // loadMethod: ({ row }) => {
+      //   console.log('-------------');
+      //   const { searchSymbolForm } = getSearchForm().getSearchFormParameter();
+      //   const parameter = {
+      //     parameter: {
+      //       ...searchSymbolForm,
+      //       'parentId@=': row.functionId,
+      //     },
+      //   };
+      //   return listApi(parameter);
+      // },
     },
     rowConfig: {
       keyField: 'functionId',
@@ -174,22 +238,23 @@
         wrapperCol: { span: 16 },
         schemas: getAddEditForm(t),
       },
-      afterSave: () => {
-        // 保存完成之后重新加载节点
-        return getTableInstance().reloadTreeExpand(unref(currentRowRef));
-      },
     },
     proxyConfig: {
       ajax: {
-        query: ({ ajaxParameter }) => {
+        query: async ({ ajaxParameter }) => {
           const parameter = {
             ...ajaxParameter,
             parameter: {
               ...ajaxParameter?.parameter,
-              'parentId@=': 0,
+              // 'parentId@=': 0,
             },
           };
-          return listApi(parameter);
+          return TreeUtils.convertList2Tree(
+            (await listApi(parameter)) || [],
+            (row) => row.functionId,
+            (row) => row.parentId,
+            0,
+          );
         },
         delete: (params) => deleteApi(params),
         save: saveApi,
@@ -205,8 +270,8 @@
             onClick: () => {
               setTypeDisabled(['function']);
               showAddModal({
-                parentId: '0',
-                parentName: '根目录',
+                isTopAdd: true,
+                parentId: 0,
               });
             },
           },