From ce93dd70e5dee287f963e0055b5501076ee2500a Mon Sep 17 00:00:00 2001
From: John Cote <jwcotejr@gmail.com>
Date: Sat, 7 Sep 2024 19:40:39 -0400
Subject: [PATCH] ToontownClient: Handle CLIENT_HEARTBEAT, CLIENT_DISCONNECT

---
 otpgo/lua/MsgTypes.lua       |  8 ++++++++
 otpgo/lua/ToontownClient.lua | 13 ++++++++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)
 create mode 100644 otpgo/lua/MsgTypes.lua

diff --git a/otpgo/lua/MsgTypes.lua b/otpgo/lua/MsgTypes.lua
new file mode 100644
index 0000000..4238706
--- /dev/null
+++ b/otpgo/lua/MsgTypes.lua
@@ -0,0 +1,8 @@
+CLIENT_DISCONNECT =     37
+
+CLIENT_HEARTBEAT =      52
+
+-- new toontown specific login message, adds last logged in, and if child account has parent acount
+CLIENT_LOGIN_TOONTOWN = 125
+
+CLIENT_DISCONNECT_GENERIC = 1
diff --git a/otpgo/lua/ToontownClient.lua b/otpgo/lua/ToontownClient.lua
index 292a378..e5b24ba 100644
--- a/otpgo/lua/ToontownClient.lua
+++ b/otpgo/lua/ToontownClient.lua
@@ -1,6 +1,17 @@
 package.path = package.path .. ";lua/?.lua"
 
+-- Load message types
+dofile("lua/MsgTypes.lua")
+
 function receiveDatagram(client, dgi)
+    -- Client received datagrams
     msgType = dgi:readUint16()
-    print(msgType)
+
+    if msgType == CLIENT_HEARTBEAT then
+        client:handleHeartbeat()
+    elseif msgType == CLIENT_DISCONNECT then
+        client:handleDisconnect()
+    else
+        client:sendDisconnect(CLIENT_DISCONNECT_GENERIC, string.format("Unknown message type: %d", msgType), true)
+    end
 end