index.ts 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. /**
  2. * Profile-named Claude Code one-shot subagent provider. Every accepted run
  3. * invokes the official Agent SDK in the delegating Session's workspace and
  4. * places the SDK-spawned real CLI under the shared subprocess owner.
  5. *
  6. * @module @deepseek-ai/dsh-subagent-claude-code
  7. */
  8. import type { Context } from '@deepseek-ai/cordis'
  9. import z from '@deepseek-ai/schemastery'
  10. import { MAX_TIMER_DELAY_MS } from '@deepseek-ai/dsh-timeout'
  11. import {
  12. assertPositiveFinite,
  13. NO_START_CAPABILITIES,
  14. resolveChildCwd,
  15. type ResolvedSubagentStartRequest,
  16. type SubagentCapabilities,
  17. type SubagentProvider,
  18. } from '@deepseek-ai/dsh-subagent'
  19. import {
  20. CLAUDE_CODE_PERMISSION_MODES,
  21. DEFAULT_CLAUDE_CODE_PERMISSION_MODE,
  22. DEFAULT_DISPOSE_GRACE_MS,
  23. claudeCodeStartupFailure,
  24. startClaudeCodeRun,
  25. type ClaudeCodePermissionMode,
  26. type ClaudeCodeRunSpec,
  27. } from './run.ts'
  28. export const name = 'subagent-claude-code'
  29. export const inject = ['subagents', 'subprocess']
  30. const DEFAULT_PROVIDER_NAME = 'claude-code'
  31. /* jscpd:ignore-start -- sibling product providers intentionally expose
  32. * overlapping deployment-owned fields without adding a shared config owner. */
  33. /** Deployment-owned permission, environment, and process-release settings. */
  34. export interface Config {
  35. /** Provider name on `ctx.subagents` (default `claude-code`). */
  36. providerName?: string
  37. /**
  38. * Explicit environment entries layered over the subprocess seam's
  39. * credential-scrubbed parent environment.
  40. */
  41. env?: Record<string, string>
  42. /**
  43. * Native non-interactive mode fixed for this Provider instance. Defaults to
  44. * `dontAsk`; `acceptEdits` accepts edits, `auto` uses the native classifier,
  45. * `plan` returns a plan without approving execution, and
  46. * `bypassPermissions` explicitly skips permission checks.
  47. */
  48. permissionMode?: ClaudeCodePermissionMode
  49. /** Grace in milliseconds for Claude Code process-tree termination. */
  50. disposeGraceMs?: number
  51. }
  52. export const Config: z<Config> = z.object({
  53. providerName: z.string().min(1).default(DEFAULT_PROVIDER_NAME),
  54. env: z.dict(z.string()).default({}),
  55. permissionMode: z.union([...CLAUDE_CODE_PERMISSION_MODES])
  56. .default(DEFAULT_CLAUDE_CODE_PERMISSION_MODE),
  57. disposeGraceMs: z.number().default(DEFAULT_DISPOSE_GRACE_MS),
  58. })
  59. type ResolvedConfig = Required<Config>
  60. /* jscpd:ignore-end */
  61. /* jscpd:ignore-start -- Cordis registration and shared-seam plumbing mirror
  62. * the Codex sibling; each product's lifecycle remains package-private. */
  63. class ClaudeCodeProvider implements SubagentProvider {
  64. readonly capabilities: SubagentCapabilities = NO_START_CAPABILITIES
  65. readonly inheritsParentContext = false
  66. constructor(
  67. readonly name: string,
  68. private readonly ctx: Context,
  69. private readonly config: ResolvedConfig,
  70. ) {}
  71. async start(request: ResolvedSubagentStartRequest) {
  72. const parentCwd = request.parent.session.header.cwd
  73. if (parentCwd === undefined) {
  74. throw new Error(
  75. 'subagent-claude-code: no working directory for the child — delegate from a parent session that has one',
  76. )
  77. }
  78. let cwd: string
  79. try {
  80. cwd = resolveChildCwd(
  81. 'subagent-claude-code',
  82. undefined,
  83. parentCwd,
  84. )
  85. } catch (error: unknown) {
  86. if (request.signal.aborted) {
  87. throw new Error(
  88. 'subagent-claude-code: request was aborted before SDK startup',
  89. )
  90. }
  91. const failure = claudeCodeStartupFailure(error)
  92. this.ctx.logger.warn(
  93. `subagent-claude-code "${this.name}": child start failed: %o`,
  94. failure,
  95. )
  96. throw failure
  97. }
  98. const spec: ClaudeCodeRunSpec = {
  99. cwd,
  100. permissionMode: this.config.permissionMode,
  101. env: this.config.env,
  102. disposeGraceMs: this.config.disposeGraceMs,
  103. spawn: spawnSpec => this.ctx.subprocess.spawn(spawnSpec),
  104. onError: (error, stopReason) => {
  105. this.ctx.logger.warn(
  106. `subagent-claude-code "${this.name}": child run failed (${stopReason}): %o`,
  107. error,
  108. )
  109. },
  110. }
  111. return startClaudeCodeRun(request, spec)
  112. }
  113. }
  114. /**
  115. * Register one Profile-named Claude Code provider.
  116. * @param ctx - context carrying shared subagent and subprocess services.
  117. * @param config - registry name, permission mode, child environment, and disposal grace.
  118. */
  119. export function apply(ctx: Context, config: Config): void {
  120. const resolved: ResolvedConfig = {
  121. providerName: config.providerName ?? DEFAULT_PROVIDER_NAME,
  122. env: config.env as Record<string, string>,
  123. permissionMode: config.permissionMode ?? DEFAULT_CLAUDE_CODE_PERMISSION_MODE,
  124. disposeGraceMs: config.disposeGraceMs as number,
  125. }
  126. assertPositiveFinite(
  127. 'subagent-claude-code',
  128. 'disposeGraceMs',
  129. resolved.disposeGraceMs,
  130. )
  131. if (resolved.disposeGraceMs > MAX_TIMER_DELAY_MS) {
  132. throw new Error(
  133. `subagent-claude-code: disposeGraceMs must be no greater than ${MAX_TIMER_DELAY_MS}`,
  134. )
  135. }
  136. ctx.subagents.registerProvider(new ClaudeCodeProvider(
  137. resolved.providerName,
  138. ctx,
  139. resolved,
  140. ))
  141. }
  142. /* jscpd:ignore-end */