add support for local environments

twyla-ai · Feb 21, 2022 · 6053664 · 6053664
1 parent 2afe0f5
commit 6053664
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 7 deletions.
diff --git a/src/tests/index.test.js b/src/tests/index.test.js
@@ -2,6 +2,33 @@ import { notificationsChannelURLFromHookURL } from '../utils';
 
 describe('notificationsChannelURLFromHookURL tests', () => {
   test('returns notificationsChannelURL, projectName & workspaceName', () => {
+    // localhost
+    const localhosts = ["127.0.0.1", "127.0.0.1:9001", "localhost", "localhost:9001"];
+
+    for (const hostname of localhosts) {
+      expect(
+        notificationsChannelURLFromHookURL(
+          `http://${hostname}/widget-hook/massive-dynamic/z10`
+        )
+      ).toEqual({
+        notificationsChannelURL:
+          `ws://${hostname}/widget-notifications/massive-dynamic/z10`,
+        projectName: 'z10',
+        workspaceName: 'massive-dynamic',
+      });
+
+      expect(
+        notificationsChannelURLFromHookURL(
+          `https://${hostname}/widget-hook/massive-dynamic/z10`
+        )
+      ).toEqual({
+        notificationsChannelURL:
+          `wss://${hostname}/widget-notifications/massive-dynamic/z10`,
+        projectName: 'z10',
+        workspaceName: 'massive-dynamic',
+      });
+    }
+
     // rawhide
     expect(
       notificationsChannelURLFromHookURL(

diff --git a/src/utils.js b/src/utils.js
@@ -15,20 +15,28 @@ export const getDefaultPayload = () => ({
  */
 export const notificationsChannelURLFromHookURL = hookURL => {
   const hookURLTokens = hookURL.split('/');
+  const hookURLParsed = new URL(hookURL);
   let environment;
   let notificationsChannelURL;
 
-  const environmentSearch = /api\.(.*)\.canvas/.exec(hookURL);
+  if (['localhost', '127.0.0.1'].includes(hookURLParsed.hostname)) {
+    notificationsChannelURL = `${hookURLParsed.protocol === 'http:' ? 'ws' : 'wss'}://${hookURLParsed.hostname}`;
+    if (hookURLParsed.port) {
+      notificationsChannelURL += `:${hookURLParsed.port}`
+    }
+  } else {
+    const environmentSearch = /api\.(.*)\.canvas/.exec(hookURL);
 
-  if (!environmentSearch) environment = 'production';
-  else [, environment] = environmentSearch;
+    if (!environmentSearch) environment = 'production';
+    else [, environment] = environmentSearch;
 
-  notificationsChannelURL = `wss://notification.${
-    !environmentSearch ? '' : `${environment}.`
-  }canvas.twyla.ai/widget-notifications/${hookURLTokens[4]}/${hookURLTokens[5]}`;
+    notificationsChannelURL = `wss://notification.${
+      !environmentSearch ? '' : `${environment}.`
+    }canvas.twyla.ai`;
+  }
 
   return {
-    notificationsChannelURL,
+    notificationsChannelURL: `${notificationsChannelURL}/widget-notifications/${hookURLTokens[4]}/${hookURLTokens[5]}`,
     workspaceName: hookURLTokens[4],
     projectName: hookURLTokens[5],
   };