From 70d41ab57753d2c13fe80cd0f5875dcb233816d3 Mon Sep 17 00:00:00 2001
From: Joshua Ashton <joshua@froggi.es>
Date: Sat, 1 Aug 2020 22:53:02 +0100
Subject: [PATCH] Misc protocol cleanup

---
 src/Protocol.cpp | 2 +-
 src/Protocol.h   | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/Protocol.cpp b/src/Protocol.cpp
index 4c79313..bb0a012 100644
--- a/src/Protocol.cpp
+++ b/src/Protocol.cpp
@@ -35,7 +35,7 @@ namespace Feather
                 uint16_t port = packet.Read<uint16_t>();
                 
                 // next desired state
-                ProtocolState intention = (ProtocolState)(packet.ReadVarInt());
+                ProtocolState intention = packet.ReadVarInt<ProtocolState>();
 
                 printf("[Protocol] Client Intention Packet: version=%d, serverIp=%s, port=%u, intention=%d\n",
                     clientProtocolVersion,
diff --git a/src/Protocol.h b/src/Protocol.h
index 0c4ff01..7a6dc9b 100644
--- a/src/Protocol.h
+++ b/src/Protocol.h
@@ -3,6 +3,7 @@
 #include "Common.h"
 #include "NetworkMessage.h"
 #include <cstdio>
+#include <cstdint>
 
 namespace Feather
 {
@@ -10,7 +11,7 @@ namespace Feather
     class PacketReader;
     class MinecraftClient;
 
-    enum class ProtocolState
+    enum class ProtocolState : int32_t
     {
         Handholding = -1,
         Play        = 0,