bun_css 0.1.1

use crate as css;
use crate::CSSString;
use crate::css_values::easing::EasingFunction;
use crate::css_values::ident::{CustomIdent, DashedIdent, is_reserved_custom_ident};
use crate::css_values::length::{LengthPercentage, LengthPercentageOrAuto};
use crate::css_values::number::{CSSNumber, CSSNumberFns};
use crate::css_values::size::Size2D;
use crate::css_values::time::Time;
use crate::{Parser, PrintErr, Printer, SmallList};
use bun_core::strings;

/// A list of animations.
pub type AnimationList = SmallList<Animation, 1>;

/// A list of animation names.
pub type AnimationNameList = SmallList<AnimationName, 1>;

/// A value for the [animation](https://drafts.csswg.org/css-animations/#animation) shorthand property.
pub struct Animation {
    /// The animation name.
    pub name: AnimationName,
    /// The animation duration.
    pub duration: Time,
    /// The easing function for the animation.
    pub timing_function: EasingFunction,
    /// The number of times the animation will run.
    pub iteration_count: AnimationIterationCount,
    /// The direction of the animation.
    pub direction: AnimationDirection,
    /// The current play state of the animation.
    pub play_state: AnimationPlayState,
    /// The animation delay.
    pub delay: Time,
    /// The animation fill mode.
    pub fill_mode: AnimationFillMode,
    /// The animation timeline.
    pub timeline: AnimationTimeline,
}

impl Animation {
    // TODO(port): PropertyFieldMap / VendorPrefixMap were comptime anonymous-struct
    // metadata consumed by reflection in the shorthand codegen. Replace these with
    // a derive macro (e.g. #[derive(Shorthand)]) that emits the field→PropertyIdTag
    // and field→has-vendor-prefix tables.
    // PORT NOTE: PropertyFieldMap dropped — `PropertyIdTag::Animation*` variants
    // are not yet generated (animation longhands are unparsed-only for now), and
    // the table was unread comptime metadata. Re-add when the variants land.

    pub const VENDOR_PREFIX_MAP: &'static [(&'static str, bool)] = &[
        ("name", true),
        ("duration", true),
        ("timing_function", true),
        ("iteration_count", true),
        ("direction", true),
        ("play_state", true),
        ("delay", true),
        ("fill_mode", true),
    ];

    pub fn parse(input: &mut Parser) -> css::Result<Self> {
        let mut name: Option<AnimationName> = None;
        let mut duration: Option<Time> = None;
        let mut timing_function: Option<EasingFunction> = None;
        let mut iteration_count: Option<AnimationIterationCount> = None;
        let mut direction: Option<AnimationDirection> = None;
        let mut play_state: Option<AnimationPlayState> = None;
        let mut delay: Option<Time> = None;
        let mut fill_mode: Option<AnimationFillMode> = None;
        let mut timeline: Option<AnimationTimeline> = None;

        loop {
            if duration.is_none() {
                if let Ok(value) = input.try_parse(Time::parse) {
                    duration = Some(value);
                    continue;
                }
            }
            if timing_function.is_none() {
                if let Ok(value) = input.try_parse(EasingFunction::parse) {
                    timing_function = Some(value);
                    continue;
                }
            }
            if delay.is_none() {
                if let Ok(value) = input.try_parse(Time::parse) {
                    delay = Some(value);
                    continue;
                }
            }
            if iteration_count.is_none() {
                if let Ok(value) = input.try_parse(AnimationIterationCount::parse) {
                    iteration_count = Some(value);
                    continue;
                }
            }
            if direction.is_none() {
                if let Ok(value) = input.try_parse(AnimationDirection::parse) {
                    direction = Some(value);
                    continue;
                }
            }
            if fill_mode.is_none() {
                if let Ok(value) = input.try_parse(AnimationFillMode::parse) {
                    fill_mode = Some(value);
                    continue;
                }
            }
            if play_state.is_none() {
                if let Ok(value) = input.try_parse(AnimationPlayState::parse) {
                    play_state = Some(value);
                    continue;
                }
            }
            if name.is_none() {
                if let Ok(value) = input.try_parse(AnimationName::parse) {
                    name = Some(value);
                    continue;
                }
            }
            if timeline.is_none() {
                if let Ok(value) = input.try_parse(AnimationTimeline::parse) {
                    timeline = Some(value);
                    continue;
                }
            }
            break;
        }

        css::Result::Ok(Animation {
            name: name.unwrap_or(AnimationName::None),
            duration: duration.unwrap_or(Time::Seconds(0.0)),
            timing_function: timing_function.unwrap_or(EasingFunction::Ease),
            iteration_count: iteration_count.unwrap_or(AnimationIterationCount::Number(1.0)),
            direction: direction.unwrap_or(AnimationDirection::Normal),
            play_state: play_state.unwrap_or(AnimationPlayState::Running),
            delay: delay.unwrap_or(Time::Seconds(0.0)),
            fill_mode: fill_mode.unwrap_or(AnimationFillMode::None),
            timeline: timeline.unwrap_or(AnimationTimeline::Auto),
        })
    }

    pub fn to_css(&self, dest: &mut Printer) -> Result<(), PrintErr> {
        // PORT NOTE: reshaped `inline .ident, .string => |name|` — Zig's inline
        // switch monomorphized over two payload types; Rust extracts the inner
        // string slice up front instead.
        let name_str: Option<&[u8]> = match &self.name {
            AnimationName::None => None,
            AnimationName::Ident(ident) => Some(ident.v()),
            // SAFETY: arena-owned slice lives for the parse session.
            AnimationName::String(s) => Some(unsafe { crate::arena_str(*s) }),
        };

        if let Some(name_str) = name_str {
            if !self.duration.is_zero() || !self.delay.is_zero() {
                self.duration.to_css(dest)?;
                dest.write_char(b' ')?;
            }

            if !self.timing_function.is_ease() || EasingFunction::is_ident(name_str) {
                self.timing_function.to_css(dest)?;
                dest.write_char(b' ')?;
            }

            if !self.delay.is_zero() {
                self.delay.to_css(dest)?;
                dest.write_char(b' ')?;
            }

            if self.iteration_count != AnimationIterationCount::default()
                || strings::eql_case_insensitive_ascii(name_str, b"infinite", true)
            {
                self.iteration_count.to_css(dest)?;
                dest.write_char(b' ')?;
            }

            if self.direction != AnimationDirection::default()
                || css::parse_utility::parse_string::<AnimationDirection>(
                    dest.arena,
                    name_str,
                    AnimationDirection::parse,
                )
                .is_ok()
            {
                self.direction.to_css(dest)?;
                dest.write_char(b' ')?;
            }

            if self.fill_mode != AnimationFillMode::default()
                || (!strings::eql_case_insensitive_ascii(name_str, b"none", true)
                    && css::parse_utility::parse_string::<AnimationFillMode>(
                        dest.arena,
                        name_str,
                        AnimationFillMode::parse,
                    )
                    .is_ok())
            {
                self.fill_mode.to_css(dest)?;
                dest.write_char(b' ')?;
            }

            if self.play_state != AnimationPlayState::default()
                || css::parse_utility::parse_string::<AnimationPlayState>(
                    dest.arena,
                    name_str,
                    AnimationPlayState::parse,
                )
                .is_ok()
            {
                self.play_state.to_css(dest)?;
                dest.write_char(b' ')?;
            }
        }

        self.name.to_css(dest)?;

        if !matches!(self.name, AnimationName::None)
            && self.timeline != AnimationTimeline::default()
        {
            dest.write_char(b' ')?;
            self.timeline.to_css(dest)?;
        }

        Ok(())
    }
}

/// A value for the [animation-name](https://drafts.csswg.org/css-animations/#animation-name) property.
// PORT NOTE: no `#[derive(PartialEq, Eq, Hash)]` — `CustomIdent`/`CSSString`
// carry raw `*const [u8]` arena pointers; derived eq/hash would compare by
// pointer. Hand-written `eql`/`hash` below compare by content.
#[derive(Clone, Copy)]
pub enum AnimationName {
    /// The `none` keyword.
    None,
    /// An identifier of a `@keyframes` rule.
    Ident(CustomIdent),
    /// A `<string>` name of a `@keyframes` rule.
    String(CSSString),
}

impl AnimationName {
    // PORT NOTE: hand-written (not `#[derive]`) because `CSSString` is a raw
    // `*const [u8]` arena pointer — generics blanket impls cover `&[u8]` but
    // not raw slices. Mirrors Zig `css.implementEql/Hash/DeepClone`.
    pub fn eql(&self, other: &Self) -> bool {
        match (self, other) {
            (AnimationName::None, AnimationName::None) => true,
            (AnimationName::Ident(a), AnimationName::Ident(b)) => {
                use crate::generics::CssEql;
                a.eql(b)
            }
            (AnimationName::String(a), AnimationName::String(b)) => {
                // SAFETY: arena-owned slices live for the parse session.
                unsafe { bun_core::eql(&**a, &**b) }
            }
            _ => false,
        }
    }

    pub fn hash(&self, hasher: &mut bun_wyhash::Wyhash) {
        match self {
            AnimationName::None => hasher.update(&0u32.to_ne_bytes()),
            AnimationName::Ident(i) => {
                hasher.update(&1u32.to_ne_bytes());
                i.hash(hasher);
            }
            AnimationName::String(s) => {
                hasher.update(&2u32.to_ne_bytes());
                // SAFETY: arena-owned slice.
                hasher.update(unsafe { crate::arena_str(*s) });
            }
        }
    }

    #[inline]
    pub fn deep_clone(&self, _bump: &bun_alloc::Arena) -> Self {
        // All payloads are `Copy` (arena slice pointers); identity copy.
        match self {
            AnimationName::None => AnimationName::None,
            AnimationName::Ident(i) => AnimationName::Ident(*i),
            AnimationName::String(s) => AnimationName::String(*s),
        }
    }

    pub fn parse(input: &mut Parser) -> css::Result<Self> {
        // PORT NOTE: ported from src/css/properties/animation.zig — `none` keyword,
        // then `<string>`, else `<custom-ident>`.
        if input
            .try_parse(|i| i.expect_ident_matching(b"none"))
            .is_ok()
        {
            return Ok(AnimationName::None);
        }
        // PORT NOTE: `expect_string` returns a slice borrowing `&mut self`, which
        // `try_parse`'s `R` type param can't carry. Erase the lifetime through a
        // raw pointer inside the closure; the slice lives in the input arena and
        // outlives this parse (CSSString = &'static [u8]).
        if let Ok(s) = input.try_parse(|i| i.expect_string().map(std::ptr::from_ref::<[u8]>)) {
            return Ok(AnimationName::String(s));
        }
        let ident = CustomIdent::parse(input)?;
        Ok(AnimationName::Ident(ident))
    }

    pub fn to_css(&self, dest: &mut Printer) -> Result<(), PrintErr> {
        let css_module_animation_enabled = if let Some(css_module) = &dest.css_module {
            css_module.config.animation
        } else {
            false
        };

        match self {
            AnimationName::None => return dest.write_str("none"),
            AnimationName::Ident(s) => {
                // SAFETY: arena-owned slice valid for 'bump.
                let name: &[u8] = unsafe { crate::arena_str(s.v) };
                if css_module_animation_enabled {
                    // PORT NOTE: reshaped for borrowck — capture arena/source_index
                    // before borrowing dest.css_module mutably.
                    let arena = dest.arena;
                    let source_index = dest.loc.source_index;
                    if let Some(css_module) = &mut dest.css_module {
                        css_module.get_reference(arena, name, source_index);
                    }
                }
                return s.to_css_with_options(dest, css_module_animation_enabled);
            }
            AnimationName::String(s) => {
                // SAFETY: arena-owned slice valid for 'bump.
                let name: &[u8] = unsafe { crate::arena_str(*s) };
                if css_module_animation_enabled {
                    // PORT NOTE: reshaped for borrowck
                    let arena = dest.arena;
                    let source_index = dest.loc.source_index;
                    if let Some(css_module) = &mut dest.css_module {
                        css_module.get_reference(arena, name, source_index);
                    }
                }

                // CSS-wide keywords and `none` cannot remove quotes
                if strings::eql_case_insensitive_ascii_check_length(name, b"none")
                    || is_reserved_custom_ident(name)
                {
                    return dest.serialize_string(name);
                }

                return dest.write_ident(name, css_module_animation_enabled);
            }
        }
    }
}

/// A value for the [animation-iteration-count](https://drafts.csswg.org/css-animations/#animation-iteration-count) property.
// TODO(port): css.DeriveParse / css.DeriveToCss were comptime mixins generating
// parse()/to_css() from variant shape. Implement as #[derive(Parse, ToCss)].
#[derive(PartialEq)]
pub enum AnimationIterationCount {
    /// The animation will repeat the specified number of times.
    Number(CSSNumber),
    /// The animation will repeat forever.
    Infinite,
}

impl AnimationIterationCount {
    // Port of `css.DeriveParse(@This()).parse` — payload (`number: f32`) declared
    // first, single void variant (`infinite`) last → try number, then ident.
    pub fn parse(input: &mut Parser) -> css::Result<Self> {
        if let Ok(v) = input.try_parse(CSSNumberFns::parse) {
            return Ok(AnimationIterationCount::Number(v));
        }
        input.expect_ident_matching(b"infinite")?;
        Ok(AnimationIterationCount::Infinite)
    }

    // Port of `css.DeriveToCss(@This()).toCss`.
    pub fn to_css(&self, dest: &mut Printer) -> Result<(), PrintErr> {
        match self {
            AnimationIterationCount::Number(n) => CSSNumberFns::to_css(*n, dest),
            AnimationIterationCount::Infinite => dest.write_str(b"infinite"),
        }
    }

    pub fn default() -> AnimationIterationCount {
        AnimationIterationCount::Number(1.0)
    }
}

/// A value for the [animation-direction](https://drafts.csswg.org/css-animations/#animation-direction) property.
#[derive(Clone, Copy, PartialEq, Eq, Hash, css::DefineEnumProperty)]
pub enum AnimationDirection {
    /// The animation is played as specified
    Normal,
    /// The animation is played in reverse.
    Reverse,
    /// The animation iterations alternate between forward and reverse.
    Alternate,
    /// The animation iterations alternate between forward and reverse, with reverse occurring first.
    AlternateReverse, // css: "alternate-reverse"
}

impl AnimationDirection {
    pub fn deep_clone(self) -> Self {
        self
    }
    pub fn default() -> AnimationDirection {
        AnimationDirection::Normal
    }
}

/// A value for the [animation-play-state](https://drafts.csswg.org/css-animations/#animation-play-state) property.
#[derive(Clone, Copy, PartialEq, Eq, Hash, css::DefineEnumProperty)]
pub enum AnimationPlayState {
    /// The animation is playing.
    Running,
    /// The animation is paused.
    Paused,
}

impl AnimationPlayState {
    pub fn deep_clone(self) -> Self {
        self
    }
    pub fn default() -> AnimationPlayState {
        AnimationPlayState::Running
    }
}

/// A value for the [animation-fill-mode](https://drafts.csswg.org/css-animations/#animation-fill-mode) property.
#[derive(Clone, Copy, PartialEq, Eq, Hash, css::DefineEnumProperty)]
pub enum AnimationFillMode {
    /// The animation has no effect while not playing.
    None,
    /// After the animation, the ending values are applied.
    Forwards,
    /// Before the animation, the starting values are applied.
    Backwards,
    /// Both forwards and backwards apply.
    Both,
}

impl AnimationFillMode {
    pub fn deep_clone(self) -> Self {
        self
    }
    pub fn default() -> AnimationFillMode {
        AnimationFillMode::None
    }
}

/// A value for the [animation-composition](https://drafts.csswg.org/css-animations-2/#animation-composition) property.
#[derive(Clone, Copy, PartialEq, Eq, Hash, css::DefineEnumProperty)]
pub enum AnimationComposition {
    /// The result of compositing the effect value with the underlying value is simply the effect value.
    Replace,
    /// The effect value is added to the underlying value.
    Add,
    /// The effect value is accumulated onto the underlying value.
    Accumulate,
}

impl AnimationComposition {
    pub fn deep_clone(self) -> Self {
        self
    }
}

/// A value for the [animation-timeline](https://drafts.csswg.org/css-animations-2/#animation-timeline) property.
pub enum AnimationTimeline {
    /// The animation's timeline is a DocumentTimeline, more specifically the default document timeline.
    Auto,
    /// The animation is not associated with a timeline.
    None,
    /// A timeline referenced by name.
    DashedIdent(DashedIdent),
    /// The scroll() function.
    Scroll(ScrollTimeline),
    /// The view() function.
    View(ViewTimeline),
}

impl AnimationTimeline {
    // Port of `css.DeriveParse(@This()).parse` — void variants (`auto`, `none`)
    // declared first → tried first via ident match; payloads follow in
    // declaration order (`DashedIdent`, `ScrollTimeline`, `ViewTimeline`).
    // Upstream `ScrollTimeline` / `ViewTimeline` carry no `parse`, so the Zig
    // `DeriveParse` instantiation is dead code (`generic.parseFor` would
    // `@compileError` if compiled — `Animation` is unreferenced in
    // properties_generated.zig). We stop at `DashedIdent` here; if scroll()/
    // view() ever become live they need real function-syntax parsing, not the
    // derived field-sequence fallback.
    pub fn parse(input: &mut Parser) -> css::Result<Self> {
        let state = input.state();
        if let Ok(ident) = input.expect_ident() {
            if strings::eql_case_insensitive_ascii_check_length(ident, b"auto") {
                return Ok(AnimationTimeline::Auto);
            }
            if strings::eql_case_insensitive_ascii_check_length(ident, b"none") {
                return Ok(AnimationTimeline::None);
            }
            input.reset(&state);
        } else {
            input.reset(&state);
        }
        DashedIdent::parse(input).map(AnimationTimeline::DashedIdent)
    }

    // Port of `css.DeriveToCss(@This()).toCss`.
    pub fn to_css(&self, dest: &mut Printer) -> Result<(), PrintErr> {
        match self {
            AnimationTimeline::Auto => dest.write_str(b"auto"),
            AnimationTimeline::None => dest.write_str(b"none"),
            AnimationTimeline::DashedIdent(d) => d.to_css(dest),
            // Upstream Zig `ScrollTimeline` / `ViewTimeline` have no `toCss`;
            // `DeriveToCss` would delegate to `generic.toCss` → `T.toCss` and
            // `@compileError` if this arm were ever instantiated. Mirror that:
            // these variants are currently unconstructible via `parse()`, and
            // emitting bare space-separated fields here would be wrong CSS
            // (spec syntax is `scroll(...)` / `view(...)`).
            AnimationTimeline::Scroll(_) | AnimationTimeline::View(_) => {
                unreachable!("ScrollTimeline / ViewTimeline have no toCss in spec (uninstantiated)")
            }
        }
    }

    pub fn default() -> AnimationTimeline {
        AnimationTimeline::Auto
    }

    pub fn is_default(&self) -> bool {
        matches!(self, AnimationTimeline::Auto)
    }
}

// PORT NOTE: hand-written `PartialEq` — `DashedIdent` carries a raw
// `*const [u8]` arena pointer; derive would compare by pointer, not content.
impl PartialEq for AnimationTimeline {
    fn eq(&self, other: &Self) -> bool {
        use crate::generics::CssEql;
        match (self, other) {
            (AnimationTimeline::Auto, AnimationTimeline::Auto) => true,
            (AnimationTimeline::None, AnimationTimeline::None) => true,
            (AnimationTimeline::DashedIdent(a), AnimationTimeline::DashedIdent(b)) => a.eql(b),
            (AnimationTimeline::Scroll(a), AnimationTimeline::Scroll(b)) => a == b,
            (AnimationTimeline::View(a), AnimationTimeline::View(b)) => {
                a.axis == b.axis && Size2D::eql(&a.inset, &b.inset)
            }
            _ => false,
        }
    }
}

/// The [scroll()](https://drafts.csswg.org/scroll-animations-1/#scroll-notation) function.
#[derive(PartialEq, Eq)]
pub struct ScrollTimeline {
    /// Specifies which element to use as the scroll container.
    pub scroller: Scroller,
    /// Specifies which axis of the scroll container to use as the progress for the timeline.
    pub axis: ScrollAxis,
}

/// The [view()](https://drafts.csswg.org/scroll-animations-1/#view-notation) function.
pub struct ViewTimeline {
    /// Specifies which axis of the scroll container to use as the progress for the timeline.
    pub axis: ScrollAxis,
    /// Provides an adjustment of the view progress visibility range.
    pub inset: Size2D<LengthPercentageOrAuto>,
}

/// A scroller, used in the `scroll()` function.
#[derive(Clone, Copy, PartialEq, Eq, Hash, css::DefineEnumProperty)]
pub enum Scroller {
    /// Specifies to use the document viewport as the scroll container.
    Root,
    /// Specifies to use the nearest ancestor scroll container.
    Nearest,
    /// Specifies to use the element's own principal box as the scroll container.
    #[css("self")]
    Self_,
}

impl Scroller {
    pub fn deep_clone(self) -> Self {
        self
    }
    pub fn default() -> Scroller {
        Scroller::Nearest
    }
}

/// A scroll axis, used in the `scroll()` function.
#[derive(Clone, Copy, PartialEq, Eq, Hash, css::DefineEnumProperty)]
pub enum ScrollAxis {
    /// Specifies to use the measure of progress along the block axis of the scroll container.
    Block,
    /// Specifies to use the measure of progress along the inline axis of the scroll container.
    Inline,
    /// Specifies to use the measure of progress along the horizontal axis of the scroll container.
    X,
    /// Specifies to use the measure of progress along the vertical axis of the scroll container.
    Y,
}

impl ScrollAxis {
    pub fn deep_clone(self) -> Self {
        self
    }
    pub fn default() -> ScrollAxis {
        ScrollAxis::Block
    }
}

/// A value for the animation-range shorthand property.
pub struct AnimationRange {
    /// The start of the animation's attachment range.
    pub start: AnimationRangeStart,
    /// The end of the animation's attachment range.
    pub end: AnimationRangeEnd,
}

/// A value for the [animation-range-start](https://drafts.csswg.org/scroll-animations/#animation-range-start) property.
pub struct AnimationRangeStart {
    pub v: AnimationAttachmentRange,
}

/// A value for the [animation-range-end](https://drafts.csswg.org/scroll-animations/#animation-range-start) property.
pub struct AnimationRangeEnd {
    pub v: AnimationAttachmentRange,
}

/// A value for the [animation-range-start](https://drafts.csswg.org/scroll-animations/#animation-range-start)
/// or [animation-range-end](https://drafts.csswg.org/scroll-animations/#animation-range-end) property.
pub enum AnimationAttachmentRange {
    /// The start of the animation's attachment range is the start of its associated timeline.
    Normal,
    /// The animation attachment range starts at the specified point on the timeline measuring from the start of the timeline.
    LengthPercentage(LengthPercentage),
    /// The animation attachment range starts at the specified point on the timeline measuring from the start of the specified named timeline range.
    TimelineRange {
        /// The name of the timeline range.
        name: TimelineRangeName,
        /// The offset from the start of the named timeline range.
        offset: LengthPercentage,
    },
}

/// A [view progress timeline range](https://drafts.csswg.org/scroll-animations/#view-timelines-ranges)
pub enum TimelineRangeName {
    /// Represents the full range of the view progress timeline.
    Cover,
    /// Represents the range during which the principal box is either fully contained by,
    /// or fully covers, its view progress visibility range within the scrollport.
    Contain,
    /// Represents the range during which the principal box is entering the view progress visibility range.
    Entry,
    /// Represents the range during which the principal box is exiting the view progress visibility range.
    Exit,
    /// Represents the range during which the principal box crosses the end border edge.
    EntryCrossing,
    /// Represents the range during which the principal box crosses the start border edge.
    ExitCrossing,
}

// ported from: src/css/properties/animation.zig