43 lines
1.0 KiB
JavaScript
43 lines
1.0 KiB
JavaScript
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];
|
|
};
|