import { z } from "zod";
import { resultSchema } from "@chopin/experiment";

export function connectorSchemas(
	runScoped: boolean,
): Record<string, z.ZodType<Record<string, unknown>>> {
	let claim = { id: z.string().uuid(), generation: z.number().int().positive() };
	if (runScoped) {
		return {
			read_experiment: z.object({}).strict(),
			submit_experiment_result: z.object({ result: resultSchema }).strict(),
		};
	}
	return {
		disconnect_workspace: z.object({}).strict(),
		wait_for_experiment: z.object({}).strict(),
		claim_experiment: z.object({ id: z.string().uuid() }).strict(),
		renew_experiment: z.object({ ...claim, progress: z.string().max(2000).optional() }).strict(),
		read_experiment: z.object(claim).strict(),
		submit_experiment_result: z.object({ ...claim, result: resultSchema }).strict(),
		complete_experiment: z.object(claim).strict(),
		fail_experiment: z.object({ ...claim, error: z.string().max(2000) }).strict(),
	};
}