chore: tmp of it

Author: zombieJ

src/NewPicker/PickerInput/RangePicker.tsx

@@ -15,7 +15,6 @@ import type {
   PickerRef,
   RangeTimeProps,
   SelectorProps,
-  SelectorRef,
   SharedHTMLAttrs,
   SharedPickerProps,
   ValueDate,
@@ -30,6 +29,7 @@ import { useFieldFormat } from './hooks/useFieldFormat';
 import useInputReadOnly from './hooks/useInputReadOnly';
 import useInvalidate from './hooks/useInvalidate';
 import useOpen from './hooks/useOpen';
+import { usePickerRef } from './hooks/usePickerRef';
 import usePresets from './hooks/usePresets';
 import useRangeActive from './hooks/useRangeActive';
 import useRangeDisabledDate from './hooks/useRangeDisabledDate';
@@ -206,17 +206,7 @@ function RangePicker<DateType extends object = any>(
   } = props;
 
   // ========================= Refs =========================
-  const selectorRef = React.useRef<SelectorRef>();
-
-  React.useImperativeHandle(ref, () => ({
-    nativeElement: selectorRef.current?.nativeElement,
-    focus: () => {
-      selectorRef.current?.focus();
-    },
-    blur: () => {
-      selectorRef.current?.blur();
-    },
-  }));
+  const selectorRef = usePickerRef(ref);
 
   // ======================== Locale ========================
   const filledLocale = useLocale(locale);

src/NewPicker/PickerInput/SinglePicker.tsx

@@ -11,7 +11,6 @@ import type {
   PanelMode,
   PickerRef,
   SelectorProps,
-  SelectorRef,
   SharedHTMLAttrs,
   SharedPickerProps,
   ValueDate,
@@ -27,6 +26,7 @@ import useRangeValue, { useInnerValue } from './hooks/useFlexibleValue';
 import useInputReadOnly from './hooks/useInputReadOnly';
 import useInvalidate from './hooks/useInvalidate';
 import useOpen from './hooks/useOpen';
+import { usePickerRef } from './hooks/usePickerRef';
 import usePresets from './hooks/usePresets';
 import useRangeActive from './hooks/useRangeActive';
 import useRangePickerValue from './hooks/useRangePickerValue';
@@ -75,15 +75,15 @@ export interface BasePickerProps<DateType extends object> extends SharedPickerPr
   onPanelChange?: (values: DateType, modes: PanelMode) => void;
 }
 
-export interface SinglePickerProps<DateType extends object> extends SharedPickerProps<DateType> {
+export interface SinglePickerProps<DateType extends object> extends BasePickerProps<DateType> {
   // Value
   value?: DateType;
   defaultValue?: DateType;
   onChange?: (date: DateType, dateString: string) => void;
   onCalendarChange?: (date: DateType, dateString: string, info: BaseInfo) => void;
 }
 
-export interface MultiplePickerProps<DateType extends object> extends SharedPickerProps<DateType> {
+export interface MultiplePickerProps<DateType extends object> extends BasePickerProps<DateType> {
   multiple: true;
 
   // Value
@@ -174,17 +174,7 @@ function Picker<DateType extends object = any>(
   } = props;
 
   // ========================= Refs =========================
-  const selectorRef = React.useRef<SelectorRef>();
-
-  React.useImperativeHandle(ref, () => ({
-    nativeElement: selectorRef.current?.nativeElement,
-    focus: () => {
-      selectorRef.current?.focus();
-    },
-    blur: () => {
-      selectorRef.current?.blur();
-    },
-  }));
+  const selectorRef = usePickerRef(ref);
 
   // ======================== Locale ========================
   const filledLocale = useLocale(locale);

src/NewPicker/PickerInput/hooks/usePickerRef.ts

@@ -0,0 +1,18 @@
+import * as React from 'react';
+import type { PickerRef, SelectorRef } from '../../interface';
+
+export function usePickerRef(ref: React.Ref<PickerRef>) {
+  const selectorRef = React.useRef<SelectorRef>();
+
+  React.useImperativeHandle(ref, () => ({
+    nativeElement: selectorRef.current?.nativeElement,
+    focus: () => {
+      selectorRef.current?.focus();
+    },
+    blur: () => {
+      selectorRef.current?.blur();
+    },
+  }));
+
+  return selectorRef;
+}