use bevy::prelude::*;
use bevy_sprite3d::Sprite3d;

#[derive(Component, Reflect, Deref, DerefMut)]
#[reflect(Component)]
pub struct AnimationTimer(Timer);

impl AnimationTimer {
    pub fn new(t: Timer) -> Self {
        Self(t)
    }
}

pub fn plugin(app: &mut App) {
    app.add_systems(Update, animate_sprite);
}

fn animate_sprite(
    mut commands: Commands,
    time: Res<Time>,
    mut query: Query<(Entity, &mut AnimationTimer, &Sprite3d, &mut Sprite)>,
) {
    for (e, mut timer, sprite_3d, mut sprite) in query.iter_mut() {
        let length = sprite_3d.texture_atlas_keys.len();
        let atlas = sprite.texture_atlas.as_mut().unwrap();

        if length > 0 {
            timer.tick(time.delta());
        }

        if timer.just_finished() {
            if atlas.index + 1 < length {
                atlas.index = (atlas.index + 1) % length;
            } else {
                commands.entity(e).despawn();
            }
        }
    }
}