fea-rs-ast 0.1.4

use std::ops::Range;

use fea_rs::typed::{AstNode as _, Tag};
use smol_str::SmolStr;

use crate::{
    Anchor, AsFea, GlyphClass, GlyphContainer, MarkClass, Metric, SHIFT, Statement, ValueRecord,
    from_anchor,
};

/// A named anchor definition. (2.e.viii)
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct AnchorDefinition {
    /// The X coordinate of the anchor
    pub x: Metric,
    /// The Y coordinate of the anchor
    pub y: Metric,
    /// The contour point index, if any
    pub contourpoint: Option<u16>,
    /// The name of the anchor
    pub name: String,
    /// The location of the anchor definition in the source FEA
    pub location: Range<usize>,
}
impl AnchorDefinition {
    /// Creates a new `Anchor` statement.
    pub fn new(
        x: Metric,
        y: Metric,
        contourpoint: Option<u16>,
        name: String,
        location: Range<usize>,
    ) -> Self {
        Self {
            x,
            y,
            contourpoint,
            name,
            location,
        }
    }
}
impl AsFea for AnchorDefinition {
    fn as_fea(&self, _indent: &str) -> String {
        let mut res = format!("anchorDef {} {}", self.x.as_fea(""), self.y.as_fea(""));
        if let Some(cp) = self.contourpoint {
            res.push_str(&format!(" contourpoint {}", cp));
        }
        res.push_str(&format!(" {};", self.name));
        res
    }
}
impl From<fea_rs::typed::AnchorDef> for AnchorDefinition {
    fn from(val: fea_rs::typed::AnchorDef) -> Self {
        let anchor_node = val
            .iter()
            .filter_map(fea_rs::typed::Anchor::cast)
            .next()
            .unwrap();
        let our_anchor: Anchor = from_anchor(anchor_node).unwrap();
        let name = val
            .iter()
            .find(|t| t.kind() == fea_rs::Kind::Ident)
            .unwrap();
        AnchorDefinition::new(
            our_anchor.x,
            our_anchor.y,
            our_anchor.contourpoint,
            name.token_text().unwrap().to_string(),
            val.node().range(),
        )
    }
}

/// A comment in a feature file
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Comment {
    /// The text of the comment, which should include the initial `#`.
    pub text: String,
}
impl Comment {
    /// Creates a new comment
    pub fn new(text: String) -> Self {
        Self { text }
    }
}
impl AsFea for Comment {
    fn as_fea(&self, _indent: &str) -> String {
        self.text.clone()
    }
}
impl From<&str> for Comment {
    fn from(text: &str) -> Self {
        Self::new(text.to_string())
    }
}

/// Example: `feature salt;`
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct FeatureReferenceStatement {
    /// The name of the referenced feature
    pub feature_name: String,
}
impl FeatureReferenceStatement {
    /// Creates a new FeatureReferenceStatement.
    pub fn new(feature_name: String) -> Self {
        Self { feature_name }
    }
}
impl AsFea for FeatureReferenceStatement {
    fn as_fea(&self, _indent: &str) -> String {
        format!("feature {};", self.feature_name)
    }
}
impl From<fea_rs::typed::FeatureRef> for FeatureReferenceStatement {
    fn from(feature: fea_rs::typed::FeatureRef) -> Self {
        Self::new(
            feature
                .iter()
                .find_map(Tag::cast)
                .unwrap()
                .text()
                .to_string(),
        )
    }
}

/// A `head` table `FontRevision` statement.
///
/// `revision` should be a number, and will be formatted to three
/// significant decimal places.
#[derive(Debug, Clone)]
pub struct FontRevisionStatement {
    /// The font revision number
    pub revision: f32,
}
impl FontRevisionStatement {
    /// Create a new `FontRevision` statement.
    pub fn new(revision: f32) -> Self {
        Self { revision }
    }
}
impl AsFea for FontRevisionStatement {
    fn as_fea(&self, _indent: &str) -> String {
        format!("FontRevision {:.3};", self.revision)
    }
}
impl From<fea_rs::typed::HeadFontRevision> for FontRevisionStatement {
    fn from(val: fea_rs::typed::HeadFontRevision) -> Self {
        let revision_token = val
            .iter()
            .find(|t| t.kind() == fea_rs::Kind::Float)
            .unwrap();
        FontRevisionStatement {
            revision: revision_token.as_token().unwrap().text.parse().unwrap(),
        }
    }
}
impl PartialEq for FontRevisionStatement {
    fn eq(&self, other: &Self) -> bool {
        (self.revision * 1000.0).round() == (other.revision * 1000.0).round()
    }
}
impl Eq for FontRevisionStatement {}

/// A glyph class definition
///
/// Example: `@UPPERCASE = [A-Z];`
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct GlyphClassDefinition {
    /// class name as a string, without initial ``@``
    pub name: String,
    /// The glyphs in the class
    pub glyphs: GlyphClass,
    /// The location of the definition in the source feature file
    pub location: Range<usize>,
}
impl GlyphClassDefinition {
    /// Create a new glyph class definition.
    pub fn new(name: String, glyphs: GlyphClass, location: Range<usize>) -> Self {
        Self {
            name,
            glyphs,
            location,
        }
    }
}
impl AsFea for GlyphClassDefinition {
    fn as_fea(&self, _indent: &str) -> String {
        format!("@{} = {};", self.name, self.glyphs.as_fea(""))
    }
}
impl From<fea_rs::typed::GlyphClassDef> for GlyphClassDefinition {
    fn from(val: fea_rs::typed::GlyphClassDef) -> Self {
        let label = val
            .iter()
            .find_map(fea_rs::typed::GlyphClassName::cast)
            .unwrap();
        let members: fea_rs::typed::GlyphClassLiteral = val
            .iter()
            .find_map(fea_rs::typed::GlyphClassLiteral::cast)
            .unwrap();
        GlyphClassDefinition {
            name: label.text().trim_start_matches('@').to_string(),
            glyphs: members.into(),
            location: val.node().range(),
        }
    }
}

/// A ``language`` statement within a feature
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct LanguageStatement {
    /// The OpenType language tag for the language
    pub tag: String,
    /// Whether to include the default language system
    pub include_dflt: bool,
    /// Whether the language is required
    pub required: bool,
}
impl LanguageStatement {
    /// Create a new `language` statement.
    pub fn new(tag: String, include_dflt: bool, required: bool) -> Self {
        Self {
            tag,
            include_dflt,
            required,
        }
    }
}
impl AsFea for LanguageStatement {
    fn as_fea(&self, _indent: &str) -> String {
        format!(
            "language {}{}{};",
            self.tag,
            if !self.include_dflt {
                " exclude_dflt"
            } else {
                ""
            },
            if self.required { " required" } else { "" },
        )
    }
}
impl From<fea_rs::typed::Language> for LanguageStatement {
    fn from(language: fea_rs::typed::Language) -> Self {
        let exclude_dflt = language
            .iter()
            .any(|t| t.kind() == fea_rs::Kind::ExcludeDfltKw);
        let required = language
            .iter()
            .any(|t| t.kind() == fea_rs::Kind::RequiredKw);
        Self::new(
            language
                .iter()
                .find_map(Tag::cast)
                .unwrap()
                .text()
                .to_string(),
            !exclude_dflt,
            required,
        )
    }
}

/// A top-level ``languagesystem`` statement.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct LanguageSystemStatement {
    /// The OpenType script tag for the script
    pub script: String,
    /// The OpenType language tag for the language
    pub language: String,
}
impl LanguageSystemStatement {
    /// Create a new `languagesystem` statement.
    pub fn new(script: String, language: String) -> Self {
        Self { script, language }
    }
}
impl AsFea for LanguageSystemStatement {
    fn as_fea(&self, _indent: &str) -> String {
        format!(
            "languagesystem {} {};",
            self.script,
            self.language.trim_ascii_end()
        )
    }
}
impl From<fea_rs::typed::LanguageSystem> for LanguageSystemStatement {
    fn from(langsys: fea_rs::typed::LanguageSystem) -> Self {
        let mut tags = langsys.iter().filter_map(Tag::cast);
        let script = tags.next().unwrap().text().to_string();
        let language = tags.next().unwrap().text().to_string();
        Self::new(script, language)
    }
}

/// Represents a ``lookup ...;`` statement to include a lookup in a feature.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct LookupReferenceStatement {
    /// The name of the lookup to include
    ///
    /// Note: unlike in Python's fontTools, this is simply the name of the
    /// lookup rather than a `LookupBlock` object.
    pub lookup_name: String,
    /// The location of the statement in the source feature file
    pub location: Range<usize>,
}
impl LookupReferenceStatement {
    /// Create a new lookup reference statement.
    pub fn new(lookup_name: String, location: Range<usize>) -> Self {
        Self {
            lookup_name,
            location,
        }
    }
}
impl AsFea for LookupReferenceStatement {
    fn as_fea(&self, _indent: &str) -> String {
        format!("lookup {};", self.lookup_name)
    }
}
impl From<fea_rs::typed::LookupRef> for LookupReferenceStatement {
    fn from(lookup_ref: fea_rs::typed::LookupRef) -> Self {
        Self::new(
            lookup_ref
                .iter()
                .find(|t| t.kind() == fea_rs::Kind::Ident)
                .unwrap()
                .token_text()
                .unwrap()
                .to_string(),
            lookup_ref.node().range(),
        )
    }
}

/// A ``script`` statement
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ScriptStatement {
    /// The OpenType script tag for the script
    pub tag: String,
}
impl ScriptStatement {
    /// Create a new `script` statement.
    pub fn new(tag: String) -> Self {
        Self { tag }
    }
}
impl AsFea for ScriptStatement {
    fn as_fea(&self, _indent: &str) -> String {
        format!("script {};", self.tag)
    }
}
impl From<fea_rs::typed::Script> for ScriptStatement {
    fn from(script: fea_rs::typed::Script) -> Self {
        Self::new(
            script
                .iter()
                .find_map(Tag::cast)
                .unwrap()
                .text()
                .to_string(),
        )
    }
}

/// Represents a subtable break
#[derive(Debug, Clone, PartialEq, Eq, Default)]
pub struct SubtableStatement;
impl SubtableStatement {
    /// Create a new `subtable;` statement.
    pub fn new() -> Self {
        Self {}
    }
}
impl AsFea for SubtableStatement {
    fn as_fea(&self, _indent: &str) -> String {
        "subtable;".to_string()
    }
}

/// A ``parameters`` statement for the `size` feature.
///
/// Example: `parameters 10.0 0;` or `parameters 10.0 0 80 120;`
///
/// Note: `range_start` and `range_end` are stored in **points** internally,
/// but the FEA format uses **decipoints** (tenths of a point). The conversion
/// is handled automatically during parsing and serialization.
#[derive(Debug, Clone, PartialEq)]
pub struct SizeParameters {
    /// Design size in points
    pub design_size: f64,
    /// Subfamily identifier
    pub subfamily_id: u16,
    /// Range start in points (FEA format stores as decipoints, divided by 10 on read)
    pub range_start: f64,
    /// Range end in points (FEA format stores as decipoints, divided by 10 on read)
    pub range_end: f64,
    /// Location in the source FEA file
    pub location: Range<usize>,
}
impl Eq for SizeParameters {}

impl SizeParameters {
    /// Create a new SizeParameters statement.
    pub fn new(
        design_size: f64,
        subfamily_id: u16,
        range_start: f64,
        range_end: f64,
        location: Range<usize>,
    ) -> Self {
        Self {
            design_size,
            subfamily_id,
            range_start,
            range_end,
            location,
        }
    }
}

impl AsFea for SizeParameters {
    fn as_fea(&self, _indent: &str) -> String {
        let mut res = format!("parameters {:.1} {}", self.design_size, self.subfamily_id);
        if self.range_start != 0.0 || self.range_end != 0.0 {
            res.push_str(&format!(
                " {} {}",
                (self.range_start * 10.0) as i32,
                (self.range_end * 10.0) as i32
            ));
        }
        res.push(';');
        res
    }
}

impl From<fea_rs::typed::Parameters> for SizeParameters {
    fn from(val: fea_rs::typed::Parameters) -> Self {
        // Helper to parse FloatLike into f64
        let parse_float = |fl: fea_rs::typed::FloatLike| -> f64 {
            match fl {
                fea_rs::typed::FloatLike::Float(f) => f.text().parse().unwrap(),
                fea_rs::typed::FloatLike::Number(n) => n.text().parse::<i16>().unwrap() as f64,
            }
        };

        // Extract design_size (first FloatLike)
        let design_size = val
            .iter()
            .find_map(fea_rs::typed::FloatLike::cast)
            .map(parse_float)
            .unwrap();

        // Extract subfamily_id (second number, after the first FloatLike)
        let subfamily_id = val
            .iter()
            .filter(|t| t.kind() == fea_rs::Kind::Number || t.kind() == fea_rs::Kind::Float)
            .nth(1)
            .and_then(fea_rs::typed::Number::cast)
            .map(|n| n.text().parse().unwrap())
            .unwrap();

        // Extract range_start (third FloatLike, if present) - FEA stores in decipoints, convert to points
        let range_start = val
            .iter()
            .filter_map(fea_rs::typed::FloatLike::cast)
            .nth(2)
            .map(|fl| parse_float(fl) / 10.0)
            .unwrap_or(0.0);

        // Extract range_end (fourth FloatLike, if present) - FEA stores in decipoints, convert to points
        let range_end = val
            .iter()
            .filter_map(fea_rs::typed::FloatLike::cast)
            .nth(3)
            .map(|fl| parse_float(fl) / 10.0)
            .unwrap_or(0.0);

        Self::new(
            design_size,
            subfamily_id,
            range_start,
            range_end,
            val.range(),
        )
    }
}

/// A variable layout conditionset.
///
/// Example:
/// ```fea
/// conditionset heavy {
///     wght 700 900;
/// } heavy;
/// ```
#[derive(Debug, Clone, PartialEq)]
pub struct ConditionSet {
    /// The name of this conditionset
    pub name: String,
    /// A map of axis tags to (min, max) userspace coordinates
    pub conditions: Vec<(String, f32, f32)>,
    /// Location in the source FEA file
    pub location: Range<usize>,
}
impl Eq for ConditionSet {}

impl ConditionSet {
    /// Create a new `conditionset` statement.
    pub fn new(name: String, conditions: Vec<(String, f32, f32)>, location: Range<usize>) -> Self {
        Self {
            name,
            conditions,
            location,
        }
    }
}

impl From<fea_rs::typed::ConditionSet> for ConditionSet {
    fn from(val: fea_rs::typed::ConditionSet) -> Self {
        // Extract the label (name)
        let name = val
            .iter()
            .find_map(|t| {
                if t.kind() == fea_rs::Kind::Label {
                    t.as_token().map(|tok| tok.text.to_string())
                } else {
                    None
                }
            })
            .unwrap();

        // Helper to parse numbers as f32
        let parse_number =
            |n: fea_rs::typed::Number| -> f32 { n.text().parse::<i16>().unwrap() as f32 };

        // Extract conditions
        let conditions: Vec<(String, f32, f32)> = val
            .iter()
            .filter_map(fea_rs::typed::Condition::cast)
            .map(|cond| {
                // Get tag
                let tag = cond
                    .iter()
                    .find_map(fea_rs::typed::Tag::cast)
                    .unwrap()
                    .text()
                    .to_string();

                // Get min and max values
                let mut numbers = cond.iter().filter_map(fea_rs::typed::Number::cast);
                let min = parse_number(numbers.next().unwrap());
                let max = parse_number(numbers.next().unwrap());

                (tag, min, max)
            })
            .collect();

        Self::new(name, conditions, val.node().range())
    }
}

impl AsFea for ConditionSet {
    fn as_fea(&self, indent: &str) -> String {
        let mut res = format!("{}conditionset {} {{\n", indent, self.name);
        for (tag, min, max) in &self.conditions {
            // Format numbers nicely - remove trailing zeros and decimal point if integer
            let format_num = |n: &f32| {
                let s = format!("{}", n);
                if s.contains('.') {
                    s.trim_end_matches('0').trim_end_matches('.').to_string()
                } else {
                    s
                }
            };
            res.push_str(&format!(
                "{}\t{} {} {};\n",
                indent,
                tag,
                format_num(min),
                format_num(max)
            ));
        }
        res.push_str(&format!("{}}}", indent));
        res.push_str(&format!(" {};\n", self.name));
        res
    }
}

/// A variable layout variation block.
///
/// Example:
/// ```fea
/// variation rvrn heavy {
///     lookup symbols_heavy;
/// } rvrn;
/// ```
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct VariationBlock {
    /// The feature tag for this variation
    pub name: SmolStr,
    /// The name of the conditionset this variation applies to
    pub conditionset: String,
    /// Statements within this variation block
    pub statements: Vec<Statement>,
    /// Whether to use extension subtables
    pub use_extension: bool,
    /// Location in the source FEA file
    pub location: Range<usize>,
}

impl VariationBlock {
    /// Create a new `variation ABCD { ... } ABCD;` statement.
    pub fn new(
        name: SmolStr,
        conditionset: String,
        statements: Vec<Statement>,
        use_extension: bool,
        location: Range<usize>,
    ) -> Self {
        Self {
            name,
            conditionset,
            statements,
            use_extension,
            location,
        }
    }
}

impl From<fea_rs::typed::FeatureVariation> for VariationBlock {
    fn from(val: fea_rs::typed::FeatureVariation) -> Self {
        // Extract the feature tag (first tag)
        let name = val
            .iter()
            .find_map(fea_rs::typed::Tag::cast)
            .map(|tag| SmolStr::new(tag.text()))
            .unwrap();

        // Extract conditionset name - it's a label/identifier after the tag
        let conditionset = val
            .iter()
            .skip_while(|t| t.kind() != fea_rs::Kind::Tag) // skip to tag
            .skip(1) // skip the tag itself
            .find_map(|t| {
                if t.kind() == fea_rs::Kind::Label || t.kind() == fea_rs::Kind::Ident {
                    t.as_token().map(|tok| tok.text.to_string())
                } else {
                    None
                }
            })
            .unwrap_or_default();

        // Check for useExtension flag
        let use_extension = val.iter().any(|t| t.kind() == fea_rs::Kind::UseExtensionKw);

        // Parse statements within the block
        let statements: Vec<Statement> = val
            .node()
            .iter_children()
            .filter_map(crate::to_statement)
            .collect();

        Self::new(
            name,
            conditionset,
            statements,
            use_extension,
            val.node().range(),
        )
    }
}

impl AsFea for VariationBlock {
    fn as_fea(&self, indent: &str) -> String {
        let mut res = format!("{}variation {} {}", indent, self.name, self.conditionset);
        if self.use_extension {
            res.push_str(" useExtension");
        }
        res.push_str(" {\n");

        let mid_indent = indent.to_string() + SHIFT;
        for stmt in &self.statements {
            res.push_str(&stmt.as_fea(&mid_indent));
            res.push('\n');
        }

        res.push_str(&format!("{}}} {};\n", indent, self.name));
        res
    }
}

/// A ``lookupflag`` statement
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct LookupFlagStatement {
    /// The value of the flag
    pub value: u16,
    /// Optional MarkAttachmentType
    pub mark_attachment: Option<GlyphContainer>,
    /// Optional UseMarkFilteringSet
    pub mark_filtering_set: Option<GlyphContainer>,
    /// Location in the source FEA file
    pub location: Range<usize>,
}

impl LookupFlagStatement {
    /// Create a new `lookupflag` statement.
    pub fn new(
        value: u16,
        mark_attachment: Option<GlyphContainer>,
        mark_filtering_set: Option<GlyphContainer>,
        location: Range<usize>,
    ) -> Self {
        Self {
            value,
            mark_attachment,
            mark_filtering_set,
            location,
        }
    }
}

impl AsFea for LookupFlagStatement {
    fn as_fea(&self, _indent: &str) -> String {
        let mut res = Vec::new();
        let flags = [
            "RightToLeft",
            "IgnoreBaseGlyphs",
            "IgnoreLigatures",
            "IgnoreMarks",
        ];
        let mut curr = 1u16;
        for flag in &flags {
            if self.value & curr != 0 {
                res.push(flag.to_string());
            }
            curr <<= 1;
        }
        if let Some(mark_attachment) = &self.mark_attachment {
            res.push(format!("MarkAttachmentType {}", mark_attachment.as_fea("")));
        }
        if let Some(mark_filtering_set) = &self.mark_filtering_set {
            res.push(format!(
                "UseMarkFilteringSet {}",
                mark_filtering_set.as_fea("")
            ));
        }
        if res.is_empty() {
            res.push("0".to_string());
        }
        format!("lookupflag {};", res.join(" "))
    }
}

impl From<fea_rs::typed::LookupFlag> for LookupFlagStatement {
    fn from(val: fea_rs::typed::LookupFlag) -> Self {
        let mut value = 0u16;
        // Check for a numeric value
        if let Some(number) = val.iter().find_map(fea_rs::typed::Number::cast) {
            value = number.text().parse().unwrap();
        } else {
            for item in val.iter() {
                match item.kind() {
                    fea_rs::Kind::RightToLeftKw => value |= 1,
                    fea_rs::Kind::IgnoreBaseGlyphsKw => value |= 2,
                    fea_rs::Kind::IgnoreLigaturesKw => value |= 4,
                    fea_rs::Kind::IgnoreMarksKw => value |= 8,
                    _ => {}
                }
            }
        }

        // Collect all items and process MarkAttachment and UseMarkFilteringSet
        let mark_attachment = val
            .iter()
            .skip_while(|k| k.kind() != fea_rs::Kind::MarkAttachmentTypeKw)
            .find_map(|gc| fea_rs::typed::GlyphClass::cast(gc).map(|g| g.into()));
        let mark_filtering_set = val
            .iter()
            .skip_while(|k| k.kind() != fea_rs::Kind::UseMarkFilteringSetKw)
            .find_map(|gc| fea_rs::typed::GlyphClass::cast(gc).map(|g| g.into()));

        LookupFlagStatement::new(value, mark_attachment, mark_filtering_set, val.range())
    }
}

/// A definition of a glyph in a mark class, associating it with an anchor point.
///
/// See the notes for [`MarkClass`] to understand how this differs from the
/// Python `fontTools` representation.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct MarkClassDefinition {
    /// The name of the mark class
    pub mark_class: MarkClass,
    /// The anchor associated with this mark class glyph
    pub anchor: crate::Anchor,
    /// The glyphs in this mark class
    pub glyphs: GlyphContainer,
}
impl MarkClassDefinition {
    /// Create a new `markClass` definition.
    pub fn new(mark_class: MarkClass, anchor: crate::Anchor, glyphs: GlyphContainer) -> Self {
        Self {
            mark_class,
            anchor,
            glyphs,
        }
    }
}
impl AsFea for MarkClassDefinition {
    fn as_fea(&self, _indent: &str) -> String {
        format!(
            "markClass {} {} @{};",
            self.glyphs.as_fea(""),
            self.anchor.as_fea(""),
            self.mark_class.name,
        )
    }
}
impl From<fea_rs::typed::MarkClassDef> for MarkClassDefinition {
    fn from(val: fea_rs::typed::MarkClassDef) -> Self {
        // Glyphs are the first GlyphOrClass
        let glyphs_node = val
            .iter()
            .find_map(fea_rs::typed::GlyphOrClass::cast)
            .unwrap();
        // Anchor is the first Anchor
        let anchor_node = val.iter().find_map(fea_rs::typed::Anchor::cast).unwrap();
        let anchor = from_anchor(anchor_node).unwrap();
        // MarkClass name is the GlyphClassName after the anchor
        let mark_class_node = val
            .iter()
            .skip_while(|t| t.kind() != fea_rs::Kind::AnchorNode)
            .find_map(fea_rs::typed::GlyphClassName::cast)
            .unwrap();
        let mark_class = MarkClass::new(mark_class_node.text().trim_start_matches('@'));
        MarkClassDefinition::new(mark_class, anchor, GlyphContainer::from(glyphs_node))
    }
}

/// Represents a named value record definition.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ValueRecordDefinition {
    /// The name of the value record
    pub name: SmolStr,
    /// The value record data
    pub value: ValueRecord,
    /// The location of the definition in the source feature file
    pub location: Range<usize>,
}
impl ValueRecordDefinition {
    /// Create a new value record definition.
    pub fn new(name: SmolStr, value: ValueRecord, location: Range<usize>) -> Self {
        Self {
            name,
            value,
            location,
        }
    }
}

impl AsFea for ValueRecordDefinition {
    fn as_fea(&self, _indent: &str) -> String {
        format!("valueRecordDef {} {};", self.value.as_fea(""), self.name)
    }
}

impl From<fea_rs::typed::ValueRecordDef> for ValueRecordDefinition {
    fn from(val: fea_rs::typed::ValueRecordDef) -> Self {
        let name = val
            .iter()
            .find(|t| t.kind() == fea_rs::Kind::Ident)
            .unwrap();
        let value_record_node = val
            .iter()
            .find_map(fea_rs::typed::ValueRecord::cast)
            .unwrap();
        ValueRecordDefinition::new(
            name.as_token().unwrap().text.clone(),
            ValueRecord::from(value_record_node),
            val.node().range(),
        )
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{GlyphContainer, GlyphName};

    #[test]
    fn test_roundtrip_lookupflag_simple() {
        const FEA: &str = "lookup test { lookupflag RightToLeft; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let lookup = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::LookupBlock::cast)
            .unwrap();
        let lookupflag = lookup
            .node()
            .iter_children()
            .find_map(fea_rs::typed::LookupFlag::cast)
            .unwrap();
        let stmt = LookupFlagStatement::from(lookupflag);
        assert_eq!(stmt.value, 1);
        assert_eq!(stmt.as_fea(""), "lookupflag RightToLeft;");
    }

    #[test]
    fn test_roundtrip_lookupflag_multiple() {
        const FEA: &str = "lookup test { lookupflag RightToLeft IgnoreMarks; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let lookup = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::LookupBlock::cast)
            .unwrap();
        let lookupflag = lookup
            .node()
            .iter_children()
            .find_map(fea_rs::typed::LookupFlag::cast)
            .unwrap();
        let stmt = LookupFlagStatement::from(lookupflag);
        assert_eq!(stmt.value, 9); // 1 + 8
        assert_eq!(stmt.as_fea(""), "lookupflag RightToLeft IgnoreMarks;");
    }

    #[test]
    fn test_roundtrip_lookupflag_zero() {
        const FEA: &str = "lookup test { lookupflag 0; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let lookup = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::LookupBlock::cast)
            .unwrap();
        let lookupflag = lookup
            .node()
            .iter_children()
            .find_map(fea_rs::typed::LookupFlag::cast)
            .unwrap();
        let stmt = LookupFlagStatement::from(lookupflag);
        assert_eq!(stmt.value, 0);
        assert_eq!(stmt.as_fea(""), "lookupflag 0;");
    }

    #[test]
    fn test_generate_lookupflag() {
        let stmt = LookupFlagStatement::new(
            10, // IgnoreBaseGlyphs (2) + IgnoreMarks (8)
            None,
            None,
            0..0,
        );
        assert_eq!(stmt.as_fea(""), "lookupflag IgnoreBaseGlyphs IgnoreMarks;");
    }

    #[test]
    fn test_generate_lookupflag_with_mark_attachment() {
        let stmt = LookupFlagStatement::new(
            0,
            Some(GlyphContainer::GlyphClass(GlyphClass::new(
                vec![
                    GlyphContainer::GlyphName(GlyphName::new("acute")),
                    GlyphContainer::GlyphName(GlyphName::new("grave")),
                ],
                0..0,
            ))),
            None,
            0..0,
        );
        assert_eq!(
            stmt.as_fea(""),
            "lookupflag MarkAttachmentType [acute grave];"
        );
    }

    // AnchorDefinition tests
    #[test]
    fn test_roundtrip_anchordef_simple() {
        const FEA: &str = "anchorDef 300 100 ANCHOR_1;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let anchor_def = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::AnchorDef::cast)
            .unwrap();
        let stmt = AnchorDefinition::from(anchor_def);
        assert_eq!(stmt.x, 300.into());
        assert_eq!(stmt.y, 100.into());
        assert_eq!(stmt.name, "ANCHOR_1");
        assert_eq!(stmt.contourpoint, None);
        assert_eq!(stmt.as_fea(""), "anchorDef 300 100 ANCHOR_1;");
    }

    #[test]
    fn test_roundtrip_anchordef_contourpoint() {
        const FEA: &str = "anchorDef 300 100 contourpoint 5 ANCHOR_1;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let anchor_def = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::AnchorDef::cast)
            .unwrap();
        let stmt = AnchorDefinition::from(anchor_def);
        assert_eq!(stmt.x, 300.into());
        assert_eq!(stmt.y, 100.into());
        assert_eq!(stmt.contourpoint, Some(5));
        assert_eq!(stmt.name, "ANCHOR_1");
        assert_eq!(
            stmt.as_fea(""),
            "anchorDef 300 100 contourpoint 5 ANCHOR_1;"
        );
    }

    #[test]
    fn test_generation_anchordef() {
        let stmt = AnchorDefinition::new(150.into(), (-50).into(), None, "BASE".to_string(), 0..0);
        assert_eq!(stmt.as_fea(""), "anchorDef 150 -50 BASE;");
    }

    // FeatureReferenceStatement tests
    #[test]
    fn test_roundtrip_featurereference() {
        const FEA: &str = "feature test { feature salt; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let feature = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::Feature::cast)
            .unwrap();
        let feature_ref = feature
            .node()
            .iter_children()
            .find_map(fea_rs::typed::FeatureRef::cast)
            .unwrap();
        let stmt = FeatureReferenceStatement::from(feature_ref);
        assert_eq!(stmt.feature_name, "salt");
        assert_eq!(stmt.as_fea(""), "feature salt;");
    }

    #[test]
    fn test_generation_featurereference() {
        let stmt = FeatureReferenceStatement::new("liga".to_string());
        assert_eq!(stmt.as_fea(""), "feature liga;");
    }

    // FontRevisionStatement tests
    #[test]
    fn test_roundtrip_fontrevision() {
        const FEA: &str = "table head { FontRevision 2.500; } head;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let table = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::HeadTable::cast)
            .unwrap();
        let font_rev = table
            .node()
            .iter_children()
            .find_map(fea_rs::typed::HeadFontRevision::cast)
            .unwrap();
        let stmt = FontRevisionStatement::from(font_rev);
        assert_eq!(stmt.revision, 2.5);
        assert_eq!(stmt.as_fea(""), "FontRevision 2.500;");
    }

    #[test]
    fn test_generation_fontrevision() {
        let stmt = FontRevisionStatement::new(1.125);
        assert_eq!(stmt.as_fea(""), "FontRevision 1.125;");
    }

    // GlyphClassDefinition tests
    #[test]
    fn test_roundtrip_glyphclassdef() {
        const FEA: &str = "@UPPERCASE = [A B C D E F];";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let glyph_class_def = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::GlyphClassDef::cast)
            .unwrap();
        let stmt = GlyphClassDefinition::from(glyph_class_def);
        assert_eq!(stmt.name, "UPPERCASE");
        assert_eq!(stmt.glyphs.glyphs.len(), 6);
        assert_eq!(stmt.as_fea(""), "@UPPERCASE = [A B C D E F];");
    }

    #[test]
    fn test_generation_glyphclassdef() {
        let glyphs = GlyphClass::new(
            vec![
                GlyphContainer::GlyphName(GlyphName::new("a")),
                GlyphContainer::GlyphName(GlyphName::new("b")),
                GlyphContainer::GlyphName(GlyphName::new("c")),
            ],
            0..0,
        );
        let stmt = GlyphClassDefinition::new("lowercase".to_string(), glyphs, 0..0);
        assert_eq!(stmt.as_fea(""), "@lowercase = [a b c];");
    }

    // LanguageStatement tests
    #[test]
    fn test_roundtrip_language() {
        const FEA: &str = "feature test { language TRK; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let feature = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::Feature::cast)
            .unwrap();
        let lang = feature
            .node()
            .iter_children()
            .find_map(fea_rs::typed::Language::cast)
            .unwrap();
        let stmt = LanguageStatement::from(lang);
        // Note: tag includes any trailing spaces from the source
        assert_eq!(stmt.as_fea(""), "language TRK;");
    }

    #[test]
    fn test_generation_language() {
        let stmt = LanguageStatement::new("DEU ".to_string(), true, false);
        assert_eq!(stmt.as_fea(""), "language DEU ;");
    }

    // LanguageSystemStatement tests
    #[test]
    fn test_roundtrip_languagesystem() {
        const FEA: &str = "languagesystem latn dflt;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let langsys = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::LanguageSystem::cast)
            .unwrap();
        let stmt = LanguageSystemStatement::from(langsys);
        assert_eq!(stmt.script, "latn");
        assert_eq!(stmt.language, "dflt");
        assert_eq!(stmt.as_fea(""), "languagesystem latn dflt;");
    }

    #[test]
    fn test_generation_languagesystem() {
        let stmt = LanguageSystemStatement::new("cyrl".to_string(), "SRB ".to_string());
        assert_eq!(stmt.as_fea(""), "languagesystem cyrl SRB;");
    }

    // ScriptStatement tests
    #[test]
    fn test_roundtrip_script() {
        const FEA: &str = "feature test { script latn; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let feature = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::Feature::cast)
            .unwrap();
        let script = feature
            .node()
            .iter_children()
            .find_map(fea_rs::typed::Script::cast)
            .unwrap();
        let stmt = ScriptStatement::from(script);
        assert_eq!(stmt.tag, "latn");
        assert_eq!(stmt.as_fea(""), "script latn;");
    }

    #[test]
    fn test_generation_script() {
        let stmt = ScriptStatement::new("arab".to_string());
        assert_eq!(stmt.as_fea(""), "script arab;");
    }

    // SubtableStatement tests
    #[test]
    fn test_generation_subtable() {
        let stmt = SubtableStatement::new();
        assert_eq!(stmt.as_fea(""), "subtable;");
    }

    // LookupReferenceStatement tests
    #[test]
    fn test_roundtrip_lookupreference() {
        const FEA: &str = "feature test { lookup myLookup; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let feature = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::Feature::cast)
            .unwrap();
        let lookup_ref = feature
            .node()
            .iter_children()
            .find_map(fea_rs::typed::LookupRef::cast)
            .unwrap();
        let stmt = LookupReferenceStatement::from(lookup_ref);
        assert_eq!(stmt.lookup_name, "myLookup");
        assert_eq!(stmt.as_fea(""), "lookup myLookup;");
    }

    #[test]
    fn test_generation_lookupreference() {
        let stmt = LookupReferenceStatement::new("anotherLookup".to_string(), 0..0);
        assert_eq!(stmt.as_fea(""), "lookup anotherLookup;");
    }

    // SizeParameters tests
    #[test]
    fn test_roundtrip_sizeparameters_simple() {
        const FEA: &str = "feature size { parameters 10.0 0; } size;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let feature = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::Feature::cast)
            .unwrap();
        let params = feature
            .node()
            .iter_children()
            .find_map(fea_rs::typed::Parameters::cast)
            .unwrap();
        let stmt = SizeParameters::from(params);
        assert_eq!(stmt.design_size, 10.0);
        assert_eq!(stmt.subfamily_id, 0);
        assert_eq!(stmt.range_start, 0.0);
        assert_eq!(stmt.range_end, 0.0);
        assert_eq!(stmt.as_fea(""), "parameters 10.0 0;");
    }

    #[test]
    fn test_roundtrip_sizeparameters_with_range() {
        const FEA: &str = "feature size { parameters 10.0 0 80 120; } size;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let feature = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::Feature::cast)
            .unwrap();
        let params = feature
            .node()
            .iter_children()
            .find_map(fea_rs::typed::Parameters::cast)
            .unwrap();
        let stmt = SizeParameters::from(params);
        assert_eq!(stmt.design_size, 10.0);
        assert_eq!(stmt.subfamily_id, 0);
        assert_eq!(stmt.range_start, 8.0); // 80 decipoints = 8.0 points
        assert_eq!(stmt.range_end, 12.0); // 120 decipoints = 12.0 points
        assert_eq!(stmt.as_fea(""), "parameters 10.0 0 80 120;");
    }

    #[test]
    fn test_generate_sizeparameters() {
        let stmt = SizeParameters::new(12.5, 1, 100.0, 150.0, 0..0);
        assert_eq!(stmt.as_fea(""), "parameters 12.5 1 1000 1500;");
    }

    #[test]
    fn test_generation_lookupflag() {
        let stmt = LookupFlagStatement::new(
            0,
            Some(GlyphContainer::GlyphClass(GlyphClass::new(
                vec![
                    GlyphContainer::GlyphName(GlyphName::new("acute")),
                    GlyphContainer::GlyphName(GlyphName::new("grave")),
                ],
                0..0,
            ))),
            None,
            0..0,
        );
        assert_eq!(
            stmt.as_fea(""),
            "lookupflag MarkAttachmentType [acute grave];"
        );
        let stmt = LookupFlagStatement::new(
            9,
            None,
            Some(GlyphContainer::GlyphClass(GlyphClass::new(
                vec![
                    GlyphContainer::GlyphName(GlyphName::new("acute")),
                    GlyphContainer::GlyphName(GlyphName::new("grave")),
                ],
                0..0,
            ))),
            0..0,
        );
        assert_eq!(
            stmt.as_fea(""),
            "lookupflag RightToLeft IgnoreMarks UseMarkFilteringSet [acute grave];"
        );
    }

    #[test]
    fn test_roundtrip_lookupflag() {
        const FEA: &str = "lookup test { lookupflag RightToLeft IgnoreMarks UseMarkFilteringSet [acute grave]; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let lookup = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::LookupBlock::cast)
            .unwrap();
        let lookupflag = lookup
            .node()
            .iter_children()
            .find_map(fea_rs::typed::LookupFlag::cast)
            .unwrap();
        let stmt = LookupFlagStatement::from(lookupflag);
        assert_eq!(stmt.value, 9); // 1 + 8
        assert_eq!(
            stmt.clone().mark_filtering_set.unwrap().as_fea(""),
            "[acute grave]"
        );
        assert_eq!(
            stmt.as_fea(""),
            "lookupflag RightToLeft IgnoreMarks UseMarkFilteringSet [acute grave];"
        );

        const FEA2: &str =
            "lookup test { lookupflag RightToLeft IgnoreMarks MarkAttachmentType @foo; } test;";
        let (parsed, _) = fea_rs::parse::parse_string(FEA2);
        let lookup = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::LookupBlock::cast)
            .unwrap();
        let lookupflag = lookup
            .node()
            .iter_children()
            .find_map(fea_rs::typed::LookupFlag::cast)
            .unwrap();
        let stmt = LookupFlagStatement::from(lookupflag);
        assert_eq!(stmt.value, 9);
        assert_eq!(
            stmt.as_fea(""),
            "lookupflag RightToLeft IgnoreMarks MarkAttachmentType @foo;"
        );
    }

    // ConditionSet tests
    #[test]
    fn test_roundtrip_conditionset_simple() {
        const FEA: &str = r#"conditionset heavy {
	wght 700 900;
} heavy;"#;
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let condset = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::ConditionSet::cast)
            .unwrap();
        let stmt = ConditionSet::from(condset);
        assert_eq!(stmt.name, "heavy");
        assert_eq!(stmt.conditions.len(), 1);
        assert_eq!(stmt.conditions[0].0, "wght");
        assert_eq!(stmt.conditions[0].1, 700.0);
        assert_eq!(stmt.conditions[0].2, 900.0);

        let output = stmt.as_fea("");
        assert!(output.contains("conditionset heavy"));
        assert!(output.contains("wght 700 900"));
    }

    #[test]
    fn test_roundtrip_conditionset_multiple_conditions() {
        const FEA: &str = r#"conditionset complex {
	wght 400 700;
	wdth 75 100;
} complex;"#;
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let condset = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::ConditionSet::cast)
            .unwrap();
        let stmt = ConditionSet::from(condset);
        assert_eq!(stmt.name, "complex");
        assert_eq!(stmt.conditions.len(), 2);
        assert_eq!(stmt.conditions[0], ("wght".to_string(), 400.0, 700.0));
        assert_eq!(stmt.conditions[1], ("wdth".to_string(), 75.0, 100.0));

        let output = stmt.as_fea("");
        assert!(output.contains("conditionset complex"));
        assert!(output.contains("wght 400 700"));
        assert!(output.contains("wdth 75 100"));
    }

    #[test]
    fn test_roundtrip_conditionset_from_file() {
        const FEA: &str = include_str!("../resources/test/variable_conditionset.fea");
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let condset = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::ConditionSet::cast)
            .unwrap();
        let stmt = ConditionSet::from(condset);
        assert_eq!(stmt.name, "heavy");
        assert_eq!(stmt.conditions.len(), 1);
        assert_eq!(stmt.conditions[0], ("wght".to_string(), 700.0, 900.0));
    }

    #[test]
    fn test_generate_conditionset() {
        let stmt = ConditionSet::new(
            "myCondition".to_string(),
            vec![
                ("wght".to_string(), 300.0, 500.0),
                ("opsz".to_string(), 8.0, 12.0),
            ],
            0..0,
        );

        let output = stmt.as_fea("");
        assert!(output.contains("conditionset myCondition"));
        assert!(output.contains("wght 300 500"));
        assert!(output.contains("opsz 8 12"));
        assert!(output.contains("} myCondition;"));
    }

    #[test]
    fn test_conditionset_integration() {
        // Test that ConditionSet can be parsed as a top-level item
        const FEA: &str = r#"languagesystem DFLT dflt;

conditionset heavy {
    wght 700 900;
} heavy;"#;

        let ff = crate::FeatureFile::new_from_fea(FEA, None::<&[&str]>, None::<&str>).unwrap();
        assert_eq!(ff.statements.len(), 2);

        // Check that conditionset is in the statements
        let cs = ff
            .statements
            .iter()
            .find_map(|item| {
                if let crate::ToplevelItem::ConditionSet(cs) = item {
                    Some(cs)
                } else {
                    None
                }
            })
            .expect("Should have found ConditionSet");

        assert_eq!(cs.name, "heavy");
        assert_eq!(cs.conditions.len(), 1);
        assert_eq!(cs.conditions[0], ("wght".to_string(), 700.0, 900.0));

        // Test round-trip
        let output = cs.as_fea("");
        assert!(output.contains("conditionset heavy"));
        assert!(output.contains("wght 700 900"));
    }

    // VariationBlock tests
    #[test]
    fn test_roundtrip_variationblock() {
        const FEA: &str = include_str!("../resources/test/variable_conditionset.fea");
        let (parsed, _) = fea_rs::parse::parse_string(FEA);
        let variation = parsed
            .root()
            .iter_children()
            .find_map(fea_rs::typed::FeatureVariation::cast)
            .unwrap();
        let stmt = VariationBlock::from(variation);

        assert_eq!(stmt.name, "rvrn");
        assert_eq!(stmt.conditionset, "heavy");
        assert_eq!(stmt.statements.len(), 1);

        let output = stmt.as_fea("");
        assert!(output.contains("variation rvrn heavy"));
        assert!(output.contains("lookup symbols_heavy"));
    }

    #[test]
    fn test_generate_variationblock() {
        let stmt = VariationBlock::new(
            "rvrn".into(),
            "myCondition".to_string(),
            vec![crate::Statement::Comment(crate::Comment::from("# Test"))],
            false,
            0..0,
        );

        let output = stmt.as_fea("");
        assert!(output.contains("variation rvrn myCondition"));
        assert!(output.contains("# Test"));
        assert!(output.contains("} rvrn;"));
    }

    #[test]
    fn test_variationblock_integration() {
        const FEA: &str = include_str!("../resources/test/variable_conditionset.fea");

        let ff = crate::FeatureFile::new_from_fea(FEA, None::<&[&str]>, None::<&str>).unwrap();

        // Should have: languagesystem, lookup, conditionset, variation
        assert!(ff.statements.len() >= 4);

        // Check that variation block is in the statements
        let vb = ff
            .statements
            .iter()
            .find_map(|item| {
                if let crate::ToplevelItem::VariationBlock(vb) = item {
                    Some(vb)
                } else {
                    None
                }
            })
            .expect("Should have found VariationBlock");

        assert_eq!(vb.name.as_str(), "rvrn");
        assert_eq!(vb.conditionset, "heavy");
    }
}