feat(shared): 新增 mergeReactDefaultProps 工具函数

Author: mengxinssfd

packages/shared/src/utils/__tests__/mergeReactDefaultProps.test.tsx

@@ -0,0 +1,41 @@
+import { render } from '@testing-library/react';
+import { mergeReactDefaultProps } from '../';
+import React from 'react';
+
+describe('mergeReactDefaultProps', () => {
+  test('base', () => {
+    const receiveProps = jest.fn();
+    const defaultProps = {
+      emptyStr: 5,
+      undef: 1,
+      null: 2,
+      zero: 4,
+      nan: 3,
+    };
+    const App: React.FC<any> = React.forwardRef<HTMLDivElement, any>(
+      (props: any, ref) => {
+        receiveProps(props);
+        return <div ref={ref}>1</div>;
+      },
+    );
+    App.defaultProps = defaultProps;
+    const props = {
+      undef: undefined,
+      emptyStr: '',
+      null: null,
+      nan: NaN,
+      zero: 0,
+    };
+    render(<App {...props} />);
+    const result = {
+      emptyStr: '',
+      null: null,
+      nan: NaN,
+      undef: 1,
+      zero: 0,
+    };
+    expect(receiveProps.mock.calls[0]).toEqual([result]);
+
+    expect(mergeReactDefaultProps(props, defaultProps as any)).toEqual(result);
+  });
+});

packages/shared/src/utils/index.ts

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

packages/shared/src/utils/mergeReactDefaultProps.ts

@@ -0,0 +1,12 @@
+import { RequiredPart } from '@tool-pack/types';
+import { defaults } from '@tool-pack/basic';
+
+/**
+ * 用于代替 react 的 defaultProps 合并功能
+ */
+export function mergeReactDefaultProps<const T, const P>(
+  props: T,
+  defaultProps: P,
+): keyof P extends keyof T ? RequiredPart<T, keyof P> : T & P {
+  return defaults({ ...props }, defaultProps) as any;
+}