feat: support nativeElement (#289)

react-component · May 17, 2024 · 6117050 · 6117050
1 parent 9b7822b
commit 6117050
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/index.tsx b/src/index.tsx
@@ -18,6 +18,7 @@ export interface CheckboxRef {
   focus: (options?: FocusOptions) => void;
   blur: () => void;
   input: HTMLInputElement | null;
+  nativeElement: HTMLElement | null;
 }
 
 export interface CheckboxProps
@@ -41,6 +42,8 @@ export const Checkbox = forwardRef<CheckboxRef, CheckboxProps>((props, ref) => {
   } = props;
 
   const inputRef = useRef<HTMLInputElement>(null);
+  const holderRef = useRef<HTMLElement>(null);
+
   const [rawValue, setRawValue] = useMergedState(defaultChecked, {
     value: checked,
   });
@@ -53,6 +56,7 @@ export const Checkbox = forwardRef<CheckboxRef, CheckboxProps>((props, ref) => {
       inputRef.current?.blur();
     },
     input: inputRef.current,
+    nativeElement: holderRef.current,
   }));
 
   const classString = classNames(prefixCls, className, {
@@ -86,7 +90,7 @@ export const Checkbox = forwardRef<CheckboxRef, CheckboxProps>((props, ref) => {
   };
 
   return (
-    <span className={classString} title={title} style={style}>
+    <span className={classString} title={title} style={style} ref={holderRef}>
       <input
         {...inputProps}
         className={`${prefixCls}-input`}

diff --git a/tests/index.test.tsx b/tests/index.test.tsx
@@ -273,4 +273,13 @@ describe('Checkbox ref', () => {
 
     expect(ref.current?.input).toBe(inputEl);
   });
+
+  it('nativeElement should work', () => {
+    const ref = React.createRef<CheckboxRef>();
+
+    const { container } = render(<Checkbox ref={ref} />);
+    const holderEl = container.querySelector('.rc-checkbox')!;
+
+    expect(ref.current?.nativeElement).toBe(holderEl);
+  });
 });