From ff3bcf9def843043769d27c06200e6df0138f03d Mon Sep 17 00:00:00 2001
From: saller <weskerli@163.com>
Date: Mon, 13 Feb 2023 10:09:27 +0800
Subject: [PATCH] fix(comp:transfer): select all status error (#1442)

`getAllSelectedKeys` in dataStrategy should be used to calculate select all status
---
 .../src/composables/useTransferSelectState.ts | 36 +++++++++++--------
 1 file changed, 21 insertions(+), 15 deletions(-)

diff --git a/packages/components/transfer/src/composables/useTransferSelectState.ts b/packages/components/transfer/src/composables/useTransferSelectState.ts
index 467ac45d5..1f47c15e6 100644
--- a/packages/components/transfer/src/composables/useTransferSelectState.ts
+++ b/packages/components/transfer/src/composables/useTransferSelectState.ts
@@ -59,29 +59,35 @@ export function useTransferSelectState(
     getKey,
   } = transferDataContext
 
-  const sourceDataCount = computed(() =>
-    props.mode === 'immediate' ? dataKeyMap.value.size : sourceDataKeys.value.size,
-  )
-  const targetDataCount = computed(() => targetDataKeys.value.size)
   const sourceCheckableDataCount = computed(() => {
+    const sourceDataCount = props.mode === 'immediate' ? dataKeyMap.value.size : sourceDataKeys.value.size
     const disabledCount = props.mode === 'immediate' ? disabledKeys.value.size : disabledSourceKeys.value.size
 
-    return sourceDataCount.value - disabledCount
+    return sourceDataCount - disabledCount
   })
   const targetCheckableDataCount = computed(() => targetDataKeys.value.size - disabledTargetKeys.value.size)
 
-  const sourceSelectAllStatus = computed(() => {
-    return {
-      checked: sourceDataCount.value >= sourceSelectedKeys.value.length && sourceSelectedKeys.value.length > 0,
-      indeterminate: sourceDataCount.value > sourceSelectedKeys.value.length && sourceSelectedKeys.value.length > 0,
-    }
-  })
-  const targetSelectAllStatus = computed(() => {
+  const getAllSelectedStatus = (isSource: boolean) => {
+    const allSelectedKeys = transferDataStrategy.value.getAllSelectedKeys(
+      true,
+      isSource ? (props.mode === 'immediate' ? dataSource.value : sourceData.value) : targetData.value,
+      new Set(),
+      new Set(),
+      getKey.value,
+    )
+    const selectedkeyCount = (isSource ? sourceSelectedKeys : targetSelectedKeys).value.length
+    const selectedKeySet = isSource ? sourceSelectedKeySet.value : targetSelectedKeySet.value
+
     return {
-      checked: targetDataCount.value >= targetSelectedKeys.value.length && targetSelectedKeys.value.length > 0,
-      indeterminate: targetDataCount.value > targetSelectedKeys.value.length && targetSelectedKeys.value.length > 0,
+      checked: selectedkeyCount > 0,
+      indeterminate:
+        (allSelectedKeys.length !== selectedkeyCount || allSelectedKeys.some(key => !selectedKeySet.has(key))) &&
+        selectedkeyCount > 0,
     }
-  })
+  }
+
+  const sourceSelectAllStatus = computed(() => getAllSelectedStatus(true))
+  const targetSelectAllStatus = computed(() => getAllSelectedStatus(false))
 
   let transferBySelectionChangeLocked = false
   watch(sourceSelectedKeySet, (currentCheckedKeys, originalCheckedKeys) => {