From 42cf50ee7521bd751f4d0f0798276e548bb83fee Mon Sep 17 00:00:00 2001
From: Logan Hunt <loganhunt@simponic.xyz>
Date: Wed, 30 Mar 2022 22:15:20 -0600
Subject: Working

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

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

diff --git a/client/utils/use_messages.js b/client/utils/use_messages.js
new file mode 100644
index 0000000..e3a501e
--- /dev/null
+++ b/client/utils/use_messages.js
@@ -0,0 +1,42 @@
+import { useState, useContext, useEffect, useRef } from 'react';
+import { AuthContext } from './auth_context';
+import { io } from 'socket.io-client';
+
+export const useMessages = (chatRoom) => {
+  const [messages, setMessages] = useState([]);
+  const messageRef = 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]);
+      });
+      return () => {
+        socket.off('new-message');
+        socket.close();
+      };
+    }
+  }, [chatRoom]);
+
+  const sendMessage = (message) => {
+    if (socket?.emit) {
+      socket.emit('message', message);
+    }
+  };
+
+  return [messages, sendMessage];
+};
-- 
cgit v1.2.3-70-g09d2