parent
05d2d51886
commit
b34d2fc3db
14 changed files with 1667 additions and 14 deletions
@ -0,0 +1,8 @@ |
||||
import Tree from './src/tree.vue'; |
||||
|
||||
/* istanbul ignore next */ |
||||
Tree.install = function(Vue) { |
||||
Vue.component(Tree.name, Tree); |
||||
}; |
||||
|
||||
export default Tree; |
||||
@ -0,0 +1,486 @@ |
||||
import objectAssign from 'element-ui/src/utils/merge'; |
||||
import { markNodeData, NODE_KEY } from './util'; |
||||
import { arrayFindIndex } from 'element-ui/src/utils/util'; |
||||
|
||||
export const getChildState = node => { |
||||
let all = true; |
||||
let none = true; |
||||
let allWithoutDisable = true; |
||||
for (let i = 0, j = node.length; i < j; i++) { |
||||
const n = node[i]; |
||||
if (n.checked !== true || n.indeterminate) { |
||||
all = false; |
||||
if (!n.disabled) { |
||||
allWithoutDisable = false; |
||||
} |
||||
} |
||||
if (n.checked !== false || n.indeterminate) { |
||||
none = false; |
||||
} |
||||
} |
||||
|
||||
return { all, none, allWithoutDisable, half: !all && !none }; |
||||
}; |
||||
|
||||
const reInitChecked = function(node) { |
||||
if (node.childNodes.length === 0) return; |
||||
|
||||
const {all, none, half} = getChildState(node.childNodes); |
||||
if (all) { |
||||
node.checked = true; |
||||
node.indeterminate = false; |
||||
} else if (half) { |
||||
node.checked = false; |
||||
node.indeterminate = true; |
||||
} else if (none) { |
||||
node.checked = false; |
||||
node.indeterminate = false; |
||||
} |
||||
|
||||
const parent = node.parent; |
||||
if (!parent || parent.level === 0) return; |
||||
|
||||
if (!node.store.checkStrictly) { |
||||
reInitChecked(parent); |
||||
} |
||||
}; |
||||
|
||||
const getPropertyFromData = function(node, prop) { |
||||
const props = node.store.props; |
||||
const data = node.data || {}; |
||||
const config = props[prop]; |
||||
|
||||
if (typeof config === 'function') { |
||||
return config(data, node); |
||||
} else if (typeof config === 'string') { |
||||
return data[config]; |
||||
} else if (typeof config === 'undefined') { |
||||
const dataProp = data[prop]; |
||||
return dataProp === undefined ? '' : dataProp; |
||||
} |
||||
}; |
||||
|
||||
let nodeIdSeed = 0; |
||||
|
||||
export default class Node { |
||||
constructor(options) { |
||||
this.id = nodeIdSeed++; |
||||
this.text = null; |
||||
this.checked = false; |
||||
this.indeterminate = false; |
||||
this.data = null; |
||||
this.expanded = false; |
||||
this.parent = null; |
||||
this.visible = true; |
||||
this.isCurrent = false; |
||||
|
||||
for (let name in options) { |
||||
if (options.hasOwnProperty(name)) { |
||||
this[name] = options[name]; |
||||
} |
||||
} |
||||
|
||||
// internal
|
||||
this.level = 0; |
||||
this.loaded = false; |
||||
this.childNodes = []; |
||||
this.loading = false; |
||||
|
||||
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.isLeaf !== 'undefined') { |
||||
const isLeaf = getPropertyFromData(this, 'isLeaf'); |
||||
if (typeof isLeaf === 'boolean') { |
||||
this.isLeafByUser = isLeaf; |
||||
} |
||||
} |
||||
|
||||
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 (!Array.isArray(this.data)) { |
||||
markNodeData(this, this.data); |
||||
} |
||||
if (!this.data) return; |
||||
const defaultExpandedKeys = store.defaultExpandedKeys; |
||||
const key = store.key; |
||||
if (key && defaultExpandedKeys && defaultExpandedKeys.indexOf(this.key) !== -1) { |
||||
this.expand(null, store.autoExpandParent); |
||||
} |
||||
|
||||
if (key && store.currentNodeKey !== undefined && this.key === store.currentNodeKey) { |
||||
store.currentNode = this; |
||||
store.currentNode.isCurrent = true; |
||||
} |
||||
|
||||
if (store.lazy) { |
||||
store._initDefaultCheckedNode(this); |
||||
} |
||||
|
||||
this.updateLeafState(); |
||||
} |
||||
|
||||
setData(data) { |
||||
if (!Array.isArray(data)) { |
||||
markNodeData(this, data); |
||||
} |
||||
|
||||
this.data = data; |
||||
this.childNodes = []; |
||||
|
||||
let children; |
||||
if (this.level === 0 && this.data instanceof Array) { |
||||
children = this.data; |
||||
} else { |
||||
children = getPropertyFromData(this, 'children') || []; |
||||
} |
||||
|
||||
for (let i = 0, j = children.length; i < j; i++) { |
||||
this.insertChild({ data: children[i] }); |
||||
} |
||||
} |
||||
|
||||
get label() { |
||||
return getPropertyFromData(this, 'label'); |
||||
} |
||||
|
||||
get key() { |
||||
const nodeKey = this.store.key; |
||||
if (this.data) return this.data[nodeKey]; |
||||
return null; |
||||
} |
||||
|
||||
get disabled() { |
||||
return getPropertyFromData(this, 'disabled'); |
||||
} |
||||
|
||||
get nextSibling() { |
||||
const parent = this.parent; |
||||
if (parent) { |
||||
const index = parent.childNodes.indexOf(this); |
||||
if (index > -1) { |
||||
return parent.childNodes[index + 1]; |
||||
} |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
get previousSibling() { |
||||
const parent = this.parent; |
||||
if (parent) { |
||||
const index = parent.childNodes.indexOf(this); |
||||
if (index > -1) { |
||||
return index > 0 ? parent.childNodes[index - 1] : null; |
||||
} |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
contains(target, deep = true) { |
||||
const walk = function(parent) { |
||||
const children = parent.childNodes || []; |
||||
let result = false; |
||||
for (let i = 0, j = children.length; i < j; i++) { |
||||
const child = children[i]; |
||||
if (child === target || (deep && walk(child))) { |
||||
result = true; |
||||
break; |
||||
} |
||||
} |
||||
return result; |
||||
}; |
||||
|
||||
return walk(this); |
||||
} |
||||
|
||||
remove() { |
||||
const parent = this.parent; |
||||
if (parent) { |
||||
parent.removeChild(this); |
||||
} |
||||
} |
||||
|
||||
insertChild(child, index, batch) { |
||||
if (!child) throw new Error('insertChild error: child is required.'); |
||||
|
||||
if (!(child instanceof Node)) { |
||||
if (!batch) { |
||||
const children = this.getChildren(true); |
||||
if (children.indexOf(child.data) === -1) { |
||||
if (typeof index === 'undefined' || index < 0) { |
||||
children.push(child.data); |
||||
} else { |
||||
children.splice(index, 0, child.data); |
||||
} |
||||
} |
||||
} |
||||
objectAssign(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(); |
||||
} |
||||
|
||||
insertBefore(child, ref) { |
||||
let index; |
||||
if (ref) { |
||||
index = this.childNodes.indexOf(ref); |
||||
} |
||||
this.insertChild(child, index); |
||||
} |
||||
|
||||
insertAfter(child, ref) { |
||||
let index; |
||||
if (ref) { |
||||
index = this.childNodes.indexOf(ref); |
||||
if (index !== -1) index += 1; |
||||
} |
||||
this.insertChild(child, index); |
||||
} |
||||
|
||||
removeChild(child) { |
||||
const children = this.getChildren() || []; |
||||
const dataIndex = children.indexOf(child.data); |
||||
if (dataIndex > -1) { |
||||
children.splice(dataIndex, 1); |
||||
} |
||||
|
||||
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.level > 0) { |
||||
parent.expanded = true; |
||||
parent = parent.parent; |
||||
} |
||||
} |
||||
this.expanded = true; |
||||
if (callback) callback(); |
||||
}; |
||||
|
||||
if (this.shouldLoadData()) { |
||||
this.loadData((data) => { |
||||
if (data instanceof Array) { |
||||
if (this.checked) { |
||||
this.setChecked(true, true); |
||||
} else if (!this.store.checkStrictly) { |
||||
reInitChecked(this); |
||||
} |
||||
done(); |
||||
} |
||||
}); |
||||
} else { |
||||
done(); |
||||
} |
||||
} |
||||
|
||||
doCreateChildren(array, defaultProps = {}) { |
||||
array.forEach((item) => { |
||||
this.insertChild(objectAssign({ data: item }, defaultProps), undefined, true); |
||||
}); |
||||
} |
||||
|
||||
collapse() { |
||||
this.expanded = false; |
||||
} |
||||
|
||||
shouldLoadData() { |
||||
return this.store.lazy === true && this.store.load && !this.loaded; |
||||
} |
||||
|
||||
updateLeafState() { |
||||
if (this.store.lazy === true && this.loaded !== true && typeof this.isLeafByUser !== 'undefined') { |
||||
this.isLeaf = this.isLeafByUser; |
||||
return; |
||||
} |
||||
const childNodes = this.childNodes; |
||||
if (!this.store.lazy || (this.store.lazy === true && this.loaded === true)) { |
||||
// this.isLeaf = !childNodes || childNodes.length === 0;
|
||||
this.isLeaf = this.isLeafByUser; |
||||
return; |
||||
} |
||||
this.isLeaf = false; |
||||
} |
||||
|
||||
setChecked(value, deep, recursion, passValue) { |
||||
this.indeterminate = value === 'half'; |
||||
this.checked = value === true; |
||||
|
||||
if (this.store.checkStrictly) return; |
||||
|
||||
if (!(this.shouldLoadData() && !this.store.checkDescendants)) { |
||||
let { all, allWithoutDisable } = getChildState(this.childNodes); |
||||
|
||||
if (!this.isLeaf && (!all && allWithoutDisable)) { |
||||
this.checked = false; |
||||
value = false; |
||||
} |
||||
|
||||
const handleDescendants = () => { |
||||
if (deep) { |
||||
const childNodes = this.childNodes; |
||||
for (let i = 0, j = childNodes.length; i < j; i++) { |
||||
const child = childNodes[i]; |
||||
passValue = passValue || value !== false; |
||||
const isCheck = child.disabled ? child.checked : passValue; |
||||
child.setChecked(isCheck, deep, true, passValue); |
||||
} |
||||
const { half, all } = getChildState(childNodes); |
||||
if (!all) { |
||||
this.checked = all; |
||||
this.indeterminate = half; |
||||
} |
||||
} |
||||
}; |
||||
|
||||
if (this.shouldLoadData()) { |
||||
// Only work on lazy load data.
|
||||
this.loadData(() => { |
||||
handleDescendants(); |
||||
reInitChecked(this); |
||||
}, { |
||||
checked: value !== false |
||||
}); |
||||
return; |
||||
} else { |
||||
handleDescendants(); |
||||
} |
||||
} |
||||
|
||||
const parent = this.parent; |
||||
if (!parent || parent.level === 0) return; |
||||
|
||||
if (!recursion) { |
||||
reInitChecked(parent); |
||||
} |
||||
} |
||||
|
||||
getChildren(forceInit = false) { // this is data
|
||||
if (this.level === 0) return this.data; |
||||
const data = this.data; |
||||
if (!data) return null; |
||||
|
||||
const props = this.store.props; |
||||
let children = 'children'; |
||||
if (props) { |
||||
children = props.children || 'children'; |
||||
} |
||||
|
||||
if (data[children] === undefined) { |
||||
data[children] = null; |
||||
} |
||||
|
||||
if (forceInit && !data[children]) { |
||||
data[children] = []; |
||||
} |
||||
|
||||
return data[children]; |
||||
} |
||||
|
||||
updateChildren() { |
||||
const newData = this.getChildren() || []; |
||||
const oldData = this.childNodes.map((node) => node.data); |
||||
|
||||
const newDataMap = {}; |
||||
const newNodes = []; |
||||
|
||||
newData.forEach((item, index) => { |
||||
const key = item[NODE_KEY]; |
||||
const isNodeExists = !!key && arrayFindIndex(oldData, data => data[NODE_KEY] === key) >= 0; |
||||
if (isNodeExists) { |
||||
newDataMap[key] = { index, data: item }; |
||||
} else { |
||||
newNodes.push({ index, data: item }); |
||||
} |
||||
}); |
||||
|
||||
if (!this.store.lazy) { |
||||
oldData.forEach((item) => { |
||||
if (!newDataMap[item[NODE_KEY]]) this.removeChildByData(item); |
||||
}); |
||||
} |
||||
|
||||
newNodes.forEach(({ index, data }) => { |
||||
this.insertChild({ data }, index); |
||||
}); |
||||
|
||||
this.updateLeafState(); |
||||
} |
||||
|
||||
loadData(callback, defaultProps = {}) { |
||||
if (this.store.lazy === true && this.store.load && !this.loaded && (!this.loading || Object.keys(defaultProps).length)) { |
||||
this.loading = true; |
||||
|
||||
const resolve = (children) => { |
||||
this.loaded = true; |
||||
this.loading = false; |
||||
this.childNodes = []; |
||||
|
||||
this.doCreateChildren(children, defaultProps); |
||||
|
||||
this.updateLeafState(); |
||||
if (callback) { |
||||
callback.call(this, children); |
||||
} |
||||
}; |
||||
|
||||
this.store.load(this, resolve); |
||||
} else { |
||||
if (callback) { |
||||
callback.call(this); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,340 @@ |
||||
import Node from './node'; |
||||
import { getNodeKey } from './util'; |
||||
|
||||
export default class TreeStore { |
||||
constructor(options) { |
||||
this.currentNode = null; |
||||
this.currentNodeKey = null; |
||||
|
||||
for (let option in options) { |
||||
if (options.hasOwnProperty(option)) { |
||||
this[option] = options[option]; |
||||
} |
||||
} |
||||
|
||||
this.nodesMap = {}; |
||||
|
||||
this.root = new Node({ |
||||
data: this.data, |
||||
store: this |
||||
}); |
||||
|
||||
if (this.lazy && this.load) { |
||||
const loadFn = this.load; |
||||
loadFn(this.root, (data) => { |
||||
this.root.doCreateChildren(data); |
||||
this._initDefaultCheckedNodes(); |
||||
}); |
||||
} else { |
||||
this._initDefaultCheckedNodes(); |
||||
} |
||||
} |
||||
|
||||
filter(value) { |
||||
const filterNodeMethod = this.filterNodeMethod; |
||||
const lazy = this.lazy; |
||||
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.some(child => child.visible); |
||||
|
||||
if (node.root) { |
||||
node.root.visible = allHidden === false; |
||||
} else { |
||||
node.visible = allHidden === false; |
||||
} |
||||
} |
||||
if (!value) return; |
||||
|
||||
if (node.visible && !node.isLeaf && !lazy) 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 : getNodeKey(this.key, data); |
||||
return this.nodesMap[key] || null; |
||||
} |
||||
|
||||
insertBefore(data, refData) { |
||||
const refNode = this.getNode(refData); |
||||
refNode.parent.insertBefore({ data }, refNode); |
||||
} |
||||
|
||||
insertAfter(data, refData) { |
||||
const refNode = this.getNode(refData); |
||||
refNode.parent.insertAfter({ data }, refNode); |
||||
} |
||||
|
||||
remove(data) { |
||||
const node = this.getNode(data); |
||||
|
||||
if (node && node.parent) { |
||||
if (node === this.currentNode) { |
||||
this.currentNode = null; |
||||
} |
||||
node.parent.removeChild(node); |
||||
} |
||||
} |
||||
|
||||
append(data, parentData) { |
||||
const parentNode = parentData ? this.getNode(parentData) : this.root; |
||||
|
||||
if (parentNode) { |
||||
parentNode.insertChild({ data }); |
||||
} |
||||
} |
||||
|
||||
_initDefaultCheckedNodes() { |
||||
const defaultCheckedKeys = this.defaultCheckedKeys || []; |
||||
const nodesMap = this.nodesMap; |
||||
|
||||
defaultCheckedKeys.forEach((checkedKey) => { |
||||
const node = nodesMap[checkedKey]; |
||||
|
||||
if (node) { |
||||
node.setChecked(true, !this.checkStrictly); |
||||
} |
||||
}); |
||||
} |
||||
|
||||
_initDefaultCheckedNode(node) { |
||||
const defaultCheckedKeys = this.defaultCheckedKeys || []; |
||||
|
||||
if (defaultCheckedKeys.indexOf(node.key) !== -1) { |
||||
node.setChecked(true, !this.checkStrictly); |
||||
} |
||||
} |
||||
|
||||
setDefaultCheckedKey(newVal) { |
||||
if (newVal !== this.defaultCheckedKeys) { |
||||
this.defaultCheckedKeys = newVal; |
||||
this._initDefaultCheckedNodes(); |
||||
} |
||||
} |
||||
|
||||
registerNode(node) { |
||||
const key = this.key; |
||||
if (!key || !node || !node.data) return; |
||||
|
||||
const nodeKey = node.key; |
||||
if (nodeKey !== undefined) this.nodesMap[node.key] = node; |
||||
} |
||||
|
||||
deregisterNode(node) { |
||||
const key = this.key; |
||||
if (!key || !node || !node.data) return; |
||||
|
||||
node.childNodes.forEach(child => { |
||||
this.deregisterNode(child); |
||||
}); |
||||
|
||||
delete this.nodesMap[node.key]; |
||||
} |
||||
|
||||
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]); |
||||
} |
||||
|
||||
_getAllNodes() { |
||||
const allNodes = []; |
||||
const nodesMap = this.nodesMap; |
||||
for (let nodeKey in nodesMap) { |
||||
if (nodesMap.hasOwnProperty(nodeKey)) { |
||||
allNodes.push(nodesMap[nodeKey]); |
||||
} |
||||
} |
||||
|
||||
return allNodes; |
||||
} |
||||
|
||||
updateChildren(key, data) { |
||||
const node = this.nodesMap[key]; |
||||
if (!node) return; |
||||
const childNodes = node.childNodes; |
||||
for (let i = childNodes.length - 1; i >= 0; i--) { |
||||
const child = childNodes[i]; |
||||
this.remove(child.data); |
||||
} |
||||
for (let i = 0, j = data.length; i < j; i++) { |
||||
const child = data[i]; |
||||
this.append(child, node.data); |
||||
} |
||||
} |
||||
|
||||
_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); |
||||
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; |
||||
} |
||||
|
||||
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); |
||||
} |
||||
} |
||||
} |
||||
|
||||
setCheckedNodes(array, leafOnly = false) { |
||||
const key = this.key; |
||||
const checkedKeys = {}; |
||||
array.forEach((item) => { |
||||
checkedKeys[(item || {})[key]] = true; |
||||
}); |
||||
|
||||
this._setCheckedKeys(key, leafOnly, checkedKeys); |
||||
} |
||||
|
||||
setCheckedKeys(keys, leafOnly = false) { |
||||
this.defaultCheckedKeys = keys; |
||||
const key = this.key; |
||||
const checkedKeys = {}; |
||||
keys.forEach((key) => { |
||||
checkedKeys[key] = true; |
||||
}); |
||||
|
||||
this._setCheckedKeys(key, leafOnly, checkedKeys); |
||||
} |
||||
|
||||
setDefaultExpandedKeys(keys) { |
||||
keys = keys || []; |
||||
this.defaultExpandedKeys = keys; |
||||
|
||||
keys.forEach((key) => { |
||||
const node = this.getNode(key); |
||||
if (node) node.expand(null, this.autoExpandParent); |
||||
}); |
||||
} |
||||
|
||||
setChecked(data, checked, deep) { |
||||
const node = this.getNode(data); |
||||
|
||||
if (node) { |
||||
node.setChecked(!!checked, deep); |
||||
} |
||||
} |
||||
|
||||
getCurrentNode() { |
||||
return this.currentNode; |
||||
} |
||||
|
||||
setCurrentNode(currentNode) { |
||||
const prevCurrentNode = this.currentNode; |
||||
if (prevCurrentNode) { |
||||
prevCurrentNode.isCurrent = false; |
||||
} |
||||
this.currentNode = currentNode; |
||||
this.currentNode.isCurrent = true; |
||||
} |
||||
|
||||
setUserCurrentNode(node) { |
||||
const key = node[this.key]; |
||||
const currNode = this.nodesMap[key]; |
||||
this.setCurrentNode(currNode); |
||||
} |
||||
|
||||
setCurrentNodeKey(key) { |
||||
if (key === null || key === undefined) { |
||||
this.currentNode && (this.currentNode.isCurrent = false); |
||||
this.currentNode = null; |
||||
return; |
||||
} |
||||
const node = this.getNode(key); |
||||
if (node) { |
||||
this.setCurrentNode(node); |
||||
} |
||||
} |
||||
}; |
||||
@ -0,0 +1,27 @@ |
||||
export const NODE_KEY = '$treeNodeId'; |
||||
|
||||
export const markNodeData = function(node, data) { |
||||
if (!data || data[NODE_KEY]) return; |
||||
Object.defineProperty(data, NODE_KEY, { |
||||
value: node.id, |
||||
enumerable: false, |
||||
configurable: false, |
||||
writable: false |
||||
}); |
||||
}; |
||||
|
||||
export const getNodeKey = function(key, data) { |
||||
if (!key) return data[NODE_KEY]; |
||||
return data[key]; |
||||
}; |
||||
|
||||
export const findNearestComponent = (element, componentName) => { |
||||
let target = element; |
||||
while (target && target.tagName !== 'BODY') { |
||||
if (target.__vue__ && target.__vue__.$options.name === componentName) { |
||||
return target.__vue__; |
||||
} |
||||
target = target.parentNode; |
||||
} |
||||
return null; |
||||
}; |
||||
@ -0,0 +1,279 @@ |
||||
<template> |
||||
<div |
||||
class="el-tree-node" |
||||
@click.stop="handleClick" |
||||
@contextmenu="($event) => this.handleContextMenu($event)" |
||||
v-show="node.visible" |
||||
:class="{ |
||||
'is-expanded': expanded, |
||||
'is-current': node.isCurrent, |
||||
'is-hidden': !node.visible, |
||||
'is-focusable': !node.disabled, |
||||
'is-checked': !node.disabled && node.checked |
||||
}" |
||||
role="treeitem" |
||||
tabindex="-1" |
||||
:aria-expanded="expanded" |
||||
:aria-disabled="node.disabled" |
||||
:aria-checked="node.checked" |
||||
:draggable="tree.draggable" |
||||
@dragstart.stop="handleDragStart" |
||||
@dragover.stop="handleDragOver" |
||||
@dragend.stop="handleDragEnd" |
||||
@drop.stop="handleDrop" |
||||
ref="node" |
||||
> |
||||
<div class="el-tree-node__content" |
||||
:style="{ 'padding-left': (node.level - 1) * tree.indent + 'px' }"> |
||||
<span |
||||
@click.stop="handleExpandIconClick" |
||||
:class="[ |
||||
{ 'is-leaf': node.isLeaf, expanded: !node.isLeaf && expanded }, |
||||
'el-tree-node__expand-icon', |
||||
tree.iconClass ? tree.iconClass : 'el-icon-caret-right' |
||||
]" |
||||
> |
||||
</span> |
||||
<el-checkbox |
||||
v-if="showCheckbox && node.data.showCheckbox" |
||||
v-model="node.checked" |
||||
:indeterminate="node.indeterminate" |
||||
:disabled="!!node.disabled" |
||||
@click.native.stop |
||||
@change="handleCheckChange" |
||||
> |
||||
</el-checkbox> |
||||
<span |
||||
v-if="node.loading" |
||||
class="el-tree-node__loading-icon el-icon-loading"> |
||||
</span> |
||||
<node-content :node="node"></node-content> |
||||
</div> |
||||
<el-collapse-transition> |
||||
<div |
||||
class="el-tree-node__children" |
||||
v-if="!renderAfterExpand || childNodeRendered" |
||||
v-show="expanded" |
||||
role="group" |
||||
:aria-expanded="expanded" |
||||
> |
||||
<el-tree-node |
||||
:render-content="renderContent" |
||||
v-for="child in node.childNodes" |
||||
:render-after-expand="renderAfterExpand" |
||||
:show-checkbox="showCheckbox" |
||||
:key="getNodeKey(child)" |
||||
:node="child" |
||||
@node-expand="handleChildNodeExpand"> |
||||
</el-tree-node> |
||||
</div> |
||||
</el-collapse-transition> |
||||
</div> |
||||
</template> |
||||
|
||||
<script type="text/jsx"> |
||||
import ElCollapseTransition from 'element-ui/src/transitions/collapse-transition'; |
||||
import ElCheckbox from 'element-ui/packages/checkbox'; |
||||
import emitter from 'element-ui/src/mixins/emitter'; |
||||
import { getNodeKey } from './model/util'; |
||||
|
||||
export default { |
||||
name: 'ElTreeNode', |
||||
|
||||
componentName: 'ElTreeNode', |
||||
|
||||
mixins: [emitter], |
||||
|
||||
props: { |
||||
node: { |
||||
default() { |
||||
return {}; |
||||
} |
||||
}, |
||||
props: {}, |
||||
renderContent: Function, |
||||
renderAfterExpand: { |
||||
type: Boolean, |
||||
default: true |
||||
}, |
||||
showCheckbox: { |
||||
type: Boolean, |
||||
default: false |
||||
} |
||||
}, |
||||
|
||||
components: { |
||||
ElCollapseTransition, |
||||
ElCheckbox, |
||||
NodeContent: { |
||||
props: { |
||||
node: { |
||||
required: true |
||||
} |
||||
}, |
||||
render(h) { |
||||
const parent = this.$parent; |
||||
const tree = parent.tree; |
||||
const node = this.node; |
||||
const { data, store } = node; |
||||
return ( |
||||
parent.renderContent |
||||
? parent.renderContent.call(parent._renderProxy, h, { _self: tree.$vnode.context, node, data, store }) |
||||
: tree.$scopedSlots.default |
||||
? tree.$scopedSlots.default({ node, data }) |
||||
: <span class="el-tree-node__label">{ node.label }</span> |
||||
); |
||||
} |
||||
} |
||||
}, |
||||
|
||||
data() { |
||||
return { |
||||
tree: null, |
||||
expanded: false, |
||||
childNodeRendered: false, |
||||
oldChecked: null, |
||||
oldIndeterminate: null |
||||
}; |
||||
}, |
||||
|
||||
watch: { |
||||
'node.indeterminate'(val) { |
||||
this.handleSelectChange(this.node.checked, val); |
||||
}, |
||||
|
||||
'node.checked'(val) { |
||||
this.handleSelectChange(val, this.node.indeterminate); |
||||
}, |
||||
|
||||
'node.expanded'(val) { |
||||
this.$nextTick(() => this.expanded = val); |
||||
if (val) { |
||||
this.childNodeRendered = true; |
||||
} |
||||
} |
||||
}, |
||||
|
||||
methods: { |
||||
getNodeKey(node) { |
||||
return getNodeKey(this.tree.nodeKey, node.data); |
||||
}, |
||||
|
||||
handleSelectChange(checked, indeterminate) { |
||||
if (this.oldChecked !== checked && this.oldIndeterminate !== indeterminate) { |
||||
this.tree.$emit('check-change', this.node.data, checked, indeterminate); |
||||
} |
||||
this.oldChecked = checked; |
||||
this.indeterminate = indeterminate; |
||||
}, |
||||
|
||||
handleClick() { |
||||
const store = this.tree.store; |
||||
store.setCurrentNode(this.node); |
||||
this.tree.$emit('current-change', store.currentNode ? store.currentNode.data : null, store.currentNode); |
||||
this.tree.currentNode = this; |
||||
if (this.tree.expandOnClickNode) { |
||||
this.handleExpandIconClick(); |
||||
} |
||||
if (this.tree.checkOnClickNode && !this.node.disabled) { |
||||
this.handleCheckChange(null, { |
||||
target: { checked: !this.node.checked } |
||||
}); |
||||
} |
||||
this.tree.$emit('node-click', this.node.data, this.node, this); |
||||
}, |
||||
|
||||
handleContextMenu(event) { |
||||
if (this.tree._events['node-contextmenu'] && this.tree._events['node-contextmenu'].length > 0) { |
||||
event.stopPropagation(); |
||||
event.preventDefault(); |
||||
} |
||||
this.tree.$emit('node-contextmenu', event, this.node.data, this.node, this); |
||||
}, |
||||
|
||||
handleExpandIconClick() { |
||||
if (this.node.isLeaf) return; |
||||
if (this.expanded) { |
||||
this.tree.$emit('node-collapse', this.node.data, this.node, this); |
||||
this.node.collapse(); |
||||
} else { |
||||
this.node.expand(); |
||||
this.$emit('node-expand', this.node.data, this.node, this); |
||||
} |
||||
}, |
||||
|
||||
handleCheckChange(value, ev) { |
||||
this.node.setChecked(ev.target.checked, !this.tree.checkStrictly); |
||||
this.$nextTick(() => { |
||||
const store = this.tree.store; |
||||
this.tree.$emit('check', this.node.data, { |
||||
checkedNodes: store.getCheckedNodes(), |
||||
checkedKeys: store.getCheckedKeys(), |
||||
halfCheckedNodes: store.getHalfCheckedNodes(), |
||||
halfCheckedKeys: store.getHalfCheckedKeys(), |
||||
}); |
||||
}); |
||||
}, |
||||
|
||||
handleChildNodeExpand(nodeData, node, instance) { |
||||
this.broadcast('ElTreeNode', 'tree-node-expand', node); |
||||
this.tree.$emit('node-expand', nodeData, node, instance); |
||||
}, |
||||
|
||||
handleDragStart(event) { |
||||
if (!this.tree.draggable) return; |
||||
this.tree.$emit('tree-node-drag-start', event, this); |
||||
}, |
||||
|
||||
handleDragOver(event) { |
||||
if (!this.tree.draggable) return; |
||||
this.tree.$emit('tree-node-drag-over', event, this); |
||||
event.preventDefault(); |
||||
}, |
||||
|
||||
handleDrop(event) { |
||||
event.preventDefault(); |
||||
}, |
||||
|
||||
handleDragEnd(event) { |
||||
if (!this.tree.draggable) return; |
||||
this.tree.$emit('tree-node-drag-end', event, this); |
||||
} |
||||
}, |
||||
|
||||
created() { |
||||
const parent = this.$parent; |
||||
|
||||
if (parent.isTree) { |
||||
this.tree = parent; |
||||
} else { |
||||
this.tree = parent.tree; |
||||
} |
||||
|
||||
const tree = this.tree; |
||||
if (!tree) { |
||||
console.warn('Can not find node\'s tree.'); |
||||
} |
||||
|
||||
const props = tree.props || {}; |
||||
const childrenKey = props['children'] || 'children'; |
||||
|
||||
this.$watch(`node.data.${childrenKey}`, () => { |
||||
this.node.updateChildren(); |
||||
}); |
||||
|
||||
if (this.node.expanded) { |
||||
this.expanded = true; |
||||
this.childNodeRendered = true; |
||||
} |
||||
|
||||
if(this.tree.accordion) { |
||||
this.$on('tree-node-expand', node => { |
||||
if(this.node !== node) { |
||||
this.node.collapse(); |
||||
} |
||||
}); |
||||
} |
||||
} |
||||
}; |
||||
</script> |
||||
@ -0,0 +1,496 @@ |
||||
<template> |
||||
<div |
||||
class="el-tree" |
||||
:class="{ |
||||
'el-tree--highlight-current': highlightCurrent, |
||||
'is-dragging': !!dragState.draggingNode, |
||||
'is-drop-not-allow': !dragState.allowDrop, |
||||
'is-drop-inner': dragState.dropType === 'inner' |
||||
}" |
||||
role="tree" |
||||
> |
||||
<el-tree-node |
||||
v-for="child in root.childNodes" |
||||
:node="child" |
||||
:props="props" |
||||
:render-after-expand="renderAfterExpand" |
||||
:show-checkbox="showCheckbox" |
||||
:key="getNodeKey(child)" |
||||
:render-content="renderContent" |
||||
@node-expand="handleNodeExpand"> |
||||
</el-tree-node> |
||||
<div class="el-tree__empty-block" v-if="isEmpty"> |
||||
<span class="el-tree__empty-text">{{ emptyText }}</span> |
||||
</div> |
||||
<div |
||||
v-show="dragState.showDropIndicator" |
||||
class="el-tree__drop-indicator" |
||||
ref="dropIndicator"> |
||||
</div> |
||||
</div> |
||||
</template> |
||||
|
||||
<script> |
||||
import TreeStore from './model/tree-store'; |
||||
import { getNodeKey, findNearestComponent } from './model/util'; |
||||
import ElTreeNode from './tree-node.vue'; |
||||
import {t} from 'element-ui/src/locale'; |
||||
import emitter from 'element-ui/src/mixins/emitter'; |
||||
import { addClass, removeClass } from 'element-ui/src/utils/dom'; |
||||
|
||||
export default { |
||||
name: 'ElTree', |
||||
|
||||
mixins: [emitter], |
||||
|
||||
components: { |
||||
ElTreeNode |
||||
}, |
||||
|
||||
data() { |
||||
return { |
||||
store: null, |
||||
root: null, |
||||
currentNode: null, |
||||
treeItems: null, |
||||
checkboxItems: [], |
||||
dragState: { |
||||
showDropIndicator: false, |
||||
draggingNode: null, |
||||
dropNode: null, |
||||
allowDrop: true |
||||
} |
||||
}; |
||||
}, |
||||
|
||||
props: { |
||||
data: { |
||||
type: Array |
||||
}, |
||||
emptyText: { |
||||
type: String, |
||||
default() { |
||||
return t('el.tree.emptyText'); |
||||
} |
||||
}, |
||||
renderAfterExpand: { |
||||
type: Boolean, |
||||
default: true |
||||
}, |
||||
nodeKey: String, |
||||
checkStrictly: Boolean, |
||||
defaultExpandAll: Boolean, |
||||
expandOnClickNode: { |
||||
type: Boolean, |
||||
default: true |
||||
}, |
||||
checkOnClickNode: Boolean, |
||||
checkDescendants: { |
||||
type: Boolean, |
||||
default: false |
||||
}, |
||||
autoExpandParent: { |
||||
type: Boolean, |
||||
default: true |
||||
}, |
||||
defaultCheckedKeys: Array, |
||||
defaultExpandedKeys: Array, |
||||
currentNodeKey: [String, Number], |
||||
renderContent: Function, |
||||
showCheckbox: { |
||||
type: Boolean, |
||||
default: false |
||||
}, |
||||
draggable: { |
||||
type: Boolean, |
||||
default: false |
||||
}, |
||||
allowDrag: Function, |
||||
allowDrop: Function, |
||||
props: { |
||||
default() { |
||||
return { |
||||
children: 'children', |
||||
label: 'label', |
||||
disabled: 'disabled' |
||||
}; |
||||
} |
||||
}, |
||||
lazy: { |
||||
type: Boolean, |
||||
default: false |
||||
}, |
||||
highlightCurrent: Boolean, |
||||
load: Function, |
||||
filterNodeMethod: Function, |
||||
accordion: Boolean, |
||||
indent: { |
||||
type: Number, |
||||
default: 18 |
||||
}, |
||||
iconClass: String |
||||
}, |
||||
|
||||
computed: { |
||||
children: { |
||||
set(value) { |
||||
this.data = value; |
||||
}, |
||||
get() { |
||||
return this.data; |
||||
} |
||||
}, |
||||
|
||||
treeItemArray() { |
||||
return Array.prototype.slice.call(this.treeItems); |
||||
}, |
||||
|
||||
isEmpty() { |
||||
const { childNodes } = this.root; |
||||
return !childNodes || childNodes.length === 0 || childNodes.every(({visible}) => !visible); |
||||
} |
||||
}, |
||||
|
||||
watch: { |
||||
defaultCheckedKeys(newVal) { |
||||
this.store.setDefaultCheckedKey(newVal); |
||||
}, |
||||
|
||||
defaultExpandedKeys(newVal) { |
||||
this.store.defaultExpandedKeys = newVal; |
||||
this.store.setDefaultExpandedKeys(newVal); |
||||
}, |
||||
|
||||
data(newVal) { |
||||
this.store.setData(newVal); |
||||
}, |
||||
|
||||
checkboxItems(val) { |
||||
Array.prototype.forEach.call(val, (checkbox) => { |
||||
checkbox.setAttribute('tabindex', -1); |
||||
}); |
||||
}, |
||||
|
||||
checkStrictly(newVal) { |
||||
this.store.checkStrictly = newVal; |
||||
} |
||||
}, |
||||
|
||||
methods: { |
||||
filter(value) { |
||||
if (!this.filterNodeMethod) throw new Error('[Tree] filterNodeMethod is required when filter'); |
||||
this.store.filter(value); |
||||
}, |
||||
|
||||
getNodeKey(node) { |
||||
return getNodeKey(this.nodeKey, node.data); |
||||
}, |
||||
|
||||
getNodePath(data) { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in getNodePath'); |
||||
const node = this.store.getNode(data); |
||||
if (!node) return []; |
||||
const path = [node.data]; |
||||
let parent = node.parent; |
||||
while (parent && parent !== this.root) { |
||||
path.push(parent.data); |
||||
parent = parent.parent; |
||||
} |
||||
return path.reverse(); |
||||
}, |
||||
|
||||
getCheckedNodes(leafOnly, includeHalfChecked) { |
||||
return this.store.getCheckedNodes(leafOnly, includeHalfChecked); |
||||
}, |
||||
|
||||
getCheckedKeys(leafOnly) { |
||||
return this.store.getCheckedKeys(leafOnly); |
||||
}, |
||||
|
||||
getCurrentNode() { |
||||
const currentNode = this.store.getCurrentNode(); |
||||
return currentNode ? currentNode.data : null; |
||||
}, |
||||
|
||||
getCurrentKey() { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in getCurrentKey'); |
||||
const currentNode = this.getCurrentNode(); |
||||
return currentNode ? currentNode[this.nodeKey] : null; |
||||
}, |
||||
|
||||
setCheckedNodes(nodes, leafOnly) { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in setCheckedNodes'); |
||||
this.store.setCheckedNodes(nodes, leafOnly); |
||||
}, |
||||
|
||||
setCheckedKeys(keys, leafOnly) { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in setCheckedKeys'); |
||||
this.store.setCheckedKeys(keys, leafOnly); |
||||
}, |
||||
|
||||
setChecked(data, checked, deep) { |
||||
this.store.setChecked(data, checked, deep); |
||||
}, |
||||
|
||||
getHalfCheckedNodes() { |
||||
return this.store.getHalfCheckedNodes(); |
||||
}, |
||||
|
||||
getHalfCheckedKeys() { |
||||
return this.store.getHalfCheckedKeys(); |
||||
}, |
||||
|
||||
setCurrentNode(node) { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in setCurrentNode'); |
||||
this.store.setUserCurrentNode(node); |
||||
}, |
||||
|
||||
setCurrentKey(key) { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in setCurrentKey'); |
||||
this.store.setCurrentNodeKey(key); |
||||
}, |
||||
|
||||
getNode(data) { |
||||
return this.store.getNode(data); |
||||
}, |
||||
|
||||
remove(data) { |
||||
this.store.remove(data); |
||||
}, |
||||
|
||||
append(data, parentNode) { |
||||
this.store.append(data, parentNode); |
||||
}, |
||||
|
||||
insertBefore(data, refNode) { |
||||
this.store.insertBefore(data, refNode); |
||||
}, |
||||
|
||||
insertAfter(data, refNode) { |
||||
this.store.insertAfter(data, refNode); |
||||
}, |
||||
|
||||
handleNodeExpand(nodeData, node, instance) { |
||||
this.broadcast('ElTreeNode', 'tree-node-expand', node); |
||||
this.$emit('node-expand', nodeData, node, instance); |
||||
}, |
||||
|
||||
updateKeyChildren(key, data) { |
||||
if (!this.nodeKey) throw new Error('[Tree] nodeKey is required in updateKeyChild'); |
||||
this.store.updateChildren(key, data); |
||||
}, |
||||
|
||||
initTabIndex() { |
||||
this.treeItems = this.$el.querySelectorAll('.is-focusable[role=treeitem]'); |
||||
this.checkboxItems = this.$el.querySelectorAll('input[type=checkbox]'); |
||||
const checkedItem = this.$el.querySelectorAll('.is-checked[role=treeitem]'); |
||||
if (checkedItem.length) { |
||||
checkedItem[0].setAttribute('tabindex', 0); |
||||
return; |
||||
} |
||||
this.treeItems[0] && this.treeItems[0].setAttribute('tabindex', 0); |
||||
}, |
||||
|
||||
handleKeydown(ev) { |
||||
const currentItem = ev.target; |
||||
if (currentItem.className.indexOf('el-tree-node') === -1) return; |
||||
const keyCode = ev.keyCode; |
||||
this.treeItems = this.$el.querySelectorAll('.is-focusable[role=treeitem]'); |
||||
const currentIndex = this.treeItemArray.indexOf(currentItem); |
||||
let nextIndex; |
||||
if ([38, 40].indexOf(keyCode) > -1) { // up、down |
||||
ev.preventDefault(); |
||||
if (keyCode === 38) { // up |
||||
nextIndex = currentIndex !== 0 ? currentIndex - 1 : 0; |
||||
} else { |
||||
nextIndex = (currentIndex < this.treeItemArray.length - 1) ? currentIndex + 1 : 0; |
||||
} |
||||
this.treeItemArray[nextIndex].focus(); // 选中 |
||||
} |
||||
if ([37, 39].indexOf(keyCode) > -1) { // left、right 展开 |
||||
ev.preventDefault(); |
||||
currentItem.click(); // 选中 |
||||
} |
||||
const hasInput = currentItem.querySelector('[type="checkbox"]'); |
||||
if ([13, 32].indexOf(keyCode) > -1 && hasInput) { // space enter选中checkbox |
||||
ev.preventDefault(); |
||||
hasInput.click(); |
||||
} |
||||
} |
||||
}, |
||||
|
||||
created() { |
||||
this.isTree = true; |
||||
|
||||
this.store = new TreeStore({ |
||||
key: this.nodeKey, |
||||
data: this.data, |
||||
lazy: this.lazy, |
||||
props: this.props, |
||||
load: this.load, |
||||
currentNodeKey: this.currentNodeKey, |
||||
checkStrictly: this.checkStrictly, |
||||
checkDescendants: this.checkDescendants, |
||||
defaultCheckedKeys: this.defaultCheckedKeys, |
||||
defaultExpandedKeys: this.defaultExpandedKeys, |
||||
autoExpandParent: this.autoExpandParent, |
||||
defaultExpandAll: this.defaultExpandAll, |
||||
filterNodeMethod: this.filterNodeMethod |
||||
}); |
||||
|
||||
this.root = this.store.root; |
||||
|
||||
let dragState = this.dragState; |
||||
this.$on('tree-node-drag-start', (event, treeNode) => { |
||||
if (typeof this.allowDrag === 'function' && !this.allowDrag(treeNode.node)) { |
||||
event.preventDefault(); |
||||
return false; |
||||
} |
||||
event.dataTransfer.effectAllowed = 'move'; |
||||
|
||||
// wrap in try catch to address IE's error when first param is 'text/plain' |
||||
try { |
||||
// setData is required for draggable to work in FireFox |
||||
// the content has to be '' so dragging a node out of the tree won't open a new tab in FireFox |
||||
event.dataTransfer.setData('text/plain', ''); |
||||
} catch (e) {} |
||||
dragState.draggingNode = treeNode; |
||||
this.$emit('node-drag-start', treeNode.node, event); |
||||
}); |
||||
|
||||
this.$on('tree-node-drag-over', (event, treeNode) => { |
||||
const dropNode = findNearestComponent(event.target, 'ElTreeNode'); |
||||
const oldDropNode = dragState.dropNode; |
||||
if (oldDropNode && oldDropNode !== dropNode) { |
||||
removeClass(oldDropNode.$el, 'is-drop-inner'); |
||||
} |
||||
const draggingNode = dragState.draggingNode; |
||||
if (!draggingNode || !dropNode) return; |
||||
|
||||
let dropPrev = true; |
||||
let dropInner = true; |
||||
let dropNext = true; |
||||
let userAllowDropInner = true; |
||||
if (typeof this.allowDrop === 'function') { |
||||
dropPrev = this.allowDrop(draggingNode.node, dropNode.node, 'prev'); |
||||
userAllowDropInner = dropInner = this.allowDrop(draggingNode.node, dropNode.node, 'inner'); |
||||
dropNext = this.allowDrop(draggingNode.node, dropNode.node, 'next'); |
||||
} |
||||
event.dataTransfer.dropEffect = dropInner ? 'move' : 'none'; |
||||
if ((dropPrev || dropInner || dropNext) && oldDropNode !== dropNode) { |
||||
if (oldDropNode) { |
||||
this.$emit('node-drag-leave', draggingNode.node, oldDropNode.node, event); |
||||
} |
||||
this.$emit('node-drag-enter', draggingNode.node, dropNode.node, event); |
||||
} |
||||
|
||||
if (dropPrev || dropInner || dropNext) { |
||||
dragState.dropNode = dropNode; |
||||
} |
||||
|
||||
if (dropNode.node.nextSibling === draggingNode.node) { |
||||
dropNext = false; |
||||
} |
||||
if (dropNode.node.previousSibling === draggingNode.node) { |
||||
dropPrev = false; |
||||
} |
||||
if (dropNode.node.contains(draggingNode.node, false)) { |
||||
dropInner = false; |
||||
} |
||||
if (draggingNode.node === dropNode.node || draggingNode.node.contains(dropNode.node)) { |
||||
dropPrev = false; |
||||
dropInner = false; |
||||
dropNext = false; |
||||
} |
||||
|
||||
const targetPosition = dropNode.$el.getBoundingClientRect(); |
||||
const treePosition = this.$el.getBoundingClientRect(); |
||||
|
||||
let dropType; |
||||
const prevPercent = dropPrev ? (dropInner ? 0.25 : (dropNext ? 0.45 : 1)) : -1; |
||||
const nextPercent = dropNext ? (dropInner ? 0.75 : (dropPrev ? 0.55 : 0)) : 1; |
||||
|
||||
let indicatorTop = -9999; |
||||
const distance = event.clientY - targetPosition.top; |
||||
if (distance < targetPosition.height * prevPercent) { |
||||
dropType = 'before'; |
||||
} else if (distance > targetPosition.height * nextPercent) { |
||||
dropType = 'after'; |
||||
} else if (dropInner) { |
||||
dropType = 'inner'; |
||||
} else { |
||||
dropType = 'none'; |
||||
} |
||||
|
||||
const iconPosition = dropNode.$el.querySelector('.el-tree-node__expand-icon').getBoundingClientRect(); |
||||
const dropIndicator = this.$refs.dropIndicator; |
||||
if (dropType === 'before') { |
||||
indicatorTop = iconPosition.top - treePosition.top; |
||||
} else if (dropType === 'after') { |
||||
indicatorTop = iconPosition.bottom - treePosition.top; |
||||
} |
||||
dropIndicator.style.top = indicatorTop + 'px'; |
||||
dropIndicator.style.left = (iconPosition.right - treePosition.left) + 'px'; |
||||
|
||||
if (dropType === 'inner') { |
||||
addClass(dropNode.$el, 'is-drop-inner'); |
||||
} else { |
||||
removeClass(dropNode.$el, 'is-drop-inner'); |
||||
} |
||||
|
||||
dragState.showDropIndicator = dropType === 'before' || dropType === 'after'; |
||||
dragState.allowDrop = dragState.showDropIndicator || userAllowDropInner; |
||||
dragState.dropType = dropType; |
||||
this.$emit('node-drag-over', draggingNode.node, dropNode.node, event); |
||||
}); |
||||
|
||||
this.$on('tree-node-drag-end', (event) => { |
||||
const { draggingNode, dropType, dropNode } = dragState; |
||||
event.preventDefault(); |
||||
event.dataTransfer.dropEffect = 'move'; |
||||
|
||||
if (draggingNode && dropNode) { |
||||
const draggingNodeCopy = { data: draggingNode.node.data }; |
||||
if (dropType !== 'none') { |
||||
draggingNode.node.remove(); |
||||
} |
||||
if (dropType === 'before') { |
||||
dropNode.node.parent.insertBefore(draggingNodeCopy, dropNode.node); |
||||
} else if (dropType === 'after') { |
||||
dropNode.node.parent.insertAfter(draggingNodeCopy, dropNode.node); |
||||
} else if (dropType === 'inner') { |
||||
dropNode.node.insertChild(draggingNodeCopy); |
||||
} |
||||
if (dropType !== 'none') { |
||||
this.store.registerNode(draggingNodeCopy); |
||||
} |
||||
|
||||
removeClass(dropNode.$el, 'is-drop-inner'); |
||||
|
||||
this.$emit('node-drag-end', draggingNode.node, dropNode.node, dropType, event); |
||||
if (dropType !== 'none') { |
||||
this.$emit('node-drop', draggingNode.node, dropNode.node, dropType, event); |
||||
} |
||||
} |
||||
if (draggingNode && !dropNode) { |
||||
this.$emit('node-drag-end', draggingNode.node, null, dropType, event); |
||||
} |
||||
|
||||
dragState.showDropIndicator = false; |
||||
dragState.draggingNode = null; |
||||
dragState.dropNode = null; |
||||
dragState.allowDrop = true; |
||||
}); |
||||
}, |
||||
|
||||
mounted() { |
||||
this.initTabIndex(); |
||||
this.$el.addEventListener('keydown', this.handleKeydown); |
||||
}, |
||||
|
||||
updated() { |
||||
this.treeItems = this.$el.querySelectorAll('[role=treeitem]'); |
||||
this.checkboxItems = this.$el.querySelectorAll('input[type=checkbox]'); |
||||
} |
||||
}; |
||||
</script> |
||||
Loading…
Reference in new issue