refactor: streamline message handling in auth and content scripts

Author: medyo

public/auth.js

@@ -1,6 +1,8 @@
+const acceptedTypes = ['TOKEN_RECEIVED', 'ERROR_RECEIVED']
 window.addEventListener('message', (event) => {
-  if (event.data.type === 'TOKEN_RECEIVED' || event.data.type === 'ERROR_RECEIVED') {
-    // Forward to content script
+  console.log('Received message from content script:', event.data)
+  if (acceptedTypes.includes(event.data.type)) {
+    // Forward to content scripte
     window.postMessage(event.data, '*')
   }
 })

public/content.js

@@ -1,17 +1,12 @@
 const script = document.createElement('script')
 script.src = chrome.runtime.getURL('auth.js')
 document.documentElement.appendChild(script)
+const acceptedTypes = ['TOKEN_RECEIVED', 'ERROR_RECEIVED']
 
 // Listen for messages from the injected script
 window.addEventListener('message', (event) => {
-  if (
-    event.source !== window ||
-    !event.data ||
-    event.data.type !== 'TOKEN_RECEIVED' ||
-    event.data.type !== 'ERROR_RECEIVED'
-  ) {
+  if (event.source !== window || !event.data || !acceptedTypes.includes(event.data.type)) {
     return
   }
-
   chrome.runtime.sendMessage({ ...event.data })
 })