ant-design/components/tree-select/index.tsx

126 lines
3.6 KiB
TypeScript
Raw Normal View History

import * as React from 'react';
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';
import { TreeSelectProps } from './interface';
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
export { TreeNode, TreeSelectProps } from './interface';
export default class TreeSelect extends React.Component<TreeSelectProps, any> {
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
};
private rcTreeSelect: any;
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',
);
}
focus() {
this.rcTreeSelect.focus();
}
blur() {
this.rcTreeSelect.blur();
}
saveTreeSelect = (node: typeof RcTreeSelect) => {
this.rcTreeSelect = node;
2018-12-07 20:02:01 +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-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" />
);
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}
/>
);
2018-12-26 16:01:00 +08:00
};
render() {
return <ConfigConsumer>{this.renderTreeSelect}</ConfigConsumer>;
}
}