@doeixd/machine 0.0.17 → 0.0.18

package/src/generators.ts

@@ -135,12 +135,12 @@ export function run<C extends any = any, M extends { context: C } & Record<strin
 
     // If the generator has returned (done), we have our final value
     if (done) {
-      return value;
+      return value as T;
     }
 
     // Otherwise, the yielded value becomes our new current state
     // This state will be sent back into the generator on the next iteration
-    current = value;
+    current = value as M;
   }
 }
 
@@ -346,10 +346,10 @@ export function runWithDebug<C extends any = any, M extends { context: C } & Rec
 
     if (done) {
       console.log('Final:', value);
-      return value;
+      return value as T;
     }
 
-    current = value;
+    current = value as M;
     stepCount++;
     logger(stepCount, current);
   }
@@ -391,10 +391,10 @@ export async function runAsync<C extends any = any, M extends { context: C } & R
     const { value, done } = await generator.next(current);
 
     if (done) {
-      return value;
+      return value as T;
     }
 
-    current = value;
+    current = value as M;
   }
 }
 

package/src/index.ts

@@ -60,6 +60,21 @@ export type AsyncTransitionArgs<M extends AsyncMachine<any, any>, K extends keyo
   ? A extends [...infer Rest, TransitionOptions] ? Rest : A
   : never;
 
+/**
+ * A helper type to define a distinct state in a state machine (a "typestate").
+ * Allows defining the context and transitions in a single generic type.
+ * @template C - The context specific to this state.
+ * @template T - The transitions available in this state.
+ */
+export type TypeState<C extends object, T extends object = {}> = Machine<C, T>;
+
+/**
+ * A helper type to define a distinct async state in a state machine.
+ * @template C - The context specific to this state.
+ * @template T - The transitions available in this state.
+ */
+export type AsyncTypeState<C extends object, T extends object = {}> = AsyncMachine<C, T>;
+
 
 /**
  * Options passed to async transition functions, including cancellation support.
@@ -332,7 +347,7 @@ export type BindTransitions<T> = {
  */
 export function createMachine<C extends object, T extends Record<string, (this: C, ...args: any[]) => any>>(
   context: C,
-  factory: (transition: (newContext: C) => Machine<C, T>) => T
+  factory: (transition: (newContext: C) => Machine<C, any>) => T
 ): Machine<C, BindTransitions<T>>;
 
 /**