1 files changed, 24 insertions, 13 deletions

diff --git a/main.lua b/main.lua
index 40acc5b..31555ee 100644
--- a/main.lua
+++ b/main.lua
@@ -1,24 +1,35 @@
+---@diagnostic disable: duplicate-set-field
+
 local funsaac = RegisterMod("funsaac", 1)
-local json = require("json")
+require("messages")
+
+local client = require("websocket").new("127.0.0.1", 12345, "/")
 
-local requestServerInfo = {
-  Id = 1,
-  ClientName = "Funsaac v.0.0.1",
-  MessageVersion = 3
-}
+CONNECTED = false
 
-local client = require("websocket").new("127.0.0.1", 12345)
 function client:onmessage(message)
-  print(message)
+  HandleResponse(ResponseContent(message))
+  if CONNECTED == true then
+    self:send(GetMessage(Messages.StartScanning))
+    CONNECTED = false
+  end
 end
+
 function client:onopen()
-  local msg = '[{"RequestServerInfo": ' .. json.encode(requestServerInfo) .. "}]"
-  print(msg)
-  self:send(msg)
+  self:send(GetMessage(Messages.RequestServerInfo))
 end
 
-local function onUpdate()
+
+local render_tick = 0
+local s = 0;
+local function onRender()
   client:update()
+
+  render_tick = render_tick + 1
+  if render_tick >= 60 then
+    render_tick = 0
+    s = s + 1
+  end
 end
 
-funsaac:AddCallback(ModCallbacks.MC_POST_UPDATE, onUpdate)
\ No newline at end of file
+funsaac:AddCallback(ModCallbacks.MC_POST_RENDER, onRender)
\ No newline at end of file