Something went wrong. Try again.
source dump of claude code forked from oppi.li/claude-code
Something went wrong. Try again.
5.7 kB · 157 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158import { z } from 'zod/v4'import { setScheduledTasksEnabled } from '../../bootstrap/state.js'import type { ValidationResult } from '../../Tool.js'import { buildTool, type ToolDef } from '../../Tool.js'import { cronToHuman, parseCronExpression } from '../../utils/cron.js'import { addCronTask, getCronFilePath, listAllCronTasks, nextCronRunMs,} from '../../utils/cronTasks.js'import { lazySchema } from '../../utils/lazySchema.js'import { semanticBoolean } from '../../utils/semanticBoolean.js'import { getTeammateContext } from '../../utils/teammateContext.js'import { buildCronCreateDescription, buildCronCreatePrompt, CRON_CREATE_TOOL_NAME, DEFAULT_MAX_AGE_DAYS, isDurableCronEnabled, isKairosCronEnabled,} from './prompt.js'import { renderCreateResultMessage, renderCreateToolUseMessage } from './UI.js'
const MAX_JOBS = 50
const inputSchema = lazySchema(() => z.strictObject({ cron: z .string() .describe( 'Standard 5-field cron expression in local time: "M H DoM Mon DoW" (e.g. "*/5 * * * *" = every 5 minutes, "30 14 28 2 *" = Feb 28 at 2:30pm local once).', ), prompt: z.string().describe('The prompt to enqueue at each fire time.'), recurring: semanticBoolean(z.boolean().optional()).describe( `true (default) = fire on every cron match until deleted or auto-expired after ${DEFAULT_MAX_AGE_DAYS} days. false = fire once at the next match, then auto-delete. Use false for "remind me at X" one-shot requests with pinned minute/hour/dom/month.`, ), durable: semanticBoolean(z.boolean().optional()).describe( 'true = persist to .claude/scheduled_tasks.json and survive restarts. false (default) = in-memory only, dies when this Claude session ends. Use true only when the user asks the task to survive across sessions.', ), }),)type InputSchema = ReturnType<typeof inputSchema>
const outputSchema = lazySchema(() => z.object({ id: z.string(), humanSchedule: z.string(), recurring: z.boolean(), durable: z.boolean().optional(), }),)type OutputSchema = ReturnType<typeof outputSchema>export type CreateOutput = z.infer<OutputSchema>
export const CronCreateTool = buildTool({ name: CRON_CREATE_TOOL_NAME, searchHint: 'schedule a recurring or one-shot prompt', maxResultSizeChars: 100_000, shouldDefer: true, get inputSchema(): InputSchema { return inputSchema() }, get outputSchema(): OutputSchema { return outputSchema() }, isEnabled() { return isKairosCronEnabled() }, toAutoClassifierInput(input) { return `${input.cron}: ${input.prompt}` }, async description() { return buildCronCreateDescription(isDurableCronEnabled()) }, async prompt() { return buildCronCreatePrompt(isDurableCronEnabled()) }, getPath() { return getCronFilePath() }, async validateInput(input): Promise<ValidationResult> { if (!parseCronExpression(input.cron)) { return { result: false, message: `Invalid cron expression '${input.cron}'. Expected 5 fields: M H DoM Mon DoW.`, errorCode: 1, } } if (nextCronRunMs(input.cron, Date.now()) === null) { return { result: false, message: `Cron expression '${input.cron}' does not match any calendar date in the next year.`, errorCode: 2, } } const tasks = await listAllCronTasks() if (tasks.length >= MAX_JOBS) { return { result: false, message: `Too many scheduled jobs (max ${MAX_JOBS}). Cancel one first.`, errorCode: 3, } } // Teammates don't persist across sessions, so a durable teammate cron // would orphan on restart (agentId would point to a nonexistent teammate). if (input.durable && getTeammateContext()) { return { result: false, message: 'durable crons are not supported for teammates (teammates do not persist across sessions)', errorCode: 4, } } return { result: true } }, async call({ cron, prompt, recurring = true, durable = false }) { // Kill switch forces session-only; schema stays stable so the model sees // no validation errors when the gate flips mid-session. const effectiveDurable = durable && isDurableCronEnabled() const id = await addCronTask( cron, prompt, recurring, effectiveDurable, getTeammateContext()?.agentId, ) // Enable the scheduler so the task fires in this session. The // useScheduledTasks hook polls this flag and will start watching // on the next tick. For durable: false tasks the file never changes // — check() reads the session store directly — but the enable flag // is still what starts the tick loop. setScheduledTasksEnabled(true) return { data: { id, humanSchedule: cronToHuman(cron), recurring, durable: effectiveDurable, }, } }, mapToolResultToToolResultBlockParam(output, toolUseID) { const where = output.durable ? 'Persisted to .claude/scheduled_tasks.json' : 'Session-only (not written to disk, dies when Claude exits)' return { tool_use_id: toolUseID, type: 'tool_result', content: output.recurring ? `Scheduled recurring job ${output.id} (${output.humanSchedule}). ${where}. Auto-expires after ${DEFAULT_MAX_AGE_DAYS} days. Use CronDelete to cancel sooner.` : `Scheduled one-shot task ${output.id} (${output.humanSchedule}). ${where}. It will fire once then auto-delete.`, } }, renderToolUseMessage: renderCreateToolUseMessage, renderToolResultMessage: renderCreateResultMessage,} satisfies ToolDef<InputSchema, CreateOutput>)