import { mo } from '../../src/index.ts';

export const generate = mo(import.meta, async function* (count: number) {
  for (let i = 1; i <= count; i++) {
    yield i;
  }
});

export const double = mo(import.meta, async function* (input: AsyncIterable<number>) {
  for await (const n of input) {
    yield n * 2;
  }
});

export const square = mo(import.meta, async function* (input: AsyncIterable<number>) {
  for await (const n of input) {
    yield n * n;
  }
});

export const toString = mo(import.meta, async function* (input: AsyncIterable<number>) {
  for await (const n of input) {
    yield `=> ${n}`;
  }
});