fix: support file: and chrome-extension: protocols in client (#2954)

Author: vespakoen

client-src/clients/SockJSClient.js

@@ -7,7 +7,8 @@ const BaseClient = require('./BaseClient');
 module.exports = class SockJSClient extends BaseClient {
   constructor(url) {
     super();
-    this.sock = new SockJS(url);
+    const sockUrl = url.replace(/^(?:chrome-extension|file)/i, 'http');
+    this.sock = new SockJS(sockUrl);
 
     this.sock.onerror = (err) => {
       log.error(err);

client-src/clients/WebsocketClient.js

@@ -8,7 +8,8 @@ const BaseClient = require('./BaseClient');
 module.exports = class WebsocketClient extends BaseClient {
   constructor(url) {
     super();
-    this.client = new WebSocket(url.replace(/^http/, 'ws'));
+    const wsUrl = url.replace(/^(?:http|chrome-extension|file)/i, 'ws');
+    this.client = new WebSocket(wsUrl);
 
     this.client.onerror = (err) => {
       log.error(err);

test/client/clients/SockJSClient.test.js

@@ -72,6 +72,16 @@ describe('SockJSClient', () => {
         done();
       }, 3000);
     });
+    it('should change the protocol from chrome-extension to http', (done) => {
+      const client = new SockJSClient('chrome-extension://localhost');
+      expect(client.sock.url).toEqual('http://localhost');
+      done();
+    });
+    it('should change the protocol from file to http', (done) => {
+      const client = new SockJSClient('file://localhost');
+      expect(client.sock.url).toEqual('http://localhost');
+      done();
+    });
   });
 
   afterAll((done) => {

test/client/clients/WebsocketClient.test.js

@@ -65,6 +65,16 @@ describe('WebsocketClient', () => {
         done();
       }, 3000);
     });
+    it('should change the protocol from chrome-extension to http', (done) => {
+      const client = new WebsocketClient('chrome-extension://localhost/');
+      expect(client.client.url).toEqual('ws://localhost/');
+      done();
+    });
+    it('should change the protocol from file to http', (done) => {
+      const client = new WebsocketClient('file://localhost/');
+      expect(client.client.url).toEqual('ws://localhost/');
+      done();
+    });
   });
 
   afterAll((done) => {