cc-audit 3.2.14

//! YAML frontmatter parser for markdown files.

/// Parses YAML frontmatter from markdown files.
pub struct FrontmatterParser;

impl FrontmatterParser {
    /// Extract frontmatter content from a markdown file.
    ///
    /// Frontmatter is delimited by `---` at the start and end.
    /// Returns None if no valid frontmatter is found.
    ///
    /// # Example
    /// ```
    /// use cc_audit::parser::FrontmatterParser;
    /// let content = "---\nname: test\n---\n# Content";
    /// let frontmatter = FrontmatterParser::extract(content);
    /// assert_eq!(frontmatter, Some("\nname: test\n"));
    /// ```
    pub fn extract(content: &str) -> Option<&str> {
        content.strip_prefix("---").and_then(|after_start| {
            after_start
                .find("---")
                .map(|end_idx| &after_start[..end_idx])
        })
    }

    /// Parse frontmatter as a YAML value.
    ///
    /// Returns None if no frontmatter exists or parsing fails.
    pub fn parse_yaml(content: &str) -> Option<serde_yml::Value> {
        Self::extract(content).and_then(|fm| serde_yml::from_str(fm.trim()).ok())
    }

    /// Parse frontmatter as a JSON value.
    ///
    /// Returns None if no frontmatter exists or parsing fails.
    pub fn parse_json(content: &str) -> Option<serde_json::Value> {
        Self::extract(content)
            .and_then(|fm| serde_yml::from_str::<serde_json::Value>(fm.trim()).ok())
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_valid_frontmatter() {
        let content = "---\nname: test\ndescription: A test\n---\n# Content";
        let result = FrontmatterParser::extract(content);
        assert_eq!(result, Some("\nname: test\ndescription: A test\n"));
    }

    #[test]
    fn test_frontmatter_with_allowed_tools() {
        let content = "---\nname: skill\nallowed-tools: Read, Write\n---\n# Skill";
        let result = FrontmatterParser::extract(content);
        assert!(result.is_some());
        assert!(result.unwrap().contains("allowed-tools"));
    }

    #[test]
    fn test_no_frontmatter() {
        let content = "# Just Markdown\nNo frontmatter here.";
        assert!(FrontmatterParser::extract(content).is_none());
    }

    #[test]
    fn test_incomplete_frontmatter() {
        let content = "---\nname: test\nNo closing dashes";
        assert!(FrontmatterParser::extract(content).is_none());
    }

    #[test]
    fn test_empty_frontmatter() {
        let content = "------\n# Content";
        let result = FrontmatterParser::extract(content);
        assert_eq!(result, Some(""));
    }

    #[test]
    fn test_frontmatter_with_nested_dashes() {
        let content = "---\nname: test\ndata: \"some---thing\"\n---\n# Content";
        let result = FrontmatterParser::extract(content);
        // Should extract up to the first closing ---
        assert!(result.is_some());
    }

    #[test]
    fn test_content_not_starting_with_dashes() {
        let content = "# Title\n---\nname: test\n---";
        assert!(FrontmatterParser::extract(content).is_none());
    }

    #[test]
    fn test_parse_yaml() {
        let content = "---\nname: test\nversion: 1.0\n---\n# Content";
        let yaml = FrontmatterParser::parse_yaml(content);
        assert!(yaml.is_some());
        let yaml = yaml.unwrap();
        assert_eq!(yaml["name"].as_str(), Some("test"));
    }

    #[test]
    fn test_parse_json() {
        let content = "---\nname: test\nversion: 1.0\n---\n# Content";
        let json = FrontmatterParser::parse_json(content);
        assert!(json.is_some());
        let json = json.unwrap();
        assert_eq!(json["name"].as_str(), Some("test"));
    }
}