/*
* This file is part of laurelin_client
* Copyright (C) 2023 Jonni Liljamo <jonni@liljamo.com>
*
* Licensed under GPL-3.0-only.
* See LICENSE for licensing information.
*/
use bevy::prelude::*;
use bevy_mod_picking::prelude::*;
use bevy_rapier3d::prelude::*;
use crate::{
api::game::{Action, Command},
game_status::PlayerState,
plugins::GameActionCreateCallEvent,
Global,
};
use super::{
card::{visual_card_kind, ClickedCard, VisualCard, VisualCardBundle},
GameData,
};
pub struct SupplyPlugin;
impl Plugin for SupplyPlugin {
fn build(&self, app: &mut App) {
app.add_event::<SpawnSupplyPilesEvent>()
.add_event::<PositionSupplyPilesEvent>()
.add_systems(
(
spawn_supply_piles.run_if(on_event::<SpawnSupplyPilesEvent>()),
apply_system_buffers,
position_supply_piles.run_if(on_event::<PositionSupplyPilesEvent>()),
)
.chain(),
)
.add_system(handle_clicked_supply_pile);
}
}
pub struct SpawnSupplyPilesEvent;
fn spawn_supply_piles(
mut commands: Commands,
game_data: Res<GameData>,
mut psp_ev_w: EventWriter<PositionSupplyPilesEvent>,
pile_query: Query<Entity, With<visual_card_kind::Supply>>,
) {
let Some(status) = &game_data.game_status else {
warn!("game_status was none");
return;
};
// despawn possible existing supply piles
for entity in pile_query.iter() {
commands.entity(entity).despawn_recursive();
}
for (index, pile) in status.supply_piles.iter().enumerate() {
commands
.spawn(VisualCardBundle {
visual_card: VisualCard {
card: pile.card.clone(),
},
rigid_body: RigidBody::Fixed,
..Default::default()
})
.insert(visual_card_kind::Supply(index))
.insert(OnPointer::<Over>::target_component_mut::<Transform>(
|_over, transform| {
transform.translation.y += 0.1;
},
))
.insert(OnPointer::<Out>::target_component_mut::<Transform>(
|_over, transform| {
transform.translation.y -= 0.1;
},
));
}
psp_ev_w.send(PositionSupplyPilesEvent);
}
pub struct PositionSupplyPilesEvent;
fn position_supply_piles(
mut pile_query: Query<(&VisualCard, &mut Transform), With<visual_card_kind::Supply>>,
) {
let mut piles: Vec<(&VisualCard, Mut<Transform>)> = pile_query.iter_mut().collect::<Vec<_>>();
piles.sort_by_key(|(vc, _t)| vc.card.name.clone());
piles.sort_by_key(|(vc, _t)| vc.card.cost);
// split piles into top and bottom row
let mid = piles.len() / 2;
let (p1, p2) = piles.split_at_mut(mid);
// keep track of offset between cards
let mut offset = 0.;
// top row
for (_, t) in p1 {
t.translation.x += offset;
offset += 1.0;
}
// reset offset when changing rows
offset = 0.;
// bottom row
for (_, t) in p2 {
t.translation.z += 1.5;
t.translation.x += offset;
offset += 1.0;
}
}
fn handle_clicked_supply_pile(
mut commands: Commands,
card_query: Query<
(Entity, &VisualCard, &visual_card_kind::Supply),
(With<visual_card_kind::Supply>, With<ClickedCard>),
>,
mut gac_ev_w: EventWriter<GameActionCreateCallEvent>,
mut game_data: ResMut<GameData>,
global: Res<Global>,
) {
if game_data.locked {
return;
}
let Ok((entity, card, card_kind)) = card_query.get_single() else {
return;
};
commands.entity(entity).remove::<ClickedCard>();
let player = game_data
.game_status
.as_ref()
.unwrap()
.players
.get(&global.user.as_ref().unwrap().id)
.unwrap();
#[allow(clippy::if_same_then_else)]
if player.state != PlayerState::BuyPhase {
// we ain't buying rn
return;
} else if player.buys == 0 {
// not enough buys
return;
} else if player.currency < card.card.cost {
// not enough currency
return;
} else if game_data
.game_status
.as_ref()
.unwrap()
.supply_piles
.get(card_kind.0)
.unwrap()
.amount
== 0
{
// no cards in supply
return;
}
game_data.locked = true;
gac_ev_w.send(GameActionCreateCallEvent {
action: Action::new(
&game_data.game.as_ref().unwrap().id,
&global.user.as_ref().unwrap().id,
&global.user.as_ref().unwrap().id,
&Command::TakeFromPile {
index: card_kind.0,
for_cost: card.card.cost,
},
None,
),
});
}