whisker-cli 0.2.2

//! Resolve a user crate's `whisker.rs` config + `Cargo.toml` metadata
//! into a [`ResolvedManifest`] the CLI can hand to `whisker-dev-server`.
//!
//! The dev-server itself is manifest-agnostic — it accepts flat
//! parameters (paths, bundle ids, application ids, …) via
//! `whisker_dev_server::Config`. Translating the user's
//! `whisker.rs::configure(&mut Config)` result into those flat
//! values is the CLI's job and lives in [`super::run`].
//!
//! ## Discovery
//!
//! [`resolve`] takes an optional explicit `Cargo.toml` path. When
//! `None`, it walks up from `cwd` looking for the first `Cargo.toml`
//! that has a `[package]` section (a `[workspace]`-only manifest at
//! the top of a virtual workspace doesn't count — we need the
//! package node).
//!
//! Once a `Cargo.toml` is found, the sibling `whisker.rs` is the
//! config source. Missing `whisker.rs` is an error: the dev-server
//! needs bundle id, application id, etc. that the file supplies.

use anyhow::{anyhow, Context, Result};
use std::path::{Path, PathBuf};
use whisker_config::Config;

use crate::probe;

/// One CLI invocation's worth of resolved user-crate state.
#[derive(Debug)]
pub struct ResolvedManifest {
    /// Directory containing the user crate's `Cargo.toml` and (next
    /// to it) `whisker.rs`. All other paths the CLI builds are
    /// relative to this.
    pub crate_dir: PathBuf,
    /// `[package].name` from `Cargo.toml`. Used by the dev-server as
    /// the cargo `-p` argument, and by `whisker-build` to find the
    /// `lib<package>.so` / `.dylib` artifact.
    pub package: String,
    /// Result of running the user's `whisker.rs::configure`. Owned
    /// (decoded from JSON) so subsequent CLI logic can pattern-match
    /// on optional fields without rerunning the probe.
    pub config: Config,
}

/// Resolve the manifest. `cargo_toml_override` (set via
/// `whisker run --manifest-path <path>`) bypasses cwd discovery.
pub fn resolve(cargo_toml_override: Option<&Path>) -> Result<ResolvedManifest> {
    let cargo_toml = match cargo_toml_override {
        Some(p) => p.to_path_buf(),
        None => {
            let cwd = std::env::current_dir().context("read cwd")?;
            find_package_cargo_toml(&cwd).ok_or_else(|| {
                anyhow!(
                    "no `[package]` Cargo.toml at or above {} — pass `--manifest-path <path>` to point at the user crate",
                    cwd.display(),
                )
            })?
        }
    };
    // Canonicalize: downstream sites (Command::current_dir,
    // find_workspace_root's upward walk) break when crate_dir is
    // relative and the workspace root coincides with the process cwd —
    // PathBuf::pop() then bottoms out at "" which feeds chdir("") and
    // surfaces as posix-spawn ENOENT.
    let cargo_toml = std::fs::canonicalize(&cargo_toml)
        .with_context(|| format!("canonicalize {}", cargo_toml.display()))?;
    let crate_dir = cargo_toml
        .parent()
        .ok_or_else(|| anyhow!("Cargo.toml has no parent dir: {}", cargo_toml.display()))?
        .to_path_buf();
    let package = parse_package_name(&cargo_toml)?;
    let whisker_rs = crate_dir.join("whisker.rs");
    if !whisker_rs.is_file() {
        anyhow::bail!(
            "no whisker.rs next to {} — every Whisker app needs a `whisker.rs` at the crate root that defines `fn configure(app: &mut Config)`",
            cargo_toml.display(),
        );
    }
    let config = probe::run(&whisker_rs, &crate_dir, &package)?;
    Ok(ResolvedManifest {
        crate_dir,
        package,
        config,
    })
}

/// Walk up from `start` looking for the first Cargo.toml with a
/// `[package]` table. A pure `[workspace]` manifest at the top of a
/// virtual workspace is skipped — we want the user-crate package,
/// not the workspace root.
fn find_package_cargo_toml(start: &Path) -> Option<PathBuf> {
    let mut cur = start.to_path_buf();
    loop {
        let cargo = cur.join("Cargo.toml");
        if cargo.is_file() {
            if let Ok(txt) = std::fs::read_to_string(&cargo) {
                if has_package_section(&txt) {
                    return Some(cargo);
                }
            }
        }
        if !cur.pop() {
            return None;
        }
    }
}

/// Cheap test for `[package]` without pulling in a full TOML parser
/// just for the walk-up phase. We do a real `toml::from_str` once
/// we've picked a winning candidate (see `parse_package_name`).
fn has_package_section(toml_text: &str) -> bool {
    toml_text.lines().any(|line| {
        let l = line.trim();
        l == "[package]" || l.starts_with("[package]") || l == "[ package ]"
    })
}

/// Parse `[package].name` from the given Cargo.toml.
fn parse_package_name(cargo_toml: &Path) -> Result<String> {
    let text = std::fs::read_to_string(cargo_toml)
        .with_context(|| format!("read {}", cargo_toml.display()))?;
    let doc: toml::Value =
        toml::from_str(&text).with_context(|| format!("parse {} as TOML", cargo_toml.display()))?;
    let name = doc
        .get("package")
        .and_then(|p| p.get("name"))
        .and_then(|n| n.as_str())
        .ok_or_else(|| {
            anyhow!(
                "{} has no [package].name (is this a virtual-workspace Cargo.toml?)",
                cargo_toml.display(),
            )
        })?;
    Ok(name.to_string())
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::sync::atomic::{AtomicU64, Ordering};

    fn unique_tempdir(label: &str) -> PathBuf {
        static SEQ: AtomicU64 = AtomicU64::new(0);
        let n = SEQ.fetch_add(1, Ordering::Relaxed);
        let pid = std::process::id();
        let p = std::env::temp_dir().join(format!("whisker-cli-manifest-{label}-{pid}-{n}"));
        let _ = std::fs::remove_dir_all(&p);
        std::fs::create_dir_all(&p).unwrap();
        p
    }

    #[test]
    fn has_package_section_detects_the_table_header() {
        assert!(has_package_section("[package]\nname = \"x\"\n"));
        assert!(has_package_section("\n\n[package]\n"));
        assert!(!has_package_section("[workspace]\nmembers = []\n"));
        assert!(!has_package_section("[package.metadata.foo]\nbar = 1\n"));
    }

    #[test]
    fn find_package_cargo_toml_skips_virtual_workspace_root() {
        let tmp = unique_tempdir("vws");
        std::fs::write(tmp.join("Cargo.toml"), "[workspace]\nmembers = [\"app\"]\n").unwrap();
        let app = tmp.join("app");
        std::fs::create_dir_all(&app).unwrap();
        std::fs::write(
            app.join("Cargo.toml"),
            "[package]\nname = \"app\"\nversion = \"0.0.0\"\n",
        )
        .unwrap();
        // From inside the member, walker should land on the member's
        // Cargo.toml, not the virtual-workspace one.
        assert_eq!(
            find_package_cargo_toml(&app).as_deref(),
            Some(app.join("Cargo.toml").as_path()),
        );
        let _ = std::fs::remove_dir_all(&tmp);
    }

    #[test]
    fn parse_package_name_reads_the_name_field() {
        let tmp = unique_tempdir("name");
        let p = tmp.join("Cargo.toml");
        std::fs::write(
            &p,
            "[package]\nname = \"my-cool-app\"\nversion = \"0.0.0\"\n",
        )
        .unwrap();
        assert_eq!(parse_package_name(&p).unwrap(), "my-cool-app");
        let _ = std::fs::remove_dir_all(&tmp);
    }
}