@beinformed/ui 1.28.3 → 1.28.4

package/src/react-client/client.js

@@ -40,10 +40,16 @@ import type { RouterHistory } from "react-router";
 import type { BeforeRenderHook } from "../redux/store/beforeRenderHooks";
 import type { Props as FallbackProps } from "../react/ErrorBoundaryFallback";
 
+type RenderFunction = () => ReactElement<ElementType>;
+type MountFunction = (
+  applicationNode: Element,
+  element: ReactElement<ElementType>
+) => void;
 type Props = {
   customReducers?: CustomReducers,
   theme?: Theme | Array<Theme>,
-  render: Function,
+  render: RenderFunction,
+  mount: MountFunction,
   beforeRenderHooks?: Array<BeforeRenderHook>,
   ErrorFallbackComponent?: ComponentType<FallbackProps>,
 };
@@ -158,9 +164,9 @@ export const addContentLoadedEvent = (
   store: ReduxStore,
   routerHistory: RouterHistory,
   theme?: Theme | Array<Theme>,
-  render: Function,
+  render: RenderFunction,
   ErrorFallbackComponent?: ComponentType<FallbackProps>,
-  mount: Function
+  mount: MountFunction
 ) => {
   window.addEventListener("DOMContentLoaded", () => {
     const applicationNode = document.querySelector("#application");
@@ -189,13 +195,13 @@ export const addContentLoadedEvent = (
  */
 const mountClient = (
   applicationNode: Element,
-  initComponent: ReactElement<ElementType>
+  element: ReactElement<ElementType>
 ) => {
   const isSSR = applicationNode.querySelector(".application");
   if (isSSR) {
-    hydrate(initComponent, applicationNode);
+    hydrate(element, applicationNode);
   } else {
-    render(initComponent, applicationNode);
+    render(element, applicationNode);
   }
 };
 
@@ -208,6 +214,7 @@ export const client = ({
   render,
   beforeRenderHooks,
   ErrorFallbackComponent,
+  mount,
 }: Props) => {
   const { store, routerHistory } = setupClient(
     customReducers,
@@ -220,6 +227,6 @@ export const client = ({
     theme,
     render,
     ErrorFallbackComponent,
-    mountClient
+    mount ?? mountClient
   );
 };