attach gitpod service to window

try to reconnect websocket

Author: AlexTugarev

components/dashboard/src/service/service.ts

@@ -4,7 +4,7 @@
  * See License-AGPL.txt in the project root for license information.
  */
 
-import { GitpodClient, GitpodServer, GitpodServerPath, GitpodService, GitpodServiceImpl } from '@gitpod/gitpod-protocol';
+import { GitpodClient, GitpodServer, GitpodServerPath, GitpodServiceImpl } from '@gitpod/gitpod-protocol';
 import { WebSocketConnectionProvider } from '@gitpod/gitpod-protocol/lib/messaging/browser/connection';
 import { createWindowMessageConnection } from '@gitpod/gitpod-protocol/lib/messaging/browser/window-connection';
 import { JsonRpcProxy, JsonRpcProxyFactory } from '@gitpod/gitpod-protocol/lib/messaging/proxy-factory';
@@ -16,6 +16,9 @@ export const gitpodHostUrl = new GitpodHostUrl(window.location.toString());
 
 function createGitpodService<C extends GitpodClient, S extends GitpodServer>() {
     let proxy: JsonRpcProxy<S>;
+    let reconnect = () => {
+        console.log("WebSocket reconnect not possible.");
+    }
     if (window.top !== window.self) {
         const connection = createWindowMessageConnection('gitpodServer', window.parent, '*');
         const factory = new JsonRpcProxyFactory<S>();
@@ -28,6 +31,12 @@ function createGitpodService<C extends GitpodClient, S extends GitpodServer>() {
             .withApi();
 
         const connectionProvider = new WebSocketConnectionProvider();
+        let _websocket: any;
+        const _createWebSocket = connectionProvider.createWebSocket;
+        connectionProvider.createWebSocket = (url) => {
+            return (_websocket = _createWebSocket(url));
+        }
+
 
         let numberOfErrors = 0;
         proxy = connectionProvider.createProxy<S>(host.toString(), undefined, {
@@ -38,18 +47,28 @@ function createGitpodService<C extends GitpodClient, S extends GitpodServer>() {
                 }
             }
         });
+
+        if (_websocket && "reconnect" in _websocket) {
+            reconnect = () => { (_websocket as any).reconnect() };
+        }
     }
     const service = new GitpodServiceImpl<C, S>(proxy);
+    (service as any).reconnect = reconnect;
     return service;
 }
 
+declare global {
+    interface Window { gitpodService?: ReturnType<typeof createGitpodService> & { reconnect?: () => void }; }
+}
 
-let gitpodService: GitpodService;
+// reuse existing service object if present
+let gitpodService = window.gitpodService || (window.gitpodService = createGitpodService());
 
+// allow to reconnect
 const reconnect = () => {
-    gitpodService = createGitpodService();
+    if (window.gitpodService?.reconnect) {
+        window.gitpodService?.reconnect();
+    }
 }
 
-reconnect();
-
 export { gitpodService, reconnect };