iris_cssom/

computed.rs

//! Style computation: cascade, inheritance, and resolved values.

use crate::StyleSheet;
use std::collections::HashMap;

/// Resolved, computed styles for a single element.
#[derive(Debug, Clone, Default)]
pub struct ComputedStyle {
    pub properties: HashMap<String, String>,
}

impl ComputedStyle {
    pub fn get(&self, prop: &str) -> Option<&str> {
        self.properties.get(prop).map(|s| s.as_str())
    }

    pub fn set(&mut self, prop: &str, value: &str) {
        self.properties.insert(prop.to_string(), value.to_string());
    }
}

/// Compute styles for a set of classes + tag, cascading from multiple stylesheets.
pub fn compute_styles(
    sheets: &[StyleSheet],
    classes: &[String],
    tag: &str,
    parent_style: Option<&ComputedStyle>,
) -> ComputedStyle {
    let mut map = HashMap::new();

    // 1. Parent inherited properties
    if let Some(parent) = parent_style {
        for (prop, val) in &parent.properties {
            if INHERITED_PROPS.contains(&prop.as_str()) {
                map.entry(prop.clone()).or_insert_with(|| val.clone());
            }
        }
    }

    // 2. Tag selectors (lowest priority)
    for sheet in sheets {
        for decl in sheet.declarations_for_tag(tag) {
            map.entry(decl.property.clone()).or_insert(decl.value);
        }
    }

    // 3. Class selectors (higher priority)
    for sheet in sheets {
        for class in classes {
            for decl in sheet.declarations_for_class(class) {
                map.insert(decl.property, decl.value);
            }
        }
    }

    ComputedStyle { properties: map }
}

/// Properties that inherit from parent by default.
const INHERITED_PROPS: &[&str] = &[
    "color", "font-family", "font-size", "font-style", "font-weight",
    "line-height", "text-align", "visibility", "cursor",
    "direction", "letter-spacing", "word-spacing", "white-space",
];

/// Parse a CSS pixel value (e.g., "16px" → 16.0).
pub fn parse_px(value: &str) -> f32 {
    let value = value.trim();
    if value.ends_with("px") {
        value[..value.len()-2].trim().parse().unwrap_or(0.0)
    } else {
        value.parse().unwrap_or(0.0)
    }
}

/// Convert CSS color string to RGBA tuple (r, g, b, a) in 0-1 range.
/// Supports: #hex, rgb(), rgba(), named colors.
pub fn parse_color(value: &str) -> (f32, f32, f32, f32) {
    let value = value.trim().to_lowercase();

    // Named colors (simplified)
    if let Some(rgba) = named_color(&value) { return rgba; }

    // #hex
    if value.starts_with('#') {
        return parse_hex_color(&value);
    }

    // rgb() / rgba()
    if value.starts_with("rgb") {
        let inner = value.trim_start_matches("rgba(").trim_start_matches("rgb(")
            .trim_end_matches(')');
        let parts: Vec<f32> = inner.split(',').filter_map(|s| {
            let trimmed = s.trim();
            if trimmed.ends_with('%') {
                trimmed[..trimmed.len()-1].parse::<f32>().ok().map(|v| v / 100.0 * 255.0)
            } else {
                trimmed.parse::<f32>().ok()
            }
        }).collect();
        if parts.len() >= 3 {
            return (parts[0] / 255.0, parts[1] / 255.0, parts[2] / 255.0,
                    parts.get(3).copied().unwrap_or(255.0) / 255.0);
        }
    }

    (0.0, 0.0, 0.0, 1.0) // default black
}

fn parse_hex_color(hex: &str) -> (f32, f32, f32, f32) {
    let hex = hex.trim_start_matches('#');
    // 支持 3 位 hex: #rgb → #rrggbb
    let expanded;
    let hex = if hex.len() == 3 {
        expanded = hex.chars()
            .flat_map(|c| std::iter::repeat(c).take(2))
            .collect::<String>();
        expanded.as_str()
    } else {
        hex
    };
    // 支持 4 位 hex: #rgba → #rrggbbaa
    let expanded4;
    let hex = if hex.len() == 4 {
        expanded4 = hex.chars()
            .flat_map(|c| std::iter::repeat(c).take(2))
            .collect::<String>();
        expanded4.as_str()
    } else {
        hex
    };
    if hex.len() < 6 {
        return (0.0, 0.0, 0.0, 1.0);
    }
    let r = u8::from_str_radix(&hex[0..2], 16).unwrap_or(0) as f32 / 255.0;
    let g = u8::from_str_radix(&hex[2..4], 16).unwrap_or(0) as f32 / 255.0;
    let b = u8::from_str_radix(&hex[4..6], 16).unwrap_or(0) as f32 / 255.0;
    let a = if hex.len() >= 8 {
        u8::from_str_radix(&hex[6..8], 16).unwrap_or(255) as f32 / 255.0
    } else { 1.0 };
    (r, g, b, a)
}

fn named_color(name: &str) -> Option<(f32, f32, f32, f32)> {
    // ✅ 统一颜色名列表，与 iris-engine 保持一致
    let (r, g, b, has_alpha) = match name {
        "red" => (1.0, 0.0, 0.0, false),
        "green" => (0.0, 0.502, 0.0, false), // ✅ 统一为 0.502（与 engine 一致）
        "blue" => (0.0, 0.0, 1.0, false),
        "white" => (1.0, 1.0, 1.0, false),
        "black" => (0.0, 0.0, 0.0, false),
        "gray" | "grey" => (0.5, 0.5, 0.5, false),
        "silver" => (0.75, 0.75, 0.75, false),
        "yellow" => (1.0, 1.0, 0.0, false),
        "orange" => (1.0, 0.65, 0.0, false),
        "purple" => (0.5, 0.0, 0.5, false),
        "pink" => (1.0, 0.75, 0.8, false),
        "brown" => (0.65, 0.16, 0.16, false),
        "transparent" => (0.0, 0.0, 0.0, true),
        // ✅ 新增与 engine 一致的颜色
        "cyan" => (0.0, 1.0, 1.0, false),
        "magenta" => (1.0, 0.0, 1.0, false),
        "maroon" => (0.5, 0.0, 0.0, false),
        "olive" => (0.5, 0.5, 0.0, false),
        "lime" => (0.0, 1.0, 0.0, false),
        "teal" => (0.0, 0.5, 0.5, false),
        "navy" => (0.0, 0.0, 0.5, false),
        "coral" => (1.0, 0.5, 0.31, false),
        "crimson" => (0.86, 0.08, 0.24, false),
        "darkgray" => (0.66, 0.66, 0.66, false),
        "darkgreen" => (0.0, 0.39, 0.0, false),
        "darkblue" => (0.0, 0.0, 0.55, false),
        "darkred" => (0.55, 0.0, 0.0, false),
        _ => return None,
    };
    Some((r, g, b, if has_alpha { 0.0 } else { 1.0 }))
}

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

    #[test]
    fn test_parse_px() {
        assert_eq!(parse_px("16px"), 16.0);
        assert_eq!(parse_px("0"), 0.0);
    }

    #[test]
    fn test_parse_hex_color() {
        let (r, g, b, a) = parse_color("#ff0000");
        assert!((r - 1.0).abs() < 0.01);
        assert!((g - 0.0).abs() < 0.01);
        assert!((a - 1.0).abs() < 0.01);
    }

    #[test]
    fn test_parse_named_color() {
        let (r, g, b, _) = parse_color("red");
        assert!((r - 1.0).abs() < 0.01);
    }

    #[test]
    fn test_compute_styles() {
        let css = ".btn { color: red; } .primary { color: blue; font-weight: bold; }";
        let sheet = crate::parser::parse_css_simple(css);
        let style = compute_styles(&[sheet], &["btn".into(), "primary".into()], "button", None);
        assert_eq!(style.get("color").unwrap(), "blue"); // primary wins over btn
        assert_eq!(style.get("font-weight").unwrap(), "bold");
    }

    #[test]
    fn test_inherited_props() {
        let parent = {
            let mut s = ComputedStyle::default();
            s.set("color", "red");
            s.set("margin", "10px"); // not inherited
            s
        };
        let sheet = StyleSheet::default();
        let child = compute_styles(&[sheet], &[], "span", Some(&parent));
        assert_eq!(child.get("color").unwrap(), "red"); // inherited
        assert!(child.get("margin").is_none()); // not inherited
    }
}