orchestral-cli 0.4.0

//! Minimal command-line composition root for the provider-neutral Generic Agent.

use std::collections::{BTreeMap, BTreeSet};
use std::io::{self, IsTerminal, Write};
use std::path::{Path, PathBuf};
use std::sync::Arc;
use std::time::{Duration, SystemTime, UNIX_EPOCH};

use anyhow::{bail, Context};
use orchestral_agent_journal_fs::FileAgentJournalStore;
use orchestral_blob_fs::FileBlobStore;
use orchestral_core::agent_connector::AgentSessionExecutionProfile;
use orchestral_core::agent_protocol::{
    reference::AgentRunStatus,
    spi::{AgentJournalStore, InMemoryAgentJournalStore},
    wire::{
        AgentCommand, AgentCommandEnvelope, AgentRunState, AgentSessionId, AgentTelemetry,
        AgentTerminalState, ApprovalDecision, BindingRequirement, CommandAckState, CommandId,
        Content, ContentBody, Digest, PendingRequest, PendingRequestPayload, ProviderBindingRef,
        RequestId, RequestResolution, ResourceBinding, ResourceBindingId, ResourceBindingMode,
        ResourceKind, ResourceRef, ResourceRevision, RunId,
    },
};
use orchestral_core::agent_session::{AgentSessionJournalStore, InMemoryAgentSessionJournalStore};
use orchestral_core::config::{
    load_config, BackendSpec, McpTransportSpec, ModelProfile, OrchestralConfig,
};
use orchestral_core::io::BlobStore;
use orchestral_core::mcp_protocol::{McpServerId, McpTransportFactory};
use orchestral_core::model_protocol::ModelBackend;
use orchestral_core::session_history::{SessionHistory, SessionOrigin};
use orchestral_core::skill_protocol::SKILL_CATALOG_RESOURCE_KIND_V1;
use orchestral_core::tool_effect::{InMemoryToolEffectJournalStore, ToolEffectJournalStore};
use orchestral_core::tool_protocol::{
    ApprovalPolicy, EffectScope, EnvironmentPolicy, FilesystemPolicy, HostApprovalVerifier,
    HostToolPolicy, InMemoryApprovalCapabilityStore, InteractiveCommandPolicy, NetworkPolicy,
    ProcessPolicy, RunToolGrant, SandboxPolicy, ToolDescriptor, ToolPolicyBounds, ToolRestriction,
    TransportLaunchPolicy,
};
use orchestral_mcp_streamable_http::{
    ResolvedCredentialHeader, StreamableHttpMcpTransportConfig, StreamableHttpMcpTransportFactory,
    DEFAULT_MAX_MCP_HTTP_FRAME_BYTES,
};
use orchestral_model_gemini::{
    GeminiAuthentication, GeminiModelBackend, GeminiModelConfig, GoogleCloudAccessTokenProvider,
};
use orchestral_model_openai::{
    OpenAiCompatibleBackend, OpenAiCompatibleConfig, OpenAiSamplingConfig, OpenAiToolResultFormat,
};
use orchestral_runtime::api::AgentApi;
use orchestral_runtime::session_history::JournalSessionHistory;
use orchestral_runtime::tools::{
    approved_host_exec_command_descriptor, guarded_apply_patch_descriptor,
    guarded_artifact_read_v2_descriptor, guarded_file_edit_descriptor,
    guarded_file_read_descriptor, guarded_file_search_descriptor, guarded_file_write_descriptor,
    guarded_session_read_descriptor, guarded_text_search_descriptor,
    workspace_exec_command_descriptor, workspace_write_stdin_descriptor,
    CommandEnvironmentSnapshot, GuardedApplyPatchExecutor, GuardedArtifactReadExecutor,
    GuardedExecCommandExecutor, GuardedFileEditExecutor, GuardedFileReadExecutor,
    GuardedFileSearchExecutor, GuardedFileWriteExecutor, GuardedSessionReadExecutor,
    GuardedTextSearchExecutor, GuardedWriteStdinExecutor,
};
use orchestral_runtime::{
    AgentClient, AgentControlEvent, AgentController, ContinuationPolicy,
    DeterministicExtractiveSessionSummarizer, GenericAgentCheckpointStore, GenericAgentConfig,
    GuardedMcpServerConfig, GuardedToolRuntime, InMemoryBlobStore,
    InMemoryGenericAgentCheckpointStore, InMemoryHostApprovalBroker, InternalGenericAgentProvider,
    McpToolsAdapterRegistry, ModelTokenMeter, ProcessSupervisor, SessionCompactionPolicy,
    StdioMcpSandboxPolicy, StdioMcpTransportFactory, ToolArtifactStore, WorkspacePermissionPolicy,
};
use tokio::io::{AsyncBufReadExt, AsyncReadExt, BufReader};
use tokio_util::sync::CancellationToken;

use crate::google_auth::{
    google_adc_is_explicitly_requested, resolve_google_vertex_auth, GoogleCredentialSource,
};
use crate::runtime::client::prepare_runtime_config_path;
use crate::runtime::ModelOverrides;
use crate::skill_command::{build_skill_setup, SkillManager};

#[derive(Clone)]
pub struct AgentRunOptions {
    pub config: Option<PathBuf>,
    pub credential_file: Option<PathBuf>,
    pub model_overrides: ModelOverrides,
    pub session_id: Option<String>,
    pub system_prompt: Option<String>,
    pub input: Option<String>,
    pub input_mode: InputMode,
    pub no_mcp: bool,
    pub mcp_config: Vec<PathBuf>,
    pub no_skills: bool,
    pub cwd: Option<PathBuf>,
    pub add_dirs: Vec<PathBuf>,
}

/// Availability of follow-up user input for a local CLI run.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, clap::ValueEnum)]
pub enum InputMode {
    /// Enable follow-up input only when stdin is a terminal.
    #[default]
    Auto,
    /// Allow replies on stdin; a piped initial prompt cannot share that stream.
    Interactive,
    /// Do not offer model-requested user input.
    None,
}

impl InputMode {
    fn resolve(self, entry: &EntryMode, stdin_is_terminal: bool) -> anyhow::Result<bool> {
        match (self, entry) {
            (Self::Interactive, EntryMode::HeadlessPipe) => bail!(
                "--input-mode interactive cannot use stdin for both the initial prompt and replies; \
                 pass the prompt as an argument to reserve stdin for replies"
            ),
            (Self::None, _) => Ok(false),
            (Self::Interactive, _) | (Self::Auto, EntryMode::Tui) => Ok(true),
            (Self::Auto, EntryMode::HeadlessPrompt(_)) => Ok(stdin_is_terminal),
            (Self::Auto, EntryMode::HeadlessPipe) => Ok(false),
        }
    }
}

#[derive(Debug, Clone)]
struct CliWorkspaceSet {
    primary: PathBuf,
    additional: Vec<PathBuf>,
}

impl CliWorkspaceSet {
    fn resolve(primary: Option<&Path>, additional: &[PathBuf]) -> anyhow::Result<Self> {
        let process_cwd = std::env::current_dir().context("resolve process directory")?;
        let requested_primary = primary.unwrap_or(&process_cwd);
        let primary = canonical_workspace_directory(requested_primary, "primary workspace")?;
        let mut seen = BTreeSet::from([primary.clone()]);
        let mut resolved_additional = Vec::new();
        for requested in additional {
            let requested = if requested.is_absolute() {
                requested.clone()
            } else {
                primary.join(requested)
            };
            let resolved = canonical_workspace_directory(&requested, "additional workspace")?;
            if seen.insert(resolved.clone()) {
                resolved_additional.push(resolved);
            }
        }
        Ok(Self {
            primary,
            additional: resolved_additional,
        })
    }

    fn roots(&self) -> impl Iterator<Item = &PathBuf> {
        std::iter::once(&self.primary).chain(self.additional.iter())
    }

    fn root_strings(&self) -> BTreeSet<String> {
        self.roots()
            .map(|root| root.to_string_lossy().into_owned())
            .collect()
    }

    fn file_tool_descriptor(&self, mut descriptor: ToolDescriptor) -> ToolDescriptor {
        // Freeze the model schema at registration from the Host workspace set.
        // The executor already defaults to the primary root when this is omitted.
        if self.additional.is_empty() {
            if let Some(properties) =
                descriptor.model_schema.input_schema["properties"].as_object_mut()
            {
                properties.remove("workspace");
            }
        }
        descriptor
    }
}

fn canonical_workspace_directory(path: &Path, label: &str) -> anyhow::Result<PathBuf> {
    let canonical = std::fs::canonicalize(path)
        .with_context(|| format!("resolve {label} '{}'", path.display()))?;
    if !canonical.is_dir() {
        bail!("{label} is not a directory: '{}'", canonical.display());
    }
    Ok(canonical)
}

#[derive(Debug, Clone, PartialEq, Eq)]
enum EntryMode {
    HeadlessPrompt(String),
    HeadlessPipe,
    Tui,
}

#[derive(Clone)]
struct CliJournalStores {
    run: Arc<dyn AgentJournalStore>,
    session: Arc<dyn AgentSessionJournalStore>,
    effect: Arc<dyn ToolEffectJournalStore>,
    checkpoint: Arc<dyn GenericAgentCheckpointStore>,
}

struct CliExecHost {
    shell: PathBuf,
    allow_host_execution: bool,
    runtime_readable_roots: Vec<PathBuf>,
    runtime_readable_files: Vec<PathBuf>,
    environment_names: BTreeSet<String>,
    environment: CommandEnvironmentSnapshot,
    network_targets: BTreeSet<String>,
}

struct CliToolComposition {
    runtime: Arc<GuardedToolRuntime<InMemoryApprovalCapabilityStore>>,
    run_grant: RunToolGrant,
    mcp_restriction: ToolRestriction,
    approval_broker: Arc<InMemoryHostApprovalBroker>,
    process_supervisor: Arc<ProcessSupervisor>,
}

#[derive(Clone)]
pub(crate) struct HostMetadata {
    pub workspaces: Vec<PathBuf>,
    pub journal_location: String,
    pub context: String,
    pub context_budget: Option<u64>,
    pub models: Vec<ModelProfile>,
    pub model_backend: BackendSpec,
    pub reasoning: orchestral_core::config::ReasoningPreference,
}

pub struct AgentHost {
    pub(crate) metadata: HostMetadata,
    journals: CliJournalStores,
    pub api: AgentApi,
    pub approvals: Arc<InMemoryHostApprovalBroker>,
    pub process_supervisor: Arc<ProcessSupervisor>,
    pub backend_name: String,
    pub model: String,
    pub workspace_root: PathBuf,
    pub execution_profile: AgentSessionExecutionProfile,
    pub session_history: crate::local_sessions::LocalSessionHistory,
    session_origin: SessionOrigin,
    pub(crate) skill_manager: SkillManager,
    controller: Arc<AgentController>,
    resources: Vec<ResourceBinding>,
    mcp_registry: McpToolsAdapterRegistry,
    // None denotes a Host with its own input channel (for example HTTP serve).
    // Local runs retain their resolved capability across TUI reconfiguration.
    local_input_available: Option<bool>,
}

impl AgentHost {
    pub fn client(&self, session_id: AgentSessionId) -> AgentClient {
        AgentClient::new(self.controller.clone(), session_id)
            .with_resources(self.resources.clone())
            .with_default_extensions(self.session_origin.extensions())
    }

    pub(crate) async fn reconfigure(&self, options: &AgentRunOptions) -> anyhow::Result<Self> {
        let mut next = build_agent_host_with_journals(
            options,
            Some(self.journals.clone()),
            self.local_input_available,
        )
        .await?;
        next.metadata.journal_location = self.metadata.journal_location.clone();
        next.session_history = self.session_history.clone();
        Ok(next)
    }

    pub(crate) async fn switch_session(&self, options: &AgentRunOptions) -> anyhow::Result<Self> {
        if matches!(
            self.session_history,
            crate::local_sessions::LocalSessionHistory::Single(_)
        ) {
            self.reconfigure(options).await
        } else {
            build_agent_host_with_journals(options, None, self.local_input_available).await
        }
    }

    pub async fn shutdown(&self) {
        self.mcp_registry.shutdown().await;
    }
}

pub async fn build_agent_host(options: &AgentRunOptions) -> anyhow::Result<AgentHost> {
    build_agent_host_with_journals(options, None, None).await
}

async fn build_agent_host_with_journals(
    options: &AgentRunOptions,
    shared: Option<CliJournalStores>,
    local_input_available: Option<bool>,
) -> anyhow::Result<AgentHost> {
    let workspaces = CliWorkspaceSet::resolve(options.cwd.as_deref(), &options.add_dirs)?;
    let config_path = prepare_runtime_config_path(
        options.config.clone(),
        &options.model_overrides,
        options.credential_file.as_deref(),
    )?;
    let config = load_config(&config_path)
        .with_context(|| format!("load Generic Agent config '{}'", config_path.display()))?;
    let (backend, profile, model, temperature) = resolve_model(&config).await?;
    let reasoning =
        crate::model_controls::resolve_reasoning(config.agent.reasoning.clone(), profile.as_ref())?;
    let (model_backend, token_meter) = build_model_backend(
        &backend,
        &model,
        temperature,
        profile.as_ref(),
        config.agent.stream_buffer,
        options.credential_file.as_deref(),
        reasoning.clone(),
    )?;

    let mut agent_config = GenericAgentConfig::new("orchestral/internal", "generic-agent");
    agent_config.input_requests_enabled =
        config.agent.input_requests_enabled && local_input_available.unwrap_or(true);
    agent_config.model_retry = config.agent.model_retry.clone();
    agent_config.context_recovery = config.agent.context_recovery.clone();
    {
        use orchestral_core::project_instructions::ProjectInstructionSource;
        use orchestral_project_instructions_fs::FileProjectInstructionSource;
        let source = FileProjectInstructionSource::new(config.agent.project_instructions.clone())?;
        agent_config.project_instructions = source
            .snapshot(&workspaces.root_strings().into_iter().collect::<Vec<_>>())
            .context("load project instruction snapshot")?;
        for document in &agent_config.project_instructions {
            tracing::info!(source = %document.source, scope = %document.scope, "loaded project instructions");
        }
    }
    agent_config.stream_buffer = config.agent.stream_buffer;
    agent_config.continuation = ContinuationPolicy {
        max_model_steps: config.agent.max_model_steps,
        max_tool_calls: config.agent.max_tool_calls,
    };
    agent_config.history_limit = config.agent.history_limit;
    let declared_context_capacity = model_backend.descriptor().capabilities.max_context_tokens;
    agent_config.max_context_tokens = declared_context_capacity
        .map_or(config.agent.max_context_tokens, |capacity| {
            capacity.min(config.agent.max_context_tokens)
        });
    agent_config.reserved_output_tokens = config.agent.reserved_output_tokens;
    agent_config.minimum_output_reserve_tokens = config.agent.minimum_output_reserve_tokens;
    if let Some(system_prompt) = options
        .system_prompt
        .clone()
        .or_else(|| config.agent.system_prompt.clone())
        .or_else(|| {
            profile
                .as_ref()
                .and_then(|profile| profile.system_prompt.clone())
        })
    {
        agent_config.system_prompt.push_str(
            "\n\nAdditional Host-configured instructions (these refine, but do not replace, the Agent contract):\n",
        );
        agent_config.system_prompt.push_str(system_prompt.trim());
    }
    let workspace_context = serde_json::json!({
        "primary": workspaces.primary,
        "additional": workspaces.additional,
        "os": std::env::consts::OS,
    });
    agent_config.system_prompt.push_str(&format!(
        "\n\n<environment_context>\n  <cwd>{}</cwd>\n  <workspace_roots>{}</workspace_roots>\n</environment_context>",
        workspaces.primary.display(),
        workspace_context
    ));
    let base_journal_root = std::path::PathBuf::from(&config.journal.root_dir);
    let journal_root = if local_input_available.is_some()
        && matches!(config.journal.backend.as_str(), "fs" | "filesystem")
    {
        let id = options
            .session_id
            .as_deref()
            .context("local Agent requires a session identity before opening storage")?;
        crate::local_sessions::writer_root(&base_journal_root, &AgentSessionId::new(id)).await?
    } else {
        base_journal_root.clone()
    };
    let journals = if let Some(shared) = shared {
        shared
    } else {
        match config.journal.backend.as_str() {
            "memory" => CliJournalStores {
                run: Arc::new(InMemoryAgentJournalStore::default()),
                session: Arc::new(InMemoryAgentSessionJournalStore::default()),
                effect: Arc::new(InMemoryToolEffectJournalStore::default()),
                checkpoint: Arc::new(InMemoryGenericAgentCheckpointStore::default()),
            },
            "filesystem" | "fs" => {
                let root = &journal_root;
                let store = Arc::new(
                    FileAgentJournalStore::open_single_writer(root)
                        .with_context(|| format!("open Agent Journal at '{}'; if resuming, close the other terminal controlling this session", root.display()))?,
                );
                CliJournalStores {
                    run: store.clone(),
                    session: store.clone(),
                    effect: store.clone(),
                    checkpoint: store,
                }
            }
            backend => bail!("unsupported Agent Journal backend for CLI: {backend}"),
        }
    };
    let CliJournalStores {
        run: run_journal,
        session: session_journal,
        effect: effect_journal,
        checkpoint: generic_checkpoint_journal,
    } = journals.clone();
    let metadata = HostMetadata {
        context_budget: declared_context_capacity.map(|_| agent_config.max_context_tokens),
        workspaces: std::iter::once(workspaces.primary.clone()).chain(workspaces.additional.clone()).collect(),
        journal_location: if config.journal.backend == "memory" { "In memory (this process only)".to_owned() } else { std::fs::canonicalize(&journal_root)?.display().to_string() },
        context: format!("Declared model/server context limit: {}\nEffective Host budget: {} tokens\nReserved output: {} tokens\nCompaction: {}\n\nLoaded project instructions (Host snapshot):\n{}",
            declared_context_capacity.map_or_else(|| "unknown (not reported or configured)".to_owned(), |capacity| format!("{capacity} tokens")),
            agent_config.max_context_tokens, config.agent.reserved_output_tokens,
            if config.agent.compaction.enabled { "automatic" } else { "disabled" },
            agent_config.project_instructions.iter().map(|doc| format!("{}\n  Scope: {}", doc.source, doc.scope)).collect::<Vec<_>>().join("\n")),
        models: config.providers.models.clone(),
        model_backend: backend.clone(),
        reasoning: reasoning.clone(),
    };
    let session_history = if matches!(config.journal.backend.as_str(), "fs" | "filesystem") {
        crate::local_sessions::LocalSessionHistory::Directory(base_journal_root)
    } else {
        crate::local_sessions::LocalSessionHistory::Single(JournalSessionHistory::new(
            run_journal.clone(),
            session_journal.clone(),
            ProviderBindingRef::new(crate::local_sessions::GENERIC_BINDING),
        ))
    };
    let mcp_configs = if options.no_mcp {
        Vec::new()
    } else {
        configured_mcp_servers(&config, &options.mcp_config, &workspaces.primary)?
    };
    let artifact_store = ToolArtifactStore::new(
        build_cli_blob_store(&config)?,
        config.artifacts.max_bytes,
        config.artifacts.summary_max_chars,
    )
    .context("configure Tool Artifact store")?
    .with_inline_output_limit(model_inline_output_limit(
        &config,
        model_backend.descriptor().capabilities.max_context_tokens,
    ));
    let artifact_reader = GuardedArtifactReadExecutor::new_session_scoped(
        artifact_store.clone(),
        run_journal.clone(),
        session_journal.clone(),
        effect_journal.clone(),
    );
    let CliToolComposition {
        runtime: tool_runtime,
        run_grant,
        mcp_restriction,
        approval_broker,
        process_supervisor,
    } = build_cli_tool_runtime(
        &config,
        &mcp_configs,
        effect_journal,
        artifact_store,
        &workspaces,
        artifact_reader,
    )?;
    // Session identity is resolved from the registered invoking Run, never
    // from model arguments or a filesystem path supplied by the model.
    let mut recall_bounds = run_grant.bounds.clone();
    recall_bounds.allowed_effects = BTreeSet::from([EffectScope::SessionRead]);
    recall_bounds.process = ProcessPolicy::default();
    recall_bounds.filesystem = FilesystemPolicy::default();
    recall_bounds.network = NetworkPolicy::default();
    recall_bounds.environment = EnvironmentPolicy::default();
    recall_bounds.allowed_credentials.clear();
    tool_runtime
        .register(
            guarded_session_read_descriptor(ToolRestriction {
                bounds: recall_bounds,
            }),
            Arc::new(GuardedSessionReadExecutor::new(
                run_journal.clone(),
                session_journal.clone(),
            )),
        )
        .context("register guarded session_read Tool")?;
    let mcp_registry = McpToolsAdapterRegistry::register(
        tool_runtime.as_ref(),
        mcp_configs,
        mcp_restriction,
        CancellationToken::new(),
    )
    .await
    .context("register guarded MCP stdio Tools")?;
    for (server, error) in mcp_registry.skipped_optional_servers() {
        tracing::warn!(server = %server, %error, "optional MCP server was unavailable");
    }
    let (skills, skill_manager) =
        build_skill_setup(&config, &workspaces.primary, options.no_skills)?;
    let provider = match skills.clone() {
        Some(skills) => {
            InternalGenericAgentProvider::new_with_tools_approval_skills_and_session_journal(
                model_backend,
                agent_config,
                tool_runtime,
                run_grant,
                approval_broker.clone(),
                skills,
                session_journal,
                token_meter,
            )
        }
        None => InternalGenericAgentProvider::new_with_tools_approval_and_session_journal(
            model_backend,
            agent_config,
            tool_runtime,
            run_grant,
            approval_broker.clone(),
            session_journal,
            token_meter,
        ),
    }
    .context("create Generic Agent provider")?;
    let provider = if config.agent.compaction.enabled {
        provider
            .with_session_compaction(
                Arc::new(
                    DeterministicExtractiveSessionSummarizer::new(
                        config.agent.compaction.summary_max_chars,
                    )
                    .context("configure deterministic Session summarizer")?,
                ),
                SessionCompactionPolicy {
                    minimum_source_records: config.agent.compaction.minimum_source_records,
                    keep_recent_records: config.agent.compaction.keep_recent_records,
                },
            )
            .context("bind Generic Agent Session compaction")?
    } else {
        provider
    };
    let provider = provider
        .with_checkpoint_store(generic_checkpoint_journal)
        .context("bind Generic Agent private checkpoint journal")?;
    let provider = Arc::new(provider);
    let controller = Arc::new(
        AgentController::with_journal_store(
            provider,
            ProviderBindingRef::new("orchestral/generic-agent"),
            run_journal,
        )
        .context("bind Generic Agent controller")?,
    );
    let mut resources = Vec::new();
    if let Some(skills) = skills.as_deref() {
        resources.push(ResourceBinding {
            binding_id: ResourceBindingId::new("cli-skill-catalog"),
            resource: ResourceRef {
                kind: ResourceKind::new(SKILL_CATALOG_RESOURCE_KIND_V1),
                id: skills.catalog().resource_id.clone(),
                revision: ResourceRevision::new(skills.catalog().revision.as_str()),
            },
            requirement: BindingRequirement::Required,
            mode: ResourceBindingMode::Snapshot,
        });
    }
    let session_origin = SessionOrigin {
        workspace: workspaces.primary.to_string_lossy().into_owned(),
        additional_workspaces: workspaces
            .additional
            .iter()
            .map(|path| path.to_string_lossy().into_owned())
            .collect(),
        model: model.clone(),
    };
    let api = AgentApi::with_resources(controller.clone(), resources.clone())
        .with_default_extensions(session_origin.extensions());
    Ok(AgentHost {
        metadata,
        journals,
        api,
        approvals: approval_broker,
        process_supervisor,
        backend_name: backend.name,
        model: model.clone(),
        workspace_root: workspaces.primary.clone(),
        execution_profile: AgentSessionExecutionProfile {
            model: Some(model),
            reasoning_effort: Some(reasoning.to_string()),
            permissions: Default::default(),
        },
        skill_manager,
        session_history,
        session_origin,
        controller,
        resources,
        mcp_registry,
        local_input_available,
    })
}

pub async fn run(mut options: AgentRunOptions) -> anyhow::Result<()> {
    let session_id = AgentSessionId::new(
        options
            .session_id
            .get_or_insert_with(|| unique_id("cli-session", 0))
            .clone(),
    );
    let stdin_is_terminal = io::stdin().is_terminal();
    let entry_mode = select_entry_mode(
        options.input.clone(),
        stdin_is_terminal,
        io::stdout().is_terminal(),
    )?;
    let local_input_available = options.input_mode.resolve(&entry_mode, stdin_is_terminal)?;
    let mut host = Arc::new(
        build_agent_host_with_journals(&options, None, Some(local_input_available)).await?,
    );
    let tui_options = options.clone();
    let client = host.client(session_id.clone());

    let result = async {
        let history = host.session_history.read(&session_id).await?;
        if let Some(history) = &history {
            crate::local_sessions::validate_workspace(history, &host.session_origin.workspace)?;
            eprintln!(
                "Resuming session {} · {}",
                session_id, history.summary.title
            );
        }
        let resumed = resume_unfinished(&client, history.as_ref()).await?;
        let history = host.session_history.read(&session_id).await?;
        match entry_mode {
            EntryMode::HeadlessPrompt(input) => {
                eprintln!("Model: {} (provider: {})", host.model, host.backend_name);
                let mut lines = BufReader::new(tokio::io::stdin()).lines();
                run_turn(
                    &client,
                    &host.approvals,
                    1,
                    input,
                    &mut lines,
                    false,
                    resumed,
                )
                .await
            }
            EntryMode::HeadlessPipe => {
                eprintln!("Model: {} (provider: {})", host.model, host.backend_name);
                let mut input = String::new();
                tokio::io::stdin()
                    .read_to_string(&mut input)
                    .await
                    .context("read piped Agent input")?;
                if input.trim().is_empty() {
                    bail!("stdin pipe did not contain an Agent prompt")
                }
                let mut lines = BufReader::new(tokio::io::stdin()).lines();
                run_turn(
                    &client,
                    &host.approvals,
                    1,
                    input,
                    &mut lines,
                    false,
                    resumed,
                )
                .await
            }
            EntryMode::Tui => {
                crate::tui::run_tui(
                    client,
                    &mut host,
                    tui_options,
                    crate::tui::TuiResume {
                        history,
                        run: resumed,
                    },
                )
                .await
            }
        }
    }
    .await;
    host.shutdown().await;
    result
}

pub(crate) async fn resume_unfinished(
    client: &AgentClient,
    history: Option<&SessionHistory>,
) -> anyhow::Result<Option<orchestral_runtime::AgentRunHandle>> {
    let Some(history) = history else {
        return Ok(None);
    };
    if history.summary.unfinished_run_ids.len() > 1 {
        bail!("session has multiple unfinished executions; cannot select one for continuation");
    }
    let mut resumed = None;
    for run_id in &history.summary.unfinished_run_ids {
        let handle = client
            .resume_run(run_id)
            .await
            .with_context(|| format!("reconcile unfinished Run {run_id}"))?;
        if !handle.inspect().await?.state.is_terminal() {
            resumed = Some(handle);
        }
    }
    Ok(resumed)
}

fn model_inline_output_limit(
    config: &OrchestralConfig,
    backend_context_tokens: Option<u64>,
) -> std::num::NonZeroU64 {
    if let Some(limit) = config.tools.max_inline_output_bytes {
        return limit;
    }
    let context = backend_context_tokens
        .unwrap_or(config.agent.max_context_tokens)
        .min(config.agent.max_context_tokens);
    // A per-result allowance, not a tokenizer estimate or a guarantee for an
    // entire conversation. Leave room for the request, history and other tools.
    let bytes = context
        .saturating_sub(config.agent.reserved_output_tokens)
        .div_ceil(4)
        .max(1);
    std::num::NonZeroU64::new(bytes).expect("inline output allowance is positive")
}

fn build_cli_tool_runtime(
    config: &OrchestralConfig,
    mcp_configs: &[GuardedMcpServerConfig],
    effect_journal: Arc<dyn ToolEffectJournalStore>,
    artifact_store: ToolArtifactStore,
    workspaces: &CliWorkspaceSet,
    artifact_reader: GuardedArtifactReadExecutor,
) -> anyhow::Result<CliToolComposition> {
    let workspace_roots = workspaces.root_strings();
    let exec_host = configured_exec_host(config)?;
    let max_output_bytes = usize::try_from(config.tools.max_output_bytes).unwrap_or(usize::MAX);
    let process_supervisor = Arc::new(
        if exec_host.is_some() {
            ProcessSupervisor::new_with_runtime_temp_root(
                max_output_bytes,
                command_runtime_temp_root(&workspace_roots)?,
            )
        } else {
            ProcessSupervisor::new(max_output_bytes)
        }
        .context("create run-scoped process supervisor")?,
    );
    let runtime_temp_root = process_supervisor
        .runtime_temp_root()
        .to_string_lossy()
        .into_owned();
    let exec_programs = exec_host
        .as_ref()
        .map(|host| BTreeSet::from([host.shell.to_string_lossy().into_owned()]))
        .unwrap_or_default();
    let exec_enabled = exec_host.is_some();
    let host_execution_enabled = exec_host
        .as_ref()
        .is_some_and(|host| host.allow_host_execution);
    let mcp_effects = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::effect_scopes)
        .collect::<BTreeSet<_>>();
    let transport_programs = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::allowed_programs)
        .collect::<BTreeSet<_>>();
    let transport_allows_children = mcp_configs
        .iter()
        .any(GuardedMcpServerConfig::allows_child_processes);
    let mcp_readable_roots = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::filesystem_read_roots)
        .collect::<BTreeSet<_>>();
    let mcp_writable_roots = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::filesystem_write_roots)
        .collect::<BTreeSet<_>>();
    let mut allowed_effects = BTreeSet::from([
        EffectScope::FilesystemRead,
        EffectScope::FilesystemWrite,
        EffectScope::ArtifactRead,
        EffectScope::SessionRead,
    ]);
    if exec_enabled {
        allowed_effects.extend([
            EffectScope::Process,
            EffectScope::Network,
            EffectScope::EnvironmentRead,
            EffectScope::ExternalSideEffect,
        ]);
    }
    if host_execution_enabled {
        allowed_effects.insert(EffectScope::HostExecution);
    }
    allowed_effects.extend(mcp_effects.iter().copied());
    let mcp_environment = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::environment_names)
        .collect::<BTreeSet<_>>();
    let mut allowed_environment = mcp_environment.clone();
    if let Some(host) = &exec_host {
        allowed_environment.extend(host.environment_names.iter().cloned());
    }
    let mcp_network_targets = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::allowed_network_targets)
        .collect::<BTreeSet<_>>();
    let mcp_allows_unrestricted_network = mcp_configs
        .iter()
        .any(GuardedMcpServerConfig::allows_unrestricted_network);
    let mut allowed_network_targets = mcp_network_targets.clone();
    if let Some(host) = &exec_host {
        allowed_network_targets.extend(host.network_targets.iter().cloned());
    }
    let allowed_credentials = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::credential_references)
        .collect::<BTreeSet<_>>();
    let mcp_sandbox_profiles = mcp_configs
        .iter()
        .flat_map(GuardedMcpServerConfig::sandbox_profiles)
        .collect::<BTreeSet<_>>();
    let mut readable_roots = workspace_roots.clone();
    readable_roots.extend(mcp_readable_roots.iter().cloned());
    let mut writable_roots = workspace_roots.clone();
    writable_roots.extend(mcp_writable_roots.iter().cloned());
    if exec_enabled {
        readable_roots.insert(runtime_temp_root.clone());
        writable_roots.insert(runtime_temp_root.clone());
    }
    let mcp_restriction = ToolRestriction {
        bounds: ToolPolicyBounds {
            allowed_effects: mcp_effects.clone(),
            // Explicit `mcp add`/Host configuration trusts the transport.
            // Per-call approval is derived from standard MCP Tool annotations.
            approval: ApprovalPolicy::NotRequired,
            sandbox: SandboxPolicy {
                required: !mcp_sandbox_profiles.is_empty(),
                allowed_profiles: mcp_sandbox_profiles.clone(),
            },
            process: ProcessPolicy {
                interactive: InteractiveCommandPolicy::default(),
                transport: TransportLaunchPolicy {
                    allowed_programs: transport_programs.clone(),
                    allow_child_processes: transport_allows_children,
                },
            },
            filesystem: FilesystemPolicy {
                readable_roots: mcp_readable_roots,
                writable_roots: mcp_writable_roots,
            },
            network: NetworkPolicy {
                allowed_targets: mcp_network_targets,
                allow_unrestricted: mcp_allows_unrestricted_network,
            },
            environment: EnvironmentPolicy {
                allowed_variables: mcp_environment,
                inherit_host_environment: false,
            },
            allowed_credentials: allowed_credentials.clone(),
            max_timeout_ms: Some(config.tools.max_timeout_ms),
            max_output_bytes: Some(config.tools.max_output_bytes),
        },
    };
    let bounds = ToolPolicyBounds {
        allowed_effects,
        approval: ApprovalPolicy::NotRequired,
        sandbox: SandboxPolicy {
            // The Host ceiling admits both sandboxed local capabilities and
            // non-process transports. Each tool lane decides whether its own
            // sandbox is mandatory.
            required: false,
            allowed_profiles: BTreeSet::from([
                "workspace_read".to_owned(),
                orchestral_runtime::tools::GUARDED_EXEC_SANDBOX_PROFILE.to_owned(),
            ])
            .union(&mcp_sandbox_profiles)
            .cloned()
            .collect(),
        },
        process: ProcessPolicy {
            interactive: InteractiveCommandPolicy {
                enabled: exec_enabled,
                command_shells: exec_programs.clone(),
                allow_child_processes: exec_enabled,
            },
            transport: TransportLaunchPolicy {
                allowed_programs: transport_programs,
                allow_child_processes: transport_allows_children,
            },
        },
        filesystem: FilesystemPolicy {
            readable_roots,
            writable_roots,
        },
        network: NetworkPolicy {
            allowed_targets: allowed_network_targets,
            allow_unrestricted: host_execution_enabled || mcp_allows_unrestricted_network,
        },
        environment: EnvironmentPolicy {
            allowed_variables: allowed_environment,
            inherit_host_environment: false,
        },
        allowed_credentials,
        max_timeout_ms: Some(config.tools.max_timeout_ms),
        max_output_bytes: Some(config.tools.max_output_bytes),
    };
    let mut workspace_bounds = bounds.clone();
    workspace_bounds.allowed_effects = BTreeSet::from([
        EffectScope::FilesystemRead,
        EffectScope::FilesystemWrite,
        EffectScope::ArtifactRead,
    ]);
    workspace_bounds.sandbox.required = true;
    workspace_bounds.sandbox.allowed_profiles = BTreeSet::from(["workspace_read".to_owned()]);
    workspace_bounds.process = ProcessPolicy::default();
    workspace_bounds.filesystem = FilesystemPolicy {
        readable_roots: workspace_roots.clone(),
        writable_roots: workspace_roots,
    };
    workspace_bounds.network = NetworkPolicy::default();
    workspace_bounds.environment = EnvironmentPolicy::default();
    workspace_bounds.allowed_credentials.clear();
    // Tool restrictions are capability-local. MCP transport programs,
    // credentials, environment names, and network targets never become
    // ambient authority for generic command execution.
    let mut exec_bounds = bounds.clone();
    exec_bounds.allowed_effects = BTreeSet::from([
        EffectScope::Process,
        EffectScope::Network,
        EffectScope::FilesystemRead,
        EffectScope::FilesystemWrite,
        EffectScope::EnvironmentRead,
        EffectScope::ExternalSideEffect,
    ]);
    if host_execution_enabled {
        exec_bounds
            .allowed_effects
            .insert(EffectScope::HostExecution);
    }
    exec_bounds.sandbox.allowed_profiles =
        BTreeSet::from([orchestral_runtime::tools::GUARDED_EXEC_SANDBOX_PROFILE.to_owned()]);
    exec_bounds.sandbox.required = true;
    exec_bounds.process.interactive = InteractiveCommandPolicy {
        enabled: true,
        command_shells: exec_programs,
        allow_child_processes: true,
    };
    exec_bounds.process.transport = TransportLaunchPolicy::default();
    exec_bounds.filesystem = workspace_bounds.filesystem.clone();
    exec_bounds
        .filesystem
        .readable_roots
        .insert(runtime_temp_root.clone());
    exec_bounds
        .filesystem
        .writable_roots
        .insert(runtime_temp_root);
    exec_bounds.network = NetworkPolicy {
        allowed_targets: exec_host
            .as_ref()
            .map(|host| host.network_targets.clone())
            .unwrap_or_default(),
        allow_unrestricted: host_execution_enabled,
    };
    exec_bounds.environment = EnvironmentPolicy {
        allowed_variables: exec_host
            .as_ref()
            .map(|host| host.environment_names.clone())
            .unwrap_or_default(),
        inherit_host_environment: false,
    };
    exec_bounds.allowed_credentials.clear();
    let mut signing_material = [0_u8; 32];
    getrandom::fill(&mut signing_material)
        .map_err(|error| anyhow::anyhow!("generate Host approval signing key: {error}"))?;
    let approval_broker = Arc::new(
        InMemoryHostApprovalBroker::new(signing_material).context("create Host approval broker")?,
    );
    let verifier =
        HostApprovalVerifier::new(signing_material, InMemoryApprovalCapabilityStore::default())
            .context("create Host approval verifier")?;
    let runtime = Arc::new(
        GuardedToolRuntime::new_with_effect_journal_and_artifacts(
            HostToolPolicy {
                bounds: bounds.clone(),
            },
            verifier,
            effect_journal,
            artifact_store.clone(),
        )
        .context("create guarded Tool Runtime")?
        .with_permission_policy(Arc::new(WorkspacePermissionPolicy)),
    );
    runtime
        .register(
            guarded_artifact_read_v2_descriptor(ToolRestriction {
                bounds: workspace_bounds.clone(),
            }),
            Arc::new(artifact_reader),
        )
        .context("register guarded artifact_read Tool")?;
    runtime
        .register(
            workspaces.file_tool_descriptor(guarded_file_read_descriptor(ToolRestriction {
                bounds: workspace_bounds.clone(),
            })),
            Arc::new(
                GuardedFileReadExecutor::new_with_roots(
                    &workspaces.primary,
                    &workspaces.additional,
                )
                .context("open file_read workspace capability")?,
            ),
        )
        .context("register guarded file_read Tool")?;
    runtime
        .register(
            workspaces.file_tool_descriptor(guarded_file_search_descriptor(ToolRestriction {
                bounds: workspace_bounds.clone(),
            })),
            Arc::new(
                GuardedFileSearchExecutor::new_with_roots(
                    &workspaces.primary,
                    &workspaces.additional,
                )
                .context("open file_search workspace capability")?,
            ),
        )
        .context("register guarded file_search Tool")?;
    runtime
        .register(
            workspaces.file_tool_descriptor(guarded_text_search_descriptor(ToolRestriction {
                bounds: workspace_bounds.clone(),
            })),
            Arc::new(
                GuardedTextSearchExecutor::new_with_roots(
                    &workspaces.primary,
                    &workspaces.additional,
                )
                .context("open text_search workspace capability")?,
            ),
        )
        .context("register guarded text_search Tool")?;
    runtime
        .register(
            workspaces.file_tool_descriptor(guarded_file_write_descriptor(ToolRestriction {
                bounds: workspace_bounds.clone(),
            })),
            Arc::new(
                GuardedFileWriteExecutor::new_with_roots(
                    &workspaces.primary,
                    &workspaces.additional,
                )
                .context("open file_write workspace capability")?,
            ),
        )
        .context("register guarded file_write Tool")?;
    runtime
        .register(
            workspaces.file_tool_descriptor(guarded_file_edit_descriptor(ToolRestriction {
                bounds: workspace_bounds.clone(),
            })),
            Arc::new(
                GuardedFileEditExecutor::new_with_roots(
                    &workspaces.primary,
                    &workspaces.additional,
                )
                .context("open file_edit workspace capability")?,
            ),
        )
        .context("register guarded file_edit Tool")?;
    runtime
        .register(
            workspaces.file_tool_descriptor(guarded_apply_patch_descriptor(ToolRestriction {
                bounds: workspace_bounds,
            })),
            Arc::new(
                GuardedApplyPatchExecutor::new_with_roots(
                    &workspaces.primary,
                    &workspaces.additional,
                )
                .context("open apply_patch workspace capability")?,
            ),
        )
        .context("register guarded apply_patch Tool")?;
    if let Some(exec_host) = exec_host {
        let restriction = ToolRestriction {
            bounds: exec_bounds.clone(),
        };
        let mut descriptor = if config.tools.exec.sandboxed_execution_enabled {
            workspace_exec_command_descriptor(restriction)
        } else {
            approved_host_exec_command_descriptor(restriction)
        };
        let executor = GuardedExecCommandExecutor::new(
            process_supervisor.clone(),
            exec_host.shell,
            exec_host.runtime_readable_roots,
            exec_host.runtime_readable_files,
            exec_host.environment,
        )
        .and_then(|executor| {
            executor.with_pipeline_exit_status(config.tools.exec.pipeline_exit_status)
        })
        .map_err(anyhow::Error::msg)
        .context("configure guarded exec_command Tool")?
        .with_sandboxed_execution_enabled(config.tools.exec.sandboxed_execution_enabled);
        descriptor.model_schema.description.push_str(if executor.enables_pipefail() {
            " Pipefail is enabled: a pipeline returns its rightmost nonzero status, including SIGPIPE from an early-exiting consumer."
        } else {
            " Pipeline exit status uses native shell behavior; a successful last stage may hide an earlier failure."
        });
        runtime
            .register(descriptor, Arc::new(executor))
            .context("register guarded exec_command Tool")?;
        runtime
            .register(
                workspace_write_stdin_descriptor(ToolRestriction {
                    bounds: exec_bounds,
                }),
                Arc::new(GuardedWriteStdinExecutor::new(process_supervisor.clone())),
            )
            .context("register guarded write_stdin Tool")?;
    } else {
        tracing::warn!("Generic Agent command execution is disabled by Host config");
    }
    Ok(CliToolComposition {
        runtime,
        run_grant: RunToolGrant { bounds },
        mcp_restriction,
        approval_broker,
        process_supervisor,
    })
}

fn command_runtime_temp_root(workspace_roots: &BTreeSet<String>) -> anyhow::Result<PathBuf> {
    // Leave room for tools that create Unix sockets below TMPDIR. The macOS
    // per-user OS temp prefix alone can consume much of sockaddr_un.sun_path.
    let candidates = vec![
        #[cfg(unix)]
        PathBuf::from("/tmp"),
        #[cfg(unix)]
        PathBuf::from("/var/tmp"),
        std::env::temp_dir(),
    ];
    let parent = candidates
        .into_iter()
        .filter_map(|path| std::fs::canonicalize(path).ok())
        .find(|path| path.is_dir() && !workspace_roots.iter().any(|root| path.starts_with(root)))
        .context("no OS temporary directory is available outside the workspace roots")?;
    let identity =
        orchestral_core::agent_protocol::wire::Digest::sha256(serde_json::to_vec(workspace_roots)?);
    Ok(parent.join(format!("orch-{}", &identity.as_str()[..24])))
}

fn build_cli_blob_store(config: &OrchestralConfig) -> anyhow::Result<Arc<dyn BlobStore>> {
    match config.artifacts.backend.trim() {
        "memory" | "in_memory" => Ok(Arc::new(InMemoryBlobStore::default())),
        "local" | "filesystem" | "fs" => Ok(Arc::new(
            FileBlobStore::open(&config.artifacts.root_dir).with_context(|| {
                format!("open Artifact BlobStore at '{}'", config.artifacts.root_dir)
            })?,
        )),
        mode => bail!("unsupported BlobStore mode for Generic Agent Artifact results: {mode}"),
    }
}

fn configured_exec_host(config: &OrchestralConfig) -> anyhow::Result<Option<CliExecHost>> {
    if !config.tools.exec.enabled {
        return Ok(None);
    }
    if !config.tools.exec.sandboxed_execution_enabled && !config.tools.exec.allow_host_execution {
        bail!("tools.exec.sandboxed_execution_enabled=false requires tools.exec.allow_host_execution=true; command approval is still required");
    }
    let configured = config
        .tools
        .exec
        .shell
        .as_deref()
        .filter(|shell| !shell.trim().is_empty());
    let shell = if let Some(shell) = configured {
        PathBuf::from(resolve_host_program(shell)?)
    } else if let Some(shell) = std::env::var_os("SHELL").filter(|shell| !shell.is_empty()) {
        PathBuf::from(resolve_host_program(&shell.to_string_lossy())?)
    } else {
        #[cfg(windows)]
        let candidates = {
            let mut paths = vec!["pwsh.exe".to_owned(), "powershell.exe".to_owned()];
            if let Some(root) = std::env::var_os("SystemRoot") {
                paths.push(
                    PathBuf::from(root)
                        .join("System32/WindowsPowerShell/v1.0/powershell.exe")
                        .to_string_lossy()
                        .into_owned(),
                );
            }
            paths.push("cmd.exe".to_owned());
            paths
        };
        #[cfg(not(windows))]
        let candidates = ["/bin/zsh", "/bin/bash", "/bin/sh"];
        candidates
            .iter()
            .find_map(|candidate| resolve_host_program(candidate).ok())
            .map(PathBuf::from)
            .context("no command shell is available; set tools.exec.shell")?
    };
    let environment_names = [
        "PATH",
        "HOME",
        "USERPROFILE",
        "SystemRoot",
        "SYSTEMROOT",
        "WINDIR",
        "COMSPEC",
        "PATHEXT",
        "APPDATA",
        "LOCALAPPDATA",
        "USER",
        "LANG",
        "LC_ALL",
        "TERM",
        "COLORTERM",
        "NO_COLOR",
        "CARGO_HOME",
        "RUSTUP_HOME",
        "XDG_CONFIG_HOME",
        "GIT_CONFIG_GLOBAL",
        "GIT_CONFIG_SYSTEM",
        "GIT_CONFIG_NOSYSTEM",
        "HTTP_PROXY",
        "HTTPS_PROXY",
        "ALL_PROXY",
        "NO_PROXY",
        "http_proxy",
        "https_proxy",
        "all_proxy",
        "no_proxy",
    ]
    .into_iter()
    .map(str::to_owned)
    .collect::<BTreeSet<_>>();
    Ok(Some(CliExecHost {
        allow_host_execution: config.tools.exec.allow_host_execution,
        runtime_readable_roots: exec_runtime_readable_roots(&shell),
        runtime_readable_files: exec_runtime_readable_files(),
        shell,
        environment: CommandEnvironmentSnapshot::capture(environment_names.iter().cloned()),
        environment_names,
        network_targets: config.tools.exec.network_targets.iter().cloned().collect(),
    }))
}

fn exec_runtime_readable_files() -> Vec<PathBuf> {
    let mut candidates = orchestral_runtime::tools::host_toolchain_readable_files()
        .into_iter()
        .collect::<BTreeSet<_>>();
    let home = std::env::var_os("HOME").map(PathBuf::from);
    if let Some(home) = &home {
        candidates.insert(home.join(".gitconfig"));
        let xdg_home = std::env::var_os("XDG_CONFIG_HOME")
            .map(PathBuf::from)
            .filter(|path| path.is_absolute())
            .unwrap_or_else(|| home.join(".config"));
        candidates.insert(xdg_home.join("git/config"));
    }
    for name in ["GIT_CONFIG_GLOBAL", "GIT_CONFIG_SYSTEM"] {
        if let Some(path) = std::env::var_os(name).map(PathBuf::from) {
            if path.is_absolute() {
                candidates.insert(path);
            }
        }
    }
    candidates
        .into_iter()
        .filter_map(|path| std::fs::canonicalize(path).ok())
        .filter(|path| path.is_file())
        .collect()
}

fn exec_runtime_readable_roots(shell: &Path) -> Vec<PathBuf> {
    let mut candidates = BTreeSet::new();
    if let Some(parent) = shell.parent() {
        candidates.insert(parent.to_path_buf());
    }
    if let Some(path) = std::env::var_os("PATH") {
        for directory in std::env::split_paths(&path).filter(|path| path.is_absolute()) {
            candidates.insert(directory.clone());
            let text = directory.to_string_lossy();
            // Version-managed runtimes may load adjacent libraries, but never
            // widen a PATH entry to the manager's entire user directory (for
            // example ~/.cargo, which can contain credentials).
            if ["/.nvm/versions/", "/.pyenv/versions/"]
                .iter()
                .any(|marker| text.contains(marker))
                && directory.file_name().is_some_and(|name| name == "bin")
            {
                if let Some(installation) = directory.parent() {
                    candidates.insert(installation.to_path_buf());
                }
            }
            if text.starts_with("/opt/homebrew/") {
                candidates.insert(PathBuf::from("/opt/homebrew"));
            }
            if text.starts_with("/nix/store/") {
                candidates.insert(PathBuf::from("/nix/store"));
            }
        }
    }
    for candidate in [
        "/bin",
        "/usr/bin",
        "/usr/sbin",
        "/usr/local",
        "/opt/homebrew",
        "/nix/store",
        "/etc",
        "/Library/Frameworks",
        "/Library/Developer/CommandLineTools",
        "/Applications/Xcode.app",
    ] {
        candidates.insert(PathBuf::from(candidate));
    }
    if let Some(home) = std::env::var_os("HOME") {
        let home = PathBuf::from(home);
        candidates.insert(home.join(".cargo/bin"));
        candidates.insert(home.join(".cargo/registry"));
        candidates.insert(home.join(".cargo/git"));
        candidates.insert(home.join(".rustup"));
    }
    candidates
        .into_iter()
        .filter_map(|path| std::fs::canonicalize(path).ok())
        .filter(|path| path.is_dir())
        .collect()
}

/// MCP and Skill remain distinct resources. Only explicit Host MCP transports
/// are composed here; discovered methods publish as guarded Tools.
fn configured_mcp_servers(
    config: &OrchestralConfig,
    cli_manifest_paths: &[PathBuf],
    workspace: &Path,
) -> anyhow::Result<Vec<GuardedMcpServerConfig>> {
    if !config.mcp.enabled {
        return Ok(Vec::new());
    }
    let mut trusted_user_paths = Vec::new();
    match crate::mcp_config::user_registry_path() {
        Ok(path) if path.is_file() => trusted_user_paths.push(path),
        Ok(_) => {}
        Err(error) => tracing::debug!(%error, "user MCP registry path is unavailable"),
    }
    let specs = crate::mcp_config::load_server_manifests(
        workspace,
        &config.mcp.servers,
        &config.mcp.import_files,
        &trusted_user_paths,
        cli_manifest_paths,
    )?;
    let mut servers = Vec::new();
    for spec in specs.iter().filter(|server| server.enabled) {
        let transport = (|| -> anyhow::Result<Arc<dyn McpTransportFactory>> {
            match &spec.transport {
                McpTransportSpec::Stdio {
                    command,
                    args,
                    env,
                    allow_child_processes,
                    allow_host_ui,
                    cwd,
                    readable_roots,
                    writable_roots,
                    network_targets,
                    allow_unrestricted_network,
                } => {
                    let program = resolve_host_program(command)?;
                    let cwd =
                        resolve_mcp_directory(workspace, cwd.as_deref().unwrap_or("."), "cwd")?;
                    let mut reads =
                        resolve_mcp_directories(workspace, readable_roots, "readable root")?;
                    reads.insert(cwd.clone());
                    let mut writes =
                        resolve_mcp_directories(workspace, writable_roots, "writable root")?;
                    let runtime_root = prepare_mcp_runtime_root(workspace, &spec.name)?;
                    writes.insert(runtime_root.clone());
                    Ok(Arc::new(StdioMcpTransportFactory::new(
                        PathBuf::from(program),
                        args.clone(),
                        env.iter()
                            .map(|(key, value)| (key.clone(), value.clone()))
                            .collect(),
                        StdioMcpSandboxPolicy::scoped(
                            cwd,
                            reads,
                            writes,
                            network_targets.iter().cloned().collect(),
                        )
                        .with_unrestricted_network(*allow_unrestricted_network)
                        .with_child_processes(*allow_child_processes)
                        .with_host_ui(*allow_host_ui)
                        .with_private_runtime_home(runtime_root),
                    )?))
                }
                McpTransportSpec::StreamableHttp {
                    endpoint,
                    credential_headers,
                    max_frame_bytes,
                } => {
                    let mut resolved = BTreeMap::new();
                    for (header, credential) in credential_headers {
                        let env_name = credential.env.trim();
                        let value = std::env::var(env_name).with_context(|| {
                            format!(
                                "resolve credential environment variable '{env_name}' for header '{header}'"
                            )
                        })?;
                        resolved.insert(
                            header.clone(),
                            ResolvedCredentialHeader {
                                reference: format!("env:{env_name}"),
                                value,
                            },
                        );
                    }
                    Ok(Arc::new(StreamableHttpMcpTransportFactory::new(
                        StreamableHttpMcpTransportConfig {
                            endpoint: endpoint.clone(),
                            credential_headers: resolved,
                            max_frame_bytes: max_frame_bytes
                                .unwrap_or(DEFAULT_MAX_MCP_HTTP_FRAME_BYTES),
                        },
                    )?))
                }
            }
        })();
        let transport = match transport {
            Ok(transport) => transport,
            Err(error) if !spec.required => {
                tracing::warn!(server = spec.name, %error, "optional MCP transport was unavailable");
                continue;
            }
            Err(error) => {
                return Err(error)
                    .with_context(|| format!("compose required MCP server '{}'", spec.name))
            }
        };
        let server = GuardedMcpServerConfig {
            server_id: McpServerId::new(spec.name.trim()),
            required: spec.required,
            transport,
            startup_timeout: Duration::from_millis(spec.startup_timeout_ms.unwrap_or(15_000)),
            // Match Codex's proven default: remote MCP operations routinely
            // include search, authentication, and service fan-out that exceed
            // a generic short Tool deadline.
            tool_timeout: Duration::from_millis(spec.tool_timeout_ms.unwrap_or(120_000)),
            enabled_tools: spec.enabled_tools.iter().cloned().collect(),
            disabled_tools: spec.disabled_tools.iter().cloned().collect(),
        };
        if let Err(error) = server.validate() {
            if !spec.required {
                tracing::warn!(server = spec.name, %error, "optional MCP server config was invalid");
                continue;
            }
            return Err(error)
                .with_context(|| format!("validate required MCP server '{}'", spec.name));
        }
        servers.push(server);
    }
    Ok(servers)
}

fn resolve_mcp_directories(
    workspace: &Path,
    configured: &[String],
    label: &str,
) -> anyhow::Result<BTreeSet<PathBuf>> {
    configured
        .iter()
        .map(|path| resolve_mcp_directory(workspace, path, label))
        .collect()
}

fn resolve_mcp_directory(
    workspace: &Path,
    configured: &str,
    label: &str,
) -> anyhow::Result<PathBuf> {
    let path = PathBuf::from(configured);
    let path = if path.is_absolute() {
        path
    } else {
        workspace.join(path)
    };
    let canonical = std::fs::canonicalize(&path)
        .with_context(|| format!("canonicalize MCP stdio {label} '{}'", path.display()))?;
    if !canonical.is_dir() {
        bail!("MCP stdio {label} '{}' is not a directory", path.display());
    }
    Ok(canonical)
}

fn prepare_mcp_runtime_root(workspace: &Path, server_name: &str) -> anyhow::Result<PathBuf> {
    let identity = Digest::sha256(server_name.as_bytes());
    let root = workspace
        .join(".orchestral/mcp")
        .join(&identity.as_str()[..16]);
    std::fs::create_dir_all(&root)
        .with_context(|| format!("create MCP runtime directory '{}'", root.display()))?;
    std::fs::canonicalize(&root)
        .with_context(|| format!("canonicalize MCP runtime directory '{}'", root.display()))
}

fn resolve_host_program(program: &str) -> anyhow::Result<String> {
    let candidate = locate_host_program(program)?;
    std::fs::canonicalize(&candidate)
        .with_context(|| format!("canonicalize executable '{}'", candidate.display()))
        .map(|path| path.to_string_lossy().to_string())
}

fn locate_host_program(program: &str) -> anyhow::Result<PathBuf> {
    let candidate = PathBuf::from(program);
    if candidate.is_absolute() {
        if host_executable_file(&candidate) {
            return Ok(candidate);
        }
        bail!("executable is unavailable: {}", candidate.display())
    }
    if program.contains(std::path::MAIN_SEPARATOR) || program.trim().is_empty() {
        bail!("program must be an absolute path or a bare executable name")
    }
    let path = std::env::var_os("PATH").context("PATH is unavailable")?;
    for directory in std::env::split_paths(&path) {
        let candidate = directory.join(program);
        if host_executable_file(&candidate) {
            return if candidate.is_absolute() {
                Ok(candidate)
            } else {
                Ok(std::env::current_dir()
                    .context("resolve current directory for relative PATH entry")?
                    .join(candidate))
            };
        }
    }
    bail!("executable was not found on Host PATH")
}

fn host_executable_file(candidate: &Path) -> bool {
    let Ok(metadata) = candidate.metadata() else {
        return false;
    };
    if !metadata.is_file() {
        return false;
    }
    #[cfg(unix)]
    {
        use std::os::unix::fs::PermissionsExt;
        metadata.permissions().mode() & 0o111 != 0
    }
    #[cfg(not(unix))]
    {
        true
    }
}

async fn resolve_model(
    config: &OrchestralConfig,
) -> anyhow::Result<(
    orchestral_core::config::BackendSpec,
    Option<ModelProfile>,
    String,
    f32,
)> {
    let profile = config
        .agent
        .model_profile
        .as_deref()
        .map(|name| {
            config
                .providers
                .get_model(name)
                .with_context(|| format!("model profile not found: {name}"))
        })
        .transpose()?;
    let backend_name = config
        .agent
        .backend
        .clone()
        .or_else(|| profile.as_ref().map(|profile| profile.backend.clone()))
        .or_else(|| config.providers.default_backend.clone());
    let backend = match backend_name {
        Some(name) => config
            .providers
            .get_backend(&name)
            .with_context(|| format!("model backend not found: {name}"))?,
        None => config
            .providers
            .get_default_backend()
            .context("no model backend configured")?,
    };
    let model = config
        .agent
        .model
        .clone()
        .or_else(|| profile.as_ref().map(|profile| profile.model.clone()));
    let (backend, model) = crate::openai_connection::resolve_model(backend, model).await?;
    let candidate = config
        .agent
        .temperature
        .or_else(|| profile.as_ref().and_then(|profile| profile.temperature))
        .unwrap_or(0.2);
    let temperature = profile
        .as_ref()
        .map(|profile| profile.clamp_temperature(candidate))
        .unwrap_or(candidate);
    Ok((backend, profile, model, temperature))
}

pub(crate) fn build_model_backend(
    backend: &BackendSpec,
    model: &str,
    temperature: f32,
    profile: Option<&ModelProfile>,
    max_buffered_events: usize,
    credential_file: Option<&std::path::Path>,
    reasoning: orchestral_core::config::ReasoningPreference,
) -> anyhow::Result<(Arc<dyn ModelBackend>, Arc<dyn ModelTokenMeter>)> {
    let reasoning_control = crate::model_controls::openai_reasoning(backend, reasoning)?;
    let max_output_tokens = profile
        .and_then(|profile| profile.max_tokens)
        .unwrap_or(8_192) as u64;
    let timeout = Duration::from_secs(
        backend
            .get_config("stream_idle_timeout_secs")
            .or_else(|| backend.get_config("timeout_secs"))
            .unwrap_or(300),
    );
    let max_context_tokens = backend.get_config("max_context_tokens");
    match backend.kind.trim().to_ascii_lowercase().as_str() {
        "google" | "gemini" => {
            let auth_mode = backend
                .get_config::<String>("auth")
                .unwrap_or_else(|| "auto".to_owned())
                .trim()
                .to_ascii_lowercase();
            if !matches!(auth_mode.as_str(), "auto" | "api_key" | "adc") {
                bail!(
                    "unsupported Google auth mode '{auth_mode}'; expected auto, api_key, or adc"
                );
            }
            let explicit_adc = auth_mode == "adc"
                || google_adc_is_explicitly_requested(credential_file, backend);
            let api_key = (auth_mode != "adc")
                .then(|| backend.resolve_api_key().ok())
                .flatten();
            let vertex_plan = if explicit_adc || api_key.is_none() {
                resolve_google_vertex_auth(credential_file, backend)?
            } else {
                None
            };
            let (authentication, endpoint) = if let Some(plan) = vertex_plan {
                let provider = match &plan.source {
                    GoogleCredentialSource::ServiceAccountFile(path) => {
                        GoogleCloudAccessTokenProvider::from_service_account_file(path)
                    }
                    GoogleCredentialSource::ApplicationDefault => {
                        GoogleCloudAccessTokenProvider::application_default()
                    }
                }
                .context("initialize Google Cloud authentication")?;
                (
                    GeminiAuthentication::AccessTokenProvider(Arc::new(provider)),
                    backend.endpoint.clone().unwrap_or_else(|| plan.endpoint()),
                )
            } else if let Some(api_key) = api_key {
                (
                    GeminiAuthentication::ApiKey(api_key),
                    backend.endpoint.clone().unwrap_or_else(|| {
                        "https://generativelanguage.googleapis.com/v1beta".to_owned()
                    }),
                )
            } else {
                bail!(
                    "no Google credentials found for backend '{}'; use --credential-file, \
                     GOOGLE_APPLICATION_CREDENTIALS, `gcloud auth application-default login`, \
                     or GOOGLE_API_KEY",
                    backend.name
                );
            };
            let backend = Arc::new(
                GeminiModelBackend::new(GeminiModelConfig {
                    backend_id: format!("google-gemini/{}", backend.name),
                    endpoint,
                    authentication,
                    model: model.to_owned(),
                    temperature,
                    thinking_level: None,
                    default_max_output_tokens: max_output_tokens,
                    max_context_tokens,
                    timeout,
                    max_buffered_events,
                })
                .context("build Gemini ModelBackend")?,
            );
            Ok((backend.clone(), backend))
        }
        "openai" | "openrouter" | "deepseek" | "groq" | "xai" | "mistral" => {
            let sampling = profile
                .and_then(|profile| profile.config.get("sampling"))
                .map(|value| serde_json::from_value::<OpenAiSamplingConfig>(value.clone()))
                .transpose()
                .context("parse model profile config.sampling")?
                .unwrap_or_default();
            let tool_result_format = profile
                .and_then(|profile| profile.config.get("tool_result_format"))
                .map(|value| serde_json::from_value::<OpenAiToolResultFormat>(value.clone()))
                .transpose()
                .context("parse model profile config.tool_result_format")?
                .unwrap_or_default();
            let api_key = crate::openai_connection::api_key(backend)?;
            let endpoint = crate::openai_connection::endpoint(backend)?.base_url().to_owned();
            let backend = Arc::new(
                OpenAiCompatibleBackend::new(OpenAiCompatibleConfig {
                    backend_id: format!("openai-compatible/{}", backend.name),
                    endpoint,
                    api_key,
                    model: model.to_owned(),
                    temperature,
                    default_max_output_tokens: max_output_tokens,
                    max_context_tokens,
                    timeout,
                    structured_output: backend
                        .get_config("structured_output")
                        .unwrap_or(true),
                    max_buffered_events,
                })
                .context("build OpenAI-compatible ModelBackend")?
                .with_sampling(sampling)
                .context("configure OpenAI-compatible sampling")?
                .with_tool_result_format(tool_result_format)
                .with_reasoning_control(reasoning_control),
            );
            Ok((backend.clone(), backend))
        }
        kind => bail!(
            "unsupported ModelBackend kind '{kind}'; supported protocol families are OpenAI-compatible and Gemini Native"
        ),
    }
}

async fn start_or_continue_turn(
    client: &AgentClient,
    resumed: Option<orchestral_runtime::AgentRunHandle>,
    input: String,
    turn: u64,
) -> anyhow::Result<(orchestral_runtime::AgentRunHandle, Option<RequestId>)> {
    if let Some(handle) = resumed {
        let view = handle.inspect().await?;
        if !view.state.is_terminal() {
            let submitted_request = view
                .pending_requests
                .iter()
                .find(|request| matches!(request.payload, PendingRequestPayload::Input { .. }))
                .map(|request| request.request_id.clone());
            let ack = if let Some(request_id) = &submitted_request {
                handle
                    .resolve_input_text(request_id.clone(), input.clone())
                    .await
            } else {
                handle.steer_text(input.clone()).await
            };
            match ack {
                Ok(ack)
                    if matches!(
                        ack.state,
                        CommandAckState::Accepted { .. } | CommandAckState::Applied { .. }
                    ) =>
                {
                    return Ok((handle, submitted_request))
                }
                Ok(ack) => {
                    if !matches!(
                        ack.state,
                        CommandAckState::Rejected { .. } | CommandAckState::Unsupported { .. }
                    ) || !handle.inspect().await?.state.is_terminal()
                    {
                        bail!(
                            "resumed Run did not accept the follow-up input: {:?}",
                            ack.state
                        );
                    }
                }
                Err(error) => return Err(error.into()),
            }
            // A recovered Run can finish while input is offered. Only an
            // explicit rejection permits a new Run; an uncertain command
            // result must never cause duplicate input.
        }
    }
    client
        .start_with_run_id(
            RunId::new(unique_id("cli-run", turn)),
            vec![Content::text(input)],
        )
        .await
        .context("start Agent Run")
        .map(|handle| (handle, None))
}

async fn run_turn(
    client: &AgentClient,
    approval_broker: &Arc<InMemoryHostApprovalBroker>,
    turn: u64,
    input: String,
    lines: &mut tokio::io::Lines<BufReader<tokio::io::Stdin>>,
    accept_unsolicited_stdin: bool,
    resumed: Option<orchestral_runtime::AgentRunHandle>,
) -> anyhow::Result<()> {
    let (handle, submitted_request) = start_or_continue_turn(client, resumed, input, turn).await?;
    let run_id = handle.run_id().clone();
    let mut events = handle.subscribe().await.context("subscribe to Agent Run")?;
    // Accepted input may precede the durable RequestResolved projection. Treat
    // the initial resume answer like later answers, instead of prompting again
    // (and cancelling on closed stdin) while that same command is being applied.
    let mut handled_requests = submitted_request.into_iter().collect::<BTreeSet<_>>();
    let mut stdin_open = true;
    let view = loop {
        let view = handle.inspect().await.context("inspect Agent Run")?;
        if is_terminal(view.state.status()) {
            break view;
        }
        if let Some(request) = view
            .pending_requests
            .iter()
            .find(|request| !handled_requests.contains(&request.request_id))
            .cloned()
        {
            handled_requests.insert(request.request_id.clone());
            if !resolve_cli_request(
                client.controller(),
                approval_broker,
                &run_id,
                request,
                lines,
            )
            .await?
            {
                continue;
            }
            continue;
        }
        tokio::select! {
            event = events.recv() => match event {
                Ok(AgentControlEvent::Telemetry(telemetry)) => match telemetry.payload {
                    AgentTelemetry::OutputDelta { .. } => {}
                    AgentTelemetry::ProgressReported { message, fraction } => {
                        if let Some(fraction) = fraction {
                            eprintln!("\n[{:.0}%] {message}", fraction * 100.0);
                        } else {
                            eprintln!("\n{message}");
                        }
                    }
                    _ => {}
                },
                Ok(AgentControlEvent::Durable(_))
                | Err(tokio::sync::broadcast::error::RecvError::Lagged(_)) => {}
                Ok(_) => {}
                Err(tokio::sync::broadcast::error::RecvError::Closed) => {
                    bail!("Agent event stream closed before terminal state")
                }
            },
            signal = tokio::signal::ctrl_c() => {
                signal.context("listen for Ctrl-C")?;
                eprintln!("\nCancelling current Agent Run...");
                if let Err(error) = handle.cancel("CLI interrupted by user").await {
                    tracing::debug!(%error, "Agent Run reached terminal while cancellation was sent");
                }
            }
            line = lines.next_line(), if stdin_open && accept_unsolicited_stdin => {
                let line = line.context("read running Agent input")?;
                let Some(line) = line else {
                    stdin_open = false;
                    continue;
                };
                let input = line.trim();
                if input.is_empty() {
                    continue;
                }
                if matches!(input, "/cancel" | "/stop") {
                    if let Err(error) = handle.cancel("CLI cancellation command").await {
                        tracing::debug!(%error, "Agent Run reached terminal while cancellation was sent");
                    }
                    continue;
                }
                let ack = handle
                    .steer_text(input.to_owned())
                    .await
                    .context("steer running Agent")?;
                match ack.state {
                    CommandAckState::Accepted { .. } | CommandAckState::Applied { .. } => {
                        eprintln!("\n[steer accepted]");
                    }
                    CommandAckState::Rejected { code, message, .. } => {
                        eprintln!("\n[steer rejected: {code:?}: {message}]");
                    }
                    CommandAckState::Unsupported { feature, .. } => {
                        eprintln!("\n[steer unsupported: {feature}]");
                    }
                    _ => eprintln!("\n[steer acknowledgement pending]"),
                }
            }
        }
    };

    match &view.state {
        AgentRunState::Terminal {
            terminal: AgentTerminalState::Delivered { .. },
        } => {
            let delivery = view
                .delivery
                .context("Delivered Run omitted its Delivery")?;
            print_content(&delivery.final_response.body)?;
        }
        AgentRunState::Terminal {
            terminal: AgentTerminalState::Failed { failure },
        } => {
            bail!(
                "Agent Run failed [{}]{}: {}",
                failure.code,
                if failure.retryable {
                    " (retryable)"
                } else {
                    ""
                },
                failure.message
            )
        }
        AgentRunState::Terminal {
            terminal: AgentTerminalState::Incomplete { reason },
        } => bail!("Agent Run incomplete: {reason:?}"),
        AgentRunState::Terminal {
            terminal: AgentTerminalState::Cancelled { reason },
        } => bail!("Agent Run cancelled: {reason}"),
        state => bail!("Agent Run stopped in unexpected state: {state:?}"),
    }
    Ok(())
}

async fn resolve_cli_request(
    controller: &Arc<AgentController>,
    approval_broker: &Arc<InMemoryHostApprovalBroker>,
    run_id: &RunId,
    request: PendingRequest,
    lines: &mut tokio::io::Lines<BufReader<tokio::io::Stdin>>,
) -> anyhow::Result<bool> {
    let resolution = match &request.payload {
        PendingRequestPayload::Input { prompt, .. } => {
            eprintln!("\nInput required:");
            for item in prompt {
                eprintln!("{}", display_content(item));
            }
            eprint!("> ");
            io::stderr().flush().context("flush input prompt")?;
            let answer = tokio::select! {
                line = lines.next_line() => line.context("read requested Agent input")?,
                signal = tokio::signal::ctrl_c() => {
                    signal.context("listen for Ctrl-C")?;
                    eprintln!("\nCancelling current Agent Run...");
                    if let Err(error) = controller.cancel(run_id, "CLI interrupted during input request").await {
                        tracing::debug!(%error, "Agent Run reached terminal while cancellation was sent");
                    }
                    return Ok(false);
                }
            };
            let Some(answer) = answer else {
                controller
                    .cancel(run_id, "CLI input closed during input request")
                    .await
                    .context("cancel Agent after stdin closed")?;
                return Ok(false);
            };
            if answer.trim().is_empty() {
                bail!("input response must not be empty")
            }
            RequestResolution::Input {
                content: vec![Content::text(answer)],
            }
        }
        PendingRequestPayload::Approval {
            requested_scope,
            session_approval_scope,
            reason,
            ..
        } => {
            if let Some(grant_ref) = approval_broker
                .approve_if_remembered(&request.request_id, approval_expiry_ms())
                .context("apply remembered Host approval")?
            {
                RequestResolution::Approval {
                    decision: ApprovalDecision::Allow,
                    grant_ref: Some(grant_ref),
                }
            } else {
                eprintln!("\nApproval required: {reason}");
                eprintln!("Effects: {}", requested_scope.join(", "));
                if session_approval_scope.is_some() {
                    eprint!("Approve? [y] once / [a] this session / [N] deny ");
                } else {
                    eprint!("Allow this exact operation? [y/N] ");
                }
                io::stderr().flush().context("flush approval prompt")?;
                let answer = tokio::select! {
                    line = lines.next_line() => line.context("read approval decision")?,
                    signal = tokio::signal::ctrl_c() => {
                        signal.context("listen for Ctrl-C")?;
                        eprintln!("\nCancelling current Agent Run...");
                        if let Err(error) = controller.cancel(run_id, "CLI interrupted during approval").await {
                            tracing::debug!(%error, "Agent Run reached terminal while cancellation was sent");
                        }
                        return Ok(false);
                    }
                };
                let answer = answer
                    .as_deref()
                    .map(str::trim)
                    .map(str::to_ascii_lowercase)
                    .unwrap_or_default();
                let approve_session = session_approval_scope.is_some()
                    && matches!(answer.as_str(), "a" | "always" | "session");
                let approve_once = matches!(answer.as_str(), "y" | "yes");
                if approve_session || approve_once {
                    let grant_ref = if approve_session {
                        approval_broker
                            .approve_for_session(&request.request_id, approval_expiry_ms())
                            .context("remember Host approval for session")?
                    } else {
                        approval_broker
                            .approve(&request.request_id, approval_expiry_ms())
                            .context("issue exact Host approval grant")?
                    };
                    RequestResolution::Approval {
                        decision: ApprovalDecision::Allow,
                        grant_ref: Some(grant_ref),
                    }
                } else {
                    RequestResolution::Approval {
                        decision: ApprovalDecision::Deny,
                        grant_ref: None,
                    }
                }
            }
        }
        PendingRequestPayload::ExternalAction { .. } => {
            bail!("CLI does not support external action requests")
        }
        _ => bail!("CLI does not support this pending request kind"),
    };
    let command = AgentCommandEnvelope::new(
        CommandId::new(unique_id("cli-request", 0)),
        run_id.clone(),
        Some(request.request_id),
        AgentCommand::ResolveRequest {
            response: resolution,
        },
    )
    .context("build request resolution command")?;
    let ack = controller
        .command(command)
        .await
        .context("resolve Agent request")?;
    match ack.state {
        CommandAckState::Accepted { .. } | CommandAckState::Applied { .. } => Ok(true),
        CommandAckState::Rejected { code, message, .. } => {
            bail!("request resolution was rejected ({code:?}): {message}")
        }
        CommandAckState::Unsupported { feature, .. } => {
            bail!("request resolution is unsupported: {feature}")
        }
        _ => bail!("request resolution returned an unknown acknowledgement state"),
    }
}

fn is_terminal(status: AgentRunStatus) -> bool {
    matches!(
        status,
        AgentRunStatus::Delivered
            | AgentRunStatus::Incomplete
            | AgentRunStatus::Cancelled
            | AgentRunStatus::Failed
    )
}

fn print_content(body: &ContentBody) -> anyhow::Result<()> {
    match body {
        ContentBody::Inline(serde_json::Value::String(text)) => println!("{text}"),
        ContentBody::Inline(value) => println!("{}", serde_json::to_string_pretty(value)?),
        ContentBody::Artifact(artifact) => {
            println!("{}", serde_json::to_string_pretty(artifact)?)
        }
        other => println!("{}", serde_json::to_string_pretty(other)?),
    }
    Ok(())
}

fn display_content(content: &Content) -> String {
    match &content.body {
        ContentBody::Inline(serde_json::Value::String(text)) => text.clone(),
        body => serde_json::to_string(body).unwrap_or_else(|_| "<unprintable content>".to_owned()),
    }
}

fn unique_id(prefix: &str, sequence: u64) -> String {
    let epoch_nanos = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .unwrap_or_default()
        .as_nanos();
    format!("{prefix}-{}-{epoch_nanos}-{sequence}", std::process::id())
}

fn approval_expiry_ms() -> i64 {
    let now_ms = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .unwrap_or_default()
        .as_millis() as i64;
    now_ms.saturating_add(5 * 60 * 1_000)
}

fn select_entry_mode(
    input: Option<String>,
    stdin_is_terminal: bool,
    stdout_is_terminal: bool,
) -> anyhow::Result<EntryMode> {
    if let Some(input) = input {
        return Ok(EntryMode::HeadlessPrompt(input));
    }
    if !stdin_is_terminal {
        return Ok(EntryMode::HeadlessPipe);
    }
    if stdout_is_terminal {
        return Ok(EntryMode::Tui);
    }
    bail!(
        "interactive mode requires a TTY on stdout; pass a prompt or pipe stdin for Headless mode"
    )
}

#[cfg(test)]
#[path = "agent/tool_capability_tests.rs"]
mod tool_capability_tests;

#[cfg(test)]
#[path = "agent/tool_output_budget_tests.rs"]
mod tool_output_budget_tests;

#[cfg(test)]
mod entry_mode_tests {
    use std::path::PathBuf;

    use super::{select_entry_mode, unique_id, CliWorkspaceSet, EntryMode, InputMode};

    #[test]
    fn input_mode_uses_the_reply_channel_not_headless_output() {
        let prompt = EntryMode::HeadlessPrompt("inspect the project".to_owned());
        assert!(!InputMode::Auto.resolve(&prompt, false).unwrap());
        assert!(InputMode::Auto.resolve(&prompt, true).unwrap());
        assert!(InputMode::Auto.resolve(&EntryMode::Tui, true).unwrap());
        assert!(InputMode::Interactive.resolve(&prompt, false).unwrap());
        assert!(!InputMode::None.resolve(&EntryMode::Tui, true).unwrap());
        assert!(!InputMode::Auto
            .resolve(&EntryMode::HeadlessPipe, false)
            .unwrap());
        assert!(InputMode::Interactive
            .resolve(&EntryMode::HeadlessPipe, false)
            .is_err());
    }

    #[test]
    fn model_profile_sampling_is_validated_before_connecting() {
        let backend = serde_json::from_value(serde_json::json!({
            "name": "local", "kind": "openai", "endpoint": "http://127.0.0.1:1/v1",
            "config": {"auth": "none"},
        }))
        .unwrap();
        for sampling in [
            serde_json::json!({"top_k": "20"}),
            serde_json::json!({"topk": 20}),
            serde_json::json!({"repetition_penalty": 0}),
        ] {
            let profile = serde_json::from_value(serde_json::json!({
                "name": "local", "backend": "local", "model": "local-model",
                "config": {"sampling": sampling},
            }))
            .unwrap();
            let error = super::build_model_backend(
                &backend,
                "local-model",
                0.6,
                Some(&profile),
                8,
                None,
                Default::default(),
            )
            .err()
            .expect("invalid model sampling must fail before HTTP");
            assert!(format!("{error:#}").contains("sampling"));
        }
    }

    #[test]
    fn model_profile_tool_result_format_defaults_to_text_and_is_strict() {
        let backend = serde_json::from_value(serde_json::json!({
            "name": "local", "kind": "openai", "endpoint": "http://127.0.0.1:1/v1",
            "config": {"auth": "none"},
        }))
        .unwrap();
        let profile = |format| {
            serde_json::from_value(serde_json::json!({
                "name": "local", "backend": "local", "model": "local-model",
                "config": {"tool_result_format": format},
            }))
            .unwrap()
        };
        for format in [
            serde_json::json!("unsupported"),
            serde_json::json!(true),
            serde_json::Value::Null,
        ] {
            let error = super::build_model_backend(
                &backend,
                "local-model",
                0.6,
                Some(&profile(format)),
                8,
                None,
                Default::default(),
            )
            .err()
            .expect("invalid result format must fail before HTTP");
            assert!(format!("{error:#}").contains("tool_result_format"));
        }
        let (_, json_meter) = super::build_model_backend(
            &backend,
            "local-model",
            0.6,
            Some(&profile(serde_json::json!("json"))),
            8,
            None,
            Default::default(),
        )
        .unwrap();
        let (_, yaml_meter) = super::build_model_backend(
            &backend,
            "local-model",
            0.6,
            Some(&profile(serde_json::json!("yaml"))),
            8,
            None,
            Default::default(),
        )
        .unwrap();
        assert_ne!(json_meter.meter_descriptor(), yaml_meter.meter_descriptor());
        let (_, text_meter) = super::build_model_backend(
            &backend,
            "local-model",
            0.6,
            Some(&profile(serde_json::json!("text"))),
            8,
            None,
            Default::default(),
        )
        .unwrap();
        assert_ne!(text_meter.meter_descriptor(), yaml_meter.meter_descriptor());
        assert_ne!(text_meter.meter_descriptor(), json_meter.meter_descriptor());
        let (_, default_meter) = super::build_model_backend(
            &backend,
            "local-model",
            0.6,
            None,
            8,
            None,
            Default::default(),
        )
        .unwrap();
        assert_eq!(
            default_meter.meter_descriptor(),
            text_meter.meter_descriptor()
        );
        let mut omitted_format = profile(serde_json::json!("json"));
        omitted_format
            .config
            .as_object_mut()
            .unwrap()
            .remove("tool_result_format");
        let (_, omitted_meter) = super::build_model_backend(
            &backend,
            "local-model",
            0.6,
            Some(&omitted_format),
            8,
            None,
            Default::default(),
        )
        .unwrap();
        assert_eq!(
            omitted_meter.meter_descriptor(),
            text_meter.meter_descriptor()
        );
    }

    #[cfg(unix)]
    #[test]
    fn disabling_sandboxed_execution_requires_a_host_execution_ceiling() {
        let mut config = orchestral_core::config::OrchestralConfig::default();
        config.tools.exec.enabled = true;
        config.tools.exec.sandboxed_execution_enabled = false;
        let error = super::configured_exec_host(&config)
            .err()
            .expect("invalid Host policy");
        assert!(error.to_string().contains("allow_host_execution=true"));
        config.tools.exec.allow_host_execution = true;
        config.tools.exec.shell = Some("/bin/sh".to_owned());
        assert!(super::configured_exec_host(&config).unwrap().is_some());
    }

    #[cfg(unix)]
    #[test]
    fn command_temp_root_is_stable_external_and_short_enough_for_unix_sockets() {
        let workspace = std::env::temp_dir().join(unique_id("temp-root-workspace", 0));
        std::fs::create_dir(&workspace).unwrap();
        let workspace = std::fs::canonicalize(workspace).unwrap();
        let roots = std::collections::BTreeSet::from([workspace.to_string_lossy().into_owned()]);
        let path = super::command_runtime_temp_root(&roots).unwrap();
        assert_eq!(path, super::command_runtime_temp_root(&roots).unwrap());
        assert!(!path.starts_with(&workspace));
        let manager =
            orchestral_runtime::ProcessSupervisor::new_with_runtime_temp_root(1024, &path).unwrap();
        // Reserve one opaque Run component and one child tool's temp directory.
        let child = manager
            .runtime_temp_root()
            .join("a".repeat(32))
            .join(".tmpabcdef");
        std::fs::create_dir_all(&child).unwrap();
        let listener = std::os::unix::net::UnixListener::bind(child.join("control.sock"))
            .expect("Host prefix must leave room for a Run and child tool's Unix socket");
        drop(listener);
        drop(manager);
        std::fs::remove_dir_all(path).unwrap();
        std::fs::remove_dir(workspace).unwrap();
    }

    #[test]
    fn explicit_prompt_is_always_headless() {
        assert_eq!(
            select_entry_mode(Some("fix it".to_owned()), true, true).unwrap(),
            EntryMode::HeadlessPrompt("fix it".to_owned())
        );
        assert_eq!(
            select_entry_mode(Some("fix it".to_owned()), false, false).unwrap(),
            EntryMode::HeadlessPrompt("fix it".to_owned())
        );
    }

    #[test]
    fn pipe_is_headless_and_interactive_ttys_use_tui() {
        assert_eq!(
            select_entry_mode(None, false, true).unwrap(),
            EntryMode::HeadlessPipe
        );
        assert_eq!(select_entry_mode(None, true, true).unwrap(), EntryMode::Tui);
    }

    #[test]
    fn interactive_stdin_without_terminal_output_is_rejected() {
        let error = select_entry_mode(None, true, false).unwrap_err();
        assert!(error.to_string().contains("requires a TTY on stdout"));
    }

    #[test]
    fn additional_workspace_paths_are_resolved_from_the_primary_workspace() {
        let root = std::env::temp_dir().join(unique_id("orchestral-workspace-set-test", 0));
        let primary = root.join("primary");
        let additional = primary.join("../shared");
        std::fs::create_dir_all(&primary).unwrap();
        std::fs::create_dir_all(&additional).unwrap();

        let workspaces =
            CliWorkspaceSet::resolve(Some(&primary), &[PathBuf::from("../shared")]).unwrap();
        assert_eq!(workspaces.primary, std::fs::canonicalize(&primary).unwrap());
        assert_eq!(
            workspaces.additional,
            [std::fs::canonicalize(&additional).unwrap()]
        );

        std::fs::remove_dir_all(root).unwrap();
    }
}