outrig-cli 0.1.0

//! `outrig init` -- three-phase orchestrator for end-to-end setup.
//!
//! Each phase is independently idempotent:
//! 1. **Global config** -- defer to [`config::init::run_with`] when
//!    `~/.outrig/config.toml` is absent; log + skip when present.
//! 2. **Repo config** -- [`repo::ensure`] writes `.agents/outrig/config.toml`
//!    against `cwd` if missing. Also reused by `outrig image add` as a
//!    fallback when run in an uninitialized repo.
//! 3. **Image loop** -- offer to scaffold image-configs by
//!    delegating to [`image::add::run_with`] in a loop.
//!
//! One [`PromptSource`] is threaded through all three phases so scripted
//! tests can drive the entire flow with a single byte stream.

pub mod prompt;
pub mod repo;

use std::path::Path;

use crate::config_init;
use crate::error::Result;
use crate::hf::{self, HfTreeFetcher};
use crate::image_setup;
use crate::init::prompt::{Field, PromptSource};
use crate::paths::global_config_path;

pub async fn run(force: bool, global_override: Option<&Path>) -> Result<()> {
    let cwd = std::env::current_dir()?;
    let mut prompt = prompt::auto();
    let mut hf = hf::auto();
    run_with(force, global_override, &cwd, &mut prompt, &mut hf).await
}

/// Drives the three-phase flow against an arbitrary `PromptSource`.
/// `cwd` anchors the repo-config phase (no walk-up; init is meant for
/// initial setup of the directory you're standing in). `hf` is the
/// HuggingFace tree-listing client used by mistralrs `model-id` prompts.
pub async fn run_with(
    force: bool,
    global_override: Option<&Path>,
    cwd: &Path,
    prompt: &mut impl PromptSource,
    hf: &mut impl HfTreeFetcher,
) -> Result<()> {
    // Phase 1: global config.
    let global_path = global_config_path(global_override);
    if global_path.exists() {
        eprintln!(
            "[outrig] using existing global config at {}",
            global_path.display()
        );
    } else {
        eprintln!(
            "[outrig] no global config found at {} -- let's create one.",
            global_path.display()
        );
        config_init::run_with(force, &global_path, prompt, hf).await?;
        eprintln!("[outrig] wrote {}", global_path.display());
    }

    // Phase 2: repo config. Returns the bootstrapped image name (if
    // we wrote the config) so phase 3's first image-add can skip its
    // name prompt.
    let mut bootstrapped_name = repo::ensure(cwd, &global_path, prompt, hf).await?;

    // Phase 3: image loop. The gate prompt is skipped on the first
    // iteration when phase 2 just bootstrapped an image -- the user
    // already chose to add one by walking through the image section,
    // so asking again would be redundant. When phase 2 short-circuited
    // on an existing config, the gate fires (re-runs may not want to
    // add an image).
    let mut first = true;
    loop {
        let should_run = if first && bootstrapped_name.is_some() {
            true
        } else if first {
            prompt.ask_bool(&ADD_FIRST_IMAGE_FIELD, true).await?
        } else {
            prompt.ask_bool(&ADD_ANOTHER_IMAGE_FIELD, false).await?
        };
        if !should_run {
            break;
        }
        let name = if first {
            bootstrapped_name.take()
        } else {
            None
        };
        image_setup::add::run_with(cwd, name, force, prompt).await?;
        first = false;
    }

    Ok(())
}

const ADD_FIRST_IMAGE_FIELD: Field = Field {
    name: "Add an image-config now?",
    description: "Yes: walk through `outrig image add` to scaffold a \
                  Dockerfile and [images.<name>] block.",
    options: &[],
    doc_link: "doc/usage/init.md",
};

const ADD_ANOTHER_IMAGE_FIELD: Field = Field {
    name: "Add another image-config?",
    description: "Yes: scaffold one more image-config via \
                  `outrig image add`. No: finish init.",
    options: &[],
    doc_link: "doc/usage/init.md",
};

/// Slice of every `Field` declared in this module, for `prompt_doc_sync.rs`.
pub const DOC_SYNC_FIELDS: &[&Field] = &[&ADD_FIRST_IMAGE_FIELD, &ADD_ANOTHER_IMAGE_FIELD];