orca/src/shared/agent-startup-plan-inputs.ts

import type { GlobalSettings } from './global-settings-types'
import type { SessionOptionValue } from './native-chat-session-options'
import type { TuiAgent } from './tui-agent'
import { resolveTuiAgentLaunchArgs, resolveTuiAgentLaunchEnv } from './tui-agent-launch-defaults'
import { resolveStartupShell, type AgentStartupShell } from './tui-agent-startup-shell'
import { hasExtraAgentArgs, mergeExtraAgentArgs } from './automation-extra-agent-args'
import { resolveLocalWindowsAgentStartupShell } from './windows-terminal-shell'

/** The settings slice a launch reads; hosts pass their whole `GlobalSettings` row. */
export type AgentStartupSettings = Partial<
  Pick<
    GlobalSettings,
    | 'agentCmdOverrides'
    | 'agentDefaultArgs'
    | 'agentDefaultEnv'
    | 'agentPermissionDefault'
    | 'agentPermissionOverrides'
    | 'terminalWindowsShell'
  >
>

/** Exactly the `buildAgentStartupPlan` / `buildAgentDraftLaunchPlan` inputs that settings decide. */
export type AgentStartupPlanInputs = {
  agent: TuiAgent
  cmdOverrides: Partial<Record<TuiAgent, string>>
  agentArgs: string | null
  agentEnv: Record<string, string>
  platform: NodeJS.Platform
  shell: AgentStartupShell | undefined
  isRemote: boolean
  sessionOptions?: Record<string, SessionOptionValue>
  sessionOptionsOverrideAgentArgs: boolean
  suppressStartupUpdatePrompt?: boolean
}

/**
 * Assembles the settings-derived half of a startup plan's inputs.
 *
 * `buildAgentStartupPlan` was already one shared implementation, but every host re-derived its
 * argument object by hand from the same four settings, and the copies had drifted — one dropped
 * `sessionOptionsOverrideAgentArgs` and so let configured `agentDefaultArgs` silently defeat an
 * explicit model pick. What varies between launches is the host (`platform`, `isRemote`), the
 * shell this PTY will actually be, and the per-launch overrides; those stay parameters. What does
 * not vary is how settings become inputs, which is this function.
 */
export function resolveAgentStartupPlanInputs(args: {
  agent: TuiAgent
  settings: AgentStartupSettings
  platform: NodeJS.Platform
  isRemote: boolean
  /** Replaces the configured default args for this launch; `null` is "no arguments". */
  agentArgs?: string | null
  /** An automation's saved extras, merged over the resolved args; invalid extras throw. */
  extraAgentArgs?: string
  /** The shell family this PTY will be; `undefined` reads it from the Settings shell, which is
   *  right only where the Settings shell is the one the PTY gets. */
  shell: AgentStartupShell | undefined
  sessionOptions?: Record<string, SessionOptionValue> | undefined
  /** Orca types into this launch after start (worker, automation, handoff), not a user. */
  suppressStartupUpdatePrompt?: boolean
}): AgentStartupPlanInputs {
  const { agent, settings, platform, isRemote, sessionOptions } = args
  // A per-launch override wins over the Settings default; `null` is "no arguments", so this
  // tests for absence rather than falsiness.
  let agentArgs =
    args.agentArgs !== undefined ? args.agentArgs : resolveTuiAgentLaunchArgs(agent, settings)
  const shell =
    args.shell ??
    resolveLocalWindowsAgentStartupShell({
      platform,
      isRemote,
      terminalWindowsShell: settings.terminalWindowsShell
    })
  if (hasExtraAgentArgs(args.extraAgentArgs)) {
    const merged = mergeExtraAgentArgs({
      agent,
      defaultArgs: agentArgs,
      extraAgentArgs: args.extraAgentArgs,
      shell: resolveStartupShell(platform, shell)
    })
    if (!merged.ok) {
      throw new Error(merged.error)
    }
    agentArgs = merged.agentArgs
  }
  return {
    agent,
    cmdOverrides: settings.agentCmdOverrides ?? {},
    agentArgs,
    agentEnv: resolveTuiAgentLaunchEnv(agent, settings, platform),
    platform,
    shell,
    isRemote,
    ...(sessionOptions ? { sessionOptions } : {}),
    // Why: session options are an explicit per-launch pick, so they outrank configured args —
    // without this the two spellings of the same flag both reach argv and the last one wins.
    sessionOptionsOverrideAgentArgs: Boolean(sessionOptions),
    ...(args.suppressStartupUpdatePrompt ? { suppressStartupUpdatePrompt: true } : {})
  }
}