mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 05:28:20 +08:00
406 lines
11 KiB
TypeScript
406 lines
11 KiB
TypeScript
/* tslint:disable jsx-no-multiline-js */
|
|
import * as React from 'react';
|
|
import * as moment from 'moment';
|
|
import { polyfill } from 'react-lifecycles-compat';
|
|
import RangeCalendar from 'rc-calendar/lib/RangeCalendar';
|
|
import RcDatePicker from 'rc-calendar/lib/Picker';
|
|
import classNames from 'classnames';
|
|
import shallowequal from 'shallowequal';
|
|
import Icon from '../icon';
|
|
import Tag from '../tag';
|
|
import warning from '../_util/warning';
|
|
import interopDefault from '../_util/interopDefault';
|
|
import { RangePickerValue, RangePickerPresetRange } from './interface';
|
|
|
|
export interface RangePickerState {
|
|
value?: RangePickerValue;
|
|
showDate?: RangePickerValue;
|
|
open?: boolean;
|
|
hoverValue?: RangePickerValue;
|
|
}
|
|
|
|
function getShowDateFromValue(value: RangePickerValue) {
|
|
const [start, end] = value;
|
|
// value could be an empty array, then we should not reset showDate
|
|
if (!start && !end) {
|
|
return;
|
|
}
|
|
const newEnd = end && end.isSame(start, 'month') ? end.clone().add(1, 'month') : end;
|
|
return [start, newEnd] as RangePickerValue;
|
|
}
|
|
|
|
function formatValue(value: moment.Moment | undefined, format: string): string {
|
|
return (value && value.format(format)) || '';
|
|
}
|
|
|
|
function pickerValueAdapter(value?: moment.Moment | RangePickerValue): RangePickerValue | undefined {
|
|
if (!value) {
|
|
return;
|
|
}
|
|
if (Array.isArray(value)) {
|
|
return value;
|
|
}
|
|
return [value, value.clone().add(1, 'month')];
|
|
}
|
|
|
|
function isEmptyArray(arr: any) {
|
|
if (Array.isArray(arr)) {
|
|
return arr.length === 0 || arr.every(i => !i);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
function fixLocale(value: RangePickerValue | undefined, localeCode: string) {
|
|
if (!localeCode) {
|
|
return;
|
|
}
|
|
if (!value || value.length === 0) {
|
|
return;
|
|
}
|
|
if (value[0]) {
|
|
value[0]!.locale(localeCode);
|
|
}
|
|
if (value[1]) {
|
|
value[1]!.locale(localeCode);
|
|
}
|
|
}
|
|
|
|
class RangePicker extends React.Component<any, RangePickerState> {
|
|
static defaultProps = {
|
|
prefixCls: 'ant-calendar',
|
|
tagPrefixCls: 'ant-tag',
|
|
allowClear: true,
|
|
showToday: false,
|
|
};
|
|
|
|
static getDerivedStateFromProps(nextProps: any, prevState: any) {
|
|
let state = null;
|
|
if ('value' in nextProps) {
|
|
const value = nextProps.value || [];
|
|
state = {
|
|
value,
|
|
};
|
|
if (!shallowequal(nextProps.value, prevState.value)) {
|
|
state = {
|
|
...state,
|
|
showDate: getShowDateFromValue(value) || prevState.showDate,
|
|
};
|
|
}
|
|
}
|
|
if (('open' in nextProps) && prevState.open !== nextProps.open) {
|
|
state = {
|
|
...state,
|
|
open: nextProps.open,
|
|
};
|
|
}
|
|
return state;
|
|
}
|
|
|
|
private picker: HTMLSpanElement;
|
|
|
|
constructor(props: any) {
|
|
super(props);
|
|
const value = props.value || props.defaultValue || [];
|
|
if (
|
|
value[0] && !interopDefault(moment).isMoment(value[0]) ||
|
|
value[1] && !interopDefault(moment).isMoment(value[1])
|
|
) {
|
|
throw new Error(
|
|
'The value/defaultValue of RangePicker must be a moment object array after `antd@2.0`, ' +
|
|
'see: https://u.ant.design/date-picker-value',
|
|
);
|
|
}
|
|
const pickerValue = !value || isEmptyArray(value) ? props.defaultPickerValue : value;
|
|
this.state = {
|
|
value,
|
|
showDate: pickerValueAdapter(pickerValue || interopDefault(moment)()),
|
|
open: props.open,
|
|
hoverValue: [],
|
|
};
|
|
}
|
|
|
|
clearSelection = (e: React.MouseEvent<HTMLElement>) => {
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
this.setState({ value: [] });
|
|
this.handleChange([]);
|
|
}
|
|
|
|
clearHoverValue = () => this.setState({ hoverValue: [] });
|
|
|
|
handleChange = (value: RangePickerValue) => {
|
|
const props = this.props;
|
|
if (!('value' in props)) {
|
|
this.setState(({ showDate }) => ({
|
|
value,
|
|
showDate: getShowDateFromValue(value) || showDate,
|
|
}));
|
|
}
|
|
props.onChange(value, [
|
|
formatValue(value[0], props.format),
|
|
formatValue(value[1], props.format),
|
|
]);
|
|
this.focus();
|
|
}
|
|
|
|
handleOpenChange = (open: boolean) => {
|
|
if (!('open' in this.props)) {
|
|
this.setState({ open });
|
|
}
|
|
|
|
if (open === false) {
|
|
this.clearHoverValue();
|
|
}
|
|
|
|
const { onOpenChange } = this.props;
|
|
if (onOpenChange) {
|
|
onOpenChange(open);
|
|
}
|
|
}
|
|
|
|
handleShowDateChange = (showDate: RangePickerValue) => this.setState({ showDate });
|
|
|
|
handleHoverChange = (hoverValue: any) => this.setState({ hoverValue });
|
|
|
|
handleRangeMouseLeave = () => {
|
|
if (this.state.open) {
|
|
this.clearHoverValue();
|
|
}
|
|
}
|
|
|
|
handleCalendarInputSelect = (value: RangePickerValue) => {
|
|
if (!value[0]) {
|
|
return;
|
|
}
|
|
this.setState(({ showDate }) => ({
|
|
value,
|
|
showDate: getShowDateFromValue(value) || showDate,
|
|
}));
|
|
}
|
|
|
|
handleRangeClick = (value: RangePickerPresetRange) => {
|
|
if (typeof value === 'function') {
|
|
value = value();
|
|
}
|
|
|
|
this.setValue(value, true);
|
|
|
|
const { onOk, onOpenChange } = this.props;
|
|
if (onOk) {
|
|
onOk(value);
|
|
}
|
|
|
|
if (onOpenChange) {
|
|
onOpenChange(false);
|
|
}
|
|
}
|
|
|
|
setValue(value: RangePickerValue, hidePanel?: boolean) {
|
|
this.handleChange(value);
|
|
if ((hidePanel || !this.props.showTime) && !('open' in this.props)) {
|
|
this.setState({ open: false });
|
|
}
|
|
}
|
|
|
|
focus() {
|
|
this.picker.focus();
|
|
}
|
|
|
|
blur() {
|
|
this.picker.blur();
|
|
}
|
|
|
|
savePicker = (node: HTMLSpanElement) => {
|
|
this.picker = node;
|
|
}
|
|
|
|
renderFooter = (...args: any[]) => {
|
|
const { prefixCls, ranges, renderExtraFooter, tagPrefixCls } = this.props;
|
|
if (!ranges && !renderExtraFooter) {
|
|
return null;
|
|
}
|
|
const customFooter = renderExtraFooter ? (
|
|
<div className={`${prefixCls}-footer-extra`} key="extra">
|
|
{renderExtraFooter(...args)}
|
|
</div>
|
|
) : null;
|
|
const operations = Object.keys(ranges || {}).map((range) => {
|
|
const value = ranges[range];
|
|
return (
|
|
<Tag
|
|
key={range}
|
|
prefixCls={tagPrefixCls}
|
|
color="blue"
|
|
onClick={() => this.handleRangeClick(value)}
|
|
onMouseEnter={() => this.setState({ hoverValue: value })}
|
|
onMouseLeave={this.handleRangeMouseLeave}
|
|
>
|
|
{range}
|
|
</Tag>
|
|
);
|
|
});
|
|
const rangeNode = (operations && operations.length > 0) ? (
|
|
<div className={`${prefixCls}-footer-extra ${prefixCls}-range-quick-selector`} key="range">
|
|
{operations}
|
|
</div>
|
|
) : null;
|
|
return [rangeNode, customFooter];
|
|
}
|
|
|
|
render() {
|
|
const { state, props } = this;
|
|
const { value, showDate, hoverValue, open } = state;
|
|
const {
|
|
prefixCls, popupStyle, style,
|
|
disabledDate, disabledTime,
|
|
showTime, showToday,
|
|
ranges, onOk, locale, localeCode, format,
|
|
dateRender, onCalendarChange, suffixIcon,
|
|
} = props;
|
|
|
|
fixLocale(value, localeCode);
|
|
fixLocale(showDate, localeCode);
|
|
|
|
warning(!('onOK' in props), 'It should be `RangePicker[onOk]`, instead of `onOK`!');
|
|
|
|
const calendarClassName = classNames({
|
|
[`${prefixCls}-time`]: showTime,
|
|
[`${prefixCls}-range-with-ranges`]: ranges,
|
|
});
|
|
|
|
// 需要选择时间时,点击 ok 时才触发 onChange
|
|
let pickerChangeHandler = {
|
|
onChange: this.handleChange,
|
|
};
|
|
let calendarProps: any = {
|
|
onOk: this.handleChange,
|
|
};
|
|
if (props.timePicker) {
|
|
pickerChangeHandler.onChange = changedValue => this.handleChange(changedValue);
|
|
} else {
|
|
calendarProps = {};
|
|
}
|
|
if ('mode' in props) {
|
|
calendarProps.mode = props.mode;
|
|
}
|
|
|
|
const startPlaceholder = ('placeholder' in props)
|
|
? props.placeholder[0] : locale.lang.rangePlaceholder[0];
|
|
const endPlaceholder = ('placeholder' in props)
|
|
? props.placeholder[1] : locale.lang.rangePlaceholder[1];
|
|
|
|
const calendar = (
|
|
<RangeCalendar
|
|
{...calendarProps}
|
|
onChange={onCalendarChange}
|
|
format={format}
|
|
prefixCls={prefixCls}
|
|
className={calendarClassName}
|
|
renderFooter={this.renderFooter}
|
|
timePicker={props.timePicker}
|
|
disabledDate={disabledDate}
|
|
disabledTime={disabledTime}
|
|
dateInputPlaceholder={[startPlaceholder, endPlaceholder]}
|
|
locale={locale.lang}
|
|
onOk={onOk}
|
|
dateRender={dateRender}
|
|
value={showDate}
|
|
onValueChange={this.handleShowDateChange}
|
|
hoverValue={hoverValue}
|
|
onHoverChange={this.handleHoverChange}
|
|
onPanelChange={props.onPanelChange}
|
|
showToday={showToday}
|
|
onInputSelect={this.handleCalendarInputSelect}
|
|
/>
|
|
);
|
|
|
|
// default width for showTime
|
|
const pickerStyle = {} as any;
|
|
if (props.showTime) {
|
|
pickerStyle.width = (style && style.width) || 350;
|
|
}
|
|
|
|
const clearIcon = (!props.disabled && props.allowClear && value && (value[0] || value[1])) ? (
|
|
<Icon
|
|
type="close-circle"
|
|
className={`${prefixCls}-picker-clear`}
|
|
onClick={this.clearSelection}
|
|
theme="filled"
|
|
/>
|
|
) : null;
|
|
|
|
const inputIcon = suffixIcon && (
|
|
React.isValidElement<{ className?: string }>(suffixIcon)
|
|
? React.cloneElement(
|
|
suffixIcon,
|
|
{
|
|
className: classNames({
|
|
[suffixIcon.props.className!]: suffixIcon.props.className,
|
|
[`${prefixCls}-picker-icon`]: true,
|
|
}),
|
|
},
|
|
) : <span className={`${prefixCls}-picker-icon`}>{suffixIcon}</span>) || (
|
|
<Icon type="calendar" className={`${prefixCls}-picker-icon`} />
|
|
);
|
|
|
|
const input = ({ value: inputValue }: { value: any }) => {
|
|
const start = inputValue[0];
|
|
const end = inputValue[1];
|
|
return (
|
|
<span className={props.pickerInputClass}>
|
|
<input
|
|
disabled={props.disabled}
|
|
readOnly
|
|
value={(start && start.format(props.format)) || ''}
|
|
placeholder={startPlaceholder}
|
|
className={`${prefixCls}-range-picker-input`}
|
|
tabIndex={-1}
|
|
/>
|
|
<span className={`${prefixCls}-range-picker-separator`}> ~ </span>
|
|
<input
|
|
disabled={props.disabled}
|
|
readOnly
|
|
value={(end && end.format(props.format)) || ''}
|
|
placeholder={endPlaceholder}
|
|
className={`${prefixCls}-range-picker-input`}
|
|
tabIndex={-1}
|
|
/>
|
|
{clearIcon}
|
|
{inputIcon}
|
|
</span>
|
|
);
|
|
};
|
|
|
|
return (
|
|
<span
|
|
ref={this.savePicker}
|
|
id={props.id}
|
|
className={classNames(props.className, props.pickerClass)}
|
|
style={{ ...style, ...pickerStyle }}
|
|
tabIndex={props.disabled ? -1 : 0}
|
|
onFocus={props.onFocus}
|
|
onBlur={props.onBlur}
|
|
onMouseEnter={props.onMouseEnter}
|
|
onMouseLeave={props.onMouseLeave}
|
|
>
|
|
<RcDatePicker
|
|
{...props}
|
|
{...pickerChangeHandler}
|
|
calendar={calendar}
|
|
value={value}
|
|
open={open}
|
|
onOpenChange={this.handleOpenChange}
|
|
prefixCls={`${prefixCls}-picker-container`}
|
|
style={popupStyle}
|
|
>
|
|
{input}
|
|
</RcDatePicker>
|
|
</span>
|
|
);
|
|
}
|
|
}
|
|
|
|
polyfill(RangePicker);
|
|
|
|
export default RangePicker;
|