closeConnection logic adding

nodejs/sender-monitor/main.js

@@ -262,9 +262,13 @@ app.on("before-quit", async () => {
         }
     }
 
-    if (webrtc.isConnected()) {
-        webrtc.closeConnection();
-        logMessage("WebRTC connection closed.");
+    if (webrtc.connected) {  // Replace `connected` with the correct property or method if different
+        try {
+            await webrtc.closeConnection();  // Make sure closeConnection is a Promise, or remove `await`
+            logMessage("WebRTC connection closed.");
+        } catch (error) {
+            logMessage(`Failed to close WebRTC connection: ${error.message}`);
+        }
     }
 
     const tempFilePath = path.join(app.getPath("temp"), "ocr-capture.png");

nodejs/sender-monitor/webrtc.js

@@ -105,6 +105,29 @@ class WebRTCConnection extends EventEmitter {
             }
         });
     }
+
+    closeConnection() {
+        Object.values(this.peerConnections).forEach((peerConnection) => {
+            peerConnection.close();
+        });
+        Object.values(this.dataChannels).forEach((dataChannel) => {
+            if (dataChannel.readyState !== "closed") {
+                dataChannel.close();
+            }
+        });
+
+        // Clear the connection objects
+        this.peerConnections = {};
+        this.dataChannels = {};
+
+        // Disconnect the WebSocket connection
+        if (this.socket.connected) {
+            this.socket.disconnect();
+        }
+
+        this.emit("disconnected");
+        console.log("WebRTC connection closed.");
+    }
 }
 
 module.exports = new WebRTCConnection();