Refactor DisconnectReason

diff --git a/src/AmongUs.Server/Net/Client.cs b/src/AmongUs.Server/Net/Client.cs
index c4bfe14df85b5bf968bb5f0043e13ff82d483856..a5c18a9e9ed4750f10a1c48af1546838b2c17927 100644 (file)
--- a/src/AmongUs.Server/Net/Client.cs
+++ b/src/AmongUs.Server/Net/Client.cs
@@ -162,7 +162,7 @@ namespace AmongUs.Server.Net
                     var playerId = message.ReadPackedInt32();
                     var reason = message.ReadByte();
 
-                    Player.Game.HandleRemovePlayer(playerId, reason);
+                    Player.Game.HandleRemovePlayer(playerId, (DisconnectReason) reason);
                     break;
                 }
                 
@@ -258,7 +258,7 @@ namespace AmongUs.Server.Net
             {
                 if (Player.Game != null)
                 {
-                    Player.Game.HandleRemovePlayer(Id, 0);
+                    Player.Game.HandleRemovePlayer(Id, DisconnectReason.ExitGame);
                 }
             }
             catch (Exception ex)

diff --git a/src/AmongUs.Server/Net/Game.cs b/src/AmongUs.Server/Net/Game.cs
index 97f9624ddf87c08c2701b7308aae9aa1a703879e..f1f9015fe24f6cf4ae13b78dc7fe8848a5847f7b 100644 (file)
--- a/src/AmongUs.Server/Net/Game.cs
+++ b/src/AmongUs.Server/Net/Game.cs
@@ -135,7 +135,7 @@ namespace AmongUs.Server.Net
             }
         }
         
-        public void HandleRemovePlayer(int playerId, byte reason)
+        public void HandleRemovePlayer(int playerId, DisconnectReason reason)
         {
             if (_players.TryRemove(playerId, out var player))
             {
@@ -160,7 +160,7 @@ namespace AmongUs.Server.Net
 
             using (var packet = MessageWriter.Get(SendOption.Reliable))
             {
-                WriteRemovePlayerMessage(packet, false, playerId, (DisconnectReason) reason);
+                WriteRemovePlayerMessage(packet, false, playerId, reason);
                 SendToAllExcept(packet, player);
             }
         }