fix: Socket url compose in entry file (#565)

Author: sunnywx

server/middleware/store.js

@@ -21,7 +21,7 @@ module.exports = async (ctx, next) => {
   if (!serverUrl.startsWith('http')) {
     serverUrl = `http://${serverUrl}`;
   }
-  if (!socketUrl.startsWith('ws://')) {
+  if (!socketUrl.startsWith('ws')) {
     socketUrl = `ws://${socketUrl}`;
   }
 

src/utils/sock-client.js

@@ -14,15 +14,15 @@ let reopenCount = 0;
 
 class SockClient extends EventEmitter {
   static composeEndpoint = (socketUrl, accessToken = '') => {
-    const re = /(\w+?:\/\/)?([^\\?]+)/;
+    const re = /wss?:\/\/([^\\?]+)/;
     const suffix = `?sid=${accessToken}`;
     const matchParts = `${socketUrl}`.match(re);
-    let wsPrefix = 'ws://';
-    if (typeof window === 'object' && window.location.protocol === 'https:') {
-      wsPrefix = 'wss://';
+
+    if (!matchParts) {
+      throw Error(`Invalid socket url: ${socketUrl}`);
     }
 
-    return `${wsPrefix}${matchParts[2]}${suffix}`;
+    return `${matchParts[0]}${suffix}`;
   };
 
   constructor(endpoint, options = {}) {