//! Demonstrates techniques for creating a hierarchy of parent and child entities.
//!
//! When [`DefaultPlugins`] are added to your app, systems are automatically added to propagate
//! [`Transform`] and [`Visibility`] from parents to children down the hierarchy,
//! resulting in a final [`GlobalTransform`] and [`InheritedVisibility`] component for each entity.

use std::{f32::consts::*, time::Duration};

use bevy::{color::palettes::css::*, ecs::relationship::RelatedSpawner, prelude::*};

fn main() {
    App::new()
        .add_plugins(DefaultPlugins)
        .add_systems(Startup, setup)
        .init_state::<Showcase>()
        .insert_resource(Delta(Duration::ZERO))
        .add_systems(OnEnter(Showcase::WithChildren), setup_with_children)
        .add_systems(OnEnter(Showcase::ChildrenSpawn), setup_children_spawn)
        .add_systems(OnEnter(Showcase::ChildrenMacro), spawn_children_macro)
        .add_systems(OnEnter(Showcase::ChildrenIter), setup_children_iter)
        .add_systems(OnEnter(Showcase::Related), setup_children_related)
        .add_systems(OnEnter(Showcase::Bsn), setup_bsn)
        .add_systems(Update, (rotate, switch_scene))
        .run();
}

#[derive(Debug, Clone, Eq, PartialEq, Hash, States, Default)]
enum Showcase {
    #[default]
    WithChildren,
    ChildrenSpawn,
    ChildrenMacro,
    ChildrenIter,
    Related,
    Bsn,
}

impl Showcase {
    fn next(&self) -> Self {
        match self {
            Showcase::WithChildren => Showcase::ChildrenSpawn,
            Showcase::ChildrenSpawn => Showcase::ChildrenMacro,
            Showcase::ChildrenMacro => Showcase::ChildrenIter,
            Showcase::ChildrenIter => Showcase::Related,
            Showcase::Related => Showcase::Bsn,
            Showcase::Bsn => Showcase::WithChildren,
        }
    }
}

fn switch_scene(
    keyboard: Res<ButtonInput<KeyCode>>,
    scene: Res<State<Showcase>>,
    mut next_scene: ResMut<NextState<Showcase>>,
) {
    if keyboard.just_pressed(KeyCode::Space) {
        info!("Switching scene");
        next_scene.set(scene.get().next());
    }
}

fn setup(mut commands: Commands) {
    commands.spawn(Camera2d);
}

#[derive(Resource)]
struct Delta(Duration);

fn setup_common(
    commands: &mut Commands,
    time: &Res<Time>,
    delta: &mut ResMut<Delta>,
    title: &str,
    stage: Showcase,
) {
    delta.0 = time.elapsed();
    commands.spawn((
        Text::new(title),
        TextFont::from_font_size(36.),
        DespawnOnExit(stage),
    ));
}

fn setup_with_children(
    mut commands: Commands,
    asset_server: Res<AssetServer>,
    time: Res<Time>,
    mut delta: ResMut<Delta>,
) {
    let texture = asset_server.load("branding/icon.png");

    setup_common(
        &mut commands,
        &time,
        &mut delta,
        "with_children()\nPress Space to continue",
        Showcase::WithChildren,
    );

    // Spawn a root entity with no parent
    let parent = commands
        .spawn((
            Sprite::from_image(texture.clone()),
            Transform::from_scale(Vec3::splat(0.75)),
            DespawnOnExit(Showcase::WithChildren),
        ))
        // With that entity as a parent, run a lambda that spawns its children
        .with_children(|parent| {
            // parent is a ChildSpawnerCommands, which has a similar API to Commands
            parent.spawn((
                Transform::from_xyz(250.0, 0.0, 0.0).with_scale(Vec3::splat(0.75)),
                Sprite {
                    image: texture.clone(),
                    color: BLUE.into(),
                    ..default()
                },
            ));
        })
        // Store parent entity for next sections
        .id();

    // Another way is to use the add_child function to add children after the parent
    // entity has already been spawned.
    let child = commands
        .spawn((
            Sprite {
                image: texture.clone(),
                color: LIME.into(),
                ..default()
            },
            Transform::from_xyz(0.0, 250.0, 0.0).with_scale(Vec3::splat(0.75)),
        ))
        .id();

    // Add child to the parent.
    commands.entity(parent).add_child(child);
}

fn setup_children_spawn(
    mut commands: Commands,
    asset_server: Res<AssetServer>,
    time: Res<Time>,
    mut delta: ResMut<Delta>,
) {
    let texture = asset_server.load("branding/icon.png");

    setup_common(
        &mut commands,
        &time,
        &mut delta,
        "Children::spawn() \nPress Space to continue",
        Showcase::ChildrenSpawn,
    );

    // Children can also be spawned using the `Children` component as part of the parent's bundle.
    commands.spawn((
        Sprite::from_image(texture.clone()),
        Transform::from_scale(Vec3::splat(0.75)),
        DespawnOnExit(Showcase::ChildrenSpawn),
        Children::spawn((
            Spawn((
                Transform::from_xyz(250.0, 0.0, 0.0).with_scale(Vec3::splat(0.75)),
                Sprite {
                    image: texture.clone(),
                    color: BLUE.into(),
                    ..default()
                },
            )),
            // since they have to be explicitly created, `Children::spawn` can
            // mix implementers of `SpawnableList` while `children!` cannot.
            SpawnWith(|spawner: &mut RelatedSpawner<'_, ChildOf>| {
                spawner.spawn((
                    Transform::from_xyz(0.0, 250.0, 0.0).with_scale(Vec3::splat(0.75)),
                    Sprite {
                        image: texture,
                        color: LIME.into(),
                        ..default()
                    },
                ));
            }),
        )),
    ));
}

fn spawn_children_macro(
    mut commands: Commands,
    asset_server: Res<AssetServer>,
    time: Res<Time>,
    mut delta: ResMut<Delta>,
) {
    let texture = asset_server.load("branding/icon.png");

    setup_common(
        &mut commands,
        &time,
        &mut delta,
        "children!() \nPress Space to continue",
        Showcase::ChildrenMacro,
    );

    // The `children!` macro provides a convenient way to define children inline with their parent.
    commands.spawn((
        Sprite::from_image(texture.clone()),
        Transform::from_scale(Vec3::splat(0.75)),
        DespawnOnExit(Showcase::ChildrenMacro),
        children![
            (
                Transform::from_xyz(250.0, 0.0, 0.0).with_scale(Vec3::splat(0.75)),
                Sprite {
                    image: texture.clone(),
                    color: BLUE.into(),
                    ..default()
                },
            ),
            (
                Transform::from_xyz(0.0, 250.0, 0.0).with_scale(Vec3::splat(0.75)),
                Sprite {
                    image: texture,
                    color: LIME.into(),
                    ..default()
                },
            )
        ],
    ));
}

fn setup_children_iter(
    mut commands: Commands,
    asset_server: Res<AssetServer>,
    time: Res<Time>,
    mut delta: ResMut<Delta>,
) {
    let texture = asset_server.load("branding/icon.png");

    setup_common(
        &mut commands,
        &time,
        &mut delta,
        "SpawnIter() \nPress Space to continue",
        Showcase::ChildrenIter,
    );

    // You can also spawn children from an iterator yielding bundles.
    let child_components = [
        (
            Transform::from_xyz(250.0, 0.0, 0.0).with_scale(Vec3::splat(0.75)),
            BLUE,
        ),
        (
            Transform::from_xyz(0.0, 250.0, 0.0).with_scale(Vec3::splat(0.75)),
            LIME,
        ),
    ];

    commands.spawn((
        Sprite::from_image(texture.clone()),
        Transform::from_scale(Vec3::splat(0.75)),
        DespawnOnExit(Showcase::ChildrenIter),
        Children::spawn(SpawnIter(child_components.into_iter().map(
            move |(transform, color)| {
                (
                    transform,
                    Sprite {
                        image: texture.clone(),
                        color: color.into(),
                        ..default()
                    },
                )
            },
        ))),
    ));
}

fn setup_children_related(
    mut commands: Commands,
    asset_server: Res<AssetServer>,
    time: Res<Time>,
    mut delta: ResMut<Delta>,
) {
    let texture = asset_server.load("branding/icon.png");

    setup_common(
        &mut commands,
        &time,
        &mut delta,
        "related!() \nPress Space to continue",
        Showcase::Related,
    );

    // You can also spawn entities with relationships other than parent/child.
    commands.spawn((
        Sprite::from_image(texture.clone()),
        Transform::from_scale(Vec3::splat(0.75)),
        DespawnOnExit(Showcase::Related),
        // the `related!` macro will spawn entities according to the `Children: RelationshipTarget` trait, but other types implementing `RelationshipTarget` can be used as well.
        related!(Children[
            (
                Transform::from_xyz(250.0, 0.0, 0.0).with_scale(Vec3::splat(0.75)),
                Sprite {
                    image: texture.clone(),
                    color: BLUE.into(),
                    ..default()
                },
            ),
            (
                Transform::from_xyz(0.0, 250.0, 0.0).with_scale(Vec3::splat(0.75)),
                Sprite {
                    image: texture,
                    color: LIME.into(),
                    ..default()
                },
            )
        ]),
    ));
}

fn setup_bsn(mut commands: Commands, time: Res<Time>, mut delta: ResMut<Delta>) {
    setup_common(
        &mut commands,
        &time,
        &mut delta,
        "BSN\nPress Space to continue",
        Showcase::Bsn,
    );

    // BSN can also define related entities inline. Asset paths are loaded when the scene is spawned.
    commands.spawn_scene(bsn! {
        Sprite { image: "branding/icon.png" }
        Transform::from_scale(Vec3::splat(0.75))
        DespawnOnExit::<Showcase>(Showcase::Bsn)
        Children [
            Transform::from_xyz(250.0, 0.0, 0.0).with_scale(Vec3::splat(0.75))
            Sprite {
                image: "branding/icon.png",
                color: BLUE,
            }
            --
            Transform::from_xyz(0.0, 250.0, 0.0).with_scale(Vec3::splat(0.75))
            Sprite {
                image: "branding/icon.png",
                color: LIME,
            }
        ]
    });
}

// A simple system to rotate the root entity, and rotate all its children separately
fn rotate(
    mut commands: Commands,
    time: Res<Time>,
    delta: Res<Delta>,
    mut parents_query: Query<(Entity, &Children), With<Sprite>>,
    mut transform_query: Query<&mut Transform, With<Sprite>>,
) {
    for (parent, children) in &mut parents_query {
        if let Ok(mut transform) = transform_query.get_mut(parent) {
            transform.rotate_z(-PI / 2. * time.delta_secs());
        }

        // To iterate through the entities children, just treat the Children component as a Vec
        // Alternatively, you could query entities that have a ChildOf component
        for child in children {
            if let Ok(mut transform) = transform_query.get_mut(*child) {
                transform.rotate_z(PI * time.delta_secs());
            }
        }

        // To demonstrate removing children, we'll remove a child after a couple of seconds.
        let elapsed = time.elapsed_secs() - delta.0.as_secs_f32();
        if elapsed >= 2.0 && children.len() == 2 {
            let child = children.last().unwrap();
            commands.entity(*child).despawn();
        }

        if elapsed >= 4.0 {
            // This will remove the entity from its parent's list of children, as well as despawn
            // any children the entity has.
            commands.entity(parent).despawn();
        }
    }
}