@doeixd/machine 1.0.1 → 1.0.2

package/dist/cjs/development/index.js

@@ -134,6 +134,32 @@ __export(src_exports, {
 });
 module.exports = __toCommonJS(src_exports);
 
+// src/internal-transitions.ts
+var TRANSITIONS_SYMBOL = Symbol.for("__machine_transitions__");
+function attachTransitions(machine, transitions) {
+  Object.defineProperty(machine, TRANSITIONS_SYMBOL, {
+    value: transitions,
+    enumerable: false,
+    configurable: false
+  });
+  return machine;
+}
+function getStoredTransitions(machine) {
+  if (!machine || typeof machine !== "object") {
+    return void 0;
+  }
+  return machine[TRANSITIONS_SYMBOL];
+}
+function snapshotOwnTransitions(source) {
+  if (!source || typeof source !== "object") {
+    return {};
+  }
+  const entries = Object.entries(source).filter(
+    ([key, value]) => key !== "context" && typeof value === "function"
+  );
+  return Object.fromEntries(entries);
+}
+
 // src/generators.ts
 function run(flow, initial) {
   const generator = flow(initial);
@@ -2100,7 +2126,18 @@ function createContextBoundMachine(initialContext, transformers) {
       }
     ])
   );
-  return Object.assign({ context: initialContext }, boundTransitions);
+  Object.values(boundTransitions).forEach((fn) => {
+    if (typeof fn === "function") {
+      Object.defineProperty(fn, "__contextBound", {
+        value: true,
+        enumerable: false
+      });
+    }
+  });
+  return attachTransitions(
+    Object.assign({ context: initialContext }, boundTransitions),
+    boundTransitions
+  );
 }
 function callWithContext(machine, transitionName, ...args) {
   const fn = machine[transitionName];
@@ -2123,13 +2160,12 @@ function createMachine(context, fnsOrFactory) {
       return createMachine(newContext, transitions2);
     };
     transitions2 = fnsOrFactory(transition);
-    return Object.assign({ context }, transitions2);
+    return attachTransitions(Object.assign({ context }, transitions2), transitions2);
   }
-  const transitions = "context" in fnsOrFactory ? Object.fromEntries(
-    Object.entries(fnsOrFactory).filter(([key]) => key !== "context")
-  ) : fnsOrFactory;
+  const stored = getStoredTransitions(fnsOrFactory);
+  const transitions = stored != null ? stored : "context" in fnsOrFactory ? snapshotOwnTransitions(fnsOrFactory) : fnsOrFactory;
   const machine = Object.assign({ context }, transitions);
-  return machine;
+  return attachTransitions(machine, transitions);
 }
 function createAsyncMachine(context, fnsOrFactory) {
   if (typeof fnsOrFactory === "function") {
@@ -2138,13 +2174,12 @@ function createAsyncMachine(context, fnsOrFactory) {
       return createAsyncMachine(newContext, transitions2);
     };
     transitions2 = fnsOrFactory(transition);
-    return Object.assign({ context }, transitions2);
+    return attachTransitions(Object.assign({ context }, transitions2), transitions2);
   }
-  const transitions = "context" in fnsOrFactory ? Object.fromEntries(
-    Object.entries(fnsOrFactory).filter(([key]) => key !== "context")
-  ) : fnsOrFactory;
+  const stored = getStoredTransitions(fnsOrFactory);
+  const transitions = stored != null ? stored : "context" in fnsOrFactory ? snapshotOwnTransitions(fnsOrFactory) : fnsOrFactory;
   const machine = Object.assign({ context }, transitions);
-  return machine;
+  return attachTransitions(machine, transitions);
 }
 function createMachineFactory() {
   return (transformers) => {
@@ -2163,8 +2198,10 @@ function createMachineFactory() {
   };
 }
 function setContext(machine, newContextOrFn) {
-  const { context, ...transitions } = machine;
-  const newContext = typeof newContextOrFn === "function" ? newContextOrFn(context) : newContextOrFn;
+  var _a;
+  const currentContext = machine.context;
+  const transitions = (_a = getStoredTransitions(machine)) != null ? _a : snapshotOwnTransitions(machine);
+  const newContext = typeof newContextOrFn === "function" ? newContextOrFn(currentContext) : newContextOrFn;
   return createMachine(newContext, transitions);
 }
 function createContext(context) {
@@ -2263,7 +2300,6 @@ var MachineBase = class {
   }
 };
 function next(m, update) {
-  const { context, ...transitions } = m;
-  return createMachine(update(context), transitions);
+  return setContext(m, (ctx) => update(ctx));
 }
 //# sourceMappingURL=index.js.map