From: miniduikboot <mini@duikbo.at>
Date: Sat, 26 Feb 2022 19:31:45 +0000 (+0100)
Subject: Change DisconnectAsync to use extension instead
X-Git-Tag: v1.7.0~10^2~1
X-Git-Url: https://git.deb.at/?a=commitdiff_plain;h=0b431dc5faa80d2150f224617295c23dd796f7cb;p=rhonda%2Fimpostor.git

Change DisconnectAsync to use extension instead
---

diff --git a/src/Impostor.Server/Net/ClientBase.cs b/src/Impostor.Server/Net/ClientBase.cs
index 74224fc..15dfe91 100644
--- a/src/Impostor.Server/Net/ClientBase.cs
+++ b/src/Impostor.Server/Net/ClientBase.cs
@@ -5,8 +5,6 @@ using Impostor.Api;
 using Impostor.Api.Innersloth;
 using Impostor.Api.Net;
 using Impostor.Api.Net.Messages;
-using Impostor.Api.Net.Messages.S2C;
-using Impostor.Hazel;
 using Impostor.Server.Net.State;
 
 namespace Impostor.Server.Net
@@ -55,15 +53,7 @@ namespace Impostor.Server.Net
 
         public async ValueTask DisconnectAsync(DisconnectReason reason, string? message = null)
         {
-            if (!Connection.IsConnected)
-            {
-                return;
-            }
-
-            using var writer = MessageWriter.Get();
-            MessageDisconnect.Serialize(writer, true, reason, message);
-
-            await Connection.DisconnectAsync(message ?? reason.ToString(), writer);
+            await Connection.CustomDisconnectAsync(message ?? reason.ToString());
         }
     }
 }