From 42b94656227686e5c3c4cfb181511d4eb09cfcdb Mon Sep 17 00:00:00 2001
From: ChrisChrome <chris@chrischro.me>
Date: Wed, 4 Sep 2024 13:51:14 -0600
Subject: [PATCH] Update socket brrr

---
 README.md     | 5 +++--
 WebSocket.lua | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index f6ed1fc..93ca2b7 100644
--- a/README.md
+++ b/README.md
@@ -10,7 +10,7 @@
 local ws = require(game.ServerScriptService:WaitForChild("WebSocket"))
 local server = ws({
 	url = "wss://echo.websocket.org/",
-	onMessage = function(message)
+	onMessage = function(socket, message)
 		print("Message received: " .. message)
 	end,
 	onError = function(error)
@@ -25,8 +25,9 @@ server.sendMessage("Hello World")
 ```lua
 local server = ws({
 	url = "wss://echo.websocket.org/",
-	onMessage = function(message)
+	onMessage = function(socket, message)
 		print("Message received: " .. message)
+		socket.close(); -- Close the socket after receiving a message, as an example
 	end,
 	onError = function(error)
 		print("Error: " .. error)
diff --git a/WebSocket.lua b/WebSocket.lua
index 5846e2c..f3687a9 100644
--- a/WebSocket.lua
+++ b/WebSocket.lua
@@ -20,7 +20,7 @@ end
 local ws = function (dict)
 	local socket = {}
 	local closed = false
-	local onMessage = dict.onMessage or function(msg)
+	local onMessage = dict.onMessage or function(socket, msg)
 		print(msg)
 	end
 	local onError = dict.onError or function(err)