inherited state, create if does not exist

symbo-ls · Jul 13, 2023 · 3aee101 · 3aee101
1 parent 88a525a
commit 3aee101
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/packages/state/inherit.js b/packages/state/inherit.js
@@ -20,11 +20,13 @@ export const getChildStateInKey = (stateKey, parentState, options = {}) => {
   for (let i = 0; i < arrLength; i++) {
     const childKey = arr[i]
     const grandChildKey = arr[i + 1]
-    const childInParent = parentState[childKey]
-    if (childInParent && childInParent[grandChildKey]) {
-      stateKey = grandChildKey
-      parentState = childInParent
-    } else return
+
+    let childInParent = parentState[childKey]
+    if (!childInParent) childInParent = parentState[childKey] = {}
+    if (!childInParent[grandChildKey]) childInParent[grandChildKey] = {}
+
+    stateKey = grandChildKey
+    parentState = childInParent
   }
   if (options.returnParent) return parentState
   return parentState[stateKey]