import z from "zod";
import { RpcContract } from "../internal/contract.ts";

export const mainContract = new RpcContract({
    echo: {
        input: z.string(),
        output: z.string(),
    },
});

export const workerContract = new RpcContract({
    add: {
        input: z.object({
            a: z.number(),
            b: z.number(),
        }),
        output: z.number(),
    },

    range: {
        generator: true,
        input: z.object({
            start: z.number(),
            end: z.number(),
        }),
        output: z.number(),
    },
});