feat(shared/utils): 新增 forwardRefs

Author: mengxinssfd

packages/shared/src/utils/__tests__/forwardRefs.test.ts

@@ -0,0 +1,16 @@
+import { renderHook } from '@testing-library/react';
+import { forwardRefs } from '../';
+import { useRef } from 'react';
+
+describe('forwardRefs', () => {
+  test('forwardRefs should work', () => {
+    const ref2 = jest.fn();
+    const hook = renderHook(() => {
+      const ref1 = useRef();
+      forwardRefs(100, ref1, ref2);
+      return ref1.current;
+    });
+    expect(hook.result.current).toBe(100);
+    expect(ref2.mock.calls[0][0]).toBe(100);
+  });
+});

packages/shared/src/utils/forwardRefs.ts

@@ -0,0 +1,21 @@
+import type { MutableRefObject, Ref } from 'react';
+
+/**
+ * 传递 ref
+ */
+export function forwardRefs(
+  value: unknown,
+  ...refs: (Ref<unknown> | undefined)[]
+): void {
+  refs.forEach((ref): void => {
+    if (!ref) return;
+    switch (typeof ref) {
+      case 'function':
+        ref(value);
+        break;
+      case 'object':
+        (ref as MutableRefObject<unknown>).current = value;
+        break;
+    }
+  });
+}

packages/shared/src/utils/index.ts

@@ -4,3 +4,4 @@ export * from './getElRealSize';
 export * from './getSizeClassName';
 export * from './getClasses';
 export * from './isSameReactEl';
+export * from './forwardRefs';