extract getNodeClassName and getAnchorClassName

plantain-00 · Feb 6, 2017 · f3ed1dc · f3ed1dc
1 parent 84f5f46
commit f3ed1dc
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 75 deletions.
diff --git a/src/angular.ts b/src/angular.ts
@@ -32,34 +32,11 @@ export class NodeComponent {
     doubleClick = new common.DoubleClick();
 
     get nodeClassName() {
-        const values = ["jstree-node"];
-        if (this.data.children && this.data.children.length > 0) {
-            if (this.data.state.opened) {
-                values.push("jstree-open");
-            } else {
-                values.push("jstree-closed");
-            }
-        } else {
-            values.push("jstree-leaf");
-        }
-        if (this.last) {
-            values.push("jstree-last");
-        }
-        return values.join(" ");
+        return common.getNodeClassName(this.data, this.last);
     }
 
     get anchorClassName() {
-        const values = ["jstree-anchor"];
-        if (this.data.state.selected) {
-            values.push("jstree-clicked");
-        }
-        if (this.data.state.disabled) {
-            values.push("jstree-disabled");
-        }
-        if (this.hovered) {
-            values.push("jstree-hovered");
-        }
-        return values.join(" ");
+        return common.getAnchorClassName(this.data, this.hovered);
     }
 
     hover(hovered: boolean) {

diff --git a/src/common.ts b/src/common.ts
@@ -38,3 +38,34 @@ export class DoubleClick {
         }
     }
 }
+
+export function getNodeClassName(data: TreeData, last: boolean) {
+    const values = ["jstree-node"];
+    if (data.children && data.children.length > 0) {
+        if (data.state.opened) {
+            values.push("jstree-open");
+        } else {
+            values.push("jstree-closed");
+        }
+    } else {
+        values.push("jstree-leaf");
+    }
+    if (last) {
+        values.push("jstree-last");
+    }
+    return values.join(" ");
+}
+
+export function getAnchorClassName(data: TreeData, hovered: boolean) {
+    const values = ["jstree-anchor"];
+    if (data.state.selected) {
+        values.push("jstree-clicked");
+    }
+    if (data.state.disabled) {
+        values.push("jstree-disabled");
+    }
+    if (hovered) {
+        values.push("jstree-hovered");
+    }
+    return values.join(" ");
+}
diff --git a/src/react.tsx b/src/react.tsx
@@ -37,34 +37,11 @@ class Node extends React.Component<{
     }
 
     get nodeClassName() {
-        const values = ["jstree-node"];
-        if (this.props.data.children && this.props.data.children.length > 0) {
-            if (this.props.data.state.opened) {
-                values.push("jstree-open");
-            } else {
-                values.push("jstree-closed");
-            }
-        } else {
-            values.push("jstree-leaf");
-        }
-        if (this.props.last) {
-            values.push("jstree-last");
-        }
-        return values.join(" ");
+        return common.getNodeClassName(this.props.data, this.props.last);
     }
 
     get anchorClassName() {
-        const values = ["jstree-anchor"];
-        if (this.props.data.state.selected) {
-            values.push("jstree-clicked");
-        }
-        if (this.props.data.state.disabled) {
-            values.push("jstree-disabled");
-        }
-        if (this.hovered) {
-            values.push("jstree-hovered");
-        }
-        return values.join(" ");
+        return common.getAnchorClassName(this.props.data, this.hovered);
     }
 
     hover(hovered: boolean) {

diff --git a/src/vue.ts b/src/vue.ts
@@ -26,34 +26,11 @@ class Node extends Vue {
     doubleClick = new common.DoubleClick();
 
     get nodeClassName() {
-        const values = ["jstree-node"];
-        if (this.data.children && this.data.children.length > 0) {
-            if (this.data.state.opened) {
-                values.push("jstree-open");
-            } else {
-                values.push("jstree-closed");
-            }
-        } else {
-            values.push("jstree-leaf");
-        }
-        if (this.last) {
-            values.push("jstree-last");
-        }
-        return values.join(" ");
+        return common.getNodeClassName(this.data, this.last);
     }
 
     get anchorClassName() {
-        const values = ["jstree-anchor"];
-        if (this.data.state.selected) {
-            values.push("jstree-clicked");
-        }
-        if (this.data.state.disabled) {
-            values.push("jstree-disabled");
-        }
-        if (this.hovered) {
-            values.push("jstree-hovered");
-        }
-        return values.join(" ");
+        return common.getAnchorClassName(this.data, this.hovered);
     }
 
     hover(hovered: boolean) {