Something went wrong. Try again.
source dump of claude code forked from oppi.li/claude-code
Something went wrong. Try again.
4.6 kB · 98 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899import { feature } from 'bun:bundle'import { ASK_USER_QUESTION_TOOL_NAME } from '../../tools/AskUserQuestionTool/prompt.js'import { ENTER_PLAN_MODE_TOOL_NAME } from '../../tools/EnterPlanModeTool/constants.js'import { EXIT_PLAN_MODE_TOOL_NAME } from '../../tools/ExitPlanModeTool/constants.js'import { FILE_READ_TOOL_NAME } from '../../tools/FileReadTool/prompt.js'import { GLOB_TOOL_NAME } from '../../tools/GlobTool/prompt.js'import { GREP_TOOL_NAME } from '../../tools/GrepTool/prompt.js'import { LIST_MCP_RESOURCES_TOOL_NAME } from '../../tools/ListMcpResourcesTool/prompt.js'import { LSP_TOOL_NAME } from '../../tools/LSPTool/prompt.js'import { SEND_MESSAGE_TOOL_NAME } from '../../tools/SendMessageTool/constants.js'import { SLEEP_TOOL_NAME } from '../../tools/SleepTool/prompt.js'import { TASK_CREATE_TOOL_NAME } from '../../tools/TaskCreateTool/constants.js'import { TASK_GET_TOOL_NAME } from '../../tools/TaskGetTool/constants.js'import { TASK_LIST_TOOL_NAME } from '../../tools/TaskListTool/constants.js'import { TASK_OUTPUT_TOOL_NAME } from '../../tools/TaskOutputTool/constants.js'import { TASK_STOP_TOOL_NAME } from '../../tools/TaskStopTool/prompt.js'import { TASK_UPDATE_TOOL_NAME } from '../../tools/TaskUpdateTool/constants.js'import { TEAM_CREATE_TOOL_NAME } from '../../tools/TeamCreateTool/constants.js'import { TEAM_DELETE_TOOL_NAME } from '../../tools/TeamDeleteTool/constants.js'import { TODO_WRITE_TOOL_NAME } from '../../tools/TodoWriteTool/constants.js'import { TOOL_SEARCH_TOOL_NAME } from '../../tools/ToolSearchTool/prompt.js'import { YOLO_CLASSIFIER_TOOL_NAME } from './yoloClassifier.js'
// Ant-only tool names: conditional require so Bun can DCE these in external builds.// Gates mirror tools.ts. Keeps the tool name strings out of cli.js./* eslint-disable @typescript-eslint/no-require-imports */const TERMINAL_CAPTURE_TOOL_NAME = feature('TERMINAL_PANEL') ? ( require('../../tools/TerminalCaptureTool/prompt.js') as typeof import('../../tools/TerminalCaptureTool/prompt.js') ).TERMINAL_CAPTURE_TOOL_NAME : nullconst OVERFLOW_TEST_TOOL_NAME = feature('OVERFLOW_TEST_TOOL') ? ( require('../../tools/OverflowTestTool/OverflowTestTool.js') as typeof import('../../tools/OverflowTestTool/OverflowTestTool.js') ).OVERFLOW_TEST_TOOL_NAME : nullconst VERIFY_PLAN_EXECUTION_TOOL_NAME = process.env.USER_TYPE === 'ant' ? ( require('../../tools/VerifyPlanExecutionTool/constants.js') as typeof import('../../tools/VerifyPlanExecutionTool/constants.js') ).VERIFY_PLAN_EXECUTION_TOOL_NAME : nullconst WORKFLOW_TOOL_NAME = feature('WORKFLOW_SCRIPTS') ? ( require('../../tools/WorkflowTool/constants.js') as typeof import('../../tools/WorkflowTool/constants.js') ).WORKFLOW_TOOL_NAME : null/* eslint-enable @typescript-eslint/no-require-imports */
/** * Tools that are safe and don't need any classifier checking. * Used by the auto mode classifier to skip unnecessary API calls. * Does NOT include write/edit tools — those are handled by the * acceptEdits fast path (allowed in CWD, classified outside CWD). */const SAFE_YOLO_ALLOWLISTED_TOOLS = new Set([ // Read-only file operations FILE_READ_TOOL_NAME, // Search / read-only GREP_TOOL_NAME, GLOB_TOOL_NAME, LSP_TOOL_NAME, TOOL_SEARCH_TOOL_NAME, LIST_MCP_RESOURCES_TOOL_NAME, 'ReadMcpResourceTool', // no exported constant // Task management (metadata only) TODO_WRITE_TOOL_NAME, TASK_CREATE_TOOL_NAME, TASK_GET_TOOL_NAME, TASK_UPDATE_TOOL_NAME, TASK_LIST_TOOL_NAME, TASK_STOP_TOOL_NAME, TASK_OUTPUT_TOOL_NAME, // Plan mode / UI ASK_USER_QUESTION_TOOL_NAME, ENTER_PLAN_MODE_TOOL_NAME, EXIT_PLAN_MODE_TOOL_NAME, // Swarm coordination (internal mailbox/team state only — teammates have // their own permission checks, so no actual security bypass). TEAM_CREATE_TOOL_NAME, // Agent cleanup TEAM_DELETE_TOOL_NAME, SEND_MESSAGE_TOOL_NAME, // Workflow orchestration — subagents go through canUseTool individually ...(WORKFLOW_TOOL_NAME ? [WORKFLOW_TOOL_NAME] : []), // Misc safe SLEEP_TOOL_NAME, // Ant-only safe tools (gates mirror tools.ts) ...(TERMINAL_CAPTURE_TOOL_NAME ? [TERMINAL_CAPTURE_TOOL_NAME] : []), ...(OVERFLOW_TEST_TOOL_NAME ? [OVERFLOW_TEST_TOOL_NAME] : []), ...(VERIFY_PLAN_EXECUTION_TOOL_NAME ? [VERIFY_PLAN_EXECUTION_TOOL_NAME] : []), // Internal classifier tool YOLO_CLASSIFIER_TOOL_NAME,])
export function isAutoModeAllowlistedTool(toolName: string): boolean { return SAFE_YOLO_ALLOWLISTED_TOOLS.has(toolName)}