mars-schema 0.27.4

//! Skill definition reading.

use crate::diagnostic::{RemovedField, unrecognized_field};
use crate::frontmatter::Frontmatter;
use crate::invocability::parse_invocability_axis;
use serde_yaml::Value;

#[derive(Debug, Clone, serde::Serialize)]
#[serde(rename_all = "kebab-case")]
pub struct SkillProfile {
    pub name: Option<String>,
    pub description: Option<String>,
    /// `metadata.type`, kept apart for callers; the reading carries it under `metadata`.
    #[serde(skip)]
    pub skill_type: Option<String>,
    pub model_invocable: bool,
    pub user_invocable: bool,
    pub license: Option<String>,
    pub compatibility: Option<String>,
    #[serde(serialize_with = "serialize_metadata")]
    pub metadata: Option<Value>,
    /// true when the source frontmatter explicitly set `model-invocable`
    #[serde(skip)]
    pub had_model_invocable_field: bool,
    #[serde(skip)]
    pub has_frontmatter: bool,
    /// Per-harness native keys (`harness-overrides:`), written only to that harness's file.
    #[serde(skip)]
    pub harness_overrides: crate::harness_overrides::HarnessOverrides,
}

/// Non-JSON YAML (tags, non-string keys) is left out, as in the reading's `frontmatter`,
/// where it carries a `yaml.non-json-value` diagnostic.
fn serialize_metadata<S: serde::Serializer>(
    value: &Option<Value>,
    serializer: S,
) -> Result<S::Ok, S::Error> {
    use serde::Serialize;
    value
        .as_ref()
        .and_then(|value| crate::ordered::OrderedValue::from_yaml(value).ok())
        .serialize(serializer)
}

/// The frontmatter keys Mars writes into a harness's native skill file itself. A
/// `harness-overrides` entry never replaces one (native lowering keeps Mars's value and reports
/// the override as dropped), so the translation's set-aside never moves one there either. The
/// Agent counterpart is [`crate::agent::LoweredField`].
pub const LOWERED_KEYS: &[&str] = &[
    "name",
    "description",
    "license",
    "metadata",
    "disable-model-invocation",
    "user-invocable",
    "alwaysApply",
];

/// Schema v1 skills carry no tool policy, so every spelling of one is a `discarded` key; the
/// foreign-dialect import and the legacy upgrade drop them. The upgrade rewrites `type` and
/// `when_to_use` and the invocability renames; the other removed keys stay for the author.
pub(crate) const REMOVED_SKILL_FIELDS: &[RemovedField] = &[
    RemovedField::discarded("tools", "remove this unused skill policy"),
    RemovedField::discarded("allowed-tools", "remove this unused skill policy"),
    RemovedField::discarded("allowed_tools", "remove this unused skill policy"),
    RemovedField::discarded("disallowed-tools", "remove this unused skill policy"),
    RemovedField::discarded("disallowed_tools", "remove this unused skill policy"),
    RemovedField::discarded("disallowedTools", "remove this unused skill policy"),
    RemovedField::replaced("type", "use `metadata.type`"),
    RemovedField::replaced("when_to_use", "fold it into `description`"),
    RemovedField::replaced("model_invocable", "use `model-invocable`"),
    RemovedField::replaced("user_invocable", "use `user-invocable`"),
    RemovedField::replaced("invocation", "use `model-invocable` and `user-invocable`"),
    RemovedField::replaced("disable-model-invocation", "use `model-invocable`"),
    RemovedField::replaced("allow_implicit_invocation", "use `model-invocable`"),
];

/// The keys that carry a skill's tool policy.
pub(crate) fn tool_keys() -> impl Iterator<Item = &'static str> {
    REMOVED_SKILL_FIELDS
        .iter()
        .filter(|field| field.discarded)
        .map(|field| field.key)
}

pub(crate) fn read_fields(
    fm: &Frontmatter,
    diags: &mut Vec<crate::diagnostic::SchemaDiagnostic>,
) -> SkillProfile {
    use crate::diagnostic::{SchemaDiagnostic, Severity, child_path, key_path};

    let mut profile = SkillProfile {
        name: None,
        description: None,
        skill_type: None,
        model_invocable: true,
        user_invocable: true,
        license: None,
        compatibility: None,
        metadata: None,
        had_model_invocable_field: false,
        has_frontmatter: fm.has_frontmatter(),
        harness_overrides: Default::default(),
    };
    let mut classification_diagnostics = Vec::new();
    let mut field_diagnostics = Vec::new();

    for key in fm.keys() {
        let value = fm.get(&key).expect("frontmatter key came from mapping");
        match key.as_str() {
            "name" | "description" | "license" | "compatibility" => {
                let parsed = if let Some(text) = value.as_str() {
                    Some(if key == "name" {
                        text.trim().to_owned()
                    } else {
                        text.to_owned()
                    })
                } else {
                    field_diagnostics.push(SchemaDiagnostic::new(
                        "field.type",
                        Severity::Error,
                        key_path(&key),
                        format!("`{key}` must be a string"),
                    ));
                    None
                };
                match key.as_str() {
                    "name" => profile.name = parsed,
                    "description" => profile.description = parsed,
                    "license" => profile.license = parsed,
                    "compatibility" => profile.compatibility = parsed,
                    _ => unreachable!(),
                }
            }
            "metadata" => {
                profile.metadata = Some(value.clone());
                if let Some(mapping) = value.as_mapping() {
                    if let Some(skill_type) = mapping.get(Value::String("type".to_string())) {
                        if let Some(skill_type) = skill_type.as_str() {
                            profile.skill_type = Some(skill_type.to_owned());
                        } else {
                            field_diagnostics.push(SchemaDiagnostic::new(
                                "field.type",
                                Severity::Error,
                                child_path("metadata", "type"),
                                "`metadata.type` must be a string",
                            ));
                        }
                    }
                } else {
                    field_diagnostics.push(SchemaDiagnostic::new(
                        "field.type",
                        Severity::Error,
                        key_path("metadata"),
                        "`metadata` must be a map",
                    ));
                }
            }
            "model-invocable" => {
                let (value, had_field, invalid) = parse_invocability_axis(Some(value));
                profile.model_invocable = value;
                profile.had_model_invocable_field = had_field;
                if invalid.is_some() {
                    field_diagnostics.push(SchemaDiagnostic::new(
                        "field.type",
                        Severity::Error,
                        key_path("model-invocable"),
                        "`model-invocable` must be true or false",
                    ));
                }
            }
            "user-invocable" => {
                let (value, _, invalid) = parse_invocability_axis(Some(value));
                profile.user_invocable = value;
                if invalid.is_some() {
                    field_diagnostics.push(SchemaDiagnostic::new(
                        "field.type",
                        Severity::Error,
                        key_path("user-invocable"),
                        "`user-invocable` must be true or false",
                    ));
                }
            }
            "harness-overrides" => {
                profile.harness_overrides = crate::harness_overrides::parse_harness_overrides(
                    "skill",
                    value,
                    &mut field_diagnostics,
                );
            }
            _ => classification_diagnostics.push(unrecognized_field(REMOVED_SKILL_FIELDS, &key)),
        }
    }

    diags.append(&mut classification_diagnostics);

    if !fm.has_frontmatter() {
        diags.push(SchemaDiagnostic::new(
            "skill.no-frontmatter",
            Severity::Warning,
            Vec::new(),
            "body-only skill has no frontmatter",
        ));
    } else {
        for (key, value) in [
            ("name", profile.name.as_deref()),
            ("description", profile.description.as_deref()),
        ] {
            if value.is_none_or(|value| value.trim().is_empty()) {
                diags.push(SchemaDiagnostic::new(
                    "field.required",
                    Severity::Error,
                    key_path(key),
                    format!("`{key}` must be a non-empty string"),
                ));
            }
        }
    }
    diags.append(&mut field_diagnostics);

    profile
}

/// Read a skill definition through the versioned schema contract.
pub fn read(content: &str) -> crate::read::DefinitionRead<SkillProfile> {
    crate::read::read_with(content, read_fields)
}