basis-cli 0.12.3

//! The attended one-shot: the JSONL stream, and the prompt `-` reads from
//! stdin.
//!
//! This is [`Route::Attended`](crate::route::Route::Attended) and nothing
//! else. That route is granted for `--json` alone (ADR-0020), so the stream
//! *is* the rendering here: a `run_started` line, whatever happened, and a
//! `run_finished` line, on stdout, for a consumer that reads the bookends.
//! Every other spelling mints a checkpoint and goes through
//! [`local`](crate::local) instead, which is where a person's terminal is
//! rendered to.
//!
//! Nothing is minted here, so there is no handle and nothing to `send` to —
//! the price of a run that leaves no trace, paid deliberately for the one
//! contract that predates checkpoints.

use std::{
    io::{self, Read},
    process::ExitCode,
};

use basis::{
    AllowAll, Approver, DenyAll, DenyAllGate, JsonlWriter, PreparedRun, RunProfile, RunSpec,
    Runtime, ShellAccess, Workspace,
};
use basis_host::ApprovalPolicy;

use crate::{approver::TerminalApprover, cli::RunArgs, exit::exit_code};

pub(crate) async fn execute_run(args: RunArgs) -> Result<ExitCode, String> {
    let prompt = prompt_from(args.prompt.clone())?;
    // Refused before anything else happens: `Workspace::open` resolves a
    // provider and spawns every server `.mcp.json` names, and a refusal that
    // has already spawned processes is not a refusal. The free functions make
    // the same check first, for the same reason.
    if prompt.trim().is_empty() {
        return Err("prompt is empty".to_string());
    }

    // Built early, before the fields below are moved out of `args` one by
    // one: this borrows the whole struct, and Rust refuses that once any
    // field has been partially moved.
    let spec = run_spec(&args, prompt);

    let workspace = match args.workspace {
        Some(path) => path,
        None => {
            std::env::current_dir().map_err(|error| format!("no working directory: {error}"))?
        }
    };

    // The same per-workspace store dir `basis spawn`'s attach path resolves
    // (`basis_tasks::Tasks::store_dir`, G4 of the whole-wave review before the
    // task layer's extraction into its own crate) — without it this route
    // fell back to mentra's process-cwd default, so a `basis "<prompt>"` and a
    // `basis spawn ...` against the same repository saw two conversation
    // stores and two memory roots instead of one. Resolved before the runtime
    // is built for the same reason `Workspace::open` resolves memory before
    // acquiring one: a directory a run's history depends on should fail
    // loudly if it cannot be established, not be silently substituted.
    let store_dir = basis_tasks::Tasks::store_dir(&workspace)
        .map_err(|error| format!("open data directory: {error}"))?;

    // The process half seeds the private runtime the workspace builds
    // (ADR-0018): which provider answers, and at which endpoint.
    let runtime = Runtime::builder().with_store_dir(store_dir);
    let builder = Workspace::builder(workspace);
    let (runtime, builder) = basis_tasks::configure_builders(
        runtime,
        builder,
        args.provider.as_deref(),
        args.base_url.as_deref(),
        args.model.as_deref(),
        ShellAccess::from_flag(!args.no_shell),
    )
    .map_err(|error| error.to_string())?;
    let builder = builder.with_runtime_builder(runtime);

    // Every consequential call is put to this one, and — bar the refusal the
    // gate below states first — nothing else decides: `always` allows, `never`
    // refuses, `prompt` asks the person.
    let approver = approver(args.approve);

    // Held open past the mint: the workspace's hooks and MCP connections live
    // exactly as long as it does, and the turn below still needs them.
    let workspace = builder.open().await.map_err(|error| error.to_string())?;

    // The stream is the whole output: every fact a caller could want — the
    // outcome, the bound that tripped, the failure's words — is a line on it,
    // so there is nothing left for this function to say afterwards.
    let run = workspace.prepare(spec).map_err(|error| error.to_string())?;
    let mut run = gated(run, args.approve);
    let report = run
        .execute_with_approver(JsonlWriter::new(io::stdout()), approver)
        .await
        .map_err(|error| error.to_string())?;

    // This route mints and never resumes, and there is no longer a durable
    // row to worry about outliving it: mentra 0.27 remembers a "…for this
    // session" answer into `PermissionRuleScope::Process` (mentra#53), a rung
    // owned by this run's own live session and never written to the store.
    // The process exiting is what ends it; nothing here has to.
    Ok(ExitCode::from(exit_code(&report)))
}

fn approver(policy: ApprovalPolicy) -> Box<dyn Approver> {
    match policy {
        ApprovalPolicy::Always => Box::new(AllowAll),
        ApprovalPolicy::Prompt => Box::new(TerminalApprover::new()),
        ApprovalPolicy::Never => Box::new(DenyAll),
    }
}

/// The session authorizer `policy` needs over the runtime's, if any.
///
/// Only `never` needs one, and this is why: mentra resolves the runtime gate's
/// `Prompt` against the conversation's remembered rules *before* the
/// [`approver`] above is consulted, so a durable Global- or Project-scope allow
/// — seeded through the session's permission handle, and cleared by nothing
/// this route does — answers ahead of [`DenyAll`] and lets a `--approve never`
/// run write. [`DenyAllGate`] states the refusal where mentra treats it as
/// final. The two say the same thing in the same words; only the layer differs.
///
/// **`always` and `prompt` install nothing**, deliberately. Both permit
/// consequential work, so a standing allow is a host saying yes in advance
/// rather than an override of anything — and installing an authorizer
/// *replaces* whatever the runtime carries rather than layering over it, which
/// for those two would cost a bound or a posture the runtime had for no gain.
/// A refusal cannot cost either: it answers from the request alone and awaits
/// nobody, so there is nothing left to wait on and nothing to lose.
///
/// This route's policy is fixed for the run's whole life, which is what makes
/// a stateless gate right here where `basis-acp` needs a stateful one: there is
/// no mode picker, and nothing between the mint above and the turn below can
/// change the answer.
fn gated(run: PreparedRun, policy: ApprovalPolicy) -> PreparedRun {
    match policy {
        ApprovalPolicy::Never => run.with_tool_authorizer(DenyAllGate),
        ApprovalPolicy::Always | ApprovalPolicy::Prompt => run,
    }
}

/// The [`RunSpec`] one attended `spawn` invocation asks for: the CLI flags
/// that shape a run, turned into the typed builder — the same mapping
/// `local::verbs::run_spec` does for the other two routes, so the two system-
/// prompt flags mean the same thing regardless of which route reaches them.
fn run_spec(args: &RunArgs, prompt: String) -> RunSpec {
    let mut spec = RunSpec::new(prompt);
    if let Some(effort) = args.effort {
        spec = spec.with_effort(effort.into());
    }

    // A bound that trips ends the run gracefully — the stream closes the way
    // it always does and committed work is kept — so setting one costs a
    // healthy run nothing.
    if let Some(deadline) = args.deadline {
        spec = spec.with_deadline(deadline.duration());
    }
    if let Some(tool_budget) = args.tool_budget {
        spec = spec.with_tool_budget(tool_budget);
    }
    if let Some(token_budget) = args.token_budget {
        spec = spec.with_token_budget(token_budget);
    }
    // `basis::RunSpec` carries host overrides through its `RunProfile`
    // rather than a `with_system_prompt` of its own — unlike
    // `basis_tasks::RunSpec` (see `local::verbs::run_spec`), which is a
    // durable task's spawn request and owns the field directly. Both routes
    // apply the same two-flag mapping from [`cli::system_prompt`].
    if let Some(system_prompt) = crate::cli::system_prompt(
        args.system_prompt.clone(),
        args.append_system_prompt.clone(),
    ) {
        spec = spec.with_profile(RunProfile::new().with_system_prompt(system_prompt));
    }
    spec
}

/// The prompt as `spawn` (or its `run` alias) was given it, reading stdin when
/// it is `-`.
///
/// Explicit rather than detected: `basis serve --acp` owns stdin for the ACP
/// server, while `basis spawn -` owns stdin for a prompt. The caller says which
/// one this is, with the command and one character (ADR-0017).
pub(crate) fn prompt_from(argument: String) -> Result<String, String> {
    match argument.as_str() {
        "-" => read_prompt(io::stdin().lock()),
        _ => Ok(argument),
    }
}

/// Reads a whole prompt, refusing an empty one.
///
/// Whole, not a line: a generated prompt spans paragraphs, and a reader that
/// stopped at the first newline would silently run a fraction of what it was
/// given. An empty stdin is refused here rather than deeper, where the message
/// would say "prompt is empty" without saying where the prompt was looked for.
fn read_prompt(mut source: impl Read) -> Result<String, String> {
    let mut prompt = String::new();
    source
        .read_to_string(&mut prompt)
        .map_err(|error| format!("could not read the prompt from stdin: {error}"))?;

    if prompt.trim().is_empty() {
        return Err(
            "no prompt on stdin: `-` reads one from stdin, and nothing arrived".to_string(),
        );
    }

    Ok(prompt)
}

#[cfg(test)]
mod tests {
    use std::{path::Path, time::Duration};

    use super::*;

    #[test]
    fn a_dash_reads_the_prompt_from_stdin() {
        let prompt = read_prompt(&b"fix the failing test\nthen push\n"[..])
            .expect("a prompt arrived on stdin");

        assert_eq!(
            prompt, "fix the failing test\nthen push\n",
            "a multi-line prompt must arrive whole, not truncated at the first newline"
        );
    }

    #[test]
    fn an_empty_stdin_says_where_the_prompt_was_looked_for() {
        let reason = read_prompt(&b"  \n"[..]).expect_err("whitespace is not a prompt");

        assert!(
            reason.contains("stdin"),
            "the reason must name stdin: {reason}"
        );
    }

    #[test]
    fn a_prompt_that_is_not_a_dash_is_taken_as_written() {
        assert_eq!(
            prompt_from("fix the failing test".to_string()).expect("a literal prompt"),
            "fix the failing test"
        );
    }

    fn run_args(flags: &[&str]) -> RunArgs {
        use clap::Parser;

        use crate::cli::{Cli, Command};

        let mut argv = vec!["basis", "spawn", "prompt"];
        argv.extend_from_slice(flags);
        let Some(Command::Spawn(parsed)) = Cli::try_parse_from(argv).expect("parses").command
        else {
            panic!("spawn parses");
        };
        parsed
    }

    /// Pins the gap this module existed to close: `--system-prompt` and
    /// `--append-system-prompt` are declared on [`RunArgs`], but until this
    /// spec builder applied them nothing on the attended route ever read the
    /// two fields — `basis spawn --json --system-prompt ...` silently ran the
    /// workspace's own prompt instead of the caller's.
    #[test]
    fn the_replace_flag_reaches_the_minted_spec() {
        let args = run_args(&["--system-prompt", "you are Acme's reviewer"]);

        let spec = run_spec(&args, "prompt".to_string());

        let expected = RunSpec::new("prompt").with_profile(RunProfile::new().with_system_prompt(
            basis::SystemPrompt::Replace("you are Acme's reviewer".to_string()),
        ));
        assert_eq!(spec, expected);
    }

    #[test]
    fn the_append_flag_reaches_the_minted_spec() {
        let args = run_args(&["--append-system-prompt", "answer in Latin"]);

        let spec = run_spec(&args, "prompt".to_string());

        let expected = RunSpec::new("prompt").with_profile(
            RunProfile::new()
                .with_system_prompt(basis::SystemPrompt::Append("answer in Latin".to_string())),
        );
        assert_eq!(spec, expected);
    }

    #[test]
    fn neither_flag_leaves_the_spec_at_the_workspace_default() {
        let args = run_args(&[]);

        let spec = run_spec(&args, "prompt".to_string());

        assert_eq!(
            spec,
            RunSpec::new("prompt"),
            "unsaid must leave the profile empty, not force a system prompt"
        );
    }

    /// The route table is what keeps this module's contract true: reaching
    /// [`execute_run`] without `--json` would put a JSONL stream where a
    /// person expected an answer.
    #[test]
    fn the_attended_route_is_reached_only_for_the_stream_it_renders() {
        use clap::Parser;

        use crate::{
            cli::{Cli, Command},
            route::{Route, route},
        };

        for flags in [
            vec![],
            vec!["--json"],
            vec!["--await"],
            vec!["--json", "--await"],
            vec!["--resumable"],
            vec!["--detached"],
        ] {
            let mut argv = vec!["basis", "spawn", "a prompt"];
            argv.extend_from_slice(&flags);
            let Some(Command::Spawn(args)) = Cli::try_parse_from(argv).expect("parses").command
            else {
                panic!("spawn parses");
            };

            for in_task in [false, true] {
                if route(&args, in_task) == Route::Attended {
                    assert!(args.json, "attended without --json: {flags:?}");
                }
            }
        }
    }

    /// A scripted turn that writes a file: one consequential call, which is all
    /// [`gated`] has anything to say about.
    fn writing_mock(workspace: &Path) -> mentra::test::MockRuntime {
        mentra::test::MockRuntime::builder()
            .model("mock-model", "openai")
            // Not permissive, and carrying the authorizer a basis-built runtime
            // would: the `Prompt` this surfaces is exactly what a remembered
            // rule gets to answer.
            .with_policy(mentra::RuntimePolicy::workspace_bounded(workspace))
            .with_tool_authorizer(basis::ApprovalGate::new())
            .tool_calls(vec![mentra::test::MockToolCall::new(
                "files",
                serde_json::json!({
                    "operations": [{ "op": "create", "path": "made.txt", "content": "hi" }]
                }),
            )])
            .text("done")
            .build()
            .expect("mock runtime builds")
    }

    /// Prepares the scripted run against `workspace`, seeds the durable allow
    /// this route cannot clear, applies `policy` the way [`execute_run`] does,
    /// and runs it. Reports whether the write happened, and how many
    /// permission requests the run raised — the second is what tells the two
    /// tests below apart from a dead seed, since a live rule and a refusing
    /// gate both leave the write undone.
    ///
    /// The rule is **Global**-scope and written before the first turn, so it is
    /// the standing override a fixed policy has to survive: nothing here clears
    /// it — this route remembers a "…for this session" answer into mentra
    /// 0.27's process-local `Process` scope, not Global, so it could not reach
    /// this rule even if it tried — and it resolves the runtime gate's
    /// `Prompt` with no permission request ever raised.
    async fn wrote_a_file_under(policy: ApprovalPolicy, workspace: &Path) -> (bool, usize) {
        let mock = writing_mock(workspace);
        let session = mock
            .runtime()
            .create_session_with_config(
                "test",
                mock.model(),
                mentra::agent::AgentConfig {
                    workspace: mentra::agent::WorkspaceConfig {
                        base_dir: workspace.to_path_buf(),
                        ..Default::default()
                    },
                    ..Default::default()
                },
            )
            .expect("session");

        let run = basis::run::prepare_with_session(
            session,
            workspace,
            "make a file",
            &basis::ContextConfig {
                file_name: "AGENTS.md".to_string(),
                global_dir: None,
                walk_parents: false,
            },
            "openai",
            "mock-model",
        )
        .expect("prepared");

        run.session()
            .permission_handle()
            .remember_rule(mentra::session::RememberedRule {
                key: mentra::session::RuleKey {
                    tool_name: "files".to_string(),
                    // Every call to the tool, which is what a host seeding a
                    // standing allow writes.
                    pattern: None,
                },
                allow: true,
                scope: mentra::session::PermissionRuleScope::Global,
                reason: None,
            })
            .expect("the rule is remembered");

        let mut run = gated(run, policy);
        let report = tokio::time::timeout(
            Duration::from_secs(10),
            run.execute_with_approver(basis::CollectingSink::new(), approver(policy)),
        )
        .await
        .expect("a refusal never waits on anyone, so this must not hang")
        .expect("the run completes");

        let asked = report
            .sink
            .events()
            .iter()
            .filter(|event| matches!(event, basis::Event::PermissionRequested { .. }))
            .count();

        (workspace.join("made.txt").exists(), asked)
    }

    #[tokio::test]
    async fn a_durable_rule_cannot_allow_what_never_refuses() {
        // The bypass this pairing exists to close. A Global-scope allow seeded
        // before the run resolves the runtime gate's `Prompt` ahead of
        // `DenyAll`, so until the gate went on, `--approve never` was a promise
        // a rule nobody in this run wrote could stand over.
        //
        // Read with its mirror below: this one alone would pass just as
        // happily against a dead rule, since `never` refuses at the approver
        // too. The mirror's `asked == 0` assertion is what actually tells a
        // live seed from an inert one — a refusal here proves nothing about
        // whether the rule was ever consulted.
        let workspace = tempfile::tempdir().expect("tempdir");
        let (written, _asked) = wrote_a_file_under(ApprovalPolicy::Never, workspace.path()).await;

        assert!(
            !written,
            "a durable allow must not outlive the `--approve never` that refuses it"
        );
    }

    #[tokio::test]
    async fn a_durable_rule_still_answers_where_the_run_would_have_allowed() {
        // The other half, and the one that actually discriminates the pair:
        // `always` installs no authorizer of its own, so the runtime's gate
        // still surfaces the call and the remembered rule still resolves it
        // with nobody asked. A dead seed would still let the write through
        // here (nothing refuses `always`), but it would cost a permission
        // request the live rule never raises — so `asked == 0` is what would
        // catch the seed above going inert, not the write itself.
        let workspace = tempfile::tempdir().expect("tempdir");
        let (written, asked) = wrote_a_file_under(ApprovalPolicy::Always, workspace.path()).await;

        assert!(
            written,
            "a seeded rule must still answer for a policy that permits the call"
        );
        assert_eq!(
            asked, 0,
            "and it must answer without asking, or the seed was never really live"
        );
    }
}