refactor main and rendered ipc

src/background/websockets.ts

@@ -0,0 +1,88 @@
+"use strict";
+
+import WebSocket from 'ws';
+
+let socket: WebSocket;
+
+const socketUrl = process.env.PLATFORM_URL;
+
+// Run every time we want to connect to backend.
+export default function useWebSockets(
+    receiveCallback: (s: string) => void,
+    openCallback?: () => void
+) {
+
+    // Returns bool (sucess or fail).
+    const sendMessage = (data: any) => {
+        console.log("WS:Sending message: ", data)
+
+        if (socket.readyState !== 1) {
+            return false
+        }
+
+        else {
+            socket.send(JSON.stringify(data))
+            return true
+        }
+
+    }
+
+    const send = async (data: Record<string, any>): Promise<boolean> => (
+        new Promise((resolve, reject) => {
+            if (socket.readyState !== 1) {
+                reject(false);
+            }
+            socket.send(JSON.stringify(data))
+            resolve(true);
+
+        }))
+
+
+    const onOpen = (_event: WebSocket.OpenEvent) => {
+
+        console.log("WS:Connected to WS Server!");
+
+        if (openCallback) openCallback();
+
+    }
+
+    const onServerMessage = (event: WebSocket.MessageEvent) => {
+
+        console.log("WS:Message received: ", event.data);
+
+        receiveCallback(event.data.toString())
+
+    }
+
+    const onClose = (_event: WebSocket.CloseEvent) => {
+        console.log("WS:Socket closed normally.")
+    }
+
+    // Reconnect automatically on error.
+    const onError = (event: WebSocket.ErrorEvent) => {
+        console.log("WS:WebSocket error: ", event.message);
+
+        console.log("Attempting reconnect in 1s.")
+        setTimeout(createSocket, 1000)
+    }
+
+    const createSocket = () => {
+        if (socketUrl)
+        socket = new WebSocket(socketUrl)
+
+        // Add listeners.
+        socket.addEventListener("open", onOpen)
+        socket.addEventListener("message", onServerMessage)
+        socket.addEventListener("close", onClose)
+        socket.addEventListener("error", onError)
+
+        console.log("WS:New socket created.")
+    }
+
+    return {
+        createSocket,
+        sendMessage,
+        send
+    }
+
+}