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,96 @@
+import { feature } from 'bun:bundle'
+import { randomUUID } from 'crypto'
+import { getSdkBetas, getSessionId } from 'src/bootstrap/state.js'
+import { DEFAULT_OUTPUT_STYLE_NAME } from 'src/constants/outputStyles.js'
+import type {
+  ApiKeySource,
+  PermissionMode,
+  SDKMessage,
+} from 'src/entrypoints/agentSdkTypes.js'
+import {
+  AGENT_TOOL_NAME,
+  LEGACY_AGENT_TOOL_NAME,
+} from 'src/tools/AgentTool/constants.js'
+import { getAnthropicApiKeyWithSource } from '../auth.js'
+import { getCwd } from '../cwd.js'
+import { getFastModeState } from '../fastMode.js'
+import { getSettings_DEPRECATED } from '../settings/settings.js'
+
+// TODO(next-minor): remove this translation once SDK consumers have migrated
+// to the 'Agent' tool name. The wire name was renamed Task → Agent in #19647,
+// but emitting the new name in init/result events broke SDK consumers on a
+// patch-level release. Keep emitting 'Task' until the next minor.
+export function sdkCompatToolName(name: string): string {
+  return name === AGENT_TOOL_NAME ? LEGACY_AGENT_TOOL_NAME : name
+}
+
+type CommandLike = { name: string; userInvocable?: boolean }
+
+export type SystemInitInputs = {
+  tools: ReadonlyArray<{ name: string }>
+  mcpClients: ReadonlyArray<{ name: string; type: string }>
+  model: string
+  permissionMode: PermissionMode
+  commands: ReadonlyArray<CommandLike>
+  agents: ReadonlyArray<{ agentType: string }>
+  skills: ReadonlyArray<CommandLike>
+  plugins: ReadonlyArray<{ name: string; path: string; source: string }>
+  fastMode: boolean | undefined
+}
+
+/**
+ * Build the `system/init` SDKMessage — the first message on the SDK stream
+ * carrying session metadata (cwd, tools, model, commands, etc.) that remote
+ * clients use to render pickers and gate UI.
+ *
+ * Called from two paths that must produce identical shapes:
+ *   - QueryEngine (spawn-bridge / print-mode / SDK) — yielded as the first
+ *     stream message per query turn
+ *   - useReplBridge (REPL Remote Control) — sent via writeSdkMessages() on
+ *     bridge connect, since REPL uses query() directly and never hits the
+ *     QueryEngine SDKMessage layer
+ */
+export function buildSystemInitMessage(inputs: SystemInitInputs): SDKMessage {
+  const settings = getSettings_DEPRECATED()
+  const outputStyle = settings?.outputStyle ?? DEFAULT_OUTPUT_STYLE_NAME
+
+  const initMessage: SDKMessage = {
+    type: 'system',
+    subtype: 'init',
+    cwd: getCwd(),
+    session_id: getSessionId(),
+    tools: inputs.tools.map(tool => sdkCompatToolName(tool.name)),
+    mcp_servers: inputs.mcpClients.map(client => ({
+      name: client.name,
+      status: client.type,
+    })),
+    model: inputs.model,
+    permissionMode: inputs.permissionMode,
+    slash_commands: inputs.commands
+      .filter(c => c.userInvocable !== false)
+      .map(c => c.name),
+    apiKeySource: getAnthropicApiKeyWithSource().source as ApiKeySource,
+    betas: getSdkBetas(),
+    claude_code_version: MACRO.VERSION,
+    output_style: outputStyle,
+    agents: inputs.agents.map(agent => agent.agentType),
+    skills: inputs.skills
+      .filter(s => s.userInvocable !== false)
+      .map(skill => skill.name),
+    plugins: inputs.plugins.map(plugin => ({
+      name: plugin.name,
+      path: plugin.path,
+      source: plugin.source,
+    })),
+    uuid: randomUUID(),
+  }
+  // Hidden from public SDK types — ant-only UDS messaging socket path
+  if (feature('UDS_INBOX')) {
+    /* eslint-disable @typescript-eslint/no-require-imports */
+    ;(initMessage as Record<string, unknown>).messaging_socket_path =
+      require('../udsMessaging.js').getUdsMessagingSocketPath()
+    /* eslint-enable @typescript-eslint/no-require-imports */
+  }
+  initMessage.fast_mode_state = getFastModeState(inputs.model, inputs.fastMode)
+  return initMessage
+}