refactor: add config for getResetStyles (#29)

* refactor: add config for getResetStyles

* test: add test case

* refactor: optimizing code logic

* chore: add comments to style reset logic

Author: YumoImer

src/util/genStyleUtils.tsx

@@ -91,7 +91,12 @@ export type CSSVarRegisterProps = {
   };
 };
 
-export type GetResetStyles<AliasToken extends TokenType> = (token: AliasToken) => CSSInterpolation;
+type GetResetStylesConfig = {
+  prefix: ReturnType<UsePrefix>;
+  csp: ReturnType<UseCSP>
+};
+
+export type GetResetStyles<AliasToken extends TokenType> = (token: AliasToken, config?: GetResetStylesConfig) => CSSInterpolation;
 
 export type GetCompUnitless<CompTokenMap extends TokenMap, AliasToken extends TokenType> = <
   C extends TokenMapKey<CompTokenMap>,
@@ -345,11 +350,14 @@ function genStyleUtils<
         order: options.order || -999,
       };
 
-      // Generate style for all need reset tags.
-      useStyleRegister(
-        { ...sharedConfig, clientOnly: false, path: ['Shared', rootPrefixCls] },
-        () => (typeof getResetStyles === 'function' ? getResetStyles(token) : []),
-      );
+      // This if statement is safe, as it will only be used if the generator has the function. It's not dynamic.
+      if (typeof getResetStyles === 'function') {
+        // Generate style for all need reset tags.
+        useStyleRegister(
+          { ...sharedConfig, clientOnly: false, path: ['Shared', rootPrefixCls] },
+          () => getResetStyles(token, { prefix: { rootPrefixCls, iconPrefixCls }, csp }),
+        );
+      }
 
       const wrapSSR = useStyleRegister(
         { ...sharedConfig, path: [concatComponent, prefixCls, iconPrefixCls] },

tests/genStyleUtils.test.tsx

@@ -88,6 +88,27 @@ describe('genStyleUtils', () => {
     });
   });
 
+  describe('genComponentStyleHook should run without getResetStyles', () => {
+    it('should generate component style hook', () => {
+      const component = 'TestComponent';
+      const styleFn = jest.fn();
+      const getDefaultToken = jest.fn();
+      const { getResetStyles, ...restMockConfig } = mockConfig;
+      const { genComponentStyleHook: genComponentStyleHookWithoutReset } = genStyleUtils<
+        TestCompTokenMap,
+        object,
+        object
+      >(restMockConfig);
+      const hook = genComponentStyleHookWithoutReset(component, styleFn, getDefaultToken);
+      const TestComponent: React.FC<SubStyleComponentProps> = ({ prefixCls, rootCls }) => {
+        hook(prefixCls, rootCls);
+        return <div data-testid="test-component">Test</div>;
+      };
+      const { container } = render(<TestComponent prefixCls="test-prefix" rootCls="test-root" />);
+      expect(() => container).not.toThrow();
+    });
+  });
+
   describe('CSSVarRegister', () => {
     it('should render CSSVarRegister component', () => {
       const CSSVarRegister: React.FC<CSSVarRegisterProps> = ({ rootCls, cssVar = {} }) => {