repolith-core 0.0.11

//! Repolith manifest schema and parser.
//!
//! The manifest (typically `repolith.toml` at the repo root) is the user-facing
//! contract: it declares which upstream nodes to track and which actions to run
//! against each.
//!
//! Parse via [`Manifest::from_toml`], which both deserializes and validates.
//!
//! [`Manifest::from_toml`]: crate::manifest::Manifest::from_toml

use crate::types::ActionId;
use serde::{Deserialize, Serialize};
use std::path::PathBuf;
use thiserror::Error;

/// Top-level `[orchestrator]` section.
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
pub struct OrchestratorMeta {
    /// Manifest schema version, expected to match `~0.1` (semver tilde range).
    pub schema_version: String,
    /// Human-readable name for the orchestrator instance (e.g. `"anatta-rs"`).
    pub name: String,
}

/// A single `[[node]]` entry — one upstream repository to manage.
///
/// **`git` and `path` semantics depend on the action kind**:
///
/// - For `kind = "git-clone"` actions, `git` is the **source** URL
///   (required) and `path` is the **destination** working tree
///   (optional, defaults to `./{node.id}`). Both fields can — and
///   typically do — coexist on the same node.
/// - For `kind = "cargo-install"` actions, `path` (when set) is the
///   **source** crate root that wins over `git`. If only `git` is
///   set, cargo is invoked with `--git <url>`. If both are set, `path`
///   takes precedence — the assumption being you've already cloned via
///   a sibling `git-clone` action.
///
/// The validator only requires that **at least one** of `git`/`path`
/// is set — the per-action interpretation above lives in the CLI's
/// manifest factory, not here.
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
pub struct NodeEntry {
    /// Unique identifier for the node within the manifest.
    pub id: String,
    /// Remote git URL. Required when [`Self::path`] is `None` and the
    /// node has a `git-clone` action; required as the cargo-install
    /// fallback when `path` is absent. URLs must use one of the
    /// allowlisted schemes (`https://`, `http://`, `ssh://`, `git@`,
    /// `file://`) and must not start with `-` — see the validator.
    pub git: Option<String>,
    /// Local path. For `git-clone` it's the destination working tree;
    /// for `cargo-install` it's the source crate root (preferred over
    /// `git`). Tilde expansion (`~/...`) is applied by the CLI before
    /// the path reaches `git` / `cargo`.
    pub path: Option<PathBuf>,
    /// Ordered list of actions to run for this node.
    #[serde(default, rename = "action")]
    pub actions: Vec<ActionEntry>,
}

/// Root manifest document — deserialized from `repolith.toml`.
///
/// `[[attached]]` blocks are accepted in the TOML for forward
/// compatibility but currently ignored (no field is collected). The
/// `attached` schema is reserved for an M2 `template_apply` action and
/// will land with that feature.
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
pub struct Manifest {
    /// `[orchestrator]` metadata (schema version, instance name).
    pub orchestrator: OrchestratorMeta,
    /// All `[[node]]` entries declared in the manifest.
    #[serde(default, rename = "node")]
    pub nodes: Vec<NodeEntry>,
}

/// Tagged variant of `[[node.action]]`, dispatched by the TOML `kind` field
/// (kebab-case: `"git-clone"`, `"cargo-install"`, `"docker"`, `"repolith"`).
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
#[serde(tag = "kind", rename_all = "kebab-case")]
pub enum ActionEntry {
    /// `kind = "git-clone"` — fetch the node's source.
    GitClone,
    /// `kind = "cargo-install"` — `cargo install` from the node's source tree.
    CargoInstall {
        /// Crate name (TOML key `crate`). Defaults to the node's `id` when absent.
        ///
        /// This is the **binary** target name (`cargo install --bin`), which
        /// may differ from the package that contains it — see [`Self::CargoInstall::package`].
        #[serde(default, rename = "crate")]
        crate_name: Option<String>,
        /// Package to select from the source, when the source contains more
        /// than one (`cargo install [CRATE]`, the positional argument).
        ///
        /// Needed for git repositories that ship several packages with
        /// binaries — test fixtures, tool workspaces — where cargo refuses
        /// to guess. **No implicit default**: deriving it from
        /// [`Self::CargoInstall::crate_name`] would break the common case
        /// where the binary name differs from the package name. Absent
        /// means no positional argument, i.e. cargo's own resolution.
        #[serde(default)]
        package: Option<String>,
        /// Cargo profile to build with (`--profile`), e.g. `"dev"` for a
        /// debug build. Absent means cargo's default, which is release.
        ///
        /// A debug build skips optimisation and is dramatically faster to
        /// produce — the right choice when iterating on a local `path`
        /// source. The binary is larger and slower at run time; that is the
        /// trade the field exists to make explicit.
        #[serde(default)]
        profile: Option<String>,
        /// Cargo features to enable.
        #[serde(default)]
        features: Vec<String>,
        /// Target install directory. Defaults to `~/.repolith/bin` when absent.
        #[serde(default)]
        install_to: Option<PathBuf>,
    },
    /// `kind = "docker"` — `docker build` an image from the node's checkout.
    ///
    /// Build-only by design: running containers is process supervision,
    /// which repolith explicitly does not do (see README "What it is NOT").
    Docker {
        /// Image tag to apply (`-t`). Required; validated against the
        /// docker reference charset and the no-leading-dash rule.
        tag: String,
        /// Dockerfile path relative to the build context. Defaults to
        /// `Dockerfile`. Must stay inside the node's checkout: relative,
        /// no `..` component, no leading dash, no control characters.
        #[serde(default)]
        dockerfile: Option<PathBuf>,
        /// Build context directory relative to the node's `path`.
        /// Defaults to the node's `path` itself. Same containment rules
        /// as `dockerfile`.
        #[serde(default)]
        context: Option<PathBuf>,
    },
    /// `kind = "repolith"` — federation: the node's checkout contains its
    /// own `repolith.toml`, executed as a nested plan
    /// (orchestrator-of-orchestrators).
    ///
    /// The nested stack keeps its **own local cache**, exactly as if
    /// `repolith sync` had been run in that directory by hand. Cycle
    /// detection and a maximum federation depth are enforced by the
    /// executing action (`repolith-engine::federation`).
    Repolith {
        /// Child manifest path relative to the node's `path`. Defaults to
        /// `repolith.toml`. Must stay inside the node's checkout: relative,
        /// no `..` component, no leading dash, no control characters.
        #[serde(default)]
        manifest: Option<PathBuf>,
    },
}

/// Errors that can occur while parsing or validating a manifest.
#[derive(Debug, Error)]
pub enum ManifestError {
    /// Underlying TOML deserialization failed (syntax error, type mismatch, unknown variant).
    #[error("toml parse: {0}")]
    Toml(#[from] toml::de::Error),
    /// `orchestrator.schema_version` does not satisfy the `~0.1` requirement.
    #[error("schema version {got}: requires ~0.1")]
    SchemaMismatch {
        /// The offending version string from the manifest.
        got: String,
    },
    /// Two or more `[[node]]` entries share the same `id`.
    #[error("duplicate node id: {0}")]
    DuplicateNodeId(String),
    /// A `[[node]]` declares neither a `git` URL nor a local `path`.
    #[error("node {0} has neither `git` nor `path`")]
    NodeMissingSource(String),
    /// A user-supplied URL would be interpreted as a CLI flag by the
    /// downstream `git` / `cargo` invocation (CWE-88, argument injection).
    #[error("invalid URL in node `{node}`: {reason}")]
    InvalidUrl {
        /// Node id that owns the offending URL.
        node: String,
        /// Why the URL was rejected (leading dash, unknown scheme, …).
        reason: String,
    },
    /// A user-supplied identifier (crate name, feature name) would be
    /// interpreted as a CLI flag or break shell argument parsing.
    #[error("invalid argument in node `{node}`: {reason}")]
    InvalidArg {
        /// Node id that owns the offending field.
        node: String,
        /// Why the value was rejected.
        reason: String,
    },
}

/// URL schemes the orchestrator considers safe to hand to `git` and `cargo`.
/// Anything else is rejected at parse time so a malicious manifest can't
/// reach the subprocess layer.
const ALLOWED_URL_SCHEMES: &[&str] = &["https://", "http://", "ssh://", "git@", "file://"];

/// Reject strings that `git` / `cargo` would interpret as a CLI flag
/// because they begin with `-`. The check is intentionally permissive
/// for everything else — the goal is defense against trivial argument
/// injection, not validation of the entire URL grammar.
fn check_no_leading_dash(value: &str, node: &str, field: &str) -> Result<(), ManifestError> {
    if value.starts_with('-') {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`{field}` starts with `-`, would be parsed as a CLI flag"),
        });
    }
    Ok(())
}

/// Reject node ids that would let the manifest escape the workspace
/// when used to construct a default clone path (`./{id}`):
///
/// - Empty ids (no directory).
/// - Path separators `/` and `\`.
/// - Drive-letter / alternate-data-stream separator `:` — on Windows
///   `id = "C:foo"` makes `PathBuf::from("./C:foo")` resolve drive-relative
///   and escape the workspace.
/// - Control characters — NUL truncates at the syscall layer (the path
///   actually opened on Unix is `./foo` not `./foo\0bar`); newlines confuse
///   downstream tooling.
/// - The literal `.` / `..` path-traversal tokens.
fn check_node_id_safe(id: &str) -> Result<(), ManifestError> {
    if id.is_empty() {
        return Err(ManifestError::InvalidArg {
            node: id.to_string(),
            reason: "node id is empty".to_string(),
        });
    }
    if let Some(c) = id.chars().find(|c| matches!(c, '/' | '\\' | ':')) {
        return Err(ManifestError::InvalidArg {
            node: id.to_string(),
            reason: format!(
                "node id `{id}` contains path separator `{c}` — would create nested, escaping, or drive-relative directories"
            ),
        });
    }
    if let Some(c) = id.chars().find(|c| c.is_control()) {
        return Err(ManifestError::InvalidArg {
            node: id.to_string(),
            reason: format!("node id contains control character U+{:04X}", c as u32),
        });
    }
    if id == "." || id == ".." {
        return Err(ManifestError::InvalidArg {
            node: id.to_string(),
            reason: format!("node id `{id}` is a path-traversal token"),
        });
    }
    Ok(())
}

/// Validate `url` against [`ALLOWED_URL_SCHEMES`], the no-leading-dash rule
/// (including for host and path components — so `ssh://-oProxyCommand=evil@h/r`
/// is rejected), and the no-control-characters rule. Empty URLs are rejected
/// to keep `git ls-remote ""` from producing confusing diagnostics.
fn check_url(url: &str, node: &str) -> Result<(), ManifestError> {
    if url.is_empty() {
        return Err(ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: "URL is empty".into(),
        });
    }
    if let Some(c) = url.chars().find(|c| c.is_control()) {
        return Err(ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: format!("URL contains control character U+{:04X}", c as u32),
        });
    }
    if url.starts_with('-') {
        return Err(ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: format!("`{url}` starts with `-`, would be parsed as a CLI flag"),
        });
    }
    let scheme = ALLOWED_URL_SCHEMES
        .iter()
        .find(|s| url.starts_with(*s))
        .ok_or_else(|| ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: format!("scheme not in allowlist {ALLOWED_URL_SCHEMES:?} (got `{url}`)"),
        })?;

    // For RFC-3986 URL schemes, parse and check host + each path segment for
    // a leading `-`. This catches `ssh://-oProxyCommand=evil@host/r.git` and
    // friends, which `starts_with('-')` on the whole URL misses.
    // `git@host:org/repo` is SCP-style — not a valid URL — so we walk it
    // manually instead.
    if *scheme == "git@" {
        check_scp_url(url, node)
    } else {
        check_rfc_url(url, node)
    }
}

fn check_rfc_url(url: &str, node: &str) -> Result<(), ManifestError> {
    let parsed = url::Url::parse(url).map_err(|e| ManifestError::InvalidUrl {
        node: node.to_string(),
        reason: format!("URL parse failed: {e}"),
    })?;
    // Userinfo: `ssh://-oProxyCommand=evil@host/r.git` parses with user
    // `-oProxyCommand=evil`, which git/ssh can be tricked into treating as
    // a flag depending on version.
    let user = parsed.username();
    if user.starts_with('-') {
        return Err(ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: format!("URL userinfo `{user}` starts with `-`"),
        });
    }
    if let Some(host) = parsed.host_str()
        && host.starts_with('-')
    {
        return Err(ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: format!("host `{host}` starts with `-`"),
        });
    }
    if let Some(segments) = parsed.path_segments() {
        for s in segments {
            if s.starts_with('-') {
                return Err(ManifestError::InvalidUrl {
                    node: node.to_string(),
                    reason: format!("path segment `{s}` starts with `-`"),
                });
            }
        }
    }
    Ok(())
}

fn check_scp_url(url: &str, node: &str) -> Result<(), ManifestError> {
    let after_at = &url["git@".len()..];
    let (host, path) = after_at.split_once(':').unwrap_or((after_at, ""));
    if host.is_empty() || host.starts_with('-') {
        return Err(ManifestError::InvalidUrl {
            node: node.to_string(),
            reason: format!("host `{host}` is empty or starts with `-`"),
        });
    }
    for segment in path.split('/') {
        if segment.starts_with('-') {
            return Err(ManifestError::InvalidUrl {
                node: node.to_string(),
                reason: format!("path segment `{segment}` starts with `-`"),
            });
        }
    }
    Ok(())
}

/// Lexical containment check for `docker` action paths (`dockerfile`,
/// `context`), mirroring [`check_node_id_safe`]. Runs at parse time with
/// no filesystem access, so it can only guarantee the *lexical* form —
/// symlink escapes are caught at run time by the action's canonicalize
/// check (defense in depth).
///
/// Rejected:
/// - Absolute paths — the whole point of these fields is "inside the
///   node's checkout".
/// - Any `..` component — lexical directory traversal.
/// - A leading `-` on the path — `docker` would parse it as a flag
///   (CWE-88), even behind `-f`/`--`.
/// - Control characters — NUL truncates at the syscall layer, newlines
///   confuse downstream tooling.
fn check_contained_path(
    path: &std::path::Path,
    node: &str,
    field: &str,
) -> Result<(), ManifestError> {
    let s = path.to_string_lossy();
    if s.is_empty() {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`{field}` is empty"),
        });
    }
    if let Some(c) = s.chars().find(|c| c.is_control()) {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`{field}` contains control character U+{:04X}", c as u32),
        });
    }
    if s.starts_with('-') {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`{field}` `{s}` starts with `-`, would be parsed as a CLI flag"),
        });
    }
    if path.is_absolute() {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`{field}` `{s}` is absolute — must stay inside the node's checkout"),
        });
    }
    if path
        .components()
        .any(|c| matches!(c, std::path::Component::ParentDir))
    {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!(
                "`{field}` `{s}` contains `..` — path traversal outside the node's checkout"
            ),
        });
    }
    Ok(())
}

/// Validate a `docker` action image tag: non-empty, no leading dash
/// (CWE-88), and restricted to the docker reference charset
/// (`[a-zA-Z0-9._:/-]`) so the value can never smuggle whitespace or
/// shell-relevant bytes into the `-t` argument.
fn check_docker_tag(tag: &str, node: &str) -> Result<(), ManifestError> {
    if tag.is_empty() {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: "`tag` is empty".to_string(),
        });
    }
    if tag.starts_with('-') {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`tag` `{tag}` starts with `-`, would be parsed as a CLI flag"),
        });
    }
    if let Some(c) = tag
        .chars()
        .find(|c| !(c.is_ascii_alphanumeric() || matches!(c, '.' | '_' | ':' | '/' | '-')))
    {
        return Err(ManifestError::InvalidArg {
            node: node.to_string(),
            reason: format!("`tag` `{tag}` contains `{c}` — allowed charset is [a-zA-Z0-9._:/-]"),
        });
    }
    Ok(())
}

/// Argv-injection sweep + per-kind requirements for one action.
///
/// Split out of [`Manifest::validate`] so that function stays under the
/// workspace's `clippy::too_many_lines` threshold — same reason
/// `classify` lives outside `Plan::compute`.
/// Argv guards for `cargo-install`: every field here ends up on cargo's
/// command line, so each one is checked for flag smuggling and for the
/// separators cargo would split on.
fn check_cargo_install(
    node: &NodeEntry,
    crate_name: Option<&String>,
    package: Option<&String>,
    profile: Option<&String>,
    features: &[String],
) -> Result<(), ManifestError> {
    if let Some(name) = crate_name {
        check_no_leading_dash(name, &node.id, "crate")?;
        if name.contains(',') {
            return Err(ManifestError::InvalidArg {
                node: node.id.clone(),
                reason: format!(
                    "`crate` name `{name}` contains `,` which would split cargo's --features list"
                ),
            });
        }
    }
    // `package` becomes cargo's positional argument, so
    // the same argv guards apply. A leading dash would be
    // read as a flag; `@` would be parsed as the version
    // spec of `CRATE@VER` and silently select something
    // else.
    if let Some(pkg) = package {
        check_no_leading_dash(pkg, &node.id, "package")?;
        if pkg.is_empty() {
            return Err(ManifestError::InvalidArg {
                node: node.id.clone(),
                reason: "`package` is empty".to_string(),
            });
        }
        if pkg.contains('@') {
            return Err(ManifestError::InvalidArg {
                node: node.id.clone(),
                reason: format!(
                    "`package` `{pkg}` contains `@`, which cargo parses as a version spec (`CRATE@VER`)"
                ),
            });
        }
    }
    // `profile` reaches cargo as `--profile <NAME>`. Cargo profile
    // names are identifier-like, so anything outside that charset is
    // either a typo or an attempt to smuggle an argument.
    if let Some(prof) = profile {
        check_no_leading_dash(prof, &node.id, "profile")?;
        if prof.is_empty() {
            return Err(ManifestError::InvalidArg {
                node: node.id.clone(),
                reason: "`profile` is empty".to_string(),
            });
        }
        if let Some(c) = prof
            .chars()
            .find(|c| !(c.is_ascii_alphanumeric() || matches!(c, '_' | '-')))
        {
            return Err(ManifestError::InvalidArg {
                node: node.id.clone(),
                reason: format!(
                    "`profile` `{prof}` contains `{c}` — allowed charset is [a-zA-Z0-9_-]"
                ),
            });
        }
    }
    for feature in features {
        check_no_leading_dash(feature, &node.id, "features entry")?;
        if feature.contains(',') {
            return Err(ManifestError::InvalidArg {
                node: node.id.clone(),
                reason: format!(
                    "feature `{feature}` contains `,` which would split cargo's --features list"
                ),
            });
        }
    }
    Ok(())
}

fn check_action(node: &NodeEntry, action: &ActionEntry) -> Result<(), ManifestError> {
    match action {
        ActionEntry::CargoInstall {
            crate_name,
            package,
            profile,
            features,
            ..
        } => check_cargo_install(
            node,
            crate_name.as_ref(),
            package.as_ref(),
            profile.as_ref(),
            features,
        )?,
        ActionEntry::Docker {
            tag,
            dockerfile,
            context,
        } => {
            check_docker_tag(tag, &node.id)?;
            if let Some(df) = dockerfile {
                check_contained_path(df, &node.id, "dockerfile")?;
            }
            if let Some(ctx) = context {
                check_contained_path(ctx, &node.id, "context")?;
            }
            // A docker build needs a checkout to build from.
            if node.path.is_none() {
                return Err(ManifestError::InvalidArg {
                    node: node.id.clone(),
                    reason: "docker action requires `path` on the node (build context base)"
                        .to_string(),
                });
            }
        }
        ActionEntry::Repolith { manifest } => {
            if let Some(m) = manifest {
                check_contained_path(m, &node.id, "manifest")?;
            }
            // Federation descends into the node's checkout.
            if node.path.is_none() {
                return Err(ManifestError::InvalidArg {
                    node: node.id.clone(),
                    reason: "repolith action requires `path` on the node (child stack root)"
                        .to_string(),
                });
            }
        }
        ActionEntry::GitClone => {}
    }
    Ok(())
}

impl Manifest {
    /// Parse and validate a manifest from a TOML source string.
    ///
    /// Returns a fully-validated [`Manifest`] on success. Validation covers
    /// schema version range, node-id uniqueness, and the per-node "must have
    /// `git` or `path`" rule.
    ///
    /// # Errors
    /// See [`ManifestError`] for the validation rules and the corresponding variants.
    pub fn from_toml(src: &str) -> Result<Self, ManifestError> {
        let m: Manifest = toml::from_str(src)?;
        m.validate()?;
        Ok(m)
    }

    /// Run all post-deserialize semantic checks. Called by [`Self::from_toml`].
    fn validate(&self) -> Result<(), ManifestError> {
        // schema_version is in 0.1.x range
        let req = semver::VersionReq::parse("~0.1").unwrap();
        let v = semver::Version::parse(&format!("{}.0", self.orchestrator.schema_version))
            .map_err(|_| ManifestError::SchemaMismatch {
                got: self.orchestrator.schema_version.clone(),
            })?;
        if !req.matches(&v) {
            return Err(ManifestError::SchemaMismatch {
                got: self.orchestrator.schema_version.clone(),
            });
        }
        // Per-node checks: id uniqueness, source presence, then the
        // argument-injection sweep on every user-controlled string field
        // that ultimately reaches a subprocess.
        let mut seen = std::collections::HashSet::new();
        for n in &self.nodes {
            check_node_id_safe(&n.id)?;
            if !seen.insert(&n.id) {
                return Err(ManifestError::DuplicateNodeId(n.id.clone()));
            }
            if n.git.is_none() && n.path.is_none() {
                return Err(ManifestError::NodeMissingSource(n.id.clone()));
            }
            if let Some(url) = &n.git {
                check_url(url, &n.id)?;
            }
            for action in &n.actions {
                check_action(n, action)?;
            }
        }
        Ok(())
    }

    /// Enumerate every action across every node as a stable [`ActionId`].
    ///
    /// Each id has the shape `"{node.id}::{action-kind}::{index}"`, where
    /// `index` is the action's position within its node (0-based). Ordering
    /// matches manifest declaration order; downstream consumers (Plan, Cache)
    /// rely on this for deterministic builds.
    #[must_use]
    pub fn action_ids(&self) -> Vec<ActionId> {
        self.nodes
            .iter()
            .flat_map(|n| {
                n.actions
                    .iter()
                    .enumerate()
                    .map(move |(i, a)| ActionId(format!("{}::{}::{}", n.id, action_kind(a), i)))
            })
            .collect()
    }
}

/// Kebab-case discriminator string for an [`ActionEntry`], matching
/// the TOML `kind` tag. Public so the CLI's manifest factory uses the
/// same single source of truth — adding a new action variant therefore
/// only needs touching this match plus the variant declaration.
#[must_use]
pub fn action_kind(a: &ActionEntry) -> &'static str {
    match a {
        ActionEntry::GitClone => "git-clone",
        ActionEntry::CargoInstall { .. } => "cargo-install",
        ActionEntry::Docker { .. } => "docker",
        ActionEntry::Repolith { .. } => "repolith",
    }
}