mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-05 09:49:57 +08:00
fb2e355f14
close #12706
266 lines
7.0 KiB
TypeScript
266 lines
7.0 KiB
TypeScript
import * as React from 'react';
|
|
import * as PropTypes from 'prop-types';
|
|
import * as moment from 'moment';
|
|
import FullCalendar from 'rc-calendar/lib/FullCalendar';
|
|
import LocaleReceiver from '../locale-provider/LocaleReceiver';
|
|
import { PREFIX_CLS } from './Constants';
|
|
import Header from './Header';
|
|
import interopDefault from '../_util/interopDefault';
|
|
import enUS from './locale/en_US';
|
|
|
|
export { HeaderProps } from './Header';
|
|
|
|
function noop() { return null; }
|
|
|
|
function zerofixed(v: number) {
|
|
if (v < 10) {
|
|
return `0${v}`;
|
|
}
|
|
return `${v}`;
|
|
}
|
|
|
|
export type CalendarMode = 'month' | 'year';
|
|
|
|
export interface CalendarProps {
|
|
prefixCls?: string;
|
|
className?: string;
|
|
value?: moment.Moment;
|
|
defaultValue?: moment.Moment;
|
|
mode?: CalendarMode;
|
|
fullscreen?: boolean;
|
|
dateCellRender?: (date: moment.Moment) => React.ReactNode;
|
|
monthCellRender?: (date: moment.Moment) => React.ReactNode;
|
|
dateFullCellRender?: (date: moment.Moment) => React.ReactNode;
|
|
monthFullCellRender?: (date: moment.Moment) => React.ReactNode;
|
|
locale?: any;
|
|
style?: React.CSSProperties;
|
|
onPanelChange?: (date?: moment.Moment, mode?: CalendarMode) => void;
|
|
onSelect?: (date?: moment.Moment) => void;
|
|
onChange?: (date?: moment.Moment) => void;
|
|
disabledDate?: (current: moment.Moment) => boolean;
|
|
validRange ?: [moment.Moment, moment.Moment];
|
|
}
|
|
|
|
export interface CalendarState {
|
|
value: moment.Moment;
|
|
mode?: CalendarMode;
|
|
}
|
|
|
|
export default class Calendar extends React.Component<CalendarProps, CalendarState> {
|
|
static defaultProps = {
|
|
locale: {},
|
|
fullscreen: true,
|
|
prefixCls: PREFIX_CLS,
|
|
mode: 'month',
|
|
onSelect: noop,
|
|
onPanelChange: noop,
|
|
onChange: noop,
|
|
};
|
|
|
|
static propTypes = {
|
|
monthCellRender: PropTypes.func,
|
|
dateCellRender: PropTypes.func,
|
|
monthFullCellRender: PropTypes.func,
|
|
dateFullCellRender: PropTypes.func,
|
|
fullscreen: PropTypes.bool,
|
|
locale: PropTypes.object,
|
|
prefixCls: PropTypes.string,
|
|
className: PropTypes.string,
|
|
style: PropTypes.object,
|
|
onPanelChange: PropTypes.func,
|
|
value: PropTypes.object,
|
|
onSelect: PropTypes.func,
|
|
onChange: PropTypes.func,
|
|
};
|
|
|
|
constructor(props: CalendarProps) {
|
|
super(props);
|
|
|
|
const value = props.value || props.defaultValue || interopDefault(moment)();
|
|
if (!interopDefault(moment).isMoment(value)) {
|
|
throw new Error(
|
|
'The value/defaultValue of Calendar must be a moment object after `antd@2.0`, ' +
|
|
'see: https://u.ant.design/calendar-value',
|
|
);
|
|
}
|
|
this.state = {
|
|
value,
|
|
mode: props.mode,
|
|
};
|
|
}
|
|
|
|
componentWillReceiveProps(nextProps: CalendarProps) {
|
|
if ('value' in nextProps) {
|
|
this.setState({
|
|
value: nextProps.value!,
|
|
});
|
|
}
|
|
if ('mode' in nextProps && nextProps.mode !== this.props.mode) {
|
|
this.setState({
|
|
mode: nextProps.mode!,
|
|
});
|
|
}
|
|
}
|
|
|
|
monthCellRender = (value: moment.Moment) => {
|
|
const { prefixCls, monthCellRender = noop as Function } = this.props;
|
|
return (
|
|
<div className={`${prefixCls}-month`}>
|
|
<div className={`${prefixCls}-value`}>
|
|
{value.localeData().monthsShort(value)}
|
|
</div>
|
|
<div className={`${prefixCls}-content`}>
|
|
{monthCellRender(value)}
|
|
</div>
|
|
</div>
|
|
);
|
|
}
|
|
|
|
dateCellRender = (value: moment.Moment) => {
|
|
const { prefixCls, dateCellRender = noop as Function } = this.props;
|
|
return (
|
|
<div className={`${prefixCls}-date`}>
|
|
<div className={`${prefixCls}-value`}>
|
|
{zerofixed(value.date())}
|
|
</div>
|
|
<div className={`${prefixCls}-content`}>
|
|
{dateCellRender(value)}
|
|
</div>
|
|
</div>
|
|
);
|
|
}
|
|
|
|
setValue = (value: moment.Moment, way: 'select' | 'changePanel') => {
|
|
if (!('value' in this.props)) {
|
|
this.setState({ value });
|
|
}
|
|
if (way === 'select') {
|
|
if (this.props.onSelect) {
|
|
this.props.onSelect(value);
|
|
}
|
|
} else if (way === 'changePanel') {
|
|
this.onPanelChange(value, this.state.mode);
|
|
}
|
|
}
|
|
|
|
setType = (type: string) => {
|
|
const mode = (type === 'date') ? 'month' : 'year';
|
|
if (this.state.mode !== mode) {
|
|
this.setState({ mode });
|
|
this.onPanelChange(this.state.value, mode);
|
|
}
|
|
}
|
|
|
|
onHeaderValueChange = (value: moment.Moment) => {
|
|
this.setValue(value, 'changePanel');
|
|
}
|
|
|
|
onHeaderTypeChange = (type: string) => {
|
|
this.setType(type);
|
|
}
|
|
|
|
onPanelChange(value: moment.Moment, mode: CalendarMode | undefined) {
|
|
const { onPanelChange, onChange } = this.props;
|
|
if (onPanelChange) {
|
|
onPanelChange(value, mode);
|
|
}
|
|
if (onChange && value !== this.state.value) {
|
|
onChange(value);
|
|
}
|
|
}
|
|
|
|
onSelect = (value: moment.Moment) => {
|
|
this.setValue(value, 'select');
|
|
}
|
|
|
|
getDateRange = (
|
|
validRange: [moment.Moment, moment.Moment],
|
|
disabledDate?: (current: moment.Moment) => boolean,
|
|
) => (current: moment.Moment) => {
|
|
if (!current) {
|
|
return false;
|
|
}
|
|
const [ startDate, endDate ] = validRange;
|
|
const inRange = !current.isBetween(startDate, endDate, 'days', '[]');
|
|
if (disabledDate) {
|
|
return (disabledDate(current) || inRange);
|
|
}
|
|
return inRange;
|
|
}
|
|
|
|
renderCalendar = (locale: any, localeCode: string) => {
|
|
const { state, props } = this;
|
|
const { value, mode } = state;
|
|
if (value && localeCode) {
|
|
value.locale(localeCode);
|
|
}
|
|
const { prefixCls, style, className, fullscreen, dateFullCellRender, monthFullCellRender } = props;
|
|
const type = (mode === 'year') ? 'month' : 'date';
|
|
|
|
let cls = className || '';
|
|
if (fullscreen) {
|
|
cls += (` ${prefixCls}-fullscreen`);
|
|
}
|
|
|
|
const monthCellRender = monthFullCellRender || this.monthCellRender;
|
|
const dateCellRender = dateFullCellRender || this.dateCellRender;
|
|
|
|
let disabledDate = props.disabledDate;
|
|
|
|
if (props.validRange) {
|
|
disabledDate = this.getDateRange(props.validRange, disabledDate);
|
|
}
|
|
|
|
return (
|
|
<div className={cls} style={style}>
|
|
<Header
|
|
fullscreen={fullscreen}
|
|
type={type}
|
|
value={value}
|
|
locale={locale.lang}
|
|
prefixCls={prefixCls}
|
|
onTypeChange={this.onHeaderTypeChange}
|
|
onValueChange={this.onHeaderValueChange}
|
|
validRange={props.validRange}
|
|
/>
|
|
<FullCalendar
|
|
{...props}
|
|
disabledDate={disabledDate}
|
|
Select={noop}
|
|
locale={locale.lang}
|
|
type={type}
|
|
prefixCls={prefixCls}
|
|
showHeader={false}
|
|
value={value}
|
|
monthCellRender={monthCellRender}
|
|
dateCellRender={dateCellRender}
|
|
onSelect={this.onSelect}
|
|
/>
|
|
</div>
|
|
);
|
|
}
|
|
|
|
getDefaultLocale = () => {
|
|
const result = {
|
|
...enUS,
|
|
...this.props.locale,
|
|
};
|
|
result.lang = {
|
|
...result.lang,
|
|
...(this.props.locale || {}).lang,
|
|
};
|
|
return result;
|
|
}
|
|
|
|
render() {
|
|
return (
|
|
<LocaleReceiver
|
|
componentName="Calendar"
|
|
defaultLocale={this.getDefaultLocale}
|
|
>
|
|
{this.renderCalendar}
|
|
</LocaleReceiver>
|
|
);
|
|
}
|
|
}
|