From 9638e018b32e785b7148dc7025b91dc356d0ff4d Mon Sep 17 00:00:00 2001
From: tangjinzhou <415800467@qq.com>
Date: Mon, 22 Oct 2018 21:24:38 +0800
Subject: [PATCH] fix: tree custom icon auguments error #223

---
 components/tree/DirectoryTree.jsx   | 2 +-
 components/vc-tree/src/TreeNode.jsx | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/components/tree/DirectoryTree.jsx b/components/tree/DirectoryTree.jsx
index 12c67659d..930267a6f 100644
--- a/components/tree/DirectoryTree.jsx
+++ b/components/tree/DirectoryTree.jsx
@@ -19,7 +19,7 @@ import { initDefaultProps, getOptionProps } from '../_util/props-util'
 //   selectedKeys?: string[];
 // }
 
-function getIcon (h, props) {
+function getIcon (props, h) {
   const { isLeaf, expanded } = props
   if (isLeaf) {
     return <Icon type='file' />
diff --git a/components/vc-tree/src/TreeNode.jsx b/components/vc-tree/src/TreeNode.jsx
index b088b08bf..88c4e511e 100644
--- a/components/vc-tree/src/TreeNode.jsx
+++ b/components/vc-tree/src/TreeNode.jsx
@@ -358,7 +358,7 @@ const TreeNode = {
             )}
           >
             {typeof currentIcon === 'function'
-              ? currentIcon(h, { ...this.$props }) : currentIcon}
+              ? currentIcon({ ...this.$props }, h) : currentIcon}
           </span>
         ) : this.renderIcon()
       } else if (loadData && loading) {