40 lines
1023 B
JavaScript
40 lines
1023 B
JavaScript
import { useEffect, useState, useCallback } from "react";
|
|
|
|
function useNetworkedState(defaultState, session, eventName) {
|
|
const [state, _setState] = useState(defaultState);
|
|
// Used to control whether the state needs to be sent to the socket
|
|
const [dirty, setDirty] = useState(false);
|
|
|
|
// Update dirty at the same time as state
|
|
const setState = useCallback((update, sync = true) => {
|
|
_setState(update);
|
|
setDirty(sync);
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
if (session.socket && dirty) {
|
|
session.socket.emit(eventName, state);
|
|
setDirty(false);
|
|
}
|
|
}, [session.socket, dirty, eventName, state]);
|
|
|
|
useEffect(() => {
|
|
function handleSocketEvent(data) {
|
|
_setState(data);
|
|
}
|
|
|
|
if (session.socket) {
|
|
session.socket.on(eventName, handleSocketEvent);
|
|
}
|
|
return () => {
|
|
if (session.socket) {
|
|
session.socket.off(eventName, handleSocketEvent);
|
|
}
|
|
};
|
|
}, [session.socket, eventName]);
|
|
|
|
return [state, setState];
|
|
}
|
|
|
|
export default useNetworkedState;
|