giallo 0.5.0

use std::fmt::Write;
use std::ops::Range;

use serde::{Deserialize, Serialize};

use crate::renderers::html::HtmlEscaped;
use crate::scope::{ScopeInterner, ScopeListId};
use crate::themes::compiled::ThemeType;
use crate::themes::css::{DARK_SUFFIX, LIGHT_SUFFIX};
use crate::themes::font_style::FontStyle;
use crate::themes::{Color, CompiledTheme, Style, ThemeVariant};
use crate::tokenizer::Token;

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
/// A token with associated styling information
pub struct HighlightedText {
    /// The text from the input string for that specific token
    pub text: String,
    /// The assigned style. It can be a single theme or dual theme if light/dark
    /// support was requested.
    pub style: ThemeVariant<Style>,
}

impl HighlightedText {
    /// Renders this highlighted text using terminal ANSI escape codes
    pub(crate) fn as_ansi(
        &self,
        theme: &ThemeVariant<&CompiledTheme>,
        theme_type: Option<ThemeType>,
        bg_color: Option<Color>,
        f: &mut String,
    ) {
        let s = self.text.as_str();

        let (style, theme) = match (self.style, theme) {
            (ThemeVariant::Single(style), ThemeVariant::Single(theme)) => (style, theme),
            (
                ThemeVariant::Dual {
                    dark: dark_style, ..
                },
                ThemeVariant::Dual {
                    dark: dark_theme, ..
                },
            ) if theme_type == Some(ThemeType::Dark) => (dark_style, dark_theme),
            (
                ThemeVariant::Dual {
                    light: light_style, ..
                },
                ThemeVariant::Dual {
                    light: light_theme, ..
                },
            ) if theme_type == Some(ThemeType::Light) => (light_style, light_theme),
            _ => unreachable!(),
        };

        let default = &theme.default_style;
        if style == *default {
            f.push_str(s);
            return;
        }

        f.push_str("\x1b[");
        if style.foreground != default.foreground {
            style.foreground.as_ansi_fg(f);
        }
        if style.background != default.background {
            style.background.as_ansi_bg(f);
        }
        style.font_style.ansi_escapes(f);
        f.push('m');

        // Highlight background color
        if let Some(bg) = bg_color {
            f.push_str("\x1b[");
            bg.as_ansi_bg(f);
            f.push('m');
        }

        f.push_str(s);
        // reset
        f.push_str("\x1b[0m");
    }

    /// Renders this highlighted text as an HTML span element with either classes or inline style.
    pub(crate) fn as_html(
        &self,
        theme: &ThemeVariant<&CompiledTheme>,
        css_class_prefix: Option<&str>,
    ) -> String {
        let escaped = HtmlEscaped(self.text.as_str());

        // CSS class mode
        if let Some(prefix) = css_class_prefix {
            return self.as_css_html(theme, prefix, &escaped);
        }

        // Inline style mode
        match (&self.style, theme) {
            (ThemeVariant::Single(style), ThemeVariant::Single(t)) => {
                let default = &t.default_style;
                if *style == *default {
                    return format!("<span>{escaped}</span>");
                }

                let mut css = String::with_capacity(30);
                if style.foreground != default.foreground {
                    css.push_str(&style.foreground.as_css_color_property());
                }
                if style.background != default.background {
                    css.push_str(&style.background.as_css_bg_color_property());
                }
                for font_attr in style.font_style.css_attributes() {
                    css.push_str(font_attr);
                }
                format!(r#"<span style="{css}">{escaped}</span>"#)
            }
            (
                ThemeVariant::Dual { light, dark },
                ThemeVariant::Dual {
                    light: lt,
                    dark: dt,
                },
            ) => {
                let light_default = &lt.default_style;
                let dark_default = &dt.default_style;

                if *light == *light_default && *dark == *dark_default {
                    return format!("<span>{escaped}</span>");
                }

                let mut css = String::with_capacity(60);

                if light.foreground != light_default.foreground
                    || dark.foreground != dark_default.foreground
                {
                    css.push_str(&Color::as_css_light_dark_color_property(
                        &light.foreground,
                        &dark.foreground,
                    ));
                }

                if light.background != light_default.background
                    || dark.background != dark_default.background
                {
                    css.push_str(&Color::as_css_light_dark_bg_color_property(
                        &light.background,
                        &dark.background,
                    ));
                }

                for font_attr in light.font_style.css_attributes() {
                    css.push_str(font_attr);
                }

                if css.is_empty() {
                    format!("<span>{escaped}</span>")
                } else {
                    format!(r#"<span style="{css}">{escaped}</span>"#)
                }
            }
            _ => unreachable!(),
        }
    }

    fn as_css_html(
        &self,
        theme: &ThemeVariant<&CompiledTheme>,
        prefix: &str,
        escaped: &HtmlEscaped,
    ) -> String {
        let mut classes = String::new();

        let mut add_classes = |style: &Style, theme: &CompiledTheme, pref: &str| {
            if let Some(id) = theme.style_map.fg_id(style.foreground) {
                if !classes.is_empty() {
                    classes.push(' ');
                }
                write!(classes, "{pref}{id}").unwrap();
            }
            if let Some(id) = theme.style_map.bg_id(style.background) {
                if !classes.is_empty() {
                    classes.push(' ');
                }
                write!(classes, "{pref}bg{id}").unwrap();
            }
            let default_fs = theme.default_style.font_style;
            if style.font_style != default_fs {
                if style.font_style.contains(FontStyle::BOLD) {
                    write!(classes, " {pref}b").unwrap();
                }
                if style.font_style.contains(FontStyle::ITALIC) {
                    write!(classes, " {pref}i").unwrap();
                }
                if style.font_style.contains(FontStyle::UNDERLINE)
                    && style.font_style.contains(FontStyle::STRIKETHROUGH)
                {
                    write!(classes, " {pref}us").unwrap();
                } else if style.font_style.contains(FontStyle::UNDERLINE) {
                    write!(classes, " {pref}u").unwrap();
                } else if style.font_style.contains(FontStyle::STRIKETHROUGH) {
                    write!(classes, " {pref}s").unwrap();
                }
                if !default_fs.is_empty() && style.font_style.is_empty() {
                    write!(classes, " {pref}fr").unwrap();
                }
            }
        };

        match (&self.style, theme) {
            (ThemeVariant::Single(style), ThemeVariant::Single(t)) => {
                add_classes(style, t, prefix);
            }
            (
                ThemeVariant::Dual { light, dark },
                ThemeVariant::Dual {
                    light: lt,
                    dark: dt,
                },
            ) => {
                let ltp = format!("{prefix}{LIGHT_SUFFIX}");
                let dtp = format!("{prefix}{DARK_SUFFIX}");
                add_classes(light, lt, &ltp);
                add_classes(dark, dt, &dtp);
            }
            _ => unreachable!(),
        }

        if classes.is_empty() {
            format!("<span>{escaped}</span>")
        } else {
            format!(r#"<span class="{classes}">{escaped}</span>"#)
        }
    }
}

/// Options for token merging behavior
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct MergingOptions {
    pub merge_whitespaces: bool,
    pub merge_same_style_tokens: bool,
}

impl Default for MergingOptions {
    fn default() -> Self {
        Self {
            merge_whitespaces: true,
            merge_same_style_tokens: true,
        }
    }
}

/// Highlighter that applies theme styles to tokenized code
#[derive(Debug, Clone)]
pub(crate) struct Highlighter<'r> {
    // 1 theme for Single, 2 for Dual
    themes: Vec<&'r CompiledTheme>,
    // Separate cache per theme (max 2)
    // The index is the interned scopes id
    cache: [Vec<Option<Style>>; 2],
    scope_interner: &'r ScopeInterner,
}

impl<'r> Highlighter<'r> {
    /// Create a new highlighter from a compiled theme
    pub(crate) fn new(theme: &'r CompiledTheme, scope_interner: &'r ScopeInterner) -> Self {
        let cache = vec![None; scope_interner.num_nodes()];
        Highlighter {
            themes: vec![theme],
            cache: [cache.clone(), cache],
            scope_interner,
        }
    }

    /// Create a new highlighter for dual themes (light and dark)
    pub(crate) fn new_dual(
        light_theme: &'r CompiledTheme,
        dark_theme: &'r CompiledTheme,
        scope_interner: &'r ScopeInterner,
    ) -> Self {
        let cache = vec![None; scope_interner.num_nodes()];
        Highlighter {
            themes: vec![light_theme, dark_theme],
            cache: [cache.clone(), cache],
            scope_interner,
        }
    }

    /// Match a scope stack against theme rules, building styles hierarchically
    /// like vscode-textmate does.
    fn match_scopes(&mut self, scopes: ScopeListId) -> ThemeVariant<Style> {
        match self.themes.len() {
            1 => {
                let style = self.match_scopes_for_theme(scopes, 0);
                ThemeVariant::Single(style)
            }
            2 => {
                let light_style = self.match_scopes_for_theme(scopes, 0);
                let dark_style = self.match_scopes_for_theme(scopes, 1);
                ThemeVariant::Dual {
                    light: light_style,
                    dark: dark_style,
                }
            }
            _ => unreachable!("Highlighter supports only 1 or 2 themes"),
        }
    }

    /// Match scopes for a specific theme index with caching
    fn match_scopes_for_theme(&mut self, scope_id: ScopeListId, theme_index: usize) -> Style {
        let cache_idx = scope_id.as_index();

        if let Some(cached) = self.cache[theme_index][cache_idx] {
            return cached;
        }

        // cache miss, we compute the style
        let theme = self.themes[theme_index];
        let mut current_style = theme.default_style;

        let scopes = self.scope_interner.get_scopes(scope_id);

        // Build up scope path incrementally, simulating vscode-textmate's approach
        // Each scope level can override the accumulated style
        for i in 1..=scopes.len() {
            let current_scope_path = &scopes[0..i];
            for rule in &theme.rules {
                if rule.selector.matches(current_scope_path) {
                    current_style = rule.style_modifier.apply_to(&current_style);
                }
            }
            // If no match found, current_style remains unchanged (inheritance!)
        }
        let result = current_style;
        self.cache[theme_index][cache_idx] = Some(result);
        result
    }

    /// Apply highlighting to tokenized lines, preserving line structure.
    pub fn highlight_tokens(
        &mut self,
        content: &str,
        tokens: Vec<Vec<Token>>,
        options: MergingOptions,
    ) -> Vec<Vec<HighlightedText>> {
        let mut result = Vec::with_capacity(tokens.len());
        let lines = content.split('\n').collect::<Vec<_>>();

        for (line_tokens, line) in tokens.into_iter().zip(lines) {
            if line_tokens.is_empty() {
                result.push(Vec::new());
                continue;
            }

            let mut line_result: Vec<(Range<usize>, ThemeVariant<Style>)> = line_tokens
                .into_iter()
                .map(|x| {
                    let style = self.match_scopes(x.scopes);
                    (x.span, style)
                })
                .collect();

            // first merge all ws by prepending to the next non-ws token
            if options.merge_whitespaces {
                let num_tokens = line_result.len();
                let mut merged: Vec<(Range<usize>, ThemeVariant<Style>)> =
                    Vec::with_capacity(num_tokens);
                let mut carry_on_range: Option<Range<usize>> = None;

                for (idx, (span, theme_style)) in line_result.into_iter().enumerate() {
                    let could_merge = !theme_style.has_decoration();
                    let token_content = &line[span.clone()];
                    let is_whitespace_with_next = could_merge
                        && token_content.chars().all(|c| c.is_whitespace())
                        && idx + 1 < num_tokens;

                    if is_whitespace_with_next {
                        // Extend or initialize the carried range
                        carry_on_range = Some(match carry_on_range {
                            Some(range) => range.start..span.end,
                            None => span.clone(),
                        });
                    } else if let Some(carried_range) = &carry_on_range {
                        // We've hit a non-whitespace token or the last token in the line
                        if could_merge {
                            // We can prepend all the WS to that token
                            merged.push((carried_range.start..span.end, theme_style))
                        } else {
                            // We need to push 2 tokens here, one for the carried WS and one
                            // for the current token
                            let ws_style = if self.themes.len() == 1 {
                                ThemeVariant::Single(self.themes[0].default_style)
                            } else {
                                ThemeVariant::Dual {
                                    light: self.themes[0].default_style,
                                    dark: self.themes[1].default_style,
                                }
                            };
                            merged.push((carried_range.clone(), ws_style));
                            merged.push((span, theme_style));
                        }
                        carry_on_range = None;
                    } else {
                        merged.push((span, theme_style));
                    }
                }

                line_result = merged;
            }

            // then merge same style tokens after we did the WS
            if options.merge_same_style_tokens && self.themes.len() == 1 {
                let num_tokens = line_result.len();
                let mut merged: Vec<(Range<usize>, ThemeVariant<Style>)> =
                    Vec::with_capacity(num_tokens);

                for (span, theme_style) in line_result {
                    if let Some((prev_span, prev_theme_style)) = merged.last_mut() {
                        if &theme_style == prev_theme_style {
                            prev_span.end = span.end;
                        } else {
                            merged.push((span, theme_style));
                        }
                    } else {
                        merged.push((span, theme_style));
                    }
                }

                line_result = merged;
            }

            // then transform into HighlightedText
            result.push(
                line_result
                    .into_iter()
                    .map(|(span, style)| HighlightedText {
                        style,
                        text: line[span].to_string(),
                    })
                    .collect(),
            );
        }

        result
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::scope::{EMPTY_SCOPE_LIST, Scope, ScopeInterner};
    use crate::themes::compiled::StyleMap;
    use crate::themes::compiled::{CompiledThemeRule, StyleModifier, ThemeType};
    use crate::themes::font_style::FontStyle;
    use crate::themes::raw::{Colors, TokenColorRule, TokenColorSettings};
    use crate::themes::selector::parse_selector;
    use crate::themes::{Color, CompiledTheme, RawTheme};
    use crate::tokenizer::Token;
    use std::ops::Range;
    use std::path::PathBuf;

    // Helper functions
    fn scope(name: &str) -> Scope {
        Scope::new(name)[0]
    }

    fn color(hex: &str) -> Color {
        Color::from_hex(hex).unwrap()
    }

    fn token(interner: &mut ScopeInterner, start: usize, end: usize, scope_name: &str) -> Token {
        Token {
            span: Range { start, end },
            scopes: interner.extend(EMPTY_SCOPE_LIST, &[scope(scope_name)]),
        }
    }

    fn test_theme() -> CompiledTheme {
        let rules = vec![
            CompiledThemeRule {
                selector: parse_selector("comment").unwrap(),
                style_modifier: StyleModifier {
                    foreground: Some(color("#6A9955")),
                    background: None,
                    font_style: Some(FontStyle::ITALIC),
                },
            },
            CompiledThemeRule {
                selector: parse_selector("keyword").unwrap(),
                style_modifier: StyleModifier {
                    foreground: Some(color("#569CD6")),
                    background: None,
                    font_style: Some(FontStyle::BOLD),
                },
            },
        ];
        let default_style = Style {
            foreground: color("#D4D4D4"),
            background: color("#1E1E1E"),
            font_style: FontStyle::default(),
        };
        let style_map = StyleMap::new(&rules, default_style);
        CompiledTheme {
            name: "Test".to_string(),
            theme_type: ThemeType::Dark,
            default_style,
            line_number_foreground: None,
            highlight_background_color: None,
            style_map,
            rules,
        }
    }

    #[test]
    fn test_match_scopes() {
        let test_theme = test_theme();
        let mut interner = ScopeInterner::default();
        let comment = interner.extend(EMPTY_SCOPE_LIST, &[scope("comment")]);
        let keyword = interner.extend(EMPTY_SCOPE_LIST, &[scope("keyword")]);
        let unknown = interner.extend(EMPTY_SCOPE_LIST, &[scope("unknown")]);
        let mut highlighter = Highlighter::new(&test_theme, &interner);

        // Test matching scopes
        let ThemeVariant::Single(comment_style) = highlighter.match_scopes(comment) else {
            unreachable!()
        };
        assert_eq!(comment_style.foreground, color("#6A9955"));
        assert_eq!(comment_style.font_style, FontStyle::ITALIC);

        let ThemeVariant::Single(keyword_style) = highlighter.match_scopes(keyword) else {
            unreachable!()
        };
        assert_eq!(keyword_style.foreground, color("#569CD6"));
        assert_eq!(keyword_style.font_style, FontStyle::BOLD);

        // Test unmatched scope returns default
        let unknown_style = highlighter.match_scopes(unknown);
        assert_eq!(
            unknown_style,
            ThemeVariant::Single(highlighter.themes[0].default_style)
        );
    }

    #[test]
    fn test_highlight_tokens() {
        let test_theme = test_theme();
        let mut interner = ScopeInterner::default();
        let tokens = vec![
            vec![
                token(&mut interner, 0, 2, "keyword"),
                token(&mut interner, 3, 8, "unknown"),
            ],
            vec![token(&mut interner, 0, 2, "comment")],
        ];
        let mut highlighter = Highlighter::new(&test_theme, &interner);
        let content = "if hello\n//";

        let highlighted = highlighter.highlight_tokens(content, tokens, MergingOptions::default());

        assert_eq!(highlighted.len(), 2);
        assert_eq!(highlighted[0].len(), 2);
        assert_eq!(highlighted[1].len(), 1);

        // Keyword token
        assert_eq!(highlighted[0][0].text, "if");
        let ThemeVariant::Single(s) = &highlighted[0][0].style else {
            unreachable!()
        };
        assert_eq!(s.foreground, color("#569CD6"));

        // Unknown token uses default
        assert_eq!(highlighted[0][1].text, "hello");
        let ThemeVariant::Single(s) = &highlighted[0][1].style else {
            unreachable!()
        };
        assert_eq!(s.foreground, color("#D4D4D4"));

        // Comment token
        assert_eq!(highlighted[1][0].text, "//");
        let ThemeVariant::Single(s) = &highlighted[1][0].style else {
            unreachable!()
        };
        assert_eq!(s.foreground, color("#6A9955"));
    }

    #[test]
    fn test_style_modifier_apply_to() {
        let base = Style {
            foreground: color("#FFFFFF"),
            background: color("#000000"),
            font_style: FontStyle::default(),
        };

        let modifier = StyleModifier {
            foreground: Some(color("#FF0000")),
            background: None,
            font_style: Some(FontStyle::BOLD),
        };

        let result = modifier.apply_to(&base);
        assert_eq!(result.foreground, color("#FF0000"));
        assert_eq!(result.background, color("#000000")); // unchanged
        assert_eq!(result.font_style, FontStyle::BOLD);
    }

    #[test]
    fn test_theme_inheritance() {
        // Create RawTheme using proper theme structure
        let raw_theme = RawTheme {
            name: "Inheritance Test".to_string(),
            kind: Some("dark".to_string()),
            colors: Colors {
                foreground: "#D4D4D4".to_string(),
                background: "#1E1E1E".to_string(),
                highlight_background: None,
                line_number_foreground: None,
            },
            token_colors: vec![
                // Parent: constant - has both foreground and fontStyle
                TokenColorRule {
                    scope: vec!["constant".to_string()],
                    settings: TokenColorSettings {
                        foreground: Some("#300000".to_string()),
                        background: None,
                        font_style: Some("italic".to_string()),
                    },
                },
                // Child: constant.numeric - only foreground (should inherit italic)
                TokenColorRule {
                    scope: vec!["constant.numeric".to_string()],
                    settings: TokenColorSettings {
                        foreground: Some("#400000".to_string()),
                        background: None,
                        font_style: None, // Should inherit italic from constant
                    },
                },
                // Grandchild: constant.numeric.hex - only fontStyle (should inherit foreground)
                TokenColorRule {
                    scope: vec!["constant.numeric.hex".to_string()],
                    settings: TokenColorSettings {
                        foreground: None, // Should inherit #400000 from constant.numeric
                        background: None,
                        font_style: Some("bold".to_string()),
                    },
                },
            ],
        };

        // Compile using proper pipeline (automatically sorts by specificity)
        let inheritance_theme = CompiledTheme::from_raw_theme(raw_theme).unwrap();
        let mut interner = ScopeInterner::default();
        let constant = interner.extend(EMPTY_SCOPE_LIST, &[scope("constant")]);
        let numeric = interner.extend(
            EMPTY_SCOPE_LIST,
            &[scope("constant"), scope("constant.numeric")],
        );
        let hex = interner.extend(
            EMPTY_SCOPE_LIST,
            &[
                scope("constant"),
                scope("constant.numeric"),
                scope("constant.numeric.hex"),
            ],
        );
        let mut highlighter = Highlighter::new(&inheritance_theme, &interner);

        // Test: constant should get its own values
        let ThemeVariant::Single(style) = highlighter.match_scopes(constant) else {
            unreachable!()
        };
        assert_eq!(style.foreground, color("#300000"));
        assert_eq!(style.font_style, FontStyle::ITALIC);

        // Test: constant.numeric should inherit fontStyle from constant but override foreground
        let ThemeVariant::Single(style) = highlighter.match_scopes(numeric) else {
            unreachable!()
        };
        assert_eq!(style.foreground, color("#400000")); // Overridden
        assert_eq!(style.font_style, FontStyle::ITALIC);

        // Test: constant.numeric.hex should inherit foreground from constant.numeric but override fontStyle
        let ThemeVariant::Single(style) = highlighter.match_scopes(hex) else {
            unreachable!()
        };
        assert_eq!(style.foreground, color("#400000")); // Should inherit from constant.numeric
        assert_eq!(style.font_style, FontStyle::BOLD); // Overridden
    }

    #[test]
    fn test_real_world_theme_inheritance() {
        // Load the Vitesse Black theme - a real production theme
        let theme_path =
            PathBuf::from("grammars-themes/packages/tm-themes/themes/vitesse-black.json");
        let raw_theme = RawTheme::load_from_file(theme_path).unwrap();
        let compiled_theme = CompiledTheme::from_raw_theme(raw_theme).unwrap();
        let mut interner = ScopeInterner::default();

        // Test real tokenizer output from ASP.NET Core Razor with invalid HTML tag
        // Token 1: '<' - HTML tag begin punctuation
        let token1_scopes = interner.extend(
            EMPTY_SCOPE_LIST,
            &[
                scope("text.aspnetcorerazor"),
                scope("meta.element.structure.svg.html"),
                scope("meta.element.object.svg.foreignObject.html"),
                scope("meta.element.other.invalid.html"),
                scope("meta.tag.other.invalid.start.html"),
                scope("punctuation.definition.tag.begin.html"),
            ],
        );

        // Token 2: 'p' - Invalid/unrecognized HTML tag name
        let token2_scopes = interner.extend(
            EMPTY_SCOPE_LIST,
            &[
                scope("text.aspnetcorerazor"),
                scope("meta.element.structure.svg.html"),
                scope("meta.element.object.svg.foreignObject.html"),
                scope("meta.element.other.invalid.html"),
                scope("meta.tag.other.invalid.start.html"),
                scope("entity.name.tag.html"),
                scope("invalid.illegal.unrecognized-tag.html"),
            ],
        );

        // Token 3: '>' - HTML tag end punctuation
        let token3_scopes = interner.extend(
            EMPTY_SCOPE_LIST,
            &[
                scope("text.aspnetcorerazor"),
                scope("meta.element.structure.svg.html"),
                scope("meta.element.object.svg.foreignObject.html"),
                scope("meta.element.other.invalid.html"),
                scope("meta.tag.other.invalid.start.html"),
                scope("punctuation.definition.tag.end.html"),
            ],
        );

        let mut highlighter = Highlighter::new(&compiled_theme, &interner);
        let style1 = highlighter.match_scopes(token1_scopes);
        let style2 = highlighter.match_scopes(token2_scopes);
        let style3 = highlighter.match_scopes(token3_scopes);

        // Verify that styles are not default (theme inheritance is working)
        assert_ne!(style1, ThemeVariant::Single(compiled_theme.default_style));
        assert_ne!(style2, ThemeVariant::Single(compiled_theme.default_style));
        assert_ne!(style3, ThemeVariant::Single(compiled_theme.default_style));

        // Token 2 should have distinct styling due to invalid.illegal scope
        // which typically gets error/warning colors in themes
        assert_ne!(style1, style2);
        assert_ne!(style2, style3);

        // Token 'p' should get pink color from invalid.illegal rule (#FDAEB7)
        let ThemeVariant::Single(s2) = &style2 else {
            unreachable!()
        };
        assert_eq!(
            s2.foreground,
            Color {
                r: 253,
                g: 174,
                b: 183,
                a: 255,
            }
        );
    }

    #[test]
    fn test_as_html_empty() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Single(test_theme.default_style),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), None);
        insta::assert_snapshot!(res, @"<span>hello</span>");
    }

    #[test]
    fn test_as_html_content_escape() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "<script></script>".to_string(),
            style: ThemeVariant::Single(test_theme.default_style),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), None);
        insta::assert_snapshot!(res, @"<span>&lt;script&gt;&lt;/script&gt;</span>");
    }

    #[test]
    fn test_as_html_hex_fg_diff() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Single(Style {
                foreground: color("#FFFF00"),
                ..test_theme.default_style
            }),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), None);
        insta::assert_snapshot!(res, @r#"<span style="color: #FFFF00;">hello</span>"#);
    }

    #[test]
    fn test_as_html_hex_bg_diff() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Single(Style {
                background: color("#FFFF00"),
                ..test_theme.default_style
            }),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), None);
        insta::assert_snapshot!(res, @r#"<span style="background-color: #FFFF00;">hello</span>"#);
    }

    #[test]
    fn test_as_html_hex_fontstyle_diff() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Single(Style {
                font_style: FontStyle::ITALIC,
                ..test_theme.default_style
            }),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), None);
        insta::assert_snapshot!(res, @r#"<span style="font-style: italic;">hello</span>"#);
    }

    #[test]
    fn test_as_html_hex_completely_different() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Single(Style {
                foreground: color("#FFFF00"),
                background: color("#FFFF00"),
                font_style: FontStyle::ITALIC,
            }),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), None);
        insta::assert_snapshot!(res, @r#"<span style="color: #FFFF00;background-color: #FFFF00;font-style: italic;">hello</span>"#);
    }

    #[test]
    fn test_as_html_dual_both_default() {
        let light = test_theme();
        let dark = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Dual {
                light: light.default_style,
                dark: dark.default_style,
            },
        };
        let res = ht.as_html(
            &ThemeVariant::Dual {
                light: &light,
                dark: &dark,
            },
            None,
        );
        insta::assert_snapshot!(res, @"<span>hello</span>");
    }

    #[test]
    fn test_as_html_dual_fg_differs() {
        let light = test_theme();
        let dark = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Dual {
                light: Style {
                    foreground: color("#FF0000"),
                    ..light.default_style
                },
                dark: Style {
                    foreground: color("#00FF00"),
                    ..dark.default_style
                },
            },
        };
        let res = ht.as_html(
            &ThemeVariant::Dual {
                light: &light,
                dark: &dark,
            },
            None,
        );
        insta::assert_snapshot!(res, @r#"<span style="color: light-dark(#FF0000, #00FF00);">hello</span>"#);
    }

    #[test]
    fn test_as_html_dual_bg_differs() {
        let light = test_theme();
        let dark = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Dual {
                light: Style {
                    background: color("#FFFFFF"),
                    ..light.default_style
                },
                dark: Style {
                    background: color("#000000"),
                    ..dark.default_style
                },
            },
        };
        let res = ht.as_html(
            &ThemeVariant::Dual {
                light: &light,
                dark: &dark,
            },
            None,
        );
        insta::assert_snapshot!(res, @r#"<span style="background-color: light-dark(#FFFFFF, #000000);">hello</span>"#);
    }

    #[test]
    fn test_as_html_dual_both_differ() {
        let light = test_theme();
        let dark = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Dual {
                light: Style {
                    foreground: color("#FF0000"),
                    background: color("#FFFFFF"),
                    font_style: FontStyle::BOLD,
                },
                dark: Style {
                    foreground: color("#00FF00"),
                    background: color("#000000"),
                    font_style: FontStyle::BOLD,
                },
            },
        };
        let res = ht.as_html(
            &ThemeVariant::Dual {
                light: &light,
                dark: &dark,
            },
            None,
        );
        insta::assert_snapshot!(res, @r#"<span style="color: light-dark(#FF0000, #00FF00);background-color: light-dark(#FFFFFF, #000000);font-weight: bold;">hello</span>"#);
    }

    #[test]
    fn test_as_html_with_css_classes() {
        let test_theme = test_theme();
        let ht = HighlightedText {
            text: "hello".to_string(),
            style: ThemeVariant::Single(Style {
                foreground: color("#6A9955"),
                font_style: FontStyle::ITALIC,
                ..test_theme.default_style
            }),
        };
        let res = ht.as_html(&ThemeVariant::Single(&test_theme), Some("g-"));
        // Should have a foreground color class and italic class
        assert!(res.contains("g-"));
        assert!(res.contains("g-i"));
        assert!(res.starts_with("<span class=\""));
    }
}