From 89219b29cf4c08816216f867eb7f340048ac35d6 Mon Sep 17 00:00:00 2001
From: cagataycivici <cagatay.civici@gmail.com>
Date: Fri, 2 Aug 2019 13:55:19 +0300
Subject: [PATCH] Added filtering to Tree

---
 src/components/tree/Tree.vue          | 94 ++++++++++++++++++++++++++-
 src/views/tree/TreeFilterDemo.vue     | 14 ++--
 src/views/tree/TreeTemplatingDemo.vue |  1 -
 3 files changed, 97 insertions(+), 12 deletions(-)

diff --git a/src/components/tree/Tree.vue b/src/components/tree/Tree.vue
index 74120de5b..d97d06bcb 100644
--- a/src/components/tree/Tree.vue
+++ b/src/components/tree/Tree.vue
@@ -6,8 +6,13 @@
                 <i :class="loadingIconClass" />
             </div>
         </template>
+        <div class="p-tree-filter-container" v-if="filter">
+            <input type="text" autocomplete="off" class="p-tree-filter p-inputtext p-component" :placeholder="filterPlaceholder"
+                @keydown="onFilterKeydown" v-model="filterValue" />
+                <span class="p-tree-filter-icon pi pi-search"></span>
+        </div>
         <ul class="p-tree-container" role="tree">
-            <TreeNode v-for="node of value" :key="node.key" :node="node" :templates="$scopedSlots" 
+            <TreeNode v-for="node of valueToRender" :key="node.key" :node="node" :templates="$scopedSlots" 
                 :expandedKeys="d_expandedKeys" @node-toggle="onNodeToggle" @node-click="onNodeClick"
                 :selectionMode="selectionMode" :selectionKeys="selectionKeys" @checkbox-change="onCheckboxChange"></TreeNode>
         </ul>
@@ -16,6 +21,7 @@
 
 <script>
 import TreeNode from './TreeNode';
+import ObjectUtils from '../utils/ObjectUtils';
 
 export default {
     props: {
@@ -46,11 +52,28 @@ export default {
         loadingIcon: {
             type: String,
             default: 'pi pi-spinner'
+        },
+        filter: {
+            type: Boolean,
+            default: false
+        },
+        filterBy: {
+            type: String,
+            default: 'label'
+        },
+        filterMode: {
+            type: String,
+            default: 'lenient'
+        },
+        filterPlaceholder: {
+            type: String,
+            default: null
         }
     },
     data() {
         return {
-            d_expandedKeys: this.expandedKeys || {}
+            d_expandedKeys: this.expandedKeys || {},
+            filterValue: null
         }
     },
     watch: {
@@ -167,6 +190,49 @@ export default {
         isChecked(node) {
             return this.selectionKeys ? this.selectionKeys[node.key] && this.selectionKeys[node.key].checked: false;
         },
+        isNodeLeaf(node) {
+            return node.leaf === false ? false : !(node.children && node.children.length);
+        },
+        onFilterKeydown(event) {
+            if (event.which === 13) {
+                event.preventDefault();
+            }
+        },
+        findFilteredNodes(node, paramsWithoutNode) {
+            if (node) {
+                let matched = false;
+                if (node.children) {
+                    let childNodes = [...node.children];
+                    node.children = [];
+                    for (let childNode of childNodes) {
+                        let copyChildNode = {...childNode};
+                        if (this.isFilterMatched(copyChildNode, paramsWithoutNode)) {
+                            matched = true;
+                            node.children.push(copyChildNode);
+                        }
+                    }
+                }
+                
+                if (matched) {
+                    return true;
+                }
+            }
+        },
+        isFilterMatched(node, {searchFields, filterText, strict}) {
+            let matched = false;
+            for(let field of searchFields) {
+                let fieldValue = String(ObjectUtils.resolveFieldData(node, field)).toLowerCase();
+                if(fieldValue.indexOf(filterText) > -1) {
+                    matched = true;
+                }
+            }
+
+            if (!matched || (strict && !this.isNodeLeaf(node))) {
+                matched = this.findFilteredNodes(node, {searchFields, filterText, strict}) || matched;
+            }
+
+            return matched;
+        }
     },
     computed: {
         containerClass() {
@@ -177,6 +243,30 @@ export default {
         },
         loadingIconClass() {
             return ['p-tree-loading-icon pi-spin', this.loadingIcon];
+        },
+        filteredValue() {
+            let filteredNodes = [];
+            const searchFields = this.filterBy.split(',');
+            const filterText = this.filterValue.trim().toLowerCase();
+            const strict = this.filterMode === 'strict';
+
+            for (let node of this.value) {
+                let _node = {...node};
+                let paramsWithoutNode = {searchFields, filterText, strict};
+                
+                if ((strict && (this.findFilteredNodes(_node, paramsWithoutNode) || this.isFilterMatched(_node, paramsWithoutNode))) ||
+                    (!strict && (this.isFilterMatched(_node, paramsWithoutNode) || this.findFilteredNodes(_node, paramsWithoutNode)))) {
+                    filteredNodes.push(_node);
+                }
+            }
+            
+            return filteredNodes;
+        },
+        valueToRender() {
+            if (this.filterValue && this.filterValue.trim().length > 0)
+                return this.filteredValue;
+            else
+                return this.value;
         }
     },
     components: {
diff --git a/src/views/tree/TreeFilterDemo.vue b/src/views/tree/TreeFilterDemo.vue
index fd2e9ef27..f9915db39 100644
--- a/src/views/tree/TreeFilterDemo.vue
+++ b/src/views/tree/TreeFilterDemo.vue
@@ -5,20 +5,16 @@
         <div class="content-section introduction">
             <div class="feature-intro">
                 <h1>Tree - Filter</h1>
-                <p>Tree is used to display hierarchical data.</p>
+                <p>Filtering updates the node based on the constraints.</p>
             </div>
         </div>
 
         <div class="content-section implementation">
-            <h3>Basic</h3>
-            <Tree :value="nodes"></Tree>
+            <h3>Lenient Filter</h3>
+            <Tree :value="nodes" :filter="true" filterMode="lenient"></Tree>
 
-            <h3>Programmatic Control</h3>
-            <div style="margin-bottom: 1em">
-                <Button type="button" icon="pi pi-plus" label="Expand All" @click="expandAll" />
-                <Button type="button" icon="pi pi-minus" label="Collapse All" @click="collapseAll" />
-            </div>
-            <Tree :value="nodes" :expandedKeys="expandedKeys"></Tree>            
+            <h3>Strict Filter</h3>
+            <Tree :value="nodes" :filter="true" filterMode="strict"></Tree>          
         </div>
 
         <TreeDoc />
diff --git a/src/views/tree/TreeTemplatingDemo.vue b/src/views/tree/TreeTemplatingDemo.vue
index 38839cde0..393dd219d 100644
--- a/src/views/tree/TreeTemplatingDemo.vue
+++ b/src/views/tree/TreeTemplatingDemo.vue
@@ -24,7 +24,6 @@
     </div>
 </template>
 <script>
-import NodeService from '../../service/NodeService';
 import TreeDoc from './TreeDoc';
 import TreeSubMenu from './TreeSubMenu';