From 1ce519d3ff786244c4eb3de9d20d139e04bd06fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A2=81=E7=81=8F?= <admin@aresn.com>
Date: Tue, 7 Jan 2020 22:27:00 +0800
Subject: [PATCH] Table tree support selectAll status

---
 examples/routers/table.vue          | 78 ++++++++++++++++++++++++++---
 src/components/table/table-head.vue | 44 ++++++++++++++--
 2 files changed, 111 insertions(+), 11 deletions(-)

diff --git a/examples/routers/table.vue b/examples/routers/table.vue
index 4d73d428..b7445908 100644
--- a/examples/routers/table.vue
+++ b/examples/routers/table.vue
@@ -1,5 +1,7 @@
 <template>
     <div style="margin: 100px;">
+        <Table border ref="selection" :columns="columns5" :data="data2"></Table>
+        <Divider></Divider>
         <Table
                 border
                 highlight-row
@@ -61,7 +63,8 @@
                         name: 'John Brown',
                         age: 18,
                         address: 'New York No. 1 Lake Park',
-                        date: '2016-10-03'
+                        date: '2016-10-03',
+                        _disabled: true,
                     },
                     {
                         id: '101',
@@ -70,20 +73,23 @@
                         address: 'London No. 1 Lake Park',
                         date: '2016-10-01',
                         _showChildren: true,
+                        _disabled: true,
                         children: [
                             {
                                 id: '10100',
                                 name: '张三',
                                 age: 18,
                                 address: '发动机莲富大厦放假了开始的',
-                                date: '2016-10-01'
+                                date: '2016-10-01',
+                                _disabled: true,
                             },
                             {
                                 id: '10101',
                                 name: '李四',
                                 age: 19,
                                 address: '风刀霜剑分离的思路开发',
-                                date: '2016-10-02'
+                                date: '2016-10-02',
+                                _disabled: true,
                             },
                             {
                                 id: '10102',
@@ -92,13 +98,15 @@
                                 address: '分离的付款就说个就是范德萨发生',
                                 date: '2016-10-03',
                                 _showChildren: true,
+                                _disabled: true,
                                 children: [
                                     {
                                         id: '10102100',
                                         name: '赵六',
                                         age: 21,
                                         address: '梵蒂冈人太热疼我',
-                                        date: '2016-10-05'
+                                        date: '2016-10-05',
+                                        _disabled: true,
                                     },
                                     {
                                         id: '10102101',
@@ -107,20 +115,23 @@
                                         address: '法第三方的范德萨范德萨发送到',
                                         date: '2016-10-06',
                                         _showChildren: true,
+                                        _disabled: true,
                                         children: [
                                             {
                                                 id: '10102101100',
                                                 name: '第九',
                                                 age: 23,
                                                 address: '9梵蒂冈人太热疼我',
-                                                date: '2016-10-07'
+                                                date: '2016-10-07',
+                                                _disabled: true,
                                             },
                                             {
                                                 id: '10102101101',
                                                 name: '第十',
                                                 age: 24,
                                                 address: '10法第三方的范德萨范德萨发送到',
-                                                date: '2016-10-08'
+                                                date: '2016-10-08',
+                                                _disabled: true,
                                             },
                                         ]
                                     },
@@ -133,14 +144,65 @@
                         name: 'Joe Black',
                         age: 30,
                         address: 'Sydney No. 1 Lake Park',
-                        date: '2016-10-02'
+                        date: '2016-10-02',
+                        _disabled: true,
                     },
                     {
                         id: '103',
                         name: 'Jon Snow',
                         age: 26,
                         address: 'Ottawa No. 2 Lake Park',
-                        date: '2016-10-04'
+                        date: '2016-10-04',
+                        _disabled: true,
+                    }
+                ],
+                columns5: [
+                    {
+                        type: 'selection',
+                        width: 60,
+                        align: 'center'
+                    },
+                    {
+                        title: 'Name',
+                        key: 'name'
+                    },
+                    {
+                        title: 'Age',
+                        key: 'age'
+                    },
+                    {
+                        title: 'Address',
+                        key: 'address'
+                    }
+                ],
+                data2: [
+                    {
+                        name: 'John Brown',
+                        age: 18,
+                        address: 'New York No. 1 Lake Park',
+                        date: '2016-10-03',
+                        _disabled: true,
+                    },
+                    {
+                        name: 'Jim Green',
+                        age: 24,
+                        address: 'London No. 1 Lake Park',
+                        date: '2016-10-01',
+                        _disabled: true
+                    },
+                    {
+                        name: 'Joe Black',
+                        age: 30,
+                        address: 'Sydney No. 1 Lake Park',
+                        date: '2016-10-02',
+                        _disabled: true
+                    },
+                    {
+                        name: 'Jon Snow',
+                        age: 26,
+                        address: 'Ottawa No. 2 Lake Park',
+                        date: '2016-10-04',
+                        _disabled: true
                     }
                 ]
             }
diff --git a/src/components/table/table-head.vue b/src/components/table/table-head.vue
index ba428185..8e2be537 100644
--- a/src/components/table/table-head.vue
+++ b/src/components/table/table-head.vue
@@ -119,13 +119,25 @@
             isSelectAll () {
                 let isSelectAll = true;
                 if (!this.data.length) isSelectAll = false;
-                if (!this.data.find(item => !item._disabled)) isSelectAll = false;    // #1751
-                for (let i = 0; i < this.data.length; i++) {
-                    if (!this.objData[this.data[i]._index]._isChecked && !this.objData[this.data[i]._index]._isDisabled) {
+
+                // 全部disabled且全false，#1751
+                let isAllDisabledAndUnSelected = true;
+
+                for (let i in this.objData) {
+                    const objData = this.objData[i];
+                    if (!objData._isChecked && !objData._isDisabled) {
                         isSelectAll = false;
                         break;
+                    } else if (objData.children && objData.children.length) {
+                        isSelectAll = this.isChildrenSelected(objData, isSelectAll);
+                    }
+                    if (!(objData._isDisabled && !objData._isChecked)) {
+                        isAllDisabledAndUnSelected = false;
+                    } else if (objData.children && objData.children.length) {
+                        isAllDisabledAndUnSelected = this.isChildrenAllDisabledAndUnSelected(objData, isAllDisabledAndUnSelected);
                     }
                 }
+                if (isAllDisabledAndUnSelected) isSelectAll = false;
 
                 return isSelectAll;
             },
@@ -331,6 +343,32 @@
             handleMouseOut () {
                 if (this.$isServer) return;
                 document.body.style.cursor = '';
+            },
+            isChildrenSelected (objData, isSelectAll) {
+                let status = isSelectAll;
+                if (objData.children && objData.children.length) {
+                    objData.children.forEach(row => {
+                        if (!row._isChecked && !row._isDisabled) {
+                            status = false;
+                        } else if (row.children && row.children.length) {
+                            status = this.isChildrenSelected(row, status);
+                        }
+                    });
+                }
+                return status;
+            },
+            isChildrenAllDisabledAndUnSelected (objData, isAllDisabledAndUnSelected) {
+                let status = isAllDisabledAndUnSelected;
+                if (objData.children && objData.children.length) {
+                    objData.children.forEach(row => {
+                        if (!(row._isDisabled && !row._isChecked)) {
+                            status = false;
+                        } else if (row.children && row.children.length) {
+                            status = this.isChildrenAllDisabledAndUnSelected(row, status);
+                        }
+                    });
+                }
+                return status;
             }
         }
     };