Merge pull request coderoad#454 from coderoad/fix/handle-origin

ShMcK · web-flow · commit 91cc73855e04 · 2020-08-19T19:47:09.000-07:00
ensure events come from cr webview
diff --git a/web-app/src/services/state/useStateMachine.tsx b/web-app/src/services/state/useStateMachine.tsx
@@ -33,12 +33,12 @@ const useStateMachine = (): Output => {
     const listener = 'message'
     // propograte channel event to state machine
     const handler = (event: any) => {
-      // NOTE: must call event.data, cannot destructure. VSCode acts odd
-      const action = event.data
-      // ignore browser events from other extensions
-      if (action.source) {
+      // ensure events are coming from coderoad webview
+      if (!event.origin.match(/^vscode-webview/)) {
         return
       }
+      // NOTE: must call event.data, cannot destructure. VSCode acts odd
+      const action = event.data
       sendWithLog(action)
     }
     window.addEventListener(listener, handler)
