2017-11-17 14:38:54 +08:00
|
|
|
import * as React from 'react';
|
2016-03-21 09:22:14 +08:00
|
|
|
import RcTreeSelect, { TreeNode, SHOW_ALL, SHOW_PARENT, SHOW_CHILD } from 'rc-tree-select';
|
2015-12-28 19:16:02 +08:00
|
|
|
import classNames from 'classnames';
|
2017-03-17 15:23:25 +08:00
|
|
|
import { TreeSelectProps } from './interface';
|
2018-12-05 19:12:18 +08:00
|
|
|
import { ConfigConsumer, ConfigConsumerProps } from '../config-provider';
|
2017-07-29 14:50:11 +08:00
|
|
|
import warning from '../_util/warning';
|
2018-08-24 14:08:06 +08:00
|
|
|
import Icon from '../icon';
|
|
|
|
import { AntTreeNodeProps } from '../tree';
|
2018-09-15 00:05:14 +08:00
|
|
|
import omit from 'omit.js';
|
2015-12-28 19:16:02 +08:00
|
|
|
|
2018-08-03 17:12:51 +08:00
|
|
|
export { TreeNode, TreeSelectProps } from './interface';
|
2016-08-01 16:35:01 +08:00
|
|
|
|
2017-10-20 14:54:38 +08:00
|
|
|
export default class TreeSelect extends React.Component<TreeSelectProps, any> {
|
2016-03-29 14:01:10 +08:00
|
|
|
static TreeNode = TreeNode;
|
|
|
|
static SHOW_ALL = SHOW_ALL;
|
|
|
|
static SHOW_PARENT = SHOW_PARENT;
|
|
|
|
static SHOW_CHILD = SHOW_CHILD;
|
|
|
|
|
|
|
|
static defaultProps = {
|
|
|
|
transitionName: 'slide-up',
|
|
|
|
choiceTransitionName: 'zoom',
|
|
|
|
showSearch: false,
|
2016-07-13 11:14:24 +08:00
|
|
|
};
|
2016-05-08 16:05:36 +08:00
|
|
|
|
2017-11-19 01:41:40 +08:00
|
|
|
private rcTreeSelect: any;
|
|
|
|
|
2017-11-20 17:03:08 +08:00
|
|
|
constructor(props: TreeSelectProps) {
|
2017-07-29 14:50:11 +08:00
|
|
|
super(props);
|
|
|
|
|
|
|
|
warning(
|
|
|
|
props.multiple !== false || !props.treeCheckable,
|
|
|
|
'`multiple` will alway be `true` when `treeCheckable` is true',
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
2017-11-19 01:41:40 +08:00
|
|
|
focus() {
|
|
|
|
this.rcTreeSelect.focus();
|
|
|
|
}
|
|
|
|
|
|
|
|
blur() {
|
|
|
|
this.rcTreeSelect.blur();
|
|
|
|
}
|
|
|
|
|
2017-11-20 17:03:08 +08:00
|
|
|
saveTreeSelect = (node: typeof RcTreeSelect) => {
|
2017-11-19 01:41:40 +08:00
|
|
|
this.rcTreeSelect = node;
|
2018-12-07 20:02:01 +08:00
|
|
|
};
|
2017-11-19 01:41:40 +08:00
|
|
|
|
2018-12-05 19:12:18 +08:00
|
|
|
renderSwitcherIcon = (prefixCls: string, { isLeaf, loading }: AntTreeNodeProps) => {
|
2018-08-24 14:08:06 +08:00
|
|
|
if (loading) {
|
2018-12-07 20:02:01 +08:00
|
|
|
return <Icon type="loading" className={`${prefixCls}-switcher-loading-icon`} />;
|
2018-08-24 14:08:06 +08:00
|
|
|
}
|
|
|
|
if (isLeaf) {
|
|
|
|
return null;
|
|
|
|
}
|
2018-12-07 20:02:01 +08:00
|
|
|
return <Icon type="caret-down" className={`${prefixCls}-switcher-icon`} />;
|
|
|
|
};
|
2018-08-24 14:08:06 +08:00
|
|
|
|
2018-12-26 16:01:00 +08:00
|
|
|
renderTreeSelect = ({
|
|
|
|
getPopupContainer: getContextPopupContainer,
|
|
|
|
getPrefixCls,
|
|
|
|
renderEmpty,
|
|
|
|
}: ConfigConsumerProps) => {
|
|
|
|
const {
|
|
|
|
prefixCls: customizePrefixCls,
|
|
|
|
className,
|
|
|
|
size,
|
|
|
|
notFoundContent,
|
|
|
|
dropdownStyle,
|
|
|
|
dropdownClassName,
|
|
|
|
suffixIcon,
|
|
|
|
getPopupContainer,
|
|
|
|
...restProps
|
|
|
|
} = this.props;
|
|
|
|
const rest = omit(restProps, ['inputIcon', 'removeIcon', 'clearIcon', 'switcherIcon']);
|
|
|
|
|
|
|
|
const prefixCls = getPrefixCls('select', customizePrefixCls);
|
|
|
|
const cls = classNames(
|
|
|
|
{
|
|
|
|
[`${prefixCls}-lg`]: size === 'large',
|
|
|
|
[`${prefixCls}-sm`]: size === 'small',
|
|
|
|
},
|
|
|
|
className,
|
|
|
|
);
|
|
|
|
|
|
|
|
let checkable = rest.treeCheckable;
|
|
|
|
if (checkable) {
|
|
|
|
checkable = <span className={`${prefixCls}-tree-checkbox-inner`} />;
|
|
|
|
}
|
2018-12-05 19:12:18 +08:00
|
|
|
|
2018-12-26 16:01:00 +08:00
|
|
|
const inputIcon = (suffixIcon &&
|
|
|
|
(React.isValidElement<{ className?: string }>(suffixIcon)
|
|
|
|
? React.cloneElement(suffixIcon)
|
|
|
|
: suffixIcon)) || <Icon type="down" className={`${prefixCls}-arrow-icon`} />;
|
|
|
|
|
|
|
|
const removeIcon = <Icon type="close" className={`${prefixCls}-remove-icon`} />;
|
|
|
|
|
|
|
|
const clearIcon = (
|
|
|
|
<Icon type="close-circle" className={`${prefixCls}-clear-icon`} theme="filled" />
|
|
|
|
);
|
2017-03-17 15:23:25 +08:00
|
|
|
|
2017-10-20 14:54:38 +08:00
|
|
|
return (
|
2018-12-26 16:01:00 +08:00
|
|
|
<RcTreeSelect
|
|
|
|
switcherIcon={(nodeProps: AntTreeNodeProps) =>
|
|
|
|
this.renderSwitcherIcon(prefixCls, nodeProps)
|
|
|
|
}
|
|
|
|
inputIcon={inputIcon}
|
|
|
|
removeIcon={removeIcon}
|
|
|
|
clearIcon={clearIcon}
|
|
|
|
{...rest}
|
|
|
|
getPopupContainer={getPopupContainer || getContextPopupContainer}
|
|
|
|
dropdownClassName={classNames(dropdownClassName, `${prefixCls}-tree-dropdown`)}
|
|
|
|
prefixCls={prefixCls}
|
|
|
|
className={cls}
|
|
|
|
dropdownStyle={{ maxHeight: '100vh', overflow: 'auto', ...dropdownStyle }}
|
|
|
|
treeCheckable={checkable}
|
|
|
|
notFoundContent={notFoundContent || renderEmpty('Select')}
|
|
|
|
ref={this.saveTreeSelect}
|
|
|
|
/>
|
2017-10-20 14:54:38 +08:00
|
|
|
);
|
2018-12-26 16:01:00 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
render() {
|
|
|
|
return <ConfigConsumer>{this.renderTreeSelect}</ConfigConsumer>;
|
2017-10-20 14:54:38 +08:00
|
|
|
}
|
|
|
|
}
|