init claude-code

2026-04-01 17:32:37 +02:00
commit 73b208c009
1902 changed files with 513237 additions and 0 deletions
@@ -0,0 +1,157 @@
+import { 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>)