fix(badge): ts type

mallfoundry · May 30, 2024 · 7a0a724 · 7a0a724
1 parent 45b8bd0
commit 7a0a724
Showing 1 changed file with 18 additions and 14 deletions.
diff --git a/packages/core/src/badge/create-badge-wrapper.tsx b/packages/core/src/badge/create-badge-wrapper.tsx
@@ -19,21 +19,25 @@ function acquireCloneElement(wrapper: ReactNode) {
 
 const createBadgeWrapper =
   <P extends StandardProps = PropsWithChildren<ViewProps>>(wrapper: ReactNode) =>
-    (props: P): JSX.Element => {
-      const { className, ...restProps } = props
-      const cloneElement = acquireCloneElement(wrapper)
+  (props: P): JSX.Element => {
+    const { className, ...restProps } = props
+    const cloneElement = acquireCloneElement(wrapper)
 
-      if (cloneElement) {
-        wrapper = cloneElement(
-          wrapper as ReactElement,
-          {
-            className: classNames(prefixClassname("badge-wrapper"), className, wrapper?.props?.className),
-            ...restProps,
-          } as IconProps,
-        )
-      }
-
-      return <BadgeWrapperContext.Provider value={{}} children={wrapper} />
+    if (cloneElement) {
+      wrapper = cloneElement(
+        wrapper as ReactElement,
+        {
+          className: classNames(
+            prefixClassname("badge-wrapper"),
+            className,
+            (wrapper as ReactElement)?.props?.className,
+          ),
+          ...restProps,
+        } as IconProps,
+      )
     }
 
+    return <BadgeWrapperContext.Provider value={{}} children={wrapper} />
+  }
+
 export default createBadgeWrapper