misp-ui-library-test 0.0.1

package/src/packages/tree/src/model/node.js

@@ -0,0 +1,348 @@
+import Vue from 'vue';
+
+let nodeIdSeed = 0;
+
+export default class Node {
+  constructor(options) {
+    this.id = nodeIdSeed++;
+    this.label = null;
+    this.checked = false;
+    this.indeterminate = false;
+    this.data = null;
+    this.expanded = false;
+    this.parent = null;
+    this.visible = true;
+    this.isCurrent = false;
+    this.isLeaf = false;
+
+    for (let name in options) {
+      if (Object.prototype.hasOwnProperty.call(options, name)) {
+        this[name] = options[name];
+      }
+    }
+
+    this.level = 0;
+    this.loaded = false;
+    this.loading = false;
+    this.childNodes = [];
+
+    if (this.parent) {
+      this.level = this.parent.level + 1;
+    }
+
+    const store = this.store;
+    if (!store) {
+      throw new Error('[Node] store is required!');
+    }
+    store.registerNode(this);
+
+    const props = store.props;
+    if (props && typeof props.label === 'function') {
+      this.label = props.label(this.data, this);
+    } else if (props && typeof props.label === 'string') {
+      this.label = this.getPropByPath(this.data, props.label);
+    }
+
+    if (props && typeof props.disabled === 'function') {
+      this.disabled = props.disabled(this.data, this);
+    } else if (props && typeof props.disabled === 'string') {
+      this.disabled = this.getPropByPath(this.data, props.disabled);
+    }
+
+    if (store.lazy !== true && this.data) {
+      this.setData(this.data);
+
+      if (store.defaultExpandAll) {
+        this.expanded = true;
+      }
+    } else if (this.level > 0 && store.lazy && store.defaultExpandAll) {
+      this.expand();
+    }
+
+    if (!this.data) return;
+
+    this.updateLeafState();
+  }
+
+  setData(data) {
+    if (!Array.isArray(data)) {
+      this.markAsCheckable();
+    }
+
+    this.data = data;
+    Vue.set(this, 'childNodes', []);
+
+    let children;
+    if (this.level === 0 && this.data instanceof Array) {
+      children = this.data;
+    } else {
+      children = this.getPropByPath(this.data, this.store.props.children) || [];
+    }
+
+    for (let i = 0, j = children.length; i < j; i++) {
+      this.insertChild({ data: children[i] });
+    }
+
+    if (this.level !== 0) {
+      this.updateLeafState();
+    }
+  }
+
+  getPropByPath(obj, path) {
+    if (!path) return null;
+    let tempObj = obj;
+    let key = path.split('.');
+    for (let i = 0; i < key.length; i++) {
+      if (!tempObj) break;
+      tempObj = tempObj[key[i]];
+    }
+    return tempObj;
+  }
+
+  markAsCheckable() {
+    const store = this.store;
+    const key = store.key;
+    if (key && store.defaultCheckedKeys && store.defaultCheckedKeys.indexOf(this.data[key]) > -1) {
+      this.setChecked(true, store.checkStrictly);
+    }
+  }
+
+  insertChild(child, index) {
+    if (!child) throw new Error('insertChild error: child is required.');
+
+    if (!(child instanceof Node)) {
+      Object.assign(child, {
+        parent: this,
+        store: this.store
+      });
+      child = new Node(child);
+    }
+
+    child.level = this.level + 1;
+
+    if (typeof index === 'undefined' || index < 0) {
+      this.childNodes.push(child);
+    } else {
+      this.childNodes.splice(index, 0, child);
+    }
+
+    this.updateLeafState();
+  }
+
+  removeChild(child) {
+    const index = this.childNodes.indexOf(child);
+
+    if (index > -1) {
+      this.store && this.store.deregisterNode(child);
+      child.parent = null;
+      this.childNodes.splice(index, 1);
+    }
+
+    this.updateLeafState();
+  }
+
+  removeChildByData(data) {
+    let targetNode = null;
+    for (let i = 0; i < this.childNodes.length; i++) {
+      if (this.childNodes[i].data === data) {
+        targetNode = this.childNodes[i];
+        break;
+      }
+    }
+
+    if (targetNode) {
+      this.removeChild(targetNode);
+    }
+  }
+
+  expand(callback, expandParent) {
+    const done = () => {
+      if (expandParent) {
+        let parent = this.parent;
+        while (parent && parent.level !== -1) {
+          Vue.set(parent, 'expanded', true);
+          parent = parent.parent;
+        }
+      }
+      Vue.set(this, 'expanded', true);
+      if (callback) callback();
+    };
+
+    if (this.shouldLoadData()) {
+      this.loadData((data) => {
+        if (data instanceof Array) {
+          if (this.checked) {
+            this.setChecked(true, true);
+          }
+          done();
+        }
+      });
+    } else {
+      done();
+    }
+  }
+
+  shouldLoadData() {
+    return this.store.lazy && this.store.load && !this.loaded && !this.loading;
+  }
+
+  doCreateChildren(array, defaultProps = {}) {
+    array.forEach((item) => {
+      this.insertChild(Object.assign({ data: item }, defaultProps));
+    });
+  }
+
+  loadData(callback) {
+    if (this.store.lazy && this.store.load && !this.loaded && !this.loading) {
+      Vue.set(this, 'loading', true);
+
+      const done = (children) => {
+        Vue.set(this, 'loaded', true);
+        Vue.set(this, 'loading', false);
+        Vue.set(this, 'childNodes', []);
+
+        if (children instanceof Array) {
+          this.doCreateChildren(children);
+        }
+
+        this.updateLeafState();
+
+        if (callback) {
+          callback.call(this, children);
+        }
+      };
+
+      this.store.load(this, done);
+    } else {
+      if (callback) {
+        callback.call(this);
+      }
+    }
+  }
+
+  collapse() {
+    Vue.set(this, 'expanded', false);
+  }
+
+
+  setChecked(value, deep, recursion, passValue) {
+    Vue.set(this, 'indeterminate', value === 'half');
+    Vue.set(this, 'checked', value === true);
+
+    if (this.store.checkStrictly) return;
+
+    if (!(this.store.checkDescendants && recursion)) {
+      if (deep) {
+        const childNodes = this.childNodes;
+        for (let i = 0, j = childNodes.length; i < j; i++) {
+          const child = childNodes[i];
+          passValue = passValue !== false ? value !== false : passValue;
+          let isCheck = child.disabled ? child.checked : passValue;
+          child.setChecked(isCheck, deep, true, passValue);
+        }
+      }
+
+      const parent = this.parent;
+      if (!parent || parent.level === -1) return;
+      parent.recalculateCheckedState();
+    }
+  }
+
+  recalculateCheckedState() {
+    if (this.store.checkStrictly) return;
+    const childNodes = this.childNodes;
+    let checkedCount = 0;
+    let indeterminateCount = 0;
+    let allCount = childNodes.length;
+
+    for (let i = 0; i < allCount; i++) {
+      const child = childNodes[i];
+      if (child.checked) {
+        checkedCount++;
+      } else if (child.indeterminate) {
+        indeterminateCount++;
+      }
+    }
+
+    if (checkedCount === allCount && allCount > 0) {
+      this.checked = true;
+      this.indeterminate = false;
+    } else if (checkedCount > 0 || indeterminateCount > 0) {
+      this.checked = false;
+      this.indeterminate = true;
+    } else {
+      this.checked = false;
+      this.indeterminate = false;
+    }
+
+    const parent = this.parent;
+    if (parent && parent.level !== -1) {
+      parent.recalculateCheckedState();
+    }
+  }
+
+  updateChildren() {
+    const newData = this.getChildren() || [];
+    const oldData = this.childNodes.map((node) => node.data);
+
+    const newDataMap = {};
+    const newNodes = [];
+
+    newData.forEach((item, index) => {
+      const key = item[this.store.key];
+      const isNodeExists = !!key && oldData.findIndex(data => data[this.store.key] === key) >= 0;
+      if (isNodeExists) {
+        newDataMap[key] = { index, data: item };
+      } else {
+        newNodes.push({ index, data: item });
+      }
+    });
+
+    if (!this.store.key) {
+      newNodes.length = 0;
+      newData.forEach((item, index) => {
+        newNodes.push({ index, data: item });
+      });
+    }
+
+    oldData.forEach((item) => {
+      if (!this.store.key) this.removeChildByData(item);
+      else {
+        const key = item[this.store.key];
+        if (!newDataMap[key]) this.removeChildByData(item);
+      }
+    });
+
+    newNodes.forEach(({ index, data }) => {
+      this.insertChild({ data }, index);
+    });
+
+    this.updateLeafState();
+  }
+
+  updateLeafState() {
+    if (this.store.lazy && this.loaded && typeof this.store.props.isLeaf !== 'function') {
+      this.isLeaf = !this.childNodes || this.childNodes.length === 0;
+      return;
+    }
+    const isLeaf = this.store.props.isLeaf ? this.getPropByPath(this.data, this.store.props.isLeaf) : undefined;
+    if (typeof isLeaf === 'boolean') {
+      this.isLeaf = isLeaf;
+    } else if (this.store.lazy && !this.loaded) {
+      this.isLeaf = false;
+    } else {
+      this.isLeaf = !this.childNodes || this.childNodes.length === 0;
+    }
+  }
+
+  getChildren() {
+    const data = this.data;
+    if (!data) return null;
+    const props = this.store.props;
+    let children = 'children';
+    if (props) {
+      children = props.children || 'children';
+    }
+    return data[children];
+  }
+}

package/src/packages/tree/src/model/tree-store.js

@@ -0,0 +1,354 @@
+import Vue from 'vue';
+import Node from './node';
+
+export default class TreeStore {
+  constructor(options) {
+    this.currentNode = null;
+    this.nodesMap = {};
+    this.root = null;
+    this.data = null;
+    this.props = {
+      children: 'children',
+      label: 'label',
+      disabled: 'disabled',
+      isLeaf: 'isLeaf'
+    };
+    this.lazy = false;
+    this.load = null;
+    this.filterNodeMethod = null;
+    this.key = null;
+    this.defaultExpandedKeys = [];
+    this.defaultCheckedKeys = [];
+    this.checkStrictly = false;
+    this.checkDescendants = false;
+    this.defaultIsAllExpanded = false;
+    this.expandAndLoadLevel = 0;
+
+    for (let name in options) {
+      if (Object.prototype.hasOwnProperty.call(options, name)) {
+        this[name] = options[name];
+      }
+    }
+
+    this.nodesMap = {};
+
+    this.root = new Node({
+      data: this.data,
+      store: this
+    });
+
+    if (this.lazy && this.load) {
+      this.root.loadData(() => {
+        this._initDefaultCheckedNodes();
+        this.setDefaultExpandedKeys(this.defaultExpandedKeys);
+      });
+    } else {
+      this._initDefaultCheckedNodes();
+      this.setDefaultExpandedKeys(this.defaultExpandedKeys);
+    }
+  }
+
+  registerNode(node) {
+    const key = this.key;
+    if (!key || !node || !node.data) return;
+
+    const nodeKey = node.data[key];
+    if (nodeKey !== undefined) this.nodesMap[nodeKey] = node;
+  }
+
+  deregisterNode(node) {
+    const key = this.key;
+    if (!key || !node || !node.data) return;
+
+    node.childNodes.forEach(child => {
+      this.deregisterNode(child);
+    });
+
+    delete this.nodesMap[node.data[key]];
+  }
+
+  _initDefaultCheckedNodes() {
+    const defaultCheckedKeys = this.defaultCheckedKeys || [];
+    const nodesMap = this.nodesMap;
+
+    defaultCheckedKeys.forEach((key) => {
+      const node = nodesMap[key];
+
+      if (node) {
+        node.setChecked(true, !this.checkStrictly);
+      }
+    });
+  }
+
+  filter(value) {
+    const filterNodeMethod = this.filterNodeMethod;
+    const traverse = function(node) {
+      const childNodes = node.root ? node.root.childNodes : node.childNodes;
+
+      childNodes.forEach((child) => {
+        child.visible = filterNodeMethod.call(child, value, child.data, child);
+        traverse(child);
+      });
+
+      if (!node.visible && childNodes.length) {
+        let allHidden = true;
+        allHidden = childNodes.every(child => !child.visible);
+
+        if (node.root) {
+          node.root.visible = allHidden === false;
+        } else {
+          node.visible = allHidden === false;
+        }
+      }
+
+      if (!value) return;
+
+      if (node.visible && !node.isLeaf && !node.root) node.expand();
+    };
+
+    traverse(this);
+  }
+
+  setData(newVal) {
+    const instanceChanged = newVal !== this.root.data;
+    if (instanceChanged) {
+      this.root.setData(newVal);
+      this._initDefaultCheckedNodes();
+    } else {
+      this.root.updateChildren();
+    }
+  }
+
+  getNode(data) {
+    if (data instanceof Node) return data;
+    const key = typeof data !== 'object' ? data : data[this.key];
+    return this.nodesMap[key] || null;
+  }
+
+  _getAllNodes() {
+    const allNodes = [];
+    const nodesMap = this.nodesMap;
+    for (let nodeKey in nodesMap) {
+      if (Object.prototype.hasOwnProperty.call(nodesMap, nodeKey)) {
+        allNodes.push(nodesMap[nodeKey]);
+      }
+    }
+    return allNodes;
+  }
+
+  _setCheckedKeys(key, leafOnly = false, checkedKeys) {
+    const allNodes = this._getAllNodes().sort((a, b) => b.level - a.level);
+    const cache = Object.create(null);
+    const keys = Object.keys(checkedKeys);
+
+    // First uncheck all nodes without deep propagation
+    allNodes.forEach(node => node.setChecked(false, false));
+
+    for (let i = 0, j = allNodes.length; i < j; i++) {
+      const node = allNodes[i];
+      const nodeKey = node.data[key].toString();
+      let checked = keys.indexOf(nodeKey) > -1;
+
+      if (!checked) {
+        if (node.checked && !cache[nodeKey]) {
+          node.setChecked(false, false);
+        }
+        continue;
+      }
+
+      // Cache parent chain to prevent them from being unchecked
+      let parent = node.parent;
+      while (parent && parent.level > 0) {
+        cache[parent.data[key]] = true;
+        parent = parent.parent;
+      }
+
+      if (node.isLeaf || this.checkStrictly) {
+        node.setChecked(true, false);
+        continue;
+      }
+      node.setChecked(true, true);
+
+      if (leafOnly) {
+        node.setChecked(false, false);
+        const traverse = function(node) {
+          const childNodes = node.childNodes;
+          childNodes.forEach((child) => {
+            if (!child.isLeaf) {
+              child.setChecked(false, false);
+            }
+            traverse(child);
+          });
+        };
+        traverse(node);
+      }
+    }
+  }
+
+  setCheckedKeys(keys, leafOnly = false) {
+    this.defaultCheckedKeys = keys;
+    const key = this.key;
+    const checkedKeys = {};
+    keys.forEach((k) => {
+      checkedKeys[k] = true;
+    });
+
+    this._setCheckedKeys(key, leafOnly, checkedKeys);
+  }
+
+  setCheckedNodes(array, leafOnly = false) {
+    const key = this.key;
+    const checkedKeys = {};
+    array.forEach((item) => {
+      checkedKeys[(item || {})[key]] = true;
+    });
+
+    this._setCheckedKeys(key, leafOnly, checkedKeys);
+  }
+
+  getCheckedNodes(leafOnly = false, includeHalfChecked = false) {
+    const checkedNodes = [];
+    const traverse = function(node) {
+      const childNodes = node.root ? node.root.childNodes : node.childNodes;
+
+      childNodes.forEach((child) => {
+        if ((child.checked || (includeHalfChecked && child.indeterminate)) && (!leafOnly || (leafOnly && child.isLeaf))) {
+          checkedNodes.push(child.data);
+        }
+
+        traverse(child);
+      });
+    };
+
+    traverse(this);
+
+    return checkedNodes;
+  }
+
+  getCheckedKeys(leafOnly = false) {
+    return this.getCheckedNodes(leafOnly).map((data) => (data || {})[this.key]);
+  }
+
+  getHalfCheckedNodes() {
+    const nodes = [];
+    const traverse = function(node) {
+      const childNodes = node.root ? node.root.childNodes : node.childNodes;
+
+      childNodes.forEach((child) => {
+        if (child.indeterminate) {
+          nodes.push(child.data);
+        }
+
+        traverse(child);
+      });
+    };
+
+    traverse(this);
+
+    return nodes;
+  }
+
+  getHalfCheckedKeys() {
+    return this.getHalfCheckedNodes().map((data) => (data || {})[this.key]);
+  }
+
+  setChecked(data, checked, deep) {
+    const node = this.getNode(data);
+
+    if (node) {
+      node.setChecked(!!checked, deep);
+    }
+  }
+
+  setDefaultExpandedKeys(keys) {
+    this.defaultExpandedKeys = keys || [];
+    const keysMap = {};
+    this.defaultExpandedKeys.forEach((key) => {
+      keysMap[key] = true;
+    });
+
+    const traverse = (node) => {
+      const childNodes = node.root ? node.root.childNodes : node.childNodes;
+
+      childNodes.forEach((child) => {
+        const key = child.data[this.key];
+        if (keysMap[key]) {
+          child.expand();
+        }
+        traverse(child);
+      });
+    };
+
+    traverse(this);
+  }
+
+  setUserCurrentNode(node) {
+    const key = node[this.key];
+    const currNode = this.nodesMap[key];
+    this.setCurrentNode(currNode);
+  }
+
+  setCurrentNodeKey(key) {
+    const node = this.nodesMap[key];
+    if (node) {
+      this.setCurrentNode(node);
+    }
+  }
+
+  remove(data) {
+    const node = this.getNode(data);
+
+    if (node && node.parent) {
+      if (this.currentNode === node) {
+        this.currentNode = null;
+      }
+      node.parent.removeChild(node);
+    }
+  }
+
+  append(data, parentData) {
+    const parentNode = parentData ? this.getNode(parentData) : this.root;
+
+    if (parentNode) {
+      parentNode.insertChild({ data });
+    }
+  }
+
+  insertBefore(data, refData) {
+    const refNode = this.getNode(refData);
+
+    if (refNode && refNode.parent) {
+      refNode.parent.insertChild({ data }, refNode.parent.childNodes.indexOf(refNode));
+    }
+  }
+
+  insertAfter(data, refData) {
+    const refNode = this.getNode(refData);
+
+    if (refNode && refNode.parent) {
+      refNode.parent.insertChild({ data }, refNode.parent.childNodes.indexOf(refNode) + 1);
+    }
+  }
+
+  updateKeyChildren(key, data) {
+    const node = this.nodesMap[key];
+    if (!node) return;
+
+    node.setData(data);
+  }
+
+  setCurrentNode(node) {
+    const prevCurrentNode = this.currentNode;
+    if (prevCurrentNode) {
+      Vue.set(prevCurrentNode, 'isCurrent', false);
+    }
+    this.currentNode = node;
+    if (node) {
+      Vue.set(this.currentNode, 'isCurrent', true);
+    }
+  }
+
+  getCurrentNode() {
+    return this.currentNode ? this.currentNode.data : null;
+  }
+}

package/src/packages/tree/src/model/util.js

@@ -0,0 +1,10 @@
+export const getNodeKey = function(key, data) {
+  if (!key) return data.id || data._id;
+  const keys = key.split('.');
+  let tempObj = data;
+  for (let i = 0; i < keys.length; i++) {
+    if (!tempObj) break;
+    tempObj = tempObj[keys[i]];
+  }
+  return tempObj;
+};