From 06fb10b79960cedc3b8c3153e217c8af2afebc39 Mon Sep 17 00:00:00 2001
From: Nicolas Sanchez <sanchez.nicolas96@gmail.com>
Date: Fri, 6 Dec 2024 19:17:15 -0300
Subject: [PATCH] Update deathmatch endgame check to fit teams feature

---
 apps/arena/lib/arena/game_updater.ex | 27 ++++++++++++++++++---------
 1 file changed, 18 insertions(+), 9 deletions(-)

diff --git a/apps/arena/lib/arena/game_updater.ex b/apps/arena/lib/arena/game_updater.ex
index 2bd09715c..55679a3dd 100644
--- a/apps/arena/lib/arena/game_updater.ex
+++ b/apps/arena/lib/arena/game_updater.ex
@@ -329,29 +329,38 @@ defmodule Arena.GameUpdater do
   end
 
   def handle_info(:deathmatch_end_game_check, state) do
-    players =
+    players_with_kills =
       state.game_state.players
       |> Enum.map(fn {player_id, player} ->
         %{kills: kills} = GameTracker.get_player_result(player_id)
-        {player_id, player, kills}
+        Map.put(player, :kills, kills)
+      end)
+
+    {winner_team, _team_kills} =
+      players_with_kills
+      |> Enum.group_by(fn player -> player.aditional_info.team end)
+      |> Enum.map(fn {team, players} ->
+        team_kills = Enum.reduce(players, 0, fn player, acc -> player.kills + acc end)
+        {team, team_kills}
       end)
-      |> Enum.sort_by(fn {_player_id, _player, kills} -> kills end, :desc)
+      |> Enum.max_by(fn {_team, team_kills} -> team_kills end)
 
-    {winner_id, winner, _kills} = Enum.at(players, 0)
+    winners =
+      Enum.filter(state.game_state.players, fn {_player_id, player} -> player.aditional_info.team == winner_team end)
 
     state =
       state
       |> put_in([:game_state, :status], :ENDED)
       |> update_in([:game_state], fn game_state ->
-        players
-        |> Enum.reduce(game_state, fn {player_id, _player, _kills}, game_state_acc ->
-          put_player_position(game_state_acc, player_id)
+        players_with_kills
+        |> Enum.reduce(game_state, fn player, game_state_acc ->
+          put_player_position(game_state_acc, player.id)
         end)
       end)
 
     PubSub.broadcast(Arena.PubSub, state.game_state.game_id, :end_game_state)
-    broadcast_game_ended(winner, state.game_state)
-    GameTracker.finish_tracking(self(), winner_id)
+    broadcast_game_ended(winners, state.game_state)
+    GameTracker.finish_tracking(self(), winner_team)
 
     Process.send_after(self(), :game_ended, state.game_config.game.shutdown_game_wait_ms)