set/src/play.rs

use bevy::prelude::*;

use crate::{deck::Card, GameState};

pub struct PlayPlugin;

impl Plugin for PlayPlugin {
    fn build(&self, app: &mut App) {
        app.add_event::<ServeCards>()
            .add_observer(serve_cards)
            .add_observer(fill_cards)
            .add_systems(OnEnter(GameState::Main), deal_table)
            .add_systems(Update, spin_cards);
    }
}

/// Marker for cards which the player has selected
#[derive(Component)]
pub(crate) struct Selected;

/// Where in the deck a card is
#[derive(Component)]
pub(crate) struct DeckOrder(pub u8);

/// Where on the board/table/play-space a card is
#[derive(Component)]
pub(crate) struct PlayLocation {
    pub x: u8,
    pub y: u8,
}

/// Which captured set a card is associated with
#[derive(Component)]
pub(crate) struct SetNumber(pub u8);

/// Debug system rotating selected cards for visual flare
fn spin_cards(mut query: Query<&mut Transform, (With<Card>, With<Selected>)>, time: Res<Time>) {
    let dt = time.delta().as_secs_f32();
    query.iter_mut().for_each(|mut t| {
        t.rotate_z(dt * 0.5);
    });
}

/// Observer system for adding/removing selected components when a card is clicked
pub(crate) fn toggle_selected(
    trigger: Trigger<Pointer<Click>>,
    mut commands: Commands,
    query: Query<&Selected>,
) {
    let e = trigger.entity();
    if query.contains(e) {
        commands.entity(e).remove::<Selected>();
    } else {
        commands.entity(e).insert(Selected);
    }
}

/// Observer system for resetting rotation when a card is de-selected
pub(crate) fn reset_rotation(
    trigger: Trigger<OnRemove, Selected>,
    mut transforms: Query<&mut Transform>,
) {
    let mut t = transforms.get_mut(trigger.entity()).unwrap();
    t.rotation = Quat::default();
}

/// Check a set when the "Set" button is clicked
pub(crate) fn check_set(
    _trigger: Trigger<Pointer<Click>>,
    mut query: Query<(Entity, &Card, &mut Visibility, &mut Transform), With<Selected>>,
    mut commands: Commands,
) {
    let mut cards = query.iter();
    if cards.len() == 3 {
        let ((_, a, _, _), (_, b, _, _), (_, c, _, _)) = (
            cards.next().unwrap(),
            cards.next().unwrap(),
            cards.next().unwrap(),
        );
        if is_set((a, b, c)) {
            query
                .iter_mut()
                .for_each(|(entity, _, mut visibility, mut transform)| {
                    *visibility = Visibility::Hidden;
                    *transform = Transform::default();
                    let set_number = 1;
                    commands
                        .entity(entity)
                        .remove::<PlayLocation>()
                        .insert(SetNumber(set_number));
                });
        }
    } else if cards.len() > 3 {
        info!("Too many cards!")
    } else if cards.len() < 3 {
        info!("Not enough cards!")
    }
}

/// Helper function to determine if three cards constitute a set
fn is_set((a, b, c): (&Card, &Card, &Card)) -> bool {
    let color = {
        ((a.color == b.color) && (b.color == c.color) && (c.color == a.color))
            || ((a.color != b.color) && (b.color != c.color) && (c.color != a.color))
    };
    let number = {
        ((a.number == b.number) && (b.number == c.number) && (c.number == a.number))
            || ((a.number != b.number) && (b.number != c.number) && (c.number != a.number))
    };
    let pattern = {
        ((a.pattern == b.pattern) && (b.pattern == c.pattern) && (c.pattern == a.pattern))
            || ((a.pattern != b.pattern) && (b.pattern != c.pattern) && (c.pattern != a.pattern))
    };
    let shape = {
        ((a.shape == b.shape) && (b.shape == c.shape) && (c.shape == a.shape))
            || ((a.shape != b.shape) && (b.shape != c.shape) && (c.shape != a.shape))
    };
    color && number && pattern && shape
}

#[derive(Event, Clone)]
struct ServeCards;

/// When requested, fill in empty spots on the board with cards
fn serve_cards(
    _trigger: Trigger<ServeCards>,
    cards: Query<(Entity, &DeckOrder)>,
    spots: Query<&PlayLocation>,
    mut commands: Commands,
) {
    info!("Serving cards");
    let mut target = (cards.iter().len() - 1) as u8;
    // Iterate over every x, y play location
    for this_x in 0..=3 {
        for this_y in 0..=3 {
            // If this spot does not have a card
            if spots
                .iter()
                .find(|PlayLocation { x, y }| *x == (this_x as u8) && *y == (this_y as u8))
                .is_none()
            {
                if let Some((e, _)) = cards.iter().find(|(_, deck_order)| deck_order.0 == target) {
                    // Set the selected card to this play location and remove from the deck
                    commands
                        .entity(e)
                        .remove::<DeckOrder>()
                        .insert(PlayLocation {
                            x: this_x,
                            y: this_y,
                        });
                    target = target - 1;
                }
            }
        }
    }
}

/// Trigger dealing cards when the game starts
fn deal_table(mut commands: Commands) {
    info!("Dealing table");
    commands.trigger(ServeCards);
}

/// Trigger serving cards when a new set is created
fn fill_cards(_trigger: Trigger<OnAdd, SetNumber>, mut commands: Commands) {
    info!("Filling cards");
    commands.trigger(ServeCards);
}

/// When a card is added to the board, place it on the screen
pub(crate) fn place_card(
    trigger: Trigger<OnAdd, PlayLocation>,
    mut query: Query<(&PlayLocation, &mut Visibility, &mut Transform)>,
) {
    let (PlayLocation { x, y }, mut visibility, mut transform) =
        query.get_mut(trigger.entity()).unwrap();

    // Set it's transform based on it's placement
    let card_size = [100.0, 160.0];
    let offset = Vec2::new(card_size[0] * 4.0, card_size[1] * 4.0) / 2.5;
    transform.translation = Vec3::new(
        ((*x as f32) * card_size[0]) - offset.x,
        ((*y as f32) * card_size[1]) - offset.y,
        0.0,
    );

    // Set it to visible
    *visibility = Visibility::Inherited;
}

pub(crate) fn check_for_sets(
    _trigger: Trigger<Pointer<Click>>,
    cards: Query<(Entity, &Card), With<PlayLocation>>,
    selected: Query<Entity, With<Selected>>,
    mut commands: Commands,
) {
    let candidate = cards
        // Iterate over all combinations of cards on the board
        .iter_combinations()
        // Only find combinations including currently selected cards
        .filter(|[(ea, _ca), (eb, _cb), (ec, _cc)]| {
            // If selected is empty, proceed
            match selected.iter().len() {
                0 => true,
                1 | 2 | 3 => {
                    let a = selected.contains(*ea);
                    let b = selected.contains(*eb);
                    let c = selected.contains(*ec);
                    match selected.iter().len() {
                        1 => a || b || c,
                        2 => a && b || b && c || a && c,
                        3 => a && b && c,
                        _ => panic!("WTF?"),
                    }
                }
                _ => panic!("Impossible!"),
            }
        })
        // return the first valid set
        .find_map(|[(ea, ca), (eb, cb), (ec, cc)]| {
            if is_set((ca, cb, cc)) {
                info!("\n\t{}\n\t{}\n\t{}", ca, cb, cc);
                Some((ea, eb, ec))
            } else {
                None
            }
        });

    if let Some((a, b, c)) = candidate {
        if !selected.contains(a) {
            commands.entity(a).insert(Selected);
        } else if !selected.contains(b) {
            commands.entity(b).insert(Selected);
        } else if !selected.contains(c) {
            commands.entity(c).insert(Selected);
        } else {
            info!("I cannot hold your hand further...");
        }
    } else {
        info!("No sets!");
    }
}