From dbb9eea25f80e7984a112409863be5191af5bf5e Mon Sep 17 00:00:00 2001
From: Logan Hunt <loganhunt@simponic.xyz>
Date: Fri, 1 Apr 2022 16:04:00 -0600
Subject: Added way too much stuff

---
 client/utils/use_chat.js | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 client/utils/use_chat.js

(limited to 'client/utils/use_chat.js')

diff --git a/client/utils/use_chat.js b/client/utils/use_chat.js
new file mode 100644
index 0000000..d5250bf
--- /dev/null
+++ b/client/utils/use_chat.js
@@ -0,0 +1,48 @@
+import { useState, useContext, useEffect, useRef } from 'react';
+import { AuthContext } from './auth_context';
+import { io } from 'socket.io-client';
+
+export const useChat = (chatRoom) => {
+  const [messages, setMessages] = useState([]);
+  const [connections, setConnections] = useState([]);
+  const messageRef = useRef([]);
+  const connectionsRef = useRef([]);
+  const [socket, setSocket] = useState({});
+  const [authToken] = useContext(AuthContext);
+
+  useEffect(() => {
+    if (chatRoom?.id) {
+      const socket = io({
+        query: {
+          chatRoomId: chatRoom.id,
+        },
+        auth: {
+          token: authToken,
+        },
+      });
+      socket.on('connect', () => {
+        setSocket(socket);
+      });
+      socket.on('new-message', (message) => {
+        messageRef.current.push(message);
+        setMessages([...messageRef.current]);
+      });
+      socket.on('userlist', ({ users }) => {
+        connectionsRef.current = users;
+        setConnections([...connectionsRef.current]);
+      });
+      return () => {
+        socket.off('new-message');
+        socket.close();
+      };
+    }
+  }, [chatRoom]);
+
+  const sendMessage = (message) => {
+    if (socket?.emit) {
+      socket.emit('message', message);
+    }
+  };
+
+  return [connections, messages, sendMessage];
+};
-- 
cgit v1.2.3-70-g09d2