From ad4c55678a7421426ee13fefe44e109fa5d496fc Mon Sep 17 00:00:00 2001
From: CoCo_Sol007 <solois.corentin@gmail.com>
Date: Wed, 3 Apr 2024 17:02:15 +0200
Subject: [PATCH] save

---
 crates/border-wars/src/map/spawnpoint.rs | 40 +++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/crates/border-wars/src/map/spawnpoint.rs b/crates/border-wars/src/map/spawnpoint.rs
index c926855..2968e70 100644
--- a/crates/border-wars/src/map/spawnpoint.rs
+++ b/crates/border-wars/src/map/spawnpoint.rs
@@ -1,5 +1,6 @@
 //! TODO
 
+use bevnet::Connection;
 use bevy::prelude::*;
 use bevy::utils::hashbrown::HashSet;
 use bevy::utils::HashMap;
@@ -23,6 +24,7 @@ fn init_spawn_point(
     mut end_map_event: EventReader<EndMapGeneration>,
     players: Query<(Entity, &Player)>,
     map: Query<(Entity, &TilePosition)>,
+    connection: Res<Connection>,
 ) {
     if end_map_event.is_empty() {
         return;
@@ -36,9 +38,16 @@ fn init_spawn_point(
         .0
         .abs();
 
+    if radius == 0 {
+        panic!("Map radius must be greater than 0");
+    }
+
     let map_hashmap: HashMap<&TilePosition, Entity> = map.iter().map(|(e, p)| (p, e)).collect();
 
     let nb_player = players.iter().count();
+
+    let mut spawnpoints = Vec::with_capacity(nb_player);
+
     for (i, target_position) in TilePosition::new(0, 0)
         .ring(radius as usize / 2)
         .enumerate()
@@ -50,6 +59,35 @@ fn init_spawn_point(
         if i % (radius as usize * 3 / nb_player) != 0 {
             continue;
         }
-        commands.entity(*target_entity).despawn();
+        spawnpoints.push((*target_entity, target_position));
+    }
+
+    let mut sorted_players = players.iter().collect::<Vec<_>>();
+    sorted_players.sort_by(compare_player);
+    spawnpoints.sort_by(compare_spawnpoint_entity);
+
+    for (i, (target_entity, target_position)) in spawnpoints.iter().enumerate() {
+        let player = sorted_players[i].1;
+        if Some(player.uuid) == connection.identifier() {
+            commands.entity(*target_entity).despawn();
+        }
+    }
+}
+
+/// TODO
+fn compare_player((_, a): &(Entity, &Player), (_, b): &(Entity, &Player)) -> std::cmp::Ordering {
+    a.uuid.cmp(&b.uuid)
+}
+
+/// TODO
+fn compare_spawnpoint_entity(
+    (_, a): &(Entity, TilePosition),
+    (_, b): &(Entity, TilePosition),
+) -> std::cmp::Ordering {
+    let r = a.0.abs().cmp(&b.0.abs());
+    if r == std::cmp::Ordering::Equal {
+        a.1.cmp(&b.1)
+    } else {
+        r
     }
 }