fix: DatePicker 中disabledTime 禁用问题 (#666)

* fix: fix-bug

* chore: remove whitespace

* feat: change logic

* feat: change logic

* feat: change logic

* test: add test case

* test: add test case

* test: add test case

* fix: fix incorrect time

Author: SnowingFox

.gitignore

@@ -28,6 +28,7 @@ lib
 es
 yarn.lock
 package-lock.json
+pnpm-lock.yaml
 coverage/
 .doc
 .history

docs/examples/time.tsx

@@ -16,7 +16,7 @@ export default () => {
         picker="date"
         showTime
         disabledTime={() => ({
-          disabledHours: () => [1, 2, 3, 4, 5, 6],
+          disabledHours: () => [0, 1, 2, 3, 4, 5, 6, 20, 21],
         })}
         locale={zhCN}
         generateConfig={momentGenerateConfig}

src/panels/DatetimePanel/index.tsx

@@ -18,6 +18,17 @@ export type DatetimePanelProps<DateType> = {
 const ACTIVE_PANEL = tuple('date', 'time');
 type ActivePanelType = (typeof ACTIVE_PANEL)[number];
 
+const findValidTime = (disabledRange: number[], maxValidTime: number) => {
+  const rangeSet = new Set(disabledRange);
+  for (let i = 0; i <= maxValidTime; i++) {
+    if (!rangeSet.has(i)) {
+      return i;
+    }
+  }
+
+  return 0
+}
+
 function DatetimePanel<DateType>(props: DatetimePanelProps<DateType>) {
   const {
     prefixCls,
@@ -111,6 +122,15 @@ function DatetimePanel<DateType>(props: DatetimePanelProps<DateType>) {
       selectedDate = generateConfig.setYear(selectedDate, generateConfig.getYear(defaultValue));
       selectedDate = generateConfig.setMonth(selectedDate, generateConfig.getMonth(defaultValue));
       selectedDate = generateConfig.setDate(selectedDate, generateConfig.getDate(defaultValue));
+    } else if (source === 'date' && value && disabledTime) {
+      const disabledTimes = disabledTime(value)
+
+      const validHour = findValidTime(disabledTimes.disabledHours?.() || [-1], 23)
+      const validMinute = findValidTime(disabledTimes.disabledMinutes?.(validHour) || [-1], 59)
+      const validSeconds = findValidTime(disabledTimes.disabledSeconds?.(validHour, validMinute) || [-1], 59)
+      selectedDate = generateConfig.setHour(selectedDate, validHour)
+      selectedDate = generateConfig.setMinute(selectedDate, validMinute)
+      selectedDate = generateConfig.setSecond(selectedDate, validSeconds)
     }
 
     if (onSelect) {

tests/disabledTime.spec.tsx

@@ -138,6 +138,27 @@ describe('Picker.DisabledTime', () => {
     jest.useRealTimers();
   });
 
+  it('disabledTime should reset correctly when date changed by click', function () {
+    const disabledTime = jest.fn((_: Moment | null, __: 'start' | 'end') => ({
+      disabledHours: () => [0, 1, 2, 3, 4, 10],
+    }));
+
+    render(
+      <MomentRangePicker
+        open
+        showTime
+        disabledTime={disabledTime}
+        defaultValue={[getMoment('1989-11-28'), getMoment('1990-09-03')]}
+      />,
+    );
+
+    expect(document.querySelector('.rc-picker-input > input').getAttribute('value')).toEqual('1989-11-28 00:00:00');
+
+    fireEvent.click(document.querySelectorAll('.rc-picker-cell-inner')[2]);
+
+    expect(document.querySelector('.rc-picker-input > input').getAttribute('value')).toEqual('1989-10-31 05:00:00');
+  });
+
   describe('warning for legacy props', () => {
     it('single', () => {
       resetWarned();