mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-24 07:28:28 +08:00
275 lines
7.7 KiB
TypeScript
275 lines
7.7 KiB
TypeScript
/* tslint:disable jsx-no-multiline-js */
|
|
import React from 'react';
|
|
import moment from 'moment';
|
|
import PropTypes from 'prop-types';
|
|
import RangeCalendar from 'rc-calendar/lib/RangeCalendar';
|
|
import RcDatePicker from 'rc-calendar/lib/Picker';
|
|
import classNames from 'classnames';
|
|
import Icon from '../icon';
|
|
import { getLocaleCode } from '../_util/getLocale';
|
|
import warning from '../_util/warning';
|
|
|
|
function getShowDateFromValue(value: moment.Moment[]): moment.Moment[] | undefined {
|
|
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];
|
|
}
|
|
|
|
function formatValue(value: moment.Moment | undefined, format: string): string {
|
|
return (value && value.format(format)) || '';
|
|
}
|
|
|
|
function pickerValueAdapter(value?: moment.Moment | moment.Moment[]): moment.Moment[] | undefined {
|
|
if (!value) { return; }
|
|
if (Array.isArray(value)) {
|
|
return value;
|
|
}
|
|
return [value, value.clone().add(1, 'month')];
|
|
}
|
|
|
|
export default class RangePicker extends React.Component<any, any> {
|
|
static contextTypes = {
|
|
antLocale: PropTypes.object,
|
|
};
|
|
static defaultProps = {
|
|
prefixCls: 'ant-calendar',
|
|
allowClear: true,
|
|
showToday: false,
|
|
};
|
|
|
|
constructor(props) {
|
|
super(props);
|
|
const value = props.value || props.defaultValue || [];
|
|
if (
|
|
value[0] && !moment.isMoment(value[0]) ||
|
|
value[1] && !moment.isMoment(value[1])
|
|
) {
|
|
throw new Error(
|
|
'The value/defaultValue of RangePicker must be a moment object array after `antd@2.0`, ' +
|
|
'see: http://u.ant.design/date-picker-value',
|
|
);
|
|
}
|
|
this.state = {
|
|
value,
|
|
open: props.open,
|
|
hoverValue: [],
|
|
};
|
|
}
|
|
|
|
componentWillReceiveProps(nextProps) {
|
|
if ('value' in nextProps) {
|
|
const value = nextProps.value || [];
|
|
this.setState({ value, showDate: getShowDateFromValue(value) });
|
|
}
|
|
if ('open' in nextProps) {
|
|
this.setState({
|
|
open: nextProps.open,
|
|
});
|
|
}
|
|
}
|
|
|
|
clearSelection = (e) => {
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
this.setState({ value: [] });
|
|
this.handleChange([]);
|
|
}
|
|
|
|
clearHoverValue = () => this.setState({ hoverValue: [] });
|
|
|
|
handleChange = (value: moment.Moment[]) => {
|
|
const props = this.props;
|
|
if (!('value' in props)) {
|
|
this.setState({ value, showDate: getShowDateFromValue(value) });
|
|
}
|
|
props.onChange(value, [
|
|
formatValue(value[0], props.format),
|
|
formatValue(value[1], props.format),
|
|
]);
|
|
}
|
|
|
|
handleOpenChange = (open) => {
|
|
this.setState({ open });
|
|
|
|
const { onOpenChange } = this.props;
|
|
if (onOpenChange) {
|
|
onOpenChange(open);
|
|
}
|
|
}
|
|
|
|
handleShowDateChange = showDate => this.setState({ showDate });
|
|
|
|
handleHoverChange = hoverValue => this.setState({ hoverValue });
|
|
|
|
setValue(value) {
|
|
this.handleChange(value);
|
|
if (!this.props.showTime) {
|
|
this.setState({ open: false });
|
|
}
|
|
}
|
|
|
|
renderFooter = (...args) => {
|
|
const { prefixCls, ranges, renderExtraFooter } = 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 (
|
|
<a
|
|
key={range}
|
|
onClick={() => this.setValue(value)}
|
|
onMouseEnter={() => this.setState({ hoverValue: value })}
|
|
onMouseLeave={this.clearHoverValue}
|
|
>
|
|
{range}
|
|
</a>
|
|
);
|
|
});
|
|
const rangeNode = (
|
|
<div className={`${prefixCls}-footer-extra ${prefixCls}-range-quick-selector`} key="range">
|
|
{operations}
|
|
</div>
|
|
);
|
|
return [rangeNode, customFooter];
|
|
}
|
|
|
|
render() {
|
|
const { state, props, context } = this;
|
|
const { value, showDate, hoverValue, open } = state;
|
|
const localeCode = getLocaleCode(context);
|
|
if (value && localeCode) {
|
|
if (value[0]) {
|
|
value[0].locale(localeCode);
|
|
}
|
|
if (value[1]) {
|
|
value[1].locale(localeCode);
|
|
}
|
|
}
|
|
|
|
const {
|
|
prefixCls, popupStyle, style,
|
|
disabledDate, disabledTime,
|
|
showTime, showToday,
|
|
ranges, onOk, locale, format,
|
|
} = props;
|
|
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 calendarHandler: Object = {
|
|
onOk: this.handleChange,
|
|
};
|
|
if (props.timePicker) {
|
|
pickerChangeHandler.onChange = changedValue => this.handleChange(changedValue);
|
|
} else {
|
|
calendarHandler = {};
|
|
}
|
|
|
|
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
|
|
{...calendarHandler}
|
|
format={format}
|
|
prefixCls={prefixCls}
|
|
className={calendarClassName}
|
|
renderFooter={this.renderFooter}
|
|
timePicker={props.timePicker}
|
|
disabledDate={disabledDate}
|
|
disabledTime={disabledTime}
|
|
dateInputPlaceholder={[startPlaceholder, endPlaceholder]}
|
|
locale={locale.lang}
|
|
onOk={onOk}
|
|
value={showDate || pickerValueAdapter(props.defaultPickerValue) || pickerValueAdapter(moment())}
|
|
onValueChange={this.handleShowDateChange}
|
|
hoverValue={hoverValue}
|
|
onHoverChange={this.handleHoverChange}
|
|
showToday={showToday}
|
|
/>
|
|
);
|
|
|
|
// default width for showTime
|
|
const pickerStyle = {} as any;
|
|
if (props.showTime) {
|
|
pickerStyle.width = (style && style.width) || 300;
|
|
}
|
|
|
|
const clearIcon = (!props.disabled && props.allowClear && value && (value[0] || value[1])) ? (
|
|
<Icon
|
|
type="cross-circle"
|
|
className={`${prefixCls}-picker-clear`}
|
|
onClick={this.clearSelection}
|
|
/>
|
|
) : null;
|
|
|
|
const input = ({ value: inputValue }) => {
|
|
const start = inputValue[0];
|
|
const end = inputValue[1];
|
|
return (
|
|
<span className={props.pickerInputClass} disabled={props.disabled}>
|
|
<input
|
|
disabled={props.disabled}
|
|
readOnly
|
|
value={(start && start.format(props.format)) || ''}
|
|
placeholder={startPlaceholder}
|
|
className={`${prefixCls}-range-picker-input`}
|
|
/>
|
|
<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`}
|
|
/>
|
|
{clearIcon}
|
|
<span className={`${prefixCls}-picker-icon`} />
|
|
</span>
|
|
);
|
|
};
|
|
|
|
return (
|
|
<span
|
|
className={classNames(props.className, props.pickerClass)}
|
|
style={{
|
|
...style,
|
|
...pickerStyle,
|
|
}}
|
|
>
|
|
<RcDatePicker
|
|
{...props}
|
|
{...pickerChangeHandler}
|
|
calendar={calendar}
|
|
value={value}
|
|
open={open}
|
|
onOpenChange={this.handleOpenChange}
|
|
prefixCls={`${prefixCls}-picker-container`}
|
|
style={popupStyle}
|
|
>
|
|
{input}
|
|
</RcDatePicker>
|
|
</span>
|
|
);
|
|
}
|
|
}
|