218 lines
7.1 KiB
Vue
218 lines
7.1 KiB
Vue
import warning from 'warning';
|
||
import { Tree as VcTree, TreeNode } from '../vc-tree';
|
||
import animation from '../_util/openAnimation';
|
||
import PropTypes from '../_util/vue-types';
|
||
import {
|
||
initDefaultProps,
|
||
getOptionProps,
|
||
filterEmpty,
|
||
getComponentFromProp,
|
||
getClass,
|
||
} from '../_util/props-util';
|
||
import { cloneElement } from '../_util/vnode';
|
||
import { ConfigConsumerProps } from '../config-provider';
|
||
import Icon from '../icon';
|
||
|
||
function TreeProps() {
|
||
return {
|
||
showLine: PropTypes.bool,
|
||
/** 是否支持多选 */
|
||
multiple: PropTypes.bool,
|
||
/** 是否自动展开父节点 */
|
||
autoExpandParent: PropTypes.bool,
|
||
/** checkable状态下节点选择完全受控(父子节点选中状态不再关联)*/
|
||
checkStrictly: PropTypes.bool,
|
||
/** 是否支持选中 */
|
||
checkable: PropTypes.bool,
|
||
/** 是否禁用树 */
|
||
disabled: PropTypes.bool,
|
||
/** 默认展开所有树节点 */
|
||
defaultExpandAll: PropTypes.bool,
|
||
/** 默认展开对应树节点 */
|
||
defaultExpandParent: PropTypes.bool,
|
||
/** 默认展开指定的树节点 */
|
||
defaultExpandedKeys: PropTypes.array,
|
||
/** (受控)展开指定的树节点 */
|
||
expandedKeys: PropTypes.array,
|
||
/** (受控)选中复选框的树节点 */
|
||
checkedKeys: PropTypes.oneOfType([
|
||
PropTypes.array,
|
||
PropTypes.shape({
|
||
checked: PropTypes.array,
|
||
halfChecked: PropTypes.array,
|
||
}).loose,
|
||
]),
|
||
/** 默认选中复选框的树节点 */
|
||
defaultCheckedKeys: PropTypes.array,
|
||
/** (受控)设置选中的树节点 */
|
||
selectedKeys: PropTypes.array,
|
||
/** 默认选中的树节点 */
|
||
defaultSelectedKeys: PropTypes.array,
|
||
selectable: PropTypes.bool,
|
||
/** 展开/收起节点时触发 */
|
||
// onExpand: (expandedKeys: string[], info: AntTreeNodeExpandedEvent) => void | PromiseLike<any>,
|
||
/** 点击复选框触发 */
|
||
// onCheck: (checkedKeys: string[] | { checked: string[]; halfChecked: string[] }, e: AntTreeNodeCheckedEvent) => void,
|
||
/** 点击树节点触发 */
|
||
// onSelect: (selectedKeys: string[], e: AntTreeNodeSelectedEvent) => void,
|
||
/** 单击树节点触发 */
|
||
// onClick: (e: React.MouseEvent<HTMLElement>, node: AntTreeNode) => void,
|
||
/** 双击树节点触发 */
|
||
// onDoubleClick: (e: React.MouseEvent<HTMLElement>, node: AntTreeNode) => void,
|
||
/** filter some AntTreeNodes as you need. it should return true */
|
||
filterAntTreeNode: PropTypes.func,
|
||
/** 异步加载数据 */
|
||
loadData: PropTypes.func,
|
||
loadedKeys: PropTypes.array,
|
||
// onLoaded: (loadedKeys: string[], info: { event: 'load', node: AntTreeNode; }) => void,
|
||
/** 响应右键点击 */
|
||
// onRightClick: (options: AntTreeNodeMouseEvent) => void,
|
||
/** 设置节点可拖拽(IE>8)*/
|
||
draggable: PropTypes.bool,
|
||
// /** 开始拖拽时调用 */
|
||
// onDragStart: (options: AntTreeNodeMouseEvent) => void,
|
||
// /** dragenter 触发时调用 */
|
||
// onDragEnter: (options: AntTreeNodeMouseEvent) => void,
|
||
// /** dragover 触发时调用 */
|
||
// onDragOver: (options: AntTreeNodeMouseEvent) => void,
|
||
// /** dragleave 触发时调用 */
|
||
// onDragLeave: (options: AntTreeNodeMouseEvent) => void,
|
||
// /** drop 触发时调用 */
|
||
// onDrop: (options: AntTreeNodeMouseEvent) => void,
|
||
showIcon: PropTypes.bool,
|
||
icon: PropTypes.func,
|
||
switcherIcon: PropTypes.any,
|
||
prefixCls: PropTypes.string,
|
||
filterTreeNode: PropTypes.func,
|
||
openAnimation: PropTypes.any,
|
||
treeNodes: PropTypes.array,
|
||
treeData: PropTypes.array,
|
||
};
|
||
}
|
||
|
||
export { TreeProps };
|
||
|
||
export default {
|
||
name: 'ATree',
|
||
model: {
|
||
prop: 'checkedKeys',
|
||
event: 'check',
|
||
},
|
||
props: initDefaultProps(TreeProps(), {
|
||
checkable: false,
|
||
showIcon: false,
|
||
openAnimation: {
|
||
on: animation,
|
||
props: { appear: null },
|
||
},
|
||
}),
|
||
inject: {
|
||
configProvider: { default: () => ConfigConsumerProps },
|
||
},
|
||
created() {
|
||
warning(
|
||
!('treeNodes' in getOptionProps(this)),
|
||
'`treeNodes` is deprecated. please use treeData instead.',
|
||
);
|
||
},
|
||
TreeNode,
|
||
methods: {
|
||
renderSwitcherIcon(prefixCls, switcherIcon, { isLeaf, expanded, loading }) {
|
||
const { showLine } = this.$props;
|
||
if (loading) {
|
||
return <Icon type="loading" class={`${prefixCls}-switcher-loading-icon`} />;
|
||
}
|
||
if (showLine) {
|
||
if (isLeaf) {
|
||
return <Icon type="file" class={`${prefixCls}-switcher-line-icon`} />;
|
||
}
|
||
return (
|
||
<Icon
|
||
type={expanded ? 'minus-square' : 'plus-square'}
|
||
class={`${prefixCls}-switcher-line-icon`}
|
||
theme="outlined"
|
||
/>
|
||
);
|
||
} else {
|
||
const switcherCls = `${prefixCls}-switcher-icon`;
|
||
if (isLeaf) {
|
||
return null;
|
||
} else if (switcherIcon) {
|
||
const switcherOriginCls = getClass(switcherIcon[0]);
|
||
return cloneElement(switcherIcon, {
|
||
class: {
|
||
[switcherCls]: true,
|
||
},
|
||
});
|
||
} else {
|
||
return <Icon type="caret-down" class={`${prefixCls}-switcher-icon`} theme="filled" />;
|
||
}
|
||
}
|
||
},
|
||
updateTreeData(treeData) {
|
||
const { $slots, $scopedSlots } = this;
|
||
return treeData.map(item => {
|
||
const {
|
||
children,
|
||
on = {},
|
||
slots = {},
|
||
scopedSlots = {},
|
||
key,
|
||
class: cls,
|
||
style,
|
||
...restProps
|
||
} = item;
|
||
const treeNodeProps = {
|
||
...restProps,
|
||
icon:
|
||
$slots[slots.icon] ||
|
||
($scopedSlots[scopedSlots.icon] && $scopedSlots[scopedSlots.icon]) ||
|
||
restProps.icon,
|
||
title:
|
||
$slots[slots.title] ||
|
||
($scopedSlots[scopedSlots.title] && $scopedSlots[scopedSlots.title](item)) ||
|
||
restProps.title,
|
||
dataRef: item,
|
||
on,
|
||
key,
|
||
class: cls,
|
||
style,
|
||
};
|
||
if (children) {
|
||
return { ...treeNodeProps, children: this.updateTreeData(children) };
|
||
}
|
||
return treeNodeProps;
|
||
});
|
||
},
|
||
},
|
||
render() {
|
||
const props = getOptionProps(this);
|
||
const { prefixCls: customizePrefixCls, showIcon, treeNodes } = props;
|
||
const getPrefixCls = this.configProvider.getPrefixCls;
|
||
const prefixCls = getPrefixCls('tree', customizePrefixCls);
|
||
const switcherIcon = getComponentFromProp(this, 'switcherIcon');
|
||
const checkable = props.checkable;
|
||
let treeData = props.treeData || treeNodes;
|
||
if (treeData) {
|
||
treeData = this.updateTreeData(treeData);
|
||
}
|
||
const vcTreeProps = {
|
||
props: {
|
||
...props,
|
||
prefixCls,
|
||
checkable: checkable ? <span class={`${prefixCls}-checkbox-inner`} /> : checkable,
|
||
children: filterEmpty(this.$slots.default || []),
|
||
__propsSymbol__: Symbol(),
|
||
switcherIcon: nodeProps => this.renderSwitcherIcon(prefixCls, switcherIcon, nodeProps),
|
||
},
|
||
on: this.$listeners,
|
||
ref: 'tree',
|
||
class: !showIcon && `${prefixCls}-icon-hide`,
|
||
};
|
||
if (treeData) {
|
||
vcTreeProps.props.treeData = treeData;
|
||
}
|
||
return <VcTree {...vcTreeProps} />;
|
||
},
|
||
};
|