mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-03 04:30:06 +08:00
136 lines
3.8 KiB
TypeScript
136 lines
3.8 KiB
TypeScript
import * as React from 'react';
|
|
import RcTreeSelect, { TreeNode, SHOW_ALL, SHOW_PARENT, SHOW_CHILD } from 'rc-tree-select';
|
|
import classNames from 'classnames';
|
|
import { TreeSelectProps, TreeNodeValue } from './interface';
|
|
import { ConfigConsumer, ConfigConsumerProps } from '../config-provider';
|
|
import warning from '../_util/warning';
|
|
import Icon from '../icon';
|
|
import { AntTreeNodeProps } from '../tree';
|
|
import omit from 'omit.js';
|
|
|
|
export { TreeNode, TreeSelectProps } from './interface';
|
|
|
|
export default class TreeSelect<T extends TreeNodeValue> extends React.Component<
|
|
TreeSelectProps<T>,
|
|
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',
|
|
};
|
|
|
|
private rcTreeSelect: any;
|
|
|
|
constructor(props: TreeSelectProps<T>) {
|
|
super(props);
|
|
|
|
warning(
|
|
props.multiple !== false || !props.treeCheckable,
|
|
'TreeSelect',
|
|
'`multiple` will alway be `true` when `treeCheckable` is true',
|
|
);
|
|
}
|
|
|
|
focus() {
|
|
this.rcTreeSelect.focus();
|
|
}
|
|
|
|
blur() {
|
|
this.rcTreeSelect.blur();
|
|
}
|
|
|
|
saveTreeSelect = (node: typeof RcTreeSelect) => {
|
|
this.rcTreeSelect = node;
|
|
};
|
|
|
|
renderSwitcherIcon = (prefixCls: string, { isLeaf, loading }: AntTreeNodeProps) => {
|
|
if (loading) {
|
|
return <Icon type="loading" className={`${prefixCls}-switcher-loading-icon`} />;
|
|
}
|
|
if (isLeaf) {
|
|
return null;
|
|
}
|
|
return <Icon type="caret-down" className={`${prefixCls}-switcher-icon`} />;
|
|
};
|
|
|
|
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,
|
|
);
|
|
|
|
// showSearch: single - false, multiple - true
|
|
let { showSearch } = restProps;
|
|
if (!('showSearch' in restProps)) {
|
|
showSearch = !!(restProps.multiple || restProps.treeCheckable);
|
|
}
|
|
|
|
let checkable = rest.treeCheckable;
|
|
if (checkable) {
|
|
checkable = <span className={`${prefixCls}-tree-checkbox-inner`} />;
|
|
}
|
|
|
|
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 (
|
|
<RcTreeSelect
|
|
switcherIcon={(nodeProps: AntTreeNodeProps) =>
|
|
this.renderSwitcherIcon(prefixCls, nodeProps)
|
|
}
|
|
inputIcon={inputIcon}
|
|
removeIcon={removeIcon}
|
|
clearIcon={clearIcon}
|
|
{...rest}
|
|
showSearch={showSearch}
|
|
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}
|
|
/>
|
|
);
|
|
};
|
|
|
|
render() {
|
|
return <ConfigConsumer>{this.renderTreeSelect}</ConfigConsumer>;
|
|
}
|
|
}
|