use bevy::prelude::*;
use bevy_enum_filter::prelude::*;

fn main() {
    App::new()
        .add_plugins(DefaultPlugins.set(WindowPlugin {
            primary_window: Some(Window {
                title: "Import Example".to_string(),
                resolution: (100., 100.).into(),
                ..default()
            }),
            ..default()
        }))
        // ! === Add the Filter === ! //
        .add_enum_filter::<components::Choice>()
        // ! === Add the Filter === ! //
        .add_systems(Startup, systems::info)
        .add_systems(
            Update,
            (
                systems::spawn,
                systems::on_spawn_a,
                systems::on_spawn_b,
                systems::on_spawn_c,
            ),
        )
        .run();
}

mod components {
    use bevy::prelude::*;
    use bevy_enum_filter::prelude::*;

    #[derive(Component, EnumFilter)]
    pub enum Choice {
        A,
        B,
        C,
    }
}

mod systems {
    use bevy::prelude::*;
    use bevy_enum_filter::prelude::*;

    // ! === Import Enum AND Filter Module === ! //
    use super::components::{choice_filters, Choice};
    // ! === Import Enum AND Filter Module === ! //

    pub fn spawn(mut commands: Commands, input: Res<Input<KeyCode>>) {
        if input.just_pressed(KeyCode::A) {
            commands.spawn((Choice::A,));
        }
        if input.just_pressed(KeyCode::B) {
            commands.spawn((Choice::B,));
        }
        if input.just_pressed(KeyCode::C) {
            commands.spawn((Choice::C,));
        }
    }

    pub fn on_spawn_a(query: Query<Entity, Added<Enum!(Choice::A)>>) {
        for _ in &query {
            println!("Spawned entity with `Choice::A`!");
        }
    }

    pub fn on_spawn_b(query: Query<Entity, Added<Enum!(Choice::B)>>) {
        for _ in &query {
            println!("Spawned entity with `Choice::B`!");
        }
    }

    pub fn on_spawn_c(query: Query<Entity, Added<Enum!(Choice::C)>>) {
        for _ in &query {
            println!("Spawned entity with `Choice::C`!");
        }
    }

    pub fn info() {
        println!(
            "Press any of the following keys to spawn an entity with that value: [`A`, `B`, or `C`]"
        );
    }
}