1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
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];
};
|
