phoxal 0.43.1

//! Data types for authored source robot manifests.
//!
//! The crate root is the version dispatcher for `robot.yaml`. Schema wire
//! types live under [`v0`]; consumers that need the v0 struct directly can
//! import [`RobotV0`] or [`v0::Robot`].
//!
//! Parsing is strict: every struct denies unknown fields, `serde_yaml`
//! natively rejects duplicate mapping keys and YAML-1.1-only booleans
//! (`yes`/`no`/`on`/`off`) coerced into typed `bool` fields, and
//! [`strict_yaml::check`] additionally rejects anchors/aliases, merge keys,
//! explicit tags, and multi-document streams before the manifest reaches
//! `serde_yaml::from_str`.

use anyhow::{Context, Result, bail};
use serde::{Deserialize, Serialize};
use std::collections::BTreeSet;
use std::path::{Path, PathBuf};

mod strict_yaml;
pub mod v0;

pub use v0::Robot as RobotV0;
pub use v0::ValidationError;

const ROBOT_FILE: &str = "robot.yaml";

/// Version dispatcher for `robot.yaml`.
///
/// On the wire, manifests use `schema: robot/v0` for the manifest grammar.
/// The single-variant enum is the future-versioning seam: a second schema
/// generation adds a sibling variant here without disturbing `v0::Robot`.
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[cfg_attr(test, derive(schemars::JsonSchema))]
#[serde(tag = "schema")]
pub enum Robot {
    #[serde(rename = "robot/v0")]
    V0(v0::Robot),
}

impl Robot {
    pub fn read_from_dir(path: impl AsRef<Path>) -> Result<Self> {
        let robot = Self::parse_from_dir(path)?;
        robot.validate().map_err(validation_error)?;
        Ok(robot)
    }

    /// Reads, composes, and validates one robot document at an explicit path.
    pub fn read_from_path(path: impl AsRef<Path>) -> Result<Self> {
        let robot = Self::parse_from_path(path)?;
        robot.validate().map_err(validation_error)?;
        Ok(robot)
    }

    pub fn read_from_string(text: &str) -> Result<Self> {
        let robot = Self::parse_from_string(text)?;
        robot.validate().map_err(validation_error)?;
        Ok(robot)
    }

    pub fn parse_from_dir(path: impl AsRef<Path>) -> Result<Self> {
        Self::parse_from_path(path.as_ref().join(ROBOT_FILE))
    }

    /// Reads one leaf robot document and composes its ordered direct parents.
    ///
    /// Parent entries are partial strict-YAML maps relative to the leaf
    /// directory. Maps merge recursively; sequences and scalar values replace.
    /// Nested composition is rejected so the leaf remains the single,
    /// deterministic authority for parent order.
    pub fn parse_from_path(path: impl AsRef<Path>) -> Result<Self> {
        let leaf_path = path
            .as_ref()
            .canonicalize()
            .with_context(|| format!("failed to resolve robot file {}", path.as_ref().display()))?;
        let root = leaf_path
            .parent()
            .context("robot file must have a parent directory")?
            .to_path_buf();
        let mut leaf = read_yaml_value(&leaf_path)?;
        let parents = take_extends(&mut leaf, &leaf_path)?;
        let mut seen = BTreeSet::new();
        let mut composed = serde_yaml::Value::Mapping(Default::default());

        for relative in parents {
            if relative.is_absolute() {
                bail!(
                    "robot extends path must be relative: {}",
                    relative.display()
                );
            }
            let parent_path = root.join(&relative).canonicalize().with_context(|| {
                format!(
                    "failed to resolve robot parent {} declared by {}",
                    relative.display(),
                    leaf_path.display()
                )
            })?;
            if !parent_path.starts_with(&root) {
                bail!(
                    "robot parent {} escapes robot directory {}",
                    relative.display(),
                    root.display()
                );
            }
            if parent_path == leaf_path {
                bail!(
                    "robot document cannot extend itself: {}",
                    relative.display()
                );
            }
            if !seen.insert(parent_path.clone()) {
                bail!("duplicate robot parent: {}", relative.display());
            }

            let mut parent = read_yaml_value(&parent_path)?;
            let nested = take_extends(&mut parent, &parent_path)?;
            if !nested.is_empty() {
                bail!(
                    "robot parent {} declares nested extends; list every parent directly in {}",
                    parent_path.display(),
                    leaf_path.display()
                );
            }
            deep_merge(&mut composed, parent);
        }
        deep_merge(&mut composed, leaf);

        serde_yaml::from_value(composed)
            .with_context(|| format!("failed to parse composed robot {}", leaf_path.display()))
    }

    pub fn parse_from_string(text: &str) -> Result<Self> {
        strict_yaml::check(text).context("failed to parse robot")?;
        let robot: Self = serde_yaml::from_str(text).with_context(|| "failed to parse robot")?;
        if !robot.as_v0().extends.is_empty() {
            bail!(
                "robot extends requires a file path; use Robot::read_from_path or Robot::read_from_dir"
            );
        }
        Ok(robot)
    }

    pub fn write_to_dir(&self, path: impl AsRef<Path>) -> Result<()> {
        let path = path.as_ref();
        std::fs::create_dir_all(path)
            .with_context(|| format!("failed to create robot directory {}", path.display()))?;
        let yaml = serde_yaml::to_string(self)?;
        std::fs::write(path.join(ROBOT_FILE), yaml).with_context(|| {
            format!(
                "failed to write robot file {}",
                path.join(ROBOT_FILE).display()
            )
        })?;
        Ok(())
    }

    pub fn validate(&self) -> std::result::Result<(), Vec<ValidationError>> {
        match self {
            Self::V0(robot) => robot.validate(),
        }
    }

    #[must_use]
    pub fn as_v0(&self) -> &v0::Robot {
        match self {
            Self::V0(robot) => robot,
        }
    }

    #[must_use]
    pub fn into_v0(self) -> v0::Robot {
        match self {
            Self::V0(robot) => robot,
        }
    }
}

fn read_yaml_value(path: &Path) -> Result<serde_yaml::Value> {
    let text = std::fs::read_to_string(path)
        .with_context(|| format!("failed to read robot file {}", path.display()))?;
    strict_yaml::check(&text)
        .with_context(|| format!("failed to parse robot file {}", path.display()))?;
    serde_yaml::from_str(&text)
        .with_context(|| format!("failed to parse robot file {}", path.display()))
}

fn take_extends(value: &mut serde_yaml::Value, path: &Path) -> Result<Vec<PathBuf>> {
    let serde_yaml::Value::Mapping(map) = value else {
        bail!("robot document {} must be a mapping", path.display());
    };
    let key = serde_yaml::Value::String("extends".to_string());
    let Some(raw) = map.remove(&key) else {
        return Ok(Vec::new());
    };
    serde_yaml::from_value(raw)
        .with_context(|| format!("invalid extends list in {}", path.display()))
}

fn deep_merge(base: &mut serde_yaml::Value, overlay: serde_yaml::Value) {
    match (base, overlay) {
        (serde_yaml::Value::Mapping(base), serde_yaml::Value::Mapping(overlay)) => {
            for (key, value) in overlay {
                match base.get_mut(&key) {
                    Some(current) => deep_merge(current, value),
                    None => {
                        base.insert(key, value);
                    }
                }
            }
        }
        (base, overlay) => *base = overlay,
    }
}

fn validation_error(errors: Vec<ValidationError>) -> anyhow::Error {
    let message = errors
        .iter()
        .map(ToString::to_string)
        .collect::<Vec<_>>()
        .join("\n");
    anyhow::anyhow!("Robot errors:\n{message}")
}

#[cfg(test)]
mod composition_tests {
    use super::Robot;

    const LEAF: &str = r#"
schema: robot/v0
extends: [base.robot.yaml, host.robot.yaml]
robot:
  id: leaf
  namespace: dev
  kinematic:
    kind: omnidirectional
    actuators: [drive.motor]
    encoders: []
  components: {}
"#;

    #[test]
    fn direct_parents_deep_merge_in_order_and_are_cleared() -> anyhow::Result<()> {
        let dir = tempfile::tempdir()?;
        std::fs::write(
            dir.path().join("base.robot.yaml"),
            r#"
robot:
  id: base
  motion_limits:
    max_linear_speed_mps: 0.5
    max_angular_speed_radps: 1.0
services:
  autonomy:
    config: { planner: base, limit: 1 }
tools:
  lidar-viz:
    config: { port: 9000, verbose: false }
"#,
        )?;
        std::fs::write(
            dir.path().join("host.robot.yaml"),
            r#"
robot:
  motion_limits:
    max_linear_speed_mps: 0.8
services:
  autonomy:
    config: { planner: host }
tools:
  lidar-viz:
    config: { verbose: true }
"#,
        )?;
        std::fs::write(dir.path().join("robot.yaml"), LEAF)?;

        let robot = Robot::read_from_dir(dir.path())?.into_v0();
        assert!(robot.extends.is_empty());
        assert_eq!(robot.robot.id, "leaf");
        assert_eq!(robot.robot.motion_limits.max_linear_speed_mps, 0.8);
        assert_eq!(robot.robot.motion_limits.max_angular_speed_radps, 1.0);
        let config = robot.services["autonomy"].config.as_ref().unwrap();
        assert_eq!(config["planner"], "host");
        assert_eq!(config["limit"], 1);
        // The tools declaration deep-merges exactly like services (#950).
        let tool_config = robot.tools["lidar-viz"].config.as_ref().unwrap();
        assert_eq!(tool_config["port"], 9000);
        assert_eq!(tool_config["verbose"], true);
        Ok(())
    }

    #[test]
    fn nested_parent_extends_is_rejected() -> anyhow::Result<()> {
        let dir = tempfile::tempdir()?;
        std::fs::write(
            dir.path().join("base.robot.yaml"),
            "extends: [other.robot.yaml]\n",
        )?;
        std::fs::write(dir.path().join("host.robot.yaml"), "{}\n")?;
        std::fs::write(dir.path().join("robot.yaml"), LEAF)?;

        let error = Robot::read_from_dir(dir.path()).expect_err("nested extends must fail");
        assert!(format!("{error:#}").contains("declares nested extends"));
        Ok(())
    }

    #[test]
    fn string_parser_rejects_unresolvable_extends() {
        let manifest = LEAF.replace(
            "  namespace: dev\n",
            "  namespace: dev\n  motion_limits:\n    max_linear_speed_mps: 0.5\n    max_angular_speed_radps: 1.0\n",
        );
        let error = Robot::parse_from_string(&manifest)
            .expect_err("string parsing cannot resolve parent paths");
        assert!(format!("{error:#}").contains(
            "robot extends requires a file path; use Robot::read_from_path or Robot::read_from_dir"
        ));
    }

    #[test]
    fn duplicate_and_escaping_parents_are_rejected() -> anyhow::Result<()> {
        let dir = tempfile::tempdir()?;
        std::fs::write(dir.path().join("base.robot.yaml"), "{}\n")?;
        std::fs::write(
            dir.path().join("robot.yaml"),
            LEAF.replace(
                "base.robot.yaml, host.robot.yaml",
                "base.robot.yaml, base.robot.yaml",
            ),
        )?;
        let duplicate = Robot::read_from_dir(dir.path()).expect_err("duplicate must fail");
        assert!(format!("{duplicate:#}").contains("duplicate robot parent"));

        std::fs::write(
            dir.path().join("robot.yaml"),
            LEAF.replace("base.robot.yaml, host.robot.yaml", "../outside.robot.yaml"),
        )?;
        std::fs::write(
            dir.path().parent().unwrap().join("outside.robot.yaml"),
            "{}\n",
        )?;
        let escaping = Robot::read_from_dir(dir.path()).expect_err("escape must fail");
        assert!(format!("{escaping:#}").contains("escapes robot directory"));
        Ok(())
    }
}

/// The editor-facing `robot.yaml` JSON Schema is a structural grammar aid.
/// `Robot` and every serde-shaped type it reaches carry a
/// `#[cfg_attr(test, derive(schemars::JsonSchema))]`, so the schema is derived
/// from the manifest structs rather than maintained separately.
/// `schema_matches_model` fails when the checked-in
/// `examples/robot.schema.json` stops matching that derived serde shape.
/// The schema is not an executable specification.
/// It does not cover `strict_yaml::check`, semantic [`Robot::validate`]
/// constraints, custom `FromStr` validation, or cross-file component and URDF
/// constraints.
///
/// Kept test-only (not a normal dependency) on purpose: `schemars` and
/// `jsonschema` are dev-dependencies of `phoxal` (see `phoxal/Cargo.toml`),
/// so the published crate carries no schema-generation code or its
/// dependency weight - only `cargo test` regenerates and checks the schema.
#[cfg(test)]
mod schema_guard {
    use super::Robot;

    /// Path to the checked-in JSON Schema, relative to this crate's manifest
    /// directory (`phoxal/`).
    const SCHEMA_PATH: &str = "../examples/robot.schema.json";

    fn generated_schema_json() -> String {
        let schema = schemars::schema_for!(Robot);
        serde_json::to_string_pretty(&schema).expect("schema serializes to JSON") + "\n"
    }

    fn schema_path() -> std::path::PathBuf {
        std::path::Path::new(env!("CARGO_MANIFEST_DIR")).join(SCHEMA_PATH)
    }

    /// Regenerate-and-diff guard: run `cargo test -p phoxal schema_guard::print_schema
    /// -- --ignored --nocapture > examples/robot.schema.json` (from the
    /// `framework/phoxal` directory) and re-check the file in when this fails.
    #[test]
    fn schema_matches_model() {
        let generated = generated_schema_json();
        let checked_in = std::fs::read_to_string(schema_path()).unwrap_or_default();

        assert_eq!(
            generated,
            checked_in,
            "examples/robot.schema.json is stale relative to the schemars-derived \
             schema for phoxal::model::robot::Robot. Regenerate it with:\n\n  \
             cargo test -p phoxal schema_guard::print_schema -- --ignored --nocapture \
             > {}\n\nthen re-check the file in.",
            schema_path().display()
        );
    }

    /// Not an assertion - an opt-in helper (`--ignored`) that prints the
    /// freshly generated schema so `schema_matches_model`'s failure message
    /// can be piped straight into `examples/robot.schema.json`.
    #[test]
    #[ignore = "prints the schema; run explicitly to regenerate examples/robot.schema.json"]
    fn print_schema() {
        print!("{}", generated_schema_json());
    }

    fn validator_for_model() -> jsonschema::Validator {
        let schema = schemars::schema_for!(Robot);
        jsonschema::validator_for(&serde_json::to_value(&schema).expect("schema is valid JSON"))
            .expect("generated schema should itself be a valid JSON Schema")
    }

    /// The schema must actually accept the checked-in example manifest - proof
    /// that the drift guard tracks a schema that validates real content, not
    /// just two frozen blobs that happen to match each other.
    #[test]
    fn schema_validates_the_hello_rover_example() {
        let manifest_dir = std::path::Path::new(env!("CARGO_MANIFEST_DIR"));
        let robot_yaml =
            std::fs::read_to_string(manifest_dir.join("../examples/hello-rover/robot.yaml"))
                .expect("examples/hello-rover/robot.yaml should be readable");
        let value: serde_json::Value = serde_yaml::from_str(&robot_yaml)
            .expect("examples/hello-rover/robot.yaml should parse as YAML");

        let validator = validator_for_model();
        let errors: Vec<_> = validator
            .iter_errors(&value)
            .map(|error| error.to_string())
            .collect();
        assert!(
            errors.is_empty(),
            "examples/hello-rover/robot.yaml should validate against robot.schema.json: {errors:?}"
        );
    }

    /// A manifest with an unknown root key must be rejected - proof that
    /// `deny_unknown_fields` really propagates into the schema's
    /// `additionalProperties: false`, not just that the happy path validates.
    #[test]
    fn schema_rejects_unknown_root_key() {
        let yaml = r#"
schema: robot/v0
robot:
  id: test-bot
  namespace: dev
  kinematic:
    kind: omnidirectional
    actuators: []
    encoders: []
  components: {}
not_a_real_key: {}
"#;
        let value: serde_json::Value = serde_yaml::from_str(yaml).expect("valid YAML");

        assert!(
            !validator_for_model().is_valid(&value),
            "schema should reject an unknown root key"
        );
    }
}