From abae7f3295846f10b69c591739bbec22d176b6fe Mon Sep 17 00:00:00 2001
From: xachary <179740385@qq.com>
Date: Thu, 11 Jan 2024 15:29:36 +0800
Subject: [PATCH] feat(BasicTable): table enable accordion expand (#3533).
 resolve #3525

---
 src/components/Table/src/BasicTable.vue       |  8 +-
 .../Table/src/hooks/useTableExpand.ts         | 77 +++++++++++++++++--
 src/components/Table/src/types/table.ts       |  1 +
 src/views/demo/table/ExpandTable.vue          |  1 +
 src/views/demo/table/TreeTable.vue            |  1 +
 src/views/demo/table/tableData.tsx            | 53 ++++++++++++-
 6 files changed, 126 insertions(+), 15 deletions(-)

diff --git a/src/components/Table/src/BasicTable.vue b/src/components/Table/src/BasicTable.vue
index 9d0ac56f..174d6d97 100644
--- a/src/components/Table/src/BasicTable.vue
+++ b/src/components/Table/src/BasicTable.vue
@@ -22,6 +22,7 @@
       v-show="getEmptyDataIsShowTable"
       @change="handleTableChange"
       @resize-column="setColumnWidth"
+      @expand="handleTableExpand"
     >
       <template #[item]="data" v-for="item in Object.keys($slots)" :key="item">
         <slot :name="item" v-bind="data || {}"></slot>
@@ -207,11 +208,8 @@
 
   const { getRowClassName } = useTableStyle(getProps, prefixCls);
 
-  const { getExpandOption, expandAll, expandRows, collapseRows, collapseAll } = useTableExpand(
-    getProps,
-    tableData,
-    emit,
-  );
+  const { getExpandOption, expandAll, expandRows, collapseRows, collapseAll, handleTableExpand } =
+    useTableExpand(getProps, tableData, emit);
 
   const handlers: InnerHandlers = {
     onColumnsChange: (data: ColumnChangeParam[]) => {
diff --git a/src/components/Table/src/hooks/useTableExpand.ts b/src/components/Table/src/hooks/useTableExpand.ts
index caaa431d..a47d1468 100644
--- a/src/components/Table/src/hooks/useTableExpand.ts
+++ b/src/components/Table/src/hooks/useTableExpand.ts
@@ -1,6 +1,6 @@
 import type { ComputedRef, Ref } from 'vue';
 import type { BasicTableProps } from '../types/table';
-import { computed, unref, ref, toRaw } from 'vue';
+import { computed, unref, ref, toRaw, nextTick } from 'vue';
 import { ROW_KEY } from '../const';
 
 export function useTableExpand(
@@ -20,8 +20,8 @@ export function useTableExpand(
   });
 
   const getExpandOption = computed(() => {
-    const { isTreeTable } = unref(propsRef);
-    if (!isTreeTable) return {};
+    const { isTreeTable, expandRowByClick } = unref(propsRef);
+    if (!isTreeTable && !expandRowByClick) return {};
 
     return {
       expandedRowKeys: unref(expandedRowKeys),
@@ -43,15 +43,15 @@ export function useTableExpand(
 
   function expandRows(keys: (string | number)[]) {
     // use row ID expands the specified table row
-    const { isTreeTable } = unref(propsRef);
-    if (!isTreeTable) return;
+    const { isTreeTable, expandRowByClick } = unref(propsRef);
+    if (!isTreeTable && !expandRowByClick) return;
     expandedRowKeys.value = [...expandedRowKeys.value, ...keys];
   }
 
   function collapseRows(keys: (string | number)[]) {
     // use row ID collapses the specified table row
-    const { isTreeTable } = unref(propsRef);
-    if (!isTreeTable) return;
+    const { isTreeTable, expandRowByClick } = unref(propsRef);
+    if (!isTreeTable && !expandRowByClick) return;
     expandedRowKeys.value = unref(expandedRowKeys).filter((key) => !keys.includes(key));
   }
 
@@ -68,5 +68,66 @@ export function useTableExpand(
     return keys;
   }
 
-  return { getExpandOption, expandAll, collapseAll, expandRows, collapseRows };
+  // 获取展开路径 keys
+  function getKeyPaths(
+    records: Recordable[],
+    rowKey: string,
+    childrenColumnName: string,
+    key: string | number,
+    paths: Array<string | number>,
+  ): boolean {
+    if (records.findIndex((record) => record[rowKey] === key) > -1) {
+      paths.push(key);
+      return true;
+    } else {
+      for (const record of records) {
+        const children = record[childrenColumnName];
+        if (
+          Array.isArray(children) &&
+          getKeyPaths(children, rowKey, childrenColumnName, key, paths)
+        ) {
+          paths.push(record[rowKey]);
+          return true;
+        }
+      }
+    }
+    return false;
+  }
+
+  // 手风琴展开
+  function expandRowAccordion(key: string | number, rowKey: string) {
+    const { childrenColumnName } = unref(propsRef);
+    const paths: Array<string | number> = [];
+    getKeyPaths(tableData.value, rowKey, childrenColumnName || 'children', key, paths);
+    expandedRowKeys.value = paths;
+  }
+
+  // 监听展开事件，用于支持手风琴展开效果
+  function handleTableExpand(expanded, record) {
+    // 手风琴开关
+    // isTreeTable 或 expandRowByClick 时支持
+    // rowKey 是字符串时支持
+    // 展开操作
+    if (
+      propsRef.value.accordion &&
+      (propsRef.value.isTreeTable || propsRef.value.expandRowByClick) &&
+      typeof getRowKey.value === 'string' &&
+      expanded
+    ) {
+      const rowKey = getRowKey.value as string;
+      nextTick(() => {
+        expandRowAccordion(record[rowKey], rowKey);
+      });
+    }
+  }
+
+  return {
+    getExpandOption,
+    expandAll,
+    collapseAll,
+    expandRows,
+    collapseRows,
+    expandRowAccordion,
+    handleTableExpand,
+  };
 }
diff --git a/src/components/Table/src/types/table.ts b/src/components/Table/src/types/table.ts
index bd5afb9c..7768e90a 100644
--- a/src/components/Table/src/types/table.ts
+++ b/src/components/Table/src/types/table.ts
@@ -148,6 +148,7 @@ export interface BasicTableProps<T = any> {
   // 点击行选中
   clickToRowSelect?: boolean;
   isTreeTable?: boolean;
+  accordion?: boolean; // isTreeTable 或 expandRowByClick 时支持
   // 自定义排序方法
   sortFn?: (sortInfo: SorterResult) => any;
   // 排序方法
diff --git a/src/views/demo/table/ExpandTable.vue b/src/views/demo/table/ExpandTable.vue
index 7f3d232d..3339bba1 100644
--- a/src/views/demo/table/ExpandTable.vue
+++ b/src/views/demo/table/ExpandTable.vue
@@ -48,6 +48,7 @@
     rowKey: 'id',
     canResize: false,
     expandRowByClick: true,
+    accordion: true, // 手风琴效果
     actionColumn: {
       width: 160,
       title: 'Action',
diff --git a/src/views/demo/table/TreeTable.vue b/src/views/demo/table/TreeTable.vue
index 489932de..5d0cef7c 100644
--- a/src/views/demo/table/TreeTable.vue
+++ b/src/views/demo/table/TreeTable.vue
@@ -17,6 +17,7 @@
   const [register, { expandAll, collapseAll, expandRows, collapseRows }] = useTable({
     title: '树形表格',
     isTreeTable: true,
+    accordion: true, // 手风琴效果
     rowSelection: {
       type: 'checkbox',
       getCheckboxProps(record: Recordable) {
diff --git a/src/views/demo/table/tableData.tsx b/src/views/demo/table/tableData.tsx
index 9d817663..9fa2b730 100644
--- a/src/views/demo/table/tableData.tsx
+++ b/src/views/demo/table/tableData.tsx
@@ -294,13 +294,62 @@ export function getTreeTableData() {
         endTime: new Date().toLocaleString(),
         children: [
           {
-            id: `l2-${index}`,
+            id: `l2-${index}-1`,
             name: 'John Brown',
-            age: `1${index}`,
+            age: `1`,
             no: `${index + 10}`,
             address: 'New York No. 1 Lake ParkNew York No. 1 Lake Park',
             beginTime: new Date().toLocaleString(),
             endTime: new Date().toLocaleString(),
+            children: [
+              {
+                id: `l3-${index}-1-1`,
+                name: 'John Brown',
+                age: `11`,
+                no: `11`,
+                address: 'New York No. 1 Lake ParkNew York No. 1 Lake Park',
+                beginTime: new Date().toLocaleString(),
+                endTime: new Date().toLocaleString(),
+              },
+              {
+                id: `l3-${index}-1-2`,
+                name: 'John Brown',
+                age: `12`,
+                no: `12`,
+                address: 'New York No. 1 Lake ParkNew York No. 1 Lake Park',
+                beginTime: new Date().toLocaleString(),
+                endTime: new Date().toLocaleString(),
+              },
+            ],
+          },
+          {
+            id: `l2-${index}-2`,
+            name: 'John Brown',
+            age: `2`,
+            no: `${index + 10}`,
+            address: 'New York No. 1 Lake ParkNew York No. 1 Lake Park',
+            beginTime: new Date().toLocaleString(),
+            endTime: new Date().toLocaleString(),
+            children: [
+              {
+                id: `l3-${index}-2-1`,
+                name: 'John Brown',
+                age: `21`,
+                no: `21`,
+                address: 'New York No. 1 Lake ParkNew York No. 1 Lake Park',
+                beginTime: new Date().toLocaleString(),
+                endTime: new Date().toLocaleString(),
+              },
+              {
+                id: `l3-${index}-2-2`,
+                name: 'John Brown',
+                age: `22`,
+                no: `22`,
+                address: 'New York No. 1 Lake ParkNew York No. 1 Lake Park',
+                beginTime: new Date().toLocaleString(),
+                endTime: new Date().toLocaleString(),
+              },
+            ],
           },
         ],
       });