From bcffcc3b3c4900ab091b7edc5c7636ca0db1494e Mon Sep 17 00:00:00 2001
From: kavin <kavin.huang@alphaliongroup.com>
Date: Thu, 10 Oct 2019 10:06:12 +0800
Subject: [PATCH] fix(#19132): fix component tree defaultExpandAll does not
 work when using treeData

---
 components/tree/DirectoryTree.tsx | 17 +++++++++++++++--
 components/tree/util.ts           |  9 +++++++++
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/components/tree/DirectoryTree.tsx b/components/tree/DirectoryTree.tsx
index bce268a8a179..803221f74e2f 100644
--- a/components/tree/DirectoryTree.tsx
+++ b/components/tree/DirectoryTree.tsx
@@ -13,7 +13,12 @@ import Tree, {
   AntTreeNodeSelectedEvent,
   AntTreeNode,
 } from './Tree';
-import { calcRangeKeys, getFullKeyList, convertDirectoryKeysToNodes } from './util';
+import {
+  calcRangeKeys,
+  getFullKeyList,
+  convertDirectoryKeysToNodes,
+  getFullKeyListByTreeData,
+} from './util';
 import Icon from '../icon';
 
 export type ExpandAction = false | 'click' | 'doubleClick';
@@ -82,7 +87,15 @@ class DirectoryTree extends React.Component<DirectoryTreeProps, DirectoryTreeSta
 
     // Expanded keys
     if (defaultExpandAll) {
-      this.state.expandedKeys = getFullKeyList(props.children);
+      const expandedKeysByChildren = getFullKeyList(props.children);
+      this.state.expandedKeys = expandedKeysByChildren;
+      if (
+        !!props.treeData &&
+        Array.isArray(expandedKeysByChildren) &&
+        expandedKeysByChildren.length === 0
+      ) {
+        this.state.expandedKeys = getFullKeyListByTreeData(props.treeData);
+      }
     } else if (defaultExpandParent) {
       this.state.expandedKeys = conductExpandParent(
         expandedKeys || defaultExpandedKeys,
diff --git a/components/tree/util.ts b/components/tree/util.ts
index d8208ab2f1fb..16385d2c66d4 100644
--- a/components/tree/util.ts
+++ b/components/tree/util.ts
@@ -101,3 +101,12 @@ export function convertDirectoryKeysToNodes(
   });
   return nodes;
 }
+
+export function getFullKeyListByTreeData(treeData: any[]): any {
+  for (const item of treeData) {
+    if (item.children) {
+      return getFullKeyListByTreeData(item.children).concat([item.key]);
+    }
+    return [item.key];
+  }
+}