Rework webrtc reconnection

src/APIManager.js

@@ -39,6 +39,16 @@ module.exports = class APIManager {
             description: 'Disconnect from the current instance, ending the WebSocket communication.',
         });
 
+        // Register a function to restart the webrtc connection
+        this.registerFunction({
+            name: 'reconnect',
+            category: 'VM_communication',
+            fn: () => {
+                this.instance.reconnect();
+            },
+            description: 'Disconnect and recreate the webrtc connection.',
+        });
+
         // Register a function to get all registered functions
         this.registerFunction({
             name: 'getRegisteredFunctions',

src/DeviceRenderer.js

@@ -351,6 +351,19 @@ module.exports = class DeviceRenderer {
         }
     }
 
+    /**
+     * Reconnect the instance. Disconnect the current instance, and then
+     * attempt to reconnect.
+     *
+     * When reconnecting, it will attempt to add the previous keyboard and mouse
+     * callbacks. (including keyboard mapping if applicable)
+     */
+    reconnect() {
+        this.disconnect();
+        this.onWebRTCReady();
+        this.store.reconnect();
+    }
+
     /**
      * Send event to the instance through the Websocket connection.
      *
@@ -572,7 +585,7 @@ module.exports = class DeviceRenderer {
         this.onConnectionStateChange = () => {
             log.debug('ConnectionState changed:', this.peerConnection.iceConnectionState);
             if (this.peerConnection.iceConnectionState === 'disconnected') {
-                this.onWebRTCReady();
+                this.reconnect();
             }
         };
         this.addListener(this.peerConnection, 'connectionstatechange', this.onConnectionStateChange);

src/store/index.js

@@ -103,7 +103,17 @@ const createStore = (instance, reducer) => {
         return unsubscribe;
     };
 
-    instance.store = {state: initialState, dispatch, subscribe, getters};
+    /**
+     * This function is used when the connection to the device is lost and restored.
+     * It will re-emit the current state to all listeners.
+     */
+    const reconnect = () => {
+        listeners.forEach(({cb}) => {
+            cb({...instance.store.state});
+        });
+    };
+
+    instance.store = {state: initialState, dispatch, subscribe, reconnect, getters};
 };
 
 const reducer = (state, action) => {