milkio 1.0.0-alpha.94 → 1.0.0-alpha.95

package/flow/index.ts

@@ -1,38 +1,60 @@
-export type MilkioFlow<T> = {
-    emit: (flow: T) => void;
-    next: () => Promise<T>;
-}
+export type MilkioFlow<T> = AsyncIterator<T | undefined> & { emit: (flow: T) => void };
+
+export function createFlow<T>(): MilkioFlow<T> {
+  const flows: Array<{
+    blank: boolean;
+    promise: Promise<T>;
+    resolve: (value?: T | PromiseLike<T>) => void;
+    reject: (reason?: any) => void;
+  }> = [];
 
-export function createFlow<T extends unknown>(): MilkioFlow<T> {
-    const flows: Array<{
-        blank: boolean;
-        promise: Promise<T>;
-        resolve: (value?: T | PromiseLike<T>) => void;
-        reject: (reason?: any) => void;
-    }> = [];
+  const iterator = {
+    emit: (flow: T) => {
+      if (flows.at(-1)?.blank === true) {
+        const item = flows.at(-1)!;
+        item.blank = false;
+        item.resolve(flow);
+        return;
+      } else {
+        const resolvers = Promise.withResolvers<T>();
+        resolvers.resolve(flow);
+        flows.push({ ...resolvers, blank: false });
+      }
+    },
+    ...({
+      async next(): Promise<IteratorResult<T>> {
+        if (flows.length === 0) {
+          const resolvers = Promise.withResolvers<T>();
+          flows.push({ ...resolvers, blank: true });
+        }
+        const flow = flows.at(0)!;
+        const result = await flow.promise;
+        flows.shift();
+        return { done: false, value: result };
+      },
+      async return(): Promise<IteratorResult<void>> {
+        for (const flow of flows) {
+          flow.blank = false;
+          flow.resolve(undefined);
+        }
+        return { done: true, value: undefined };
+      },
+      async throw(err: any): Promise<IteratorResult<void>> {
+        if (flows.length === 0) {
+          const resolvers = Promise.withResolvers<T>();
+          flows.push({ ...resolvers, blank: true });
+        }
+        for (const flow of flows) {
+          flow.blank = false;
+          flow.reject(err);
+        }
+        return { done: true, value: undefined };
+      },
+    } satisfies AsyncIterator<unknown>),
+    [Symbol.asyncIterator]() {
+      return this;
+    },
+  };
 
-    return {
-        emit: (flow: T) => {
-            if (flows.at(-1)?.blank === true) {
-                const item = flows.at(-1)!;
-                item.blank = false;
-                item.resolve(flow);
-                return;
-            } else {
-                const resolvers = Promise.withResolvers<T>();
-                resolvers.resolve(flow);
-                flows.push({ ...resolvers, blank: false });
-            }
-        },
-        next: async () => {
-            if (flows.length === 0) {
-                const resolvers = Promise.withResolvers<T>();
-                flows.push({ ...resolvers, blank: true });
-            }
-            const flow = flows.at(0)!;
-            const result = await flow.promise;
-            flows.shift();
-            return result;
-        },
-    }
-}
+  return iterator as MilkioFlow<T>;
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "milkio",
   "type": "module",
-  "version": "1.0.0-alpha.94",
+  "version": "1.0.0-alpha.95",
   "module": "index.ts",
   "scripts": {},
   "peerDependencies": {