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