locode_exec/
run.rs

1//! Session assembly + drive (plan §3.2): resolve inputs, build host/pack/
2//! provider through the facade, run the engine, emit per `--output-format`.
3
4use std::io::Read;
5use std::process::ExitCode;
6use std::sync::Arc;
7
8use locode_core::{
9    CacheHint, EngineConfig, EventSink, FnSink, Host, HostConfig, NullSink, PackContext,
10    PathPolicy, ProviderInit, ProviderRegistry, SamplingArgs, Session, grok,
11};
12
13use crate::cli::{Cli, Harness, OutputFormat};
14use crate::output;
15
16/// A pre-run failure (config/setup — no report exists yet): stderr + exit 1.
17pub struct PreRunError(pub String);
18
19impl<E: std::fmt::Display> From<E> for PreRunError {
20    fn from(e: E) -> Self {
21        PreRunError(e.to_string())
22    }
23}
24
25/// Build and drive one session; returns the process exit code.
26///
27/// Every terminal state of a *started* run yields a report (the engine's
28/// `run()` is infallible) — only pre-run setup can fail here.
29///
30/// # Errors
31/// [`PreRunError`] on config/setup failures before a run exists (bad `--cwd`,
32/// unknown/misconfigured provider, empty prompt): stderr + exit 1, nothing on
33/// stdout.
34pub async fn run(cli: Cli, providers: &ProviderRegistry) -> Result<ExitCode, PreRunError> {
35    // ---- 0. SIGTERM handler (ADR-0018, Task 21): installed before any
36    //         pre-run work so a pre-run SIGTERM exits 1 cleanly; armed with
37    //         the session's cancel handle once one exists. ----
38    #[cfg(unix)]
39    let cancel_slot = crate::signal::install_sigterm();
40
41    // ---- 1. Prompt: positional, or stdin when absent / `-`. ----
42    let prompt = resolve_prompt(cli.prompt.as_deref())?;
43
44    // ---- 2. Workspace root: canonicalize FIRST, then hand the SAME canonical
45    //         path to the host (jail root), the engine (cwd), and the pack
46    //         (prompt context) — they must agree (STATUS concern #7). ----
47    let cwd = match cli.cwd {
48        Some(dir) => dir,
49        None => std::env::current_dir()?,
50    };
51    let cwd = std::fs::canonicalize(&cwd)
52        .map_err(|e| PreRunError(format!("--cwd {}: {e}", cwd.display())))?;
53
54    let mut host_config = HostConfig::new(&cwd);
55    if cli.dangerously_skip_permissions {
56        host_config.path_policy = PathPolicy::Unrestricted;
57    }
58    let host = Arc::new(Host::new(host_config)?);
59
60    // ---- 3. Pack: tools + preamble (system prompt + <user_info> prefix). ----
61    let pack = locode_core::resolve(cli.harness.as_str())?;
62    let registry = pack.build_registry(&host);
63    let pack_ctx = PackContext {
64        cwd: cwd.clone(),
65        os: std::env::consts::OS.to_string(),
66        shell: std::env::var("SHELL").unwrap_or_else(|_| "/bin/sh".to_string()),
67        date: chrono::Local::now().format("%Y-%m-%d").to_string(),
68        headless: true,
69        strip_identity: cli.strip_identity,
70    };
71    let preamble = pack.preamble(&pack_ctx);
72
73    // The grok system prompt directs the model to the <user_query> tag — wrap
74    // the prompt the way the harness expects (pack-specific shaping).
75    let user_prompt = match cli.harness {
76        Harness::Grok => grok::prompt::user_query(&prompt),
77    };
78
79    // ---- 4. Provider: registry-resolved (ADR-0015); unknown names and
80    //         factory failures (missing env, …) fail BEFORE driving the loop. ----
81    let session_id = new_session_id();
82    let built = providers
83        .build(
84            &cli.api_schema,
85            &ProviderInit {
86                session_id: session_id.clone(),
87            },
88        )
89        .map_err(|e| PreRunError(e.to_string()))?;
90    let (provider, model) = (built.provider, built.model);
91
92    // ---- 5. Engine config + event sink per output mode. ----
93    let config = EngineConfig {
94        session_id,
95        harness: cli.harness.as_str().to_string(),
96        api_schema: provider.api_schema().to_string(),
97        model,
98        cwd: cwd.clone(),
99        workspace_root: cwd,
100        max_turns: cli.max_turns,
101        sampling_args: SamplingArgs::default(),
102        cache_hint: CacheHint::Standard,
103        ..EngineConfig::default()
104    };
105    let sink: Box<dyn EventSink> = match cli.output_format {
106        // stream-json writes each event live; the terminal `result` event
107        // carries the same Report as json mode.
108        OutputFormat::StreamJson => Box::new(FnSink(|event| output::write_json_line(&event))),
109        // json/text only want the final report — events are dropped.
110        OutputFormat::Json | OutputFormat::Text => Box::new(NullSink),
111    };
112
113    // ---- 6. Drive to a terminal state (infallible) and emit the artifact. ----
114    let mut session = Session::new(provider, registry, preamble, config, sink);
115    #[cfg(unix)]
116    crate::signal::arm(&cancel_slot, session.cancel_handle());
117    let report = session.run_text(user_prompt).await;
118
119    match cli.output_format {
120        OutputFormat::Json => output::write_json_line(&report),
121        OutputFormat::Text => output::write_text(report.final_message.as_deref().unwrap_or("")),
122        OutputFormat::StreamJson => {} // the result event already streamed
123    }
124    Ok(output::exit_code(report.status))
125}
126
127/// Positional prompt, or stdin when absent / `-` (Codex's convention;
128/// positional XOR stdin per the plan addendum). Empty → usage error.
129fn resolve_prompt(arg: Option<&str>) -> Result<String, PreRunError> {
130    let prompt = match arg {
131        Some("-") | None => {
132            let mut buf = String::new();
133            std::io::stdin().read_to_string(&mut buf)?;
134            buf
135        }
136        Some(text) => text.to_string(),
137    };
138    let prompt = prompt.trim().to_string();
139    if prompt.is_empty() {
140        return Err(PreRunError(
141            "no prompt: pass it as the positional argument or on stdin".to_string(),
142        ));
143    }
144    Ok(prompt)
145}
146
147/// A unique-enough session id for a headless run (no uuid dep in v0).
148fn new_session_id() -> String {
149    let now = std::time::SystemTime::now()
150        .duration_since(std::time::UNIX_EPOCH)
151        .map_or(0, |d| d.as_millis());
152    format!("sess-{now}-{}", std::process::id())
153}
locode_exec/run.rs

locode_exec/
run.rs
