pi-crew 0.9.50 → 0.9.52

package/src/extension/team-tool/handle-schedule.ts

@@ -1,332 +0,0 @@
-import * as crypto from "node:crypto";
-import { humanizeSchedule, nextRunTime, parseSchedule } from "../../runtime/scheduler.ts";
-import { type CrewSettings, updateCrewSettings } from "../../runtime/settings-store.ts";
-import type { TeamToolParamsValue } from "../../schema/team-tool-schema.ts";
-import type { PiTeamsToolResult } from "../tool-result.ts";
-import { result, type TeamContext } from "./context.ts";
-
-// Global key for cross-module scheduler access.
-const CREW_SCHEDULER_KEY = Symbol.for("pi-crew:scheduler");
-type SchedulerRef = {
-	add(job: import("../../runtime/scheduler.ts").ScheduledJob): void;
-	list(): import("../../runtime/scheduler.ts").ScheduledJob[];
-	remove(id: string): boolean;
-	update(
-		id: string,
-		patch: Partial<import("../../runtime/scheduler.ts").ScheduledJob>,
-	): import("../../runtime/scheduler.ts").ScheduledJob | undefined;
-};
-
-function getCrewScheduler(): SchedulerRef | undefined {
-	return (globalThis as Record<symbol | string, unknown>)[CREW_SCHEDULER_KEY] as SchedulerRef | undefined;
-}
-
-export function registerCrewScheduler(scheduler: SchedulerRef): void {
-	(globalThis as Record<symbol | string, unknown>)[CREW_SCHEDULER_KEY] = scheduler;
-}
-
-interface ScheduleParams {
-	team?: string;
-	goal?: string;
-	task?: string;
-	cron?: string;
-	interval?: number;
-	once?: number | string;
-}
-
-function buildScheduleSpec(params: ScheduleParams): {
-	spec: import("../../runtime/scheduler.ts").ScheduleSpec;
-	schedule: string;
-	scheduleType: import("../../runtime/scheduler.ts").ScheduleType;
-	intervalMs?: number;
-} {
-	// Priority: cron > interval > once
-	if (params.cron) {
-		const parsed = parseSchedule(params.cron);
-		if ("error" in parsed) throw new Error(parsed.error);
-		return {
-			spec: parsed,
-			schedule: params.cron,
-			scheduleType: "cron" as const,
-		};
-	}
-	if (params.interval !== undefined && (!Number.isFinite(params.interval) || params.interval <= 0)) {
-		throw new Error("interval must be a positive finite number");
-	}
-	if (params.once !== undefined) {
-		const ts = typeof params.once === "number" ? params.once : Date.parse(String(params.once));
-		if (!Number.isFinite(ts)) throw new Error("once must be a valid timestamp");
-	}
-	if (params.interval !== undefined) {
-		const specStr = `${params.interval}ms`;
-		const spec = parseSchedule(specStr);
-		if ("error" in spec) throw new Error(spec.error);
-		return {
-			spec,
-			schedule: specStr,
-			scheduleType: "interval" as const,
-			intervalMs: params.interval,
-		};
-	}
-	if (params.once !== undefined) {
-		const ts = typeof params.once === "number" ? new Date(params.once).toISOString() : params.once;
-		const parsed = parseSchedule(ts);
-		if ("error" in parsed) throw new Error(parsed.error);
-		return { spec: parsed, schedule: ts, scheduleType: "once" as const };
-	}
-	throw new Error("schedule requires one of: cron, interval, or once.");
-}
-
-function getSubAction(params: TeamToolParamsValue): string | undefined {
-	const raw = params.subAction ?? (params.config as { subAction?: unknown } | undefined)?.subAction;
-	return typeof raw === "string" ? raw.toLowerCase() : undefined;
-}
-
-function getJobIdParam(params: TeamToolParamsValue): string {
-	const fromTop = typeof params.jobId === "string" ? params.jobId : "";
-	const fromConfig = ((params.config as { jobId?: unknown } | undefined)?.jobId as string | undefined) ?? "";
-	return (fromTop || fromConfig).trim();
-}
-
-export function handleSchedule(params: TeamToolParamsValue, ctx: TeamContext): PiTeamsToolResult {
-	// Route subactions FIRST so remove/disable/enable don't require a goal.
-	const subAction = getSubAction(params);
-	if (subAction === "remove" || subAction === "delete") {
-		return handleRemoveScheduled(params, ctx);
-	}
-	if (subAction === "disable" || subAction === "enable" || subAction === "update") {
-		return handleUpdateScheduled(params, ctx);
-	}
-
-	const team = params.team ?? "default";
-	const goal = params.goal ?? params.task ?? "";
-	if (!goal) return result("Schedule requires goal or task.", { action: "schedule", status: "error" }, true);
-
-	let specResult: ReturnType<typeof buildScheduleSpec>;
-	try {
-		specResult = buildScheduleSpec({
-			team,
-			goal,
-			cron: params.cron,
-			interval: params.interval,
-			once: params.once as ScheduleParams["once"],
-		});
-	} catch (err) {
-		const msg = err instanceof Error ? err.message : String(err);
-		return result(msg, { action: "schedule", status: "error" }, true);
-	}
-
-	const { spec, schedule, scheduleType, intervalMs } = specResult;
-	const next = nextRunTime(spec);
-	if ("error" in next) return result(next.error, { action: "schedule", status: "error" }, true);
-
-	// Build the ScheduledJob
-	const job: import("../../runtime/scheduler.ts").ScheduledJob = {
-		id: crypto.randomUUID(),
-		name: `${team}: ${goal.slice(0, 60)}`,
-		description: `Scheduled run for team '${team}'`,
-		schedule,
-		scheduleType,
-		intervalMs,
-		subagentType: "team",
-		prompt: JSON.stringify({ action: "run", team, goal }),
-		enabled: true,
-		createdAt: new Date().toISOString(),
-		nextRun: next.toISOString(),
-		runCount: 0,
-	};
-
-	const scheduler = getCrewScheduler();
-	if (!scheduler) {
-		// Persist even if scheduler isn't running yet — register.ts loads them on startup.
-		persistScheduledJob(ctx.cwd, job);
-		return result(
-			[
-				`Scheduled job created (scheduler not yet running — will activate on next session start):`,
-				`  Job ID: ${job.id}`,
-				`  Team: ${team}`,
-				`  Goal: ${goal}`,
-				`  Schedule: ${humanizeSchedule(spec)}`,
-				`  Next run: ${next.toISOString()}`,
-			].join("\n"),
-			{
-				action: "schedule",
-				status: "ok",
-				data: {
-					jobId: job.id,
-					team,
-					goal,
-					schedule: humanizeSchedule(spec),
-					nextRun: next.toISOString(),
-					pending: true,
-				},
-			},
-		);
-	}
-
-	scheduler.add(job);
-	persistScheduledJob(ctx.cwd, job);
-
-	return result(
-		[
-			`Scheduled job registered.`,
-			`  Job ID: ${job.id}`,
-			`  Team: ${team}`,
-			`  Goal: ${goal}`,
-			`  Schedule: ${humanizeSchedule(spec)}`,
-			`  Next run: ${next.toISOString()}`,
-		].join("\n"),
-		{
-			action: "schedule",
-			status: "ok",
-			data: {
-				jobId: job.id,
-				team,
-				goal,
-				schedule: humanizeSchedule(spec),
-				nextRun: next.toISOString(),
-			},
-		},
-	);
-}
-
-function scheduledJobsOf(settings: CrewSettings): import("../../runtime/scheduler.ts").ScheduledJob[] {
-	return Array.isArray(settings.scheduledJobs) ? (settings.scheduledJobs as import("../../runtime/scheduler.ts").ScheduledJob[]) : [];
-}
-
-function persistScheduledJob(cwd: string, job: import("../../runtime/scheduler.ts").ScheduledJob): void {
-	try {
-		updateCrewSettings(cwd, (settings) => ({
-			...settings,
-			scheduledJobs: [...scheduledJobsOf(settings), job],
-		}));
-	} catch {
-		/* best-effort persistence */
-	}
-}
-
-/** Update an existing scheduled job in persistent settings. */
-export function persistScheduledJobUpdate(cwd: string, job: import("../../runtime/scheduler.ts").ScheduledJob): void {
-	try {
-		updateCrewSettings(cwd, (settings) => ({
-			...settings,
-			scheduledJobs: scheduledJobsOf(settings).map((j) => (j.id === job.id ? job : j)),
-		}));
-	} catch {
-		/* best-effort persistence */
-	}
-}
-
-/** Remove a scheduled job from persistent settings. */
-function persistScheduledJobRemove(cwd: string, jobId: string): void {
-	try {
-		updateCrewSettings(cwd, (settings) => ({
-			...settings,
-			scheduledJobs: scheduledJobsOf(settings).filter((j) => j.id !== jobId),
-		}));
-	} catch {
-		/* best-effort persistence */
-	}
-}
-
-export function handleListScheduled(_params: TeamToolParamsValue, ctx: TeamContext): PiTeamsToolResult {
-	const scheduler = getCrewScheduler();
-	if (!scheduler) return result("Scheduler not running.", { action: "scheduled", status: "error" }, true);
-	const jobs = scheduler.list();
-	if (jobs.length === 0)
-		return result("No scheduled jobs.", {
-			action: "scheduled",
-			status: "ok",
-		});
-	const lines: string[] = [`Scheduled jobs (${jobs.length}):`];
-	for (const job of jobs) {
-		lines.push(
-			`  [${job.id}] ${job.name}`,
-			`    Schedule: ${job.schedule} (${job.scheduleType})`,
-			`    Enabled: ${job.enabled}`,
-			`    Next run: ${job.nextRun ?? "(unscheduled)"}`,
-			`    Runs: ${job.runCount}, Last: ${job.lastRun ?? "(never)"} [${job.lastStatus ?? "?"}]`,
-		);
-		if (job.spawnedRunIds && job.spawnedRunIds.length > 0) {
-			lines.push(`    Spawned runs: ${job.spawnedRunIds.join(", ")}`);
-		}
-	}
-	return result(lines.join("\n"), { action: "scheduled", status: "ok" });
-}
-
-/**
- * Remove a scheduled job. Mirrors the `cancel` action for runs.
- *
- * Usage:
- *   team action='schedule' subAction='remove' jobId='<uuid>'
- *   team action='schedule' subAction='delete' jobId='<uuid>'
- */
-export function handleRemoveScheduled(params: TeamToolParamsValue, ctx: TeamContext): PiTeamsToolResult {
-	const jobId = getJobIdParam(params);
-	if (!jobId) {
-		return result(
-			"subAction=remove requires jobId. Usage: team action='schedule' subAction='remove' jobId='<uuid>'",
-			{ action: "schedule", status: "error" },
-			true,
-		);
-	}
-	const scheduler = getCrewScheduler();
-	if (!scheduler) {
-		return result("Scheduler not running.", { action: "schedule", status: "error" }, true);
-	}
-	const removed = scheduler.remove(jobId);
-	persistScheduledJobRemove(ctx.cwd, jobId);
-	if (!removed) {
-		return result(`No scheduled job with id '${jobId}'.`, { action: "schedule", status: "error" }, true);
-	}
-	return result([`Scheduled job removed.`, `  Job ID: ${jobId}`].join("\n"), {
-		action: "schedule",
-		status: "ok",
-		data: { jobId, removed: true },
-	});
-}
-
-/**
- * Update a scheduled job — toggle enabled flag or apply patches.
- *
- * Usage:
- *   team action='schedule' subAction='disable' jobId='<uuid>'
- *   team action='schedule' subAction='enable'  jobId='<uuid>'
- *   team action='schedule' subAction='update'  jobId='<uuid>' cron='0 9 * * *'
- */
-export function handleUpdateScheduled(params: TeamToolParamsValue, ctx: TeamContext): PiTeamsToolResult {
-	const jobId = getJobIdParam(params);
-	if (!jobId) {
-		return result(
-			"subAction=update requires jobId. Usage: team action='schedule' subAction='update|enable|disable' jobId='<uuid>'",
-			{ action: "schedule", status: "error" },
-			true,
-		);
-	}
-	const subAction = typeof params.subAction === "string" ? params.subAction.toLowerCase() : "update";
-	const scheduler = getCrewScheduler();
-	if (!scheduler) {
-		return result("Scheduler not running.", { action: "schedule", status: "error" }, true);
-	}
-
-	const patch: Partial<import("../../runtime/scheduler.ts").ScheduledJob> = {};
-	if (subAction === "disable") patch.enabled = false;
-	if (subAction === "enable") patch.enabled = true;
-	// For generic "update", allow patch of cron / interval / goal.
-	if (subAction === "update") {
-		if (typeof params.cron === "string") patch.schedule = params.cron;
-		if (typeof params.goal === "string" || typeof params.task === "string") {
-			patch.description = typeof params.task === "string" ? params.task : params.goal;
-		}
-	}
-
-	const updated = scheduler.update(jobId, patch);
-	if (!updated) {
-		return result(`No scheduled job with id '${jobId}'.`, { action: "schedule", status: "error" }, true);
-	}
-	persistScheduledJobUpdate(ctx.cwd, updated);
-	return result(
-		[`Scheduled job updated.`, `  Job ID: ${jobId}`, `  Enabled: ${updated.enabled}`, `  Schedule: ${updated.schedule}`].join("\n"),
-		{ action: "schedule", status: "ok", data: { jobId, enabled: updated.enabled, schedule: updated.schedule } },
-	);
-}