use bevy::prelude::*;
use crate::states::main_menu::{MainMenuNewGameButton, MainMenuSettingsButton};
use crate::ui::button_click::ButtonClickMessage;
use crate::ui::click::handle_click_system;

#[derive(SystemSet, Debug, Clone, PartialEq, Eq, Hash)]
pub struct MainMenuButtonSet;

pub struct MainMenuUiPlugin;

impl Plugin for MainMenuUiPlugin {
    fn build(&self, app: &mut App) {
        app.add_message::<ButtonClickMessage<MainMenuNewGameButton>>()
            .add_message::<ButtonClickMessage<MainMenuSettingsButton>>()
            .add_systems(
                Update,
                (
                    handle_click_system::<MainMenuNewGameButton>,
                    handle_click_system::<MainMenuSettingsButton>,
                ).in_set(MainMenuButtonSet)
            );
    }
}