xevion/dynamic-preauth
1
0
Fork 0
mirror of https://github.com/Xevion/dynamic-preauth.git synced 2025-12-05 23:14:53 -06:00
Code Issues Packages Projects Releases Wiki Activity

Setup websocket reconnection logic

Browse Source
This commit is contained in:
Xevion
2025-01-01 11:21:32 -06:00
parent d2ec94f578
commit e8fe3e8ec9
1 changed files with 43 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
frontend/src/components/useSocket.ts
View File

@@ -25,6 +25,7 @@ function useSocket(): UseSocketResult {
const [downloads, setDownloads] = useState<Download[] | null>(null); const [downloads, setDownloads] = useState<Download[] | null>(null);
const [executables, setExecutables] = useState<Executable[] | null>(null); const [executables, setExecutables] = useState<Executable[] | null>(null);
const socketRef = useRef<WebSocket | null>(null); const socketRef = useRef<WebSocket | null>(null);
const allowReconnectRef = useRef<boolean>(true);
function deleteDownload(download_token: number) { function deleteDownload(download_token: number) {
if (socketRef.current == null) { if (socketRef.current == null) {
@@ -43,40 +44,56 @@ function useSocket(): UseSocketResult {
} }
useEffect(() => { useEffect(() => {
const socket = new WebSocket( function connect() {
(window.location.protocol === "https:" ? "wss://" : "ws://") + const socket = new WebSocket(
(import.meta.env.DEV ? "localhost:5800" : window.location.host) + (window.location.protocol === "https:" ? "wss://" : "ws://") +
"/ws" (import.meta.env.DEV ? "localhost:5800" : window.location.host) +
); "/ws"
socketRef.current = socket; );
socketRef.current = socket;
socket.onmessage = (event) => { socket.onmessage = (event) => {
const data = JSON.parse(event.data); const data = JSON.parse(event.data);
if (data.type == undefined) if (data.type == undefined)
throw new Error("Received message without type"); throw new Error("Received message without type");
switch (data.type) { switch (data.type) {
case "state": case "state":
setId(data.session.id as number); setId(data.session.id as number);
setDownloads(data.session.downloads as Download[]); setDownloads(data.session.downloads as Download[]);
break; break;
case "executables": case "executables":
setExecutables(data.executables as Executable[]); setExecutables(data.executables as Executable[]);
break; break;
default: default:
console.warn("Received unknown message type", data.type); console.warn("Received unknown message type", data.type);
} }
}; };
socket.onclose = (event) => { socket.onclose = (event) => {
console.log("WebSocket connection closed", event); console.log("WebSocket connection closed", event);
};
socketRef.current = null;
if (allowReconnectRef.current) {
setId(null);
setDownloads(null);
setExecutables(null);
setTimeout(() => {
connect();
}, 3000);
}
};
}
connect();
return () => { return () => {
// Close the socket when the component is unmounted // Close the socket when the component is unmounted
console.log("Unmounting, closing WebSocket connection"); console.log("Unmounting, closing WebSocket connection");
socket.close(); socketRef.current?.close();
allowReconnectRef.current = false;
}; };
}, []); }, []);