forked from
oppi.li/claude-code
source dump of claude code
1import { z } from 'zod/v4'
2import { buildTool, type ToolDef } from '../../Tool.js'
3import { cronToHuman } from '../../utils/cron.js'
4import { listAllCronTasks } from '../../utils/cronTasks.js'
5import { truncate } from '../../utils/format.js'
6import { lazySchema } from '../../utils/lazySchema.js'
7import { getTeammateContext } from '../../utils/teammateContext.js'
8import {
9 buildCronListPrompt,
10 CRON_LIST_DESCRIPTION,
11 CRON_LIST_TOOL_NAME,
12 isDurableCronEnabled,
13 isKairosCronEnabled,
14} from './prompt.js'
15import { renderListResultMessage, renderListToolUseMessage } from './UI.js'
16
17const inputSchema = lazySchema(() => z.strictObject({}))
18type InputSchema = ReturnType<typeof inputSchema>
19
20const outputSchema = lazySchema(() =>
21 z.object({
22 jobs: z.array(
23 z.object({
24 id: z.string(),
25 cron: z.string(),
26 humanSchedule: z.string(),
27 prompt: z.string(),
28 recurring: z.boolean().optional(),
29 durable: z.boolean().optional(),
30 }),
31 ),
32 }),
33)
34type OutputSchema = ReturnType<typeof outputSchema>
35export type ListOutput = z.infer<OutputSchema>
36
37export const CronListTool = buildTool({
38 name: CRON_LIST_TOOL_NAME,
39 searchHint: 'list active cron jobs',
40 maxResultSizeChars: 100_000,
41 shouldDefer: true,
42 get inputSchema(): InputSchema {
43 return inputSchema()
44 },
45 get outputSchema(): OutputSchema {
46 return outputSchema()
47 },
48 isEnabled() {
49 return isKairosCronEnabled()
50 },
51 isConcurrencySafe() {
52 return true
53 },
54 isReadOnly() {
55 return true
56 },
57 async description() {
58 return CRON_LIST_DESCRIPTION
59 },
60 async prompt() {
61 return buildCronListPrompt(isDurableCronEnabled())
62 },
63 async call() {
64 const allTasks = await listAllCronTasks()
65 // Teammates only see their own crons; team lead (no ctx) sees all.
66 const ctx = getTeammateContext()
67 const tasks = ctx
68 ? allTasks.filter(t => t.agentId === ctx.agentId)
69 : allTasks
70 const jobs = tasks.map(t => ({
71 id: t.id,
72 cron: t.cron,
73 humanSchedule: cronToHuman(t.cron),
74 prompt: t.prompt,
75 ...(t.recurring ? { recurring: true } : {}),
76 ...(t.durable === false ? { durable: false } : {}),
77 }))
78 return { data: { jobs } }
79 },
80 mapToolResultToToolResultBlockParam(output, toolUseID) {
81 return {
82 tool_use_id: toolUseID,
83 type: 'tool_result',
84 content:
85 output.jobs.length > 0
86 ? output.jobs
87 .map(
88 j =>
89 `${j.id} — ${j.humanSchedule}${j.recurring ? ' (recurring)' : ' (one-shot)'}${j.durable === false ? ' [session-only]' : ''}: ${truncate(j.prompt, 80, true)}`,
90 )
91 .join('\n')
92 : 'No scheduled jobs.',
93 }
94 },
95 renderToolUseMessage: renderListToolUseMessage,
96 renderToolResultMessage: renderListResultMessage,
97} satisfies ToolDef<InputSchema, ListOutput>)