Add: child container ref changing on parent change

Merge pull request #30 from BerkliumBirb/main
vovaspace · Feb 6, 2023 · 570c4bd · 570c4bd
2 parents b5115b7 + a554a68
commit 570c4bd
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/brandi-react/src/container/ContainerProvider.tsx b/packages/brandi-react/src/container/ContainerProvider.tsx
@@ -10,9 +10,15 @@ export const ContainerProvider: React.FunctionComponent<{
   isolated?: boolean;
 }> = ({ children, container, isolated = false }) => {
   const parentContainer = useContainer(false);
-  const clonedContainer = React.useMemo(() => container.clone(), [container]);
 
-  if (!isolated) clonedContainer.extend(parentContainer);
+  const extend = !isolated ? parentContainer : null;
+
+  const clonedContainer = React.useMemo(() => {
+    const cloned = container.clone()
+    if (extend) cloned.extend(extend)
+    return cloned
+  }, [container, extend]);
+
 
   return (
     <ContainerContext.Provider value={clonedContainer}>