@clinebot/core 0.0.35 → 0.0.36

package/src/cron/scheduler.ts

@@ -0,0 +1,637 @@
+import {
+	createSessionId,
+	type HubScheduleCreateInput,
+	type HubScheduleUpdateInput,
+	type ScheduleExecutionRecord,
+	type ScheduleExecutionStatus,
+	type ScheduleRecord,
+} from "@clinebot/shared";
+
+function createHubId(prefix: string): string {
+	return createSessionId(`${prefix}_`);
+}
+
+function parseCronField(
+	token: string,
+	min: number,
+	max: number,
+	names?: readonly string[],
+): number[] {
+	const results = new Set<number>();
+
+	function resolveValue(raw: string): number {
+		const lower = raw.toLowerCase();
+		if (names) {
+			const index = names.indexOf(lower);
+			if (index !== -1) {
+				return index + min;
+			}
+		}
+		const value = Number(raw);
+		if (!Number.isInteger(value) || value < min || value > max) {
+			throw new Error(`Invalid cron value "${raw}" for range [${min}-${max}]`);
+		}
+		return value;
+	}
+
+	for (const part of token.split(",")) {
+		if (part === "*") {
+			for (let value = min; value <= max; value += 1) {
+				results.add(value);
+			}
+			continue;
+		}
+
+		const stepSeparator = part.indexOf("/");
+		if (stepSeparator !== -1) {
+			const rangePart = part.slice(0, stepSeparator);
+			const step = Number(part.slice(stepSeparator + 1));
+			if (!Number.isInteger(step) || step < 1) {
+				throw new Error(`Invalid step "${part.slice(stepSeparator + 1)}"`);
+			}
+			let from = min;
+			let to = max;
+			if (rangePart !== "*") {
+				const dashIndex = rangePart.indexOf("-");
+				if (dashIndex !== -1) {
+					from = resolveValue(rangePart.slice(0, dashIndex));
+					to = resolveValue(rangePart.slice(dashIndex + 1));
+				} else {
+					from = resolveValue(rangePart);
+				}
+			}
+			for (let value = from; value <= to; value += step) {
+				results.add(value);
+			}
+			continue;
+		}
+
+		const dashIndex = part.indexOf("-");
+		if (dashIndex !== -1) {
+			const from = resolveValue(part.slice(0, dashIndex));
+			const to = resolveValue(part.slice(dashIndex + 1));
+			for (let value = from; value <= to; value += 1) {
+				results.add(value);
+			}
+			continue;
+		}
+
+		results.add(resolveValue(part));
+	}
+
+	return [...results].sort((left, right) => left - right);
+}
+
+const MONTH_NAMES = [
+	"jan",
+	"feb",
+	"mar",
+	"apr",
+	"may",
+	"jun",
+	"jul",
+	"aug",
+	"sep",
+	"oct",
+	"nov",
+	"dec",
+] as const;
+
+const DOW_NAMES = ["sun", "mon", "tue", "wed", "thu", "fri", "sat"] as const;
+
+export interface ParsedCron {
+	minutes: number[];
+	hours: number[];
+	daysOfMonth: number[];
+	months: number[];
+	daysOfWeek: number[];
+}
+
+function getRequiredField(
+	fields: readonly string[],
+	index: number,
+	pattern: string,
+): string {
+	const value = fields[index];
+	if (typeof value !== "string") {
+		throw new Error(
+			`Invalid cron pattern "${pattern}": missing field ${index + 1}`,
+		);
+	}
+	return value;
+}
+
+function getFirstCronValue(values: readonly number[], label: string): number {
+	const value = values[0];
+	if (typeof value !== "number") {
+		throw new Error(`Invalid cron pattern: no values parsed for ${label}`);
+	}
+	return value;
+}
+
+export function parseCron(pattern: string): ParsedCron {
+	const fields = pattern.trim().split(/\s+/);
+	if (fields.length !== 5) {
+		throw new Error(
+			`Invalid cron pattern "${pattern}": expected 5 fields, got ${fields.length}`,
+		);
+	}
+	return {
+		minutes: parseCronField(getRequiredField(fields, 0, pattern), 0, 59),
+		hours: parseCronField(getRequiredField(fields, 1, pattern), 0, 23),
+		daysOfMonth: parseCronField(getRequiredField(fields, 2, pattern), 1, 31),
+		months: parseCronField(
+			getRequiredField(fields, 3, pattern),
+			1,
+			12,
+			MONTH_NAMES,
+		),
+		daysOfWeek: parseCronField(
+			getRequiredField(fields, 4, pattern),
+			0,
+			6,
+			DOW_NAMES,
+		),
+	};
+}
+
+export function validateCronPattern(pattern: string): void {
+	parseCron(pattern);
+}
+
+export function getNextCronTime(pattern: string, after: number): number {
+	const cron = parseCron(pattern);
+	let next = new Date(after);
+	next.setSeconds(0, 0);
+	next = new Date(next.getTime() + 60_000);
+
+	const limit = new Date(after);
+	limit.setFullYear(limit.getFullYear() + 4);
+
+	while (next <= limit) {
+		const month = next.getMonth() + 1;
+		const dayOfMonth = next.getDate();
+		const dayOfWeek = next.getDay();
+		const hour = next.getHours();
+		const minute = next.getMinutes();
+
+		if (!cron.months.includes(month)) {
+			const targetMonth =
+				cron.months.find((value) => value > month) ??
+				getFirstCronValue(cron.months, "months");
+			const yearDelta = targetMonth <= month ? 1 : 0;
+			next = new Date(
+				next.getFullYear() + yearDelta,
+				targetMonth - 1,
+				1,
+				0,
+				0,
+				0,
+				0,
+			);
+			continue;
+		}
+
+		if (
+			!cron.daysOfMonth.includes(dayOfMonth) ||
+			!cron.daysOfWeek.includes(dayOfWeek)
+		) {
+			next = new Date(
+				next.getFullYear(),
+				next.getMonth(),
+				dayOfMonth + 1,
+				0,
+				0,
+				0,
+				0,
+			);
+			continue;
+		}
+
+		if (!cron.hours.includes(hour)) {
+			const targetHour =
+				cron.hours.find((value) => value > hour) ??
+				getFirstCronValue(cron.hours, "hours");
+			const dayDelta = targetHour <= hour ? 1 : 0;
+			next = new Date(
+				next.getFullYear(),
+				next.getMonth(),
+				next.getDate() + dayDelta,
+				targetHour,
+				0,
+				0,
+				0,
+			);
+			continue;
+		}
+
+		if (!cron.minutes.includes(minute)) {
+			const targetMinute =
+				cron.minutes.find((value) => value > minute) ??
+				getFirstCronValue(cron.minutes, "minutes");
+			const hourDelta = targetMinute <= minute ? 1 : 0;
+			next = new Date(
+				next.getFullYear(),
+				next.getMonth(),
+				next.getDate(),
+				next.getHours() + hourDelta,
+				targetMinute,
+				0,
+				0,
+			);
+			continue;
+		}
+
+		return next.getTime();
+	}
+
+	throw new Error(
+		`No cron occurrence found within 4 years for pattern "${pattern}"`,
+	);
+}
+
+export class ScheduleStore {
+	private schedules = new Map<string, ScheduleRecord>();
+	private executions = new Map<string, ScheduleExecutionRecord>();
+
+	load(
+		schedules: ScheduleRecord[] | undefined,
+		executions: ScheduleExecutionRecord[] | undefined,
+	): void {
+		this.schedules.clear();
+		this.executions.clear();
+		for (const schedule of schedules ?? []) {
+			this.schedules.set(schedule.scheduleId, { ...schedule });
+		}
+		for (const execution of executions ?? []) {
+			this.executions.set(execution.executionId, { ...execution });
+		}
+	}
+
+	snapshotSchedules(): ScheduleRecord[] {
+		return [...this.schedules.values()].map((record) => ({ ...record }));
+	}
+
+	snapshotExecutions(): ScheduleExecutionRecord[] {
+		return [...this.executions.values()].map((record) => ({ ...record }));
+	}
+
+	create(input: HubScheduleCreateInput, now: number): ScheduleRecord {
+		validateCronPattern(input.cronPattern);
+		const nextRunAt = getNextCronTime(input.cronPattern, now);
+		const record: ScheduleRecord = {
+			scheduleId: createHubId("sched"),
+			name: input.name,
+			cronPattern: input.cronPattern,
+			prompt: input.prompt,
+			workspaceRoot: input.workspaceRoot,
+			enabled: input.enabled ?? true,
+			createdAt: now,
+			updatedAt: now,
+			nextRunAt,
+			runtimeOptions: input.runtimeOptions
+				? { ...input.runtimeOptions }
+				: undefined,
+			metadata: input.metadata ? { ...input.metadata } : undefined,
+		};
+		this.schedules.set(record.scheduleId, record);
+		return { ...record };
+	}
+
+	get(scheduleId: string): ScheduleRecord | undefined {
+		const record = this.schedules.get(scheduleId);
+		return record ? { ...record } : undefined;
+	}
+
+	list(): ScheduleRecord[] {
+		return [...this.schedules.values()].map((record) => ({ ...record }));
+	}
+
+	update(input: HubScheduleUpdateInput, now: number): ScheduleRecord {
+		const existing = this.schedules.get(input.scheduleId);
+		if (!existing) {
+			throw new Error(`Schedule "${input.scheduleId}" not found`);
+		}
+		const cronPattern = input.cronPattern ?? existing.cronPattern;
+		if (input.cronPattern) {
+			validateCronPattern(input.cronPattern);
+		}
+		const enabled = input.enabled ?? existing.enabled;
+		const nextRunAt =
+			(input.cronPattern || input.enabled !== undefined) && enabled
+				? getNextCronTime(cronPattern, now)
+				: enabled
+					? existing.nextRunAt
+					: undefined;
+		const updated: ScheduleRecord = {
+			...existing,
+			name: input.name ?? existing.name,
+			cronPattern,
+			prompt: input.prompt ?? existing.prompt,
+			enabled,
+			updatedAt: now,
+			nextRunAt,
+			runtimeOptions:
+				input.runtimeOptions !== undefined
+					? { ...input.runtimeOptions }
+					: existing.runtimeOptions,
+			metadata:
+				input.metadata !== undefined
+					? { ...input.metadata }
+					: existing.metadata,
+		};
+		this.schedules.set(updated.scheduleId, updated);
+		return { ...updated };
+	}
+
+	setEnabled(
+		scheduleId: string,
+		enabled: boolean,
+		now: number,
+	): ScheduleRecord {
+		const existing = this.schedules.get(scheduleId);
+		if (!existing) {
+			throw new Error(`Schedule "${scheduleId}" not found`);
+		}
+		const updated: ScheduleRecord = {
+			...existing,
+			enabled,
+			updatedAt: now,
+			nextRunAt: enabled
+				? getNextCronTime(existing.cronPattern, now)
+				: undefined,
+		};
+		this.schedules.set(scheduleId, updated);
+		return { ...updated };
+	}
+
+	delete(scheduleId: string): void {
+		if (!this.schedules.has(scheduleId)) {
+			throw new Error(`Schedule "${scheduleId}" not found`);
+		}
+		this.schedules.delete(scheduleId);
+	}
+
+	recordTriggered(scheduleId: string, now: number): ScheduleRecord {
+		const existing = this.schedules.get(scheduleId);
+		if (!existing) {
+			throw new Error(`Schedule "${scheduleId}" not found`);
+		}
+		const updated: ScheduleRecord = {
+			...existing,
+			lastRunAt: now,
+			updatedAt: now,
+			nextRunAt: getNextCronTime(existing.cronPattern, now),
+		};
+		this.schedules.set(scheduleId, updated);
+		return { ...updated };
+	}
+
+	getDueSchedules(now: number): ScheduleRecord[] {
+		return [...this.schedules.values()]
+			.filter(
+				(record) =>
+					record.enabled &&
+					record.nextRunAt !== undefined &&
+					record.nextRunAt <= now,
+			)
+			.map((record) => ({ ...record }));
+	}
+
+	startExecution(
+		scheduleId: string,
+		sessionId: string,
+		now: number,
+	): ScheduleExecutionRecord {
+		const record: ScheduleExecutionRecord = {
+			executionId: createHubId("sexec"),
+			scheduleId,
+			sessionId,
+			triggeredAt: now,
+			startedAt: now,
+			status: "running",
+		};
+		this.executions.set(record.executionId, record);
+		return { ...record };
+	}
+
+	completeExecution(
+		executionId: string,
+		status: Exclude<ScheduleExecutionStatus, "running">,
+		now: number,
+		errorMessage?: string,
+	): ScheduleExecutionRecord {
+		const existing = this.executions.get(executionId);
+		if (!existing) {
+			throw new Error(`Execution "${executionId}" not found`);
+		}
+		const updated: ScheduleExecutionRecord = {
+			...existing,
+			status,
+			endedAt: now,
+			...(errorMessage ? { errorMessage } : {}),
+		};
+		this.executions.set(executionId, updated);
+		return { ...updated };
+	}
+
+	listExecutions(scheduleId?: string): ScheduleExecutionRecord[] {
+		const records = [...this.executions.values()];
+		return (
+			scheduleId
+				? records.filter((record) => record.scheduleId === scheduleId)
+				: records
+		)
+			.map((record) => ({ ...record }))
+			.sort((left, right) => right.triggeredAt - left.triggeredAt);
+	}
+}
+
+export interface SchedulerTriggerContext {
+	scheduleId: string;
+	sessionId: string;
+	executionId: string;
+}
+
+export interface SchedulerCallbacks {
+	onTrigger(record: ScheduleRecord): Promise<{ sessionId: string }>;
+	onExecutionCompleted(
+		context: SchedulerTriggerContext,
+		execution: ScheduleExecutionRecord,
+	): void;
+	onExecutionFailed(
+		context: SchedulerTriggerContext,
+		execution: ScheduleExecutionRecord,
+		error: unknown,
+	): void;
+	onScheduleUpdated(record: ScheduleRecord): void;
+	onPersist(): Promise<void>;
+}
+
+export interface HubSchedulerOptions {
+	store: ScheduleStore;
+	callbacks: SchedulerCallbacks;
+	now?: () => number;
+	pollIntervalMs?: number;
+}
+
+const DEFAULT_POLL_INTERVAL_MS = 15_000;
+
+export class HubScheduler {
+	private readonly store: ScheduleStore;
+	private readonly callbacks: SchedulerCallbacks;
+	private readonly now: () => number;
+	private readonly pollIntervalMs: number;
+	private timer: ReturnType<typeof setInterval> | undefined;
+	private readonly runningExecutions = new Map<
+		string,
+		SchedulerTriggerContext
+	>();
+
+	constructor(options: HubSchedulerOptions) {
+		this.store = options.store;
+		this.callbacks = options.callbacks;
+		this.now = options.now ?? (() => Date.now());
+		this.pollIntervalMs = Math.max(
+			1_000,
+			Math.floor(options.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS),
+		);
+	}
+
+	start(): void {
+		if (this.timer !== undefined) {
+			return;
+		}
+		this.timer = setInterval(() => {
+			void this.poll();
+		}, this.pollIntervalMs);
+		void this.poll();
+	}
+
+	stop(): void {
+		if (this.timer === undefined) {
+			return;
+		}
+		clearInterval(this.timer);
+		this.timer = undefined;
+	}
+
+	async triggerNow(scheduleId: string): Promise<ScheduleExecutionRecord> {
+		const record = this.store.get(scheduleId);
+		if (!record) {
+			throw new Error(`Schedule "${scheduleId}" not found`);
+		}
+		return await this.executeSchedule(record);
+	}
+
+	async notifySessionCompleted(
+		sessionId: string,
+		failed: boolean,
+		errorMessage?: string,
+	): Promise<void> {
+		const entry = [...this.runningExecutions.entries()].find(
+			([, context]) => context.sessionId === sessionId,
+		);
+		if (!entry) {
+			return;
+		}
+		const [executionId, context] = entry;
+		this.runningExecutions.delete(executionId);
+		const status: Exclude<ScheduleExecutionStatus, "running"> = failed
+			? "failed"
+			: "completed";
+		const execution = this.store.completeExecution(
+			executionId,
+			status,
+			this.now(),
+			errorMessage,
+		);
+		if (failed) {
+			this.callbacks.onExecutionFailed(
+				context,
+				execution,
+				new Error(errorMessage),
+			);
+		} else {
+			this.callbacks.onExecutionCompleted(context, execution);
+		}
+		await this.callbacks.onPersist();
+	}
+
+	get scheduleStore(): ScheduleStore {
+		return this.store;
+	}
+
+	private async poll(): Promise<void> {
+		const due = this.store.getDueSchedules(this.now());
+		for (const record of due) {
+			const alreadyRunning = [...this.runningExecutions.values()].some(
+				(context) => context.scheduleId === record.scheduleId,
+			);
+			if (alreadyRunning) {
+				continue;
+			}
+			void this.executeSchedule(record);
+		}
+	}
+
+	private async executeSchedule(
+		record: ScheduleRecord,
+	): Promise<ScheduleExecutionRecord> {
+		const startedAt = this.now();
+
+		let sessionId: string;
+		try {
+			const result = await this.callbacks.onTrigger(record);
+			sessionId = result.sessionId;
+		} catch (error) {
+			const execution = this.store.startExecution(
+				record.scheduleId,
+				"",
+				startedAt,
+			);
+			const failed = this.store.completeExecution(
+				execution.executionId,
+				"failed",
+				this.now(),
+				error instanceof Error ? error.message : String(error),
+			);
+			const updatedSchedule = this.store.recordTriggered(
+				record.scheduleId,
+				startedAt,
+			);
+			this.callbacks.onScheduleUpdated(updatedSchedule);
+			this.callbacks.onExecutionFailed(
+				{
+					scheduleId: record.scheduleId,
+					sessionId: "",
+					executionId: execution.executionId,
+				},
+				failed,
+				error,
+			);
+			await this.callbacks.onPersist();
+			return failed;
+		}
+
+		const execution = this.store.startExecution(
+			record.scheduleId,
+			sessionId,
+			startedAt,
+		);
+		this.runningExecutions.set(execution.executionId, {
+			scheduleId: record.scheduleId,
+			sessionId,
+			executionId: execution.executionId,
+		});
+
+		const updatedSchedule = this.store.recordTriggered(
+			record.scheduleId,
+			startedAt,
+		);
+		this.callbacks.onScheduleUpdated(updatedSchedule);
+		await this.callbacks.onPersist();
+
+		return execution;
+	}
+}