From e2f4fb4c5820fc4fe1e9c6f7090b1f5b38b2cf14 Mon Sep 17 00:00:00 2001
From: Richard Neumann <mail@richard-neumann.de>
Date: Mon, 24 Jul 2023 14:54:27 +0200
Subject: [PATCH] Directly return payload as string

---
 rcon/battleye/client.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/rcon/battleye/client.py b/rcon/battleye/client.py
index f1e2511..2309512 100644
--- a/rcon/battleye/client.py
+++ b/rcon/battleye/client.py
@@ -51,7 +51,7 @@ class Client(BaseClient, socket_type=SOCK_DGRAM):
             )).type
         ].from_bytes(header, data[8:])
 
-    def receive(self, file: BufferedWriter) -> Response | bytes:
+    def receive(self, file: BufferedWriter) -> Response | str:
         """Receive a message."""
         server_messages = set()
 
@@ -66,12 +66,12 @@ class Client(BaseClient, socket_type=SOCK_DGRAM):
         if not server_messages:
             return response
 
-        return b''.join(
-            msg.payload for msg in
+        return ''.join(
+            msg.message for msg in
             sorted(server_messages, key=lambda msg: msg.seq)
         )
 
-    def communicate(self, request: Request) -> Response | bytes:
+    def communicate(self, request: Request) -> Response | str:
         """Send a request and receive a response."""
         with self._socket.makefile('wb') as file:
             file.write(bytes(request))
@@ -86,6 +86,4 @@ class Client(BaseClient, socket_type=SOCK_DGRAM):
 
     def run(self, command: str, *args: str) -> str:
         """Execute a command and return the text message."""
-        return self.communicate(
-            CommandRequest.from_command(command, *args)
-        ).message
+        return self.communicate(CommandRequest.from_command(command, *args))