monologue-trees/bin/ui-wtf.rs

use bevy::{
    input::{keyboard::KeyboardInput, ButtonState},
    prelude::*,
    window::PrimaryWindow,
};
use monologue_trees::{debug::*, ui::*};

fn main() {
    App::new()
        .add_plugins((
            DefaultPlugins.set(WindowPlugin {
                primary_window: Some(Window {
                    title: "UI WTF".into(),
                    resolution: (640., 480.).into(),
                    ..default()
                }),
                ..default()
            }),
            GameUiPlugin,
        ))
        .init_resource::<Icon>()
        .add_systems(Startup, init_ui)
        .add_systems(Update, (cursors, container))
        .run();
}

const CURSORS: [CursorIcon; 35] = [
    CursorIcon::Default,
    CursorIcon::Crosshair,
    CursorIcon::Hand,
    CursorIcon::Arrow,
    CursorIcon::Move,
    CursorIcon::Text,
    CursorIcon::Wait,
    CursorIcon::Help,
    CursorIcon::Progress,
    CursorIcon::NotAllowed,
    CursorIcon::ContextMenu,
    CursorIcon::Cell,
    CursorIcon::VerticalText,
    CursorIcon::Alias,
    CursorIcon::Copy,
    CursorIcon::NoDrop,
    CursorIcon::Grab,
    CursorIcon::Grabbing,
    CursorIcon::AllScroll,
    CursorIcon::ZoomIn,
    CursorIcon::ZoomOut,
    CursorIcon::EResize,
    CursorIcon::NResize,
    CursorIcon::NeResize,
    CursorIcon::NwResize,
    CursorIcon::SResize,
    CursorIcon::SeResize,
    CursorIcon::SwResize,
    CursorIcon::WResize,
    CursorIcon::EwResize,
    CursorIcon::NsResize,
    CursorIcon::NeswResize,
    CursorIcon::NwseResize,
    CursorIcon::ColResize,
    CursorIcon::RowResize,
];

#[derive(Debug, Component, Resource, Default)]
struct Icon(CursorIcon);

fn init_ui(mut commands: Commands) {
    commands.spawn((
        Camera2dBundle { ..default() },
        UiCameraConfig { show_ui: true },
    ));

    commands
        .spawn(NodeBundle {
            style: Style {
                height: Val::Percent(100.0),
                width: Val::Percent(100.0),
                justify_content: JustifyContent::Center,
                align_items: AlignItems::Center,
                ..default()
            },
            background_color: BackgroundColor(Color::GRAY),
            ..default()
        })
        .with_children(|parent| {
            parent
                .spawn((
                    GameUi::Nav,
                    // Name::new("Game Nav"),
                    NodeBundle { ..default() },
                ))
                .with_children(|parent| {
                    parent
                        .spawn((
                            GameUi::Tab,
                            Name::new("Grow/Shrink Tab"),
                            NodeBundle { ..default() },
                        ))
                        .with_children(|parent| {
                            parent.spawn((
                                Container,
                                GameUi::Set,
                                // Name::new("Grow/Shrink Set"),
                                NodeBundle { ..default() },
                            ));
                        });

                    parent
                        .spawn((
                            GameUi::Tab,
                            Name::new("Cursor Icons Tab"),
                            NodeBundle { ..default() },
                        ))
                        .with_children(|parent| {
                            parent
                                .spawn((
                                    GameUi::Set,
                                    // Name::new("Cursor Icons Set"),
                                    NodeBundle { ..default() },
                                ))
                                .with_children(|parent| {
                                    CURSORS.iter().for_each(|&icon| {
                                        parent.spawn((
                                            GameUi::Button,
                                            Name::new(format!("{:?}", icon)),
                                            NodeBundle { ..default() },
                                            Icon(icon),
                                        ));
                                    });
                                });
                        });
                });
        });
}

fn cursors(
    events: Query<(&Interaction, &Icon), (Changed<Interaction>, With<Button>)>,
    mut primary_window: Query<&mut Window, With<PrimaryWindow>>,
    mut curr: ResMut<Icon>,
) {
    events.iter().for_each(|(&interaction, &ref icon)| {
        let mut window = primary_window.single_mut();

        match interaction {
            Interaction::Hovered => {
                (*window).cursor.icon = icon.0.clone();
            }
            Interaction::Pressed => {
                curr.0 = icon.0.clone();
            }
            Interaction::None => {
                (*window).cursor.icon = curr.0.clone();
            }
        }
    })
}

#[derive(Debug, Component)]
struct Container;

fn container(
    mut events: EventReader<KeyboardInput>,
    mut commands: Commands,
    root: Query<Entity, With<Container>>,
    children: Query<&Children, With<Container>>,
) {
    events.iter().for_each(
        |KeyboardInput {
             key_code, state, ..
         }| {
            match (key_code, state) {
                (Some(KeyCode::Up), ButtonState::Pressed) => {
                    commands.entity(root.single()).with_children(|parent| {
                        parent.spawn((
                            GameUi::Button,
                            Name::new("asdfwtf"),
                            NodeBundle { ..default() },
                        ));
                    });
                }
                (Some(KeyCode::Down), ButtonState::Pressed) => {
                    children.single().iter().last().iter().for_each(|&&child| {
                        commands.entity(child).despawn_recursive();
                    });
                }
                _ => (),
            }
        },
    )
}