chore: allow custom renderer to receive function or react node (#291)

jeremybarbet · Dec 3, 2020 · 26a4c14 · 26a4c14
1 parent 6de9822
commit 26a4c14
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/index.tsx b/src/index.tsx
@@ -672,6 +672,9 @@ const ModalizeBase = (
     );
   };
 
+  const renderElement = (Element: React.ReactNode): JSX.Element =>
+    typeof Element === 'function' ? Element() : Element;
+
   const renderComponent = (
     component: React.ReactNode,
     name: 'header' | 'footer' | 'floating',
@@ -680,8 +683,6 @@ const ModalizeBase = (
       return null;
     }
 
-    const renderElement = (Element: React.ReactNode): JSX.Element =>
-      typeof Element === 'function' ? Element() : Element;
     const tag = renderElement(component);
 
     /**
@@ -757,7 +758,8 @@ const ModalizeBase = (
     }
 
     if (customRenderer) {
-      return React.cloneElement(customRenderer, { ...opts });
+      const tag = renderElement(customRenderer);
+      return React.cloneElement(tag, { ...opts });
     }
 
     return (