From 94d177cc836e44df8f27795bde09e79c88974ddd Mon Sep 17 00:00:00 2001
From: Sergio Crisostomo <sergiosbox@gmail.com>
Date: Sat, 14 Oct 2017 09:47:33 +0200
Subject: [PATCH 1/3] add width to expand icon so we can click on it easier

---
 src/styles/components/tree.less | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/styles/components/tree.less b/src/styles/components/tree.less
index 60a43632..0d9a37e4 100644
--- a/src/styles/components/tree.less
+++ b/src/styles/components/tree.less
@@ -38,7 +38,10 @@
     }
     &-arrow{
         cursor: pointer;
-        i{
+        width: 12px;
+        text-align: center;
+        display: inline-block;
+        i {
             transition: all @transition-time @ease-in-out;
         }
         &-open{
@@ -56,4 +59,4 @@
             cursor: @cursor-disabled;
         }
     }
-}
\ No newline at end of file
+}

From d44420be721153feb22e150d3b05ecccfdc703d7 Mon Sep 17 00:00:00 2001
From: Sergio Crisostomo <sergiosbox@gmail.com>
Date: Sat, 14 Oct 2017 09:49:13 +0200
Subject: [PATCH 2/3] refactor and make indeterminate accessible outside tree

---
 src/components/tree/node.vue |  54 +++--------
 src/components/tree/tree.vue | 179 +++++++++++++++++++++--------------
 2 files changed, 118 insertions(+), 115 deletions(-)

diff --git a/src/components/tree/node.vue b/src/components/tree/node.vue
index 261b89d0..63e38f9d 100644
--- a/src/components/tree/node.vue
+++ b/src/components/tree/node.vue
@@ -1,6 +1,6 @@
 <template>
     <collapse-transition>
-        <ul :class="classes" v-show="visible">
+        <ul :class="classes">
             <li>
                 <span :class="arrowClasses" @click="handleExpand">
                     <Icon type="arrow-right-b"></Icon>
@@ -8,15 +8,15 @@
                 <Checkbox
                         v-if="showCheckbox"
                         :value="data.checked"
-                        :indeterminate="indeterminate"
+                        :indeterminate="data.indeterminate"
                         :disabled="data.disabled || data.disableCheckbox"
                         @click.native.prevent="handleCheck"></Checkbox>
                 <span :class="titleClasses" v-html="data.title" @click="handleSelect"></span>
                 <Tree-node
+                        v-if="data.expand"
                         v-for="item in data.children"
                         :key="item.nodeKey"
                         :data="item"
-                        :visible="data.expand"
                         :multiple="multiple"
                         :show-checkbox="showCheckbox">
                 </Tree-node>
@@ -29,7 +29,6 @@
     import Icon from '../icon/icon.vue';
     import CollapseTransition from '../base/collapse-transition';
     import Emitter from '../../mixins/emitter';
-    import { findComponentsDownward } from '../../utils/assist';
 
     const prefixCls = 'ivu-tree';
 
@@ -51,16 +50,11 @@
             showCheckbox: {
                 type: Boolean,
                 default: false
-            },
-            visible: {
-                type: Boolean,
-                default: false
             }
         },
         data () {
             return {
-                prefixCls: prefixCls,
-                indeterminate: false
+                prefixCls: prefixCls
             };
         },
         computed: {
@@ -103,40 +97,16 @@
             },
             handleSelect () {
                 if (this.data.disabled) return;
-                if (this.data.selected) {
-                    this.data.selected = false;
-                } else if (this.multiple) {
-                    this.$set(this.data, 'selected', !this.data.selected);
-                } else {
-                    this.dispatch('Tree', 'selected', this.data);
-                }
-                this.dispatch('Tree', 'on-selected');
+                this.dispatch('Tree', 'on-selected', this.data.nodeKey);
             },
             handleCheck () {
-                if (this.disabled) return;
-                const checked = !this.data.checked;
-                if (!checked || this.indeterminate) {
-                    findComponentsDownward(this, 'TreeNode').forEach(node => node.data.checked = false);
-                } else {
-                    findComponentsDownward(this, 'TreeNode').forEach(node => node.data.checked = true);
-                }
-                this.data.checked = checked;
-                this.dispatch('Tree', 'checked');
-                this.dispatch('Tree', 'on-checked');
-            },
-            setIndeterminate () {
-                this.indeterminate = this.data.checked ? false : findComponentsDownward(this, 'TreeNode').some(node => node.data.checked);
+                if (this.data.disabled) return;
+                const changes = {
+                    checked: !this.data.checked && !this.data.indeterminate,
+                    nodeKey: this.data.nodeKey
+                };
+                this.dispatch('Tree', 'on-check', changes);
             }
-        },
-        created () {
-            // created node.vue first, mounted tree.vue second
-            if (!this.data.checked) this.$set(this.data, 'checked', false);
-        },
-        mounted () {
-            this.$on('indeterminate', () => {
-                this.broadcast('TreeNode', 'indeterminate');
-                this.setIndeterminate();
-            });
         }
     };
-</script>
\ No newline at end of file
+</script>
diff --git a/src/components/tree/tree.vue b/src/components/tree/tree.vue
index 51134175..eee3f305 100644
--- a/src/components/tree/tree.vue
+++ b/src/components/tree/tree.vue
@@ -1,26 +1,23 @@
 <template>
     <div :class="prefixCls">
         <Tree-node
-            v-for="item in data"
+            v-for="item in stateTree"
             :key="item.nodeKey"
             :data="item"
             visible
             :multiple="multiple"
             :show-checkbox="showCheckbox">
         </Tree-node>
-        <div :class="[prefixCls + '-empty']" v-if="!data.length">{{ localeEmptyText }}</div>
+        <div :class="[prefixCls + '-empty']" v-if="!stateTree.length">{{ localeEmptyText }}</div>
     </div>
 </template>
 <script>
     import TreeNode from './node.vue';
-    import { findComponentsDownward } from '../../utils/assist';
     import Emitter from '../../mixins/emitter';
     import Locale from '../../mixins/locale';
 
     const prefixCls = 'ivu-tree';
 
-    let key = 1;
-
     export default {
         name: 'Tree',
         mixins: [ Emitter, Locale ],
@@ -46,92 +43,128 @@
         },
         data () {
             return {
-                prefixCls: prefixCls
+                prefixCls: prefixCls,
+                stateTree: JSON.parse(JSON.stringify(this.data)),
+                flatState: [],
             };
         },
+        watch: {
+            data(){
+                this.stateTree = JSON.parse(JSON.stringify(this.data));
+                this.flatState = this.compileFlatState();
+                this.rebuildTree();
+            }
+        },
         computed: {
             localeEmptyText () {
-                if (this.emptyText === undefined) {
+                if (typeof this.emptyText === 'undefined') {
                     return this.t('i.tree.emptyText');
                 } else {
                     return this.emptyText;
                 }
-            }
+            },
         },
         methods: {
+            compileFlatState () { // so we have always a relation parent/children of each node
+                let keyCounter = 0;
+                const flatTree = [];
+                function flattenChildren(node, parent) {
+                    node.nodeKey = keyCounter++;
+                    flatTree[node.nodeKey] = { node: node, nodeKey: node.nodeKey };
+                    if (typeof parent != 'undefined') {
+                        flatTree[node.nodeKey].parent = parent.nodeKey;
+                        flatTree[parent.nodeKey].children.push(node.nodeKey);
+                    }
+
+                    if (node.children) {
+                        flatTree[node.nodeKey].children = [];
+                        node.children.forEach(child => flattenChildren(child, node));
+                    }
+                }
+                this.stateTree.forEach(rootNode => {
+                    flattenChildren(rootNode);
+                });
+                return flatTree;
+            },
+            updateTreeUp(nodeKey){
+                const parentKey = this.flatState[nodeKey].parent;
+                if (typeof parentKey == 'undefined') return;
+
+                const node = this.flatState[nodeKey].node;
+                const parent = this.flatState[parentKey].node;
+                if (node.checked == parent.checked && node.indeterminate == parent.indeterminate) return; // no need to update upwards
+
+                if (node.checked == true) {
+                    this.$set(parent, 'checked', parent.children.every(node => node.checked));
+                    this.$set(parent, 'indeterminate', !parent.checked);
+                } else {
+                    this.$set(parent, 'checked', false);
+                    this.$set(parent, 'indeterminate', parent.children.some(node => node.checked || node.indeterminate));
+                }
+                this.updateTreeUp(parentKey);
+            },
+            rebuildTree () { // only called when `data` prop changes
+                const checkedNodes = this.getCheckedNodes();
+                checkedNodes.forEach(node => {
+                    this.updateTreeDown(node, {checked: true});
+                    // propagate upwards
+                    const parentKey = this.flatState[node.nodeKey].parent;
+                    if (!parentKey) return;
+                    const parent = this.flatState[parentKey].node;
+                    const childHasCheckSetter = typeof node.checked != 'undefined' && node.checked;
+                    if (childHasCheckSetter && parent.checked != node.checked) {
+                        this.updateTreeUp(node.nodeKey); // update tree upwards
+                    }
+                });
+            },
+
             getSelectedNodes () {
-                const nodes = findComponentsDownward(this, 'TreeNode');
-                return nodes.filter(node => node.data.selected).map(node => node.data);
+                /* public API */
+                return this.flatState.filter(obj => obj.node.selected).map(obj => obj.node);
             },
             getCheckedNodes () {
-                const nodes = findComponentsDownward(this, 'TreeNode');
-                return nodes.filter(node => node.data.checked).map(node => node.data);
+                /* public API */
+                return this.flatState.filter(obj => obj.node.checked).map(obj => obj.node);
             },
-            updateData (isInit = true) {
-                // init checked status
-                function reverseChecked(data) {
-                    if (!data.nodeKey) data.nodeKey = key++;
-                    if (data.children && data.children.length) {
-                        let checkedLength = 0;
-                        data.children.forEach(node => {
-                            if (node.children) node = reverseChecked(node);
-                            if (node.checked) checkedLength++;
-                        });
-                        if (isInit) {
-                            if (checkedLength >= data.children.length) data.checked = true;
-                        } else {
-                            data.checked = checkedLength >= data.children.length;
-                        }
-                        return data;
-                    } else {
-                        return data;
-                    }
+            updateTreeDown(node, changes = {}) {
+                for (let key in changes) {
+                    this.$set(node, key, changes[key]);
                 }
-                
-                function forwardChecked(data) {
-                    if (data.children) {
-                        data.children.forEach(node => {
-                            if (data.checked) node.checked = true;
-                            if (node.children) node = forwardChecked(node);
-                        });
-                        return data;
-                    } else {
-                        return data;
-                    }
+                if (node.children) {
+                    node.children.forEach(child => {
+                        this.updateTreeDown(child, changes);
+                    });
                 }
-                this.data.map(node => reverseChecked(node)).map(node => forwardChecked(node));
-                this.broadcast('TreeNode', 'indeterminate');
+            },
+            handleSelect (nodeKey) {
+                const node = this.flatState[nodeKey].node;
+                if (!this.multiple){ // reset selected
+                    const currentSelectedKey = this.flatState.findIndex(obj => obj.node.selected);
+                    if (currentSelectedKey >= 0) this.$set(this.flatState[currentSelectedKey].node, 'selected', false);
+                }
+                this.$set(node, 'selected', !node.selected);
+
+                this.$emit('on-select-change', this.getSelectedNodes());
+            },
+            handleCheck({ checked, nodeKey }) {
+                const node = this.flatState[nodeKey].node;
+                this.$set(node, 'checked', checked);
+                this.$set(node, 'indeterminate', false);
+
+                this.updateTreeUp(nodeKey); // propagate up
+                this.updateTreeDown(node, {checked, indeterminate: false}); // reset `indeterminate` when going down
+
+                this.$emit('on-check-change', this.getCheckedNodes());
             }
         },
+        created(){
+            this.flatState = this.compileFlatState();
+            this.rebuildTree();
+        },
         mounted () {
-            this.updateData();
-            this.$on('selected', ori => {
-                const nodes = findComponentsDownward(this, 'TreeNode');
-                nodes.forEach(node => {
-                    this.$set(node.data, 'selected', false);
-                });
-                this.$set(ori, 'selected', true);
-            });
-            this.$on('on-selected', () => {
-                this.$emit('on-select-change', this.getSelectedNodes());
-            });
-            this.$on('checked', () => {
-                this.updateData(false);
-            });
-            this.$on('on-checked', () => {
-                this.$emit('on-check-change', this.getCheckedNodes());
-            });
-            this.$on('toggle-expand', (payload) => {
-                this.$emit('on-toggle-expand', payload);
-            });
-        },
-        watch: {
-            data () {
-                this.$nextTick(() => {
-                    this.updateData();
-                    this.broadcast('TreeNode', 'indeterminate');
-                });
-            }
+            this.$on('on-check', this.handleCheck);
+            this.$on('on-selected', this.handleSelect);
+            this.$on('toggle-expand', node => this.$emit('on-toggle-expand', node));
         }
     };
-</script>
\ No newline at end of file
+</script>

From 0a8f9b43f6497c8dc0376e738b314304f12ed770 Mon Sep 17 00:00:00 2001
From: Sergio Crisostomo <sergiosbox@gmail.com>
Date: Tue, 24 Oct 2017 09:30:09 +0200
Subject: [PATCH 3/3] Keep original data pointers and correct if clause

---
 src/components/tree/tree.vue | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/components/tree/tree.vue b/src/components/tree/tree.vue
index eee3f305..4395da3c 100644
--- a/src/components/tree/tree.vue
+++ b/src/components/tree/tree.vue
@@ -44,13 +44,13 @@
         data () {
             return {
                 prefixCls: prefixCls,
-                stateTree: JSON.parse(JSON.stringify(this.data)),
+                stateTree: this.data,
                 flatState: [],
             };
         },
         watch: {
             data(){
-                this.stateTree = JSON.parse(JSON.stringify(this.data));
+                this.stateTree = this.data;
                 this.flatState = this.compileFlatState();
                 this.rebuildTree();
             }
@@ -109,7 +109,7 @@
                     this.updateTreeDown(node, {checked: true});
                     // propagate upwards
                     const parentKey = this.flatState[node.nodeKey].parent;
-                    if (!parentKey) return;
+                    if (!parentKey && parentKey !== 0) return;
                     const parent = this.flatState[parentKey].node;
                     const childHasCheckSetter = typeof node.checked != 'undefined' && node.checked;
                     if (childHasCheckSetter && parent.checked != node.checked) {