feat: support nativeElement (#1047)

Author: zombieJ

src/BaseSelect/index.tsx

@@ -75,6 +75,7 @@ export interface BaseSelectRef {
   focus: (options?: FocusOptions) => void;
   blur: () => void;
   scrollTo: ScrollTo;
+  nativeElement: HTMLElement;
 }
 
 export interface BaseSelectPrivateProps {
@@ -329,6 +330,7 @@ const BaseSelect = React.forwardRef<BaseSelectRef, BaseSelectProps>((props, ref)
     focus: selectorRef.current?.focus,
     blur: selectorRef.current?.blur,
     scrollTo: (arg) => listRef.current?.scrollTo(arg),
+    nativeElement: containerRef.current || selectorDomRef.current,
   }));
 
   // ========================== Search Value ==========================

tests/React.test.tsx

@@ -1,5 +1,5 @@
 import * as React from 'react';
-import Select from '../src';
+import Select, { type BaseSelectRef } from '../src';
 import { injectRunAllTimers } from './utils/common';
 import { render } from '@testing-library/react';
 
@@ -22,4 +22,22 @@ describe('React', () => {
     expect(errSpy).not.toHaveBeenCalled();
     errSpy.mockRestore();
   });
+
+  describe('nativeElement', () => {
+    it('work', () => {
+      const selectRef = React.createRef<BaseSelectRef>();
+      const { container } = render(<Select ref={selectRef} />);
+
+      expect(selectRef.current?.nativeElement).toBe(container.querySelector('.rc-select'));
+    });
+
+    it('getRawInputElement', () => {
+      const selectRef = React.createRef<BaseSelectRef>();
+      const { container } = render(
+        <Select ref={selectRef} getRawInputElement={() => <a className="bamboo" />} />,
+      );
+
+      expect(selectRef.current?.nativeElement).toBe(container.querySelector('a.bamboo'));
+    });
+  });
 });