From: AeonLucid <aeonlucid@gmail.com>
Date: Sun, 20 Sep 2020 22:46:04 +0000 (+0200)
Subject: Refactor DisconnectReason
X-Git-Tag: v1.0.0~10
X-Git-Url: https://git.deb.at/?a=commitdiff_plain;h=2a0cc2814342f5de5796e4ca0e8694ecef184399;p=rhonda%2Fimpostor.git

Refactor DisconnectReason
---

diff --git a/src/AmongUs.Server/Net/Client.cs b/src/AmongUs.Server/Net/Client.cs
index c4bfe14..a5c18a9 100644
--- 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 97f9624..f1f9015 100644
--- 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);
             }
         }