@rectify-dev/core 2.0.3 → 2.1.0

package/dist/index.cjs

@@ -247,6 +247,7 @@ var PlacementFlag = 1 << 0;
 var UpdateFlag = 1 << 1;
 var DeletionFlag = 1 << 2;
 var MoveFlag = 1 << 3;
+var RefFlag = 1 << 4;
 
 // ../rectify-reconciler/src/RectifyFiberLanes.ts
 var NoLanes = 0;
@@ -367,7 +368,8 @@ var createFiber = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((workTag, pendingProp
     childLanes: NoLanes,
     subtreeFlags: NoFlags,
     flags: NoFlags,
-    memoizedState: null
+    memoizedState: null,
+    refCleanup: null
   };
 }, "createFiber");
 var createHostRootFiber = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((containerDom) => {
@@ -394,6 +396,7 @@ var createWorkInProgress = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((current, pe
   }
   wip.memoizedProps = current.memoizedProps;
   wip.memoizedState = current.memoizedState;
+  wip.refCleanup = current.refCleanup;
   wip.return = current.return;
   wip.child = current.child;
   wip.sibling = current.sibling;
@@ -482,23 +485,14 @@ function useState(initialState) {
   }
   const hookIndex = getHookIndex();
   nextHookIndex();
-  let state = fiber.memoizedState;
-  let prevHook = null;
-  for (let i = 0; i < hookIndex; i++) {
-    prevHook = state;
-    state = state?.next ?? null;
-  }
+  let { hook: state, prevHook } = getHookSlot(fiber, hookIndex);
   if (!state) {
     state = {
       memoizedState: getInitialState(initialState),
       queue: null,
       next: null
     };
-    if (prevHook) {
-      prevHook.next = state;
-    } else {
-      fiber.memoizedState = state;
-    }
+    attachHook(fiber, state, prevHook);
   }
   let update = state.queue;
   while (update) {
@@ -523,6 +517,42 @@ function useState(initialState) {
 }
 chunkAJJIEZ7G_cjs.__name(useState, "useState");
 var RectifyHookUseState_default = useState;
+function useReducer(reducer, initialArg, init) {
+  const fiber = getFiberRendering();
+  if (!fiber) {
+    throw new Error("useReducer must be used within a function component.");
+  }
+  const hookIndex = getHookIndex();
+  nextHookIndex();
+  let { hook, prevHook } = getHookSlot(fiber, hookIndex);
+  if (!hook) {
+    const initialState = shared.isFunction(init) ? init(initialArg) : initialArg;
+    hook = { memoizedState: initialState, queue: null, next: null };
+    attachHook(fiber, hook, prevHook);
+  }
+  let update = hook.queue;
+  while (update) {
+    hook.memoizedState = reducer(hook.memoizedState, update.action);
+    update = update.next;
+  }
+  hook.queue = null;
+  const dispatch = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((action) => {
+    const update2 = { action, next: null };
+    if (!hook.queue) {
+      hook.queue = update2;
+    } else {
+      let last = hook.queue;
+      while (last.next) {
+        last = last.next;
+      }
+      last.next = update2;
+    }
+    scheduleRerender(fiber);
+  }, "dispatch");
+  return [hook.memoizedState, dispatch];
+}
+chunkAJJIEZ7G_cjs.__name(useReducer, "useReducer");
+var RectifyHookUseReducer_default = useReducer;
 
 // ../rectify-hook/src/RectifyHookDeps.ts
 var depsChanged = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((prev, next) => {
@@ -633,20 +663,11 @@ function useRef(initialValue) {
   }
   const hookIndex = getHookIndex();
   nextHookIndex();
-  let hook = fiber.memoizedState;
-  let prevHook = null;
-  for (let i = 0; i < hookIndex; i++) {
-    prevHook = hook;
-    hook = hook?.next ?? null;
-  }
+  const { hook, prevHook } = getHookSlot(fiber, hookIndex);
   if (!hook) {
     const ref = { current: initialValue };
     const newHook = { memoizedState: ref, queue: null, next: null };
-    if (prevHook) {
-      prevHook.next = newHook;
-    } else {
-      fiber.memoizedState = newHook;
-    }
+    attachHook(fiber, newHook, prevHook);
     return ref;
   }
   return hook.memoizedState;
@@ -662,20 +683,10 @@ function useMemo(factory, deps) {
   }
   const hookIndex = getHookIndex();
   nextHookIndex();
-  let hook = fiber.memoizedState;
-  let prevHook = null;
-  for (let i = 0; i < hookIndex; i++) {
-    prevHook = hook;
-    hook = hook?.next ?? null;
-  }
+  const { hook, prevHook } = getHookSlot(fiber, hookIndex);
   if (!hook) {
     const state = { value: factory(), deps };
-    const newHook = { memoizedState: state, queue: null, next: null };
-    if (prevHook) {
-      prevHook.next = newHook;
-    } else {
-      fiber.memoizedState = newHook;
-    }
+    attachHook(fiber, { memoizedState: state, queue: null, next: null }, prevHook);
     return state.value;
   }
   const prev = hook.memoizedState;
@@ -1044,6 +1055,9 @@ var reuseOrCreate = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((oldFiber, element,
   if (hasPropsChanged(oldFiber.memoizedProps, element.props)) {
     addFlagToFiber(newFiber, UpdateFlag);
   }
+  if ((oldFiber.memoizedProps?.ref ?? null) !== (element.props?.ref ?? null)) {
+    addFlagToFiber(newFiber, RefFlag);
+  }
   newFiber.return = wip;
   return newFiber;
 }, "reuseOrCreate");
@@ -1141,7 +1155,7 @@ var reconcileChildren = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((wip, children)
 }, "reconcileChildren");
 
 // ../rectify-reconciler/src/RectifyFiberCommitWork.ts
-var MutationMask = PlacementFlag | UpdateFlag | MoveFlag;
+var MutationMask = PlacementFlag | UpdateFlag | MoveFlag | RefFlag;
 var commitWork = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((finishedWork) => {
   if (finishedWork.deletions?.length) {
     finishedWork.deletions.forEach(removeHostTree);
@@ -1166,13 +1180,25 @@ var syncMemoizedProps = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((wip) => {
 }, "syncMemoizedProps");
 var attachRef = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((wip) => {
   const ref = wip.pendingProps?.ref;
-  if (ref && typeof ref === "object" && "current" in ref) {
+  if (!ref) return;
+  if (typeof ref === "function") {
+    const cleanup = ref(wip.stateNode);
+    wip.refCleanup = typeof cleanup === "function" ? cleanup : null;
+  } else if (typeof ref === "object" && "current" in ref) {
     ref.current = wip.stateNode;
   }
 }, "attachRef");
 var detachRef = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((fiber) => {
+  if (fiber.refCleanup) {
+    fiber.refCleanup();
+    fiber.refCleanup = null;
+    return;
+  }
   const ref = fiber.memoizedProps?.ref;
-  if (ref && typeof ref === "object" && "current" in ref) {
+  if (!ref) return;
+  if (typeof ref === "function") {
+    ref(null);
+  } else if (typeof ref === "object" && "current" in ref) {
     ref.current = null;
   }
 }, "detachRef");
@@ -1216,9 +1242,13 @@ var commitMutationHostComponent = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((wip)
   if (hasFlagOnFiber(wip, UpdateFlag)) {
     applyPropsToDom(wip.stateNode, wip.memoizedProps, wip.pendingProps);
     precacheFiberNode(wip, wip.stateNode);
-    attachRef(wip);
     removeFlagFromFiber(wip, UpdateFlag);
   }
+  if (hasFlagOnFiber(wip, RefFlag)) {
+    detachRef(wip);
+    attachRef(wip);
+    removeFlagFromFiber(wip, RefFlag);
+  }
 }, "commitMutationHostComponent");
 var commitMutationHostText = /* @__PURE__ */ chunkAJJIEZ7G_cjs.__name((wip) => {
   if (!wip.stateNode) {
@@ -1435,6 +1465,7 @@ exports.useContext = useContext;
 exports.useEffect = RectifyHookUseEffect_default;
 exports.useLayoutEffect = RectifyHookUseLayoutEffect_default;
 exports.useMemo = RectifyHookUseMemo_default;
+exports.useReducer = RectifyHookUseReducer_default;
 exports.useRef = RectifyHookUseRef_default;
 exports.useState = RectifyHookUseState_default;
 //# sourceMappingURL=index.cjs.map