fix: support file: and chrome-extension: protocols in client

webpack · Jan 3, 2021 · f20a98d · f20a98d
1 parent ca9e45d
commit f20a98d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/client-src/clients/SockJSClient.js b/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);

diff --git a/client-src/clients/WebsocketClient.js b/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);