wow, chrome-extension MUCH improved - websockets

chrome-extension/inject.js

@@ -0,0 +1,92 @@
+// Jackbox Chat Tracker - Injected Script v3.2.0
+// This script intercepts Kosmi's WebSocket connection to capture chat messages
+// NOTE: This runs in the PAGE CONTEXT, not extension context!
+
+// Store original WebSocket constructor
+const OriginalWebSocket = window.WebSocket;
+
+// Hook WebSocket constructor to capture Kosmi's WebSocket
+window.WebSocket = function(url, protocols) {
+  const socket = new OriginalWebSocket(url, protocols);
+  
+  // Check if this is Kosmi's GraphQL WebSocket
+  if (url.includes('engine.kosmi.io') || url.includes('gql-ws')) {
+    console.log('[Jackbox Chat Tracker] WebSocket hook active');
+    
+    // Notify content script that we found the socket
+    window.postMessage({
+      type: 'JACKBOX_TRACKER_SOCKET_FOUND',
+      url: url
+    }, '*');
+    
+    // Method 1: Hook addEventListener (for event listeners added later)
+    const originalAddEventListener = socket.addEventListener.bind(socket);
+    socket.addEventListener = function(type, listener, options) {
+      if (type === 'message') {
+        const wrappedListener = function(event) {
+          // Forward to content script
+          try {
+            const data = JSON.parse(event.data);
+            window.postMessage({
+              type: 'JACKBOX_TRACKER_WS_MESSAGE',
+              data: data
+            }, '*');
+          } catch (e) {
+            // Not JSON, skip
+          }
+          
+          // Then forward to original listener
+          return listener.call(this, event);
+        };
+        return originalAddEventListener(type, wrappedListener, options);
+      }
+      return originalAddEventListener(type, listener, options);
+    };
+    
+    // Method 2: Intercept the onmessage property setter (for direct assignment)
+    let realOnMessage = null;
+    const descriptor = Object.getOwnPropertyDescriptor(WebSocket.prototype, 'onmessage');
+    
+    Object.defineProperty(socket, 'onmessage', {
+      get: function() {
+        return realOnMessage;
+      },
+      set: function(handler) {
+        // Create wrapped handler
+        realOnMessage = function(event) {
+          // Forward to content script
+          try {
+            const data = JSON.parse(event.data);
+            window.postMessage({
+              type: 'JACKBOX_TRACKER_WS_MESSAGE',
+              data: data
+            }, '*');
+          } catch (e) {
+            // Not JSON, skip
+          }
+          
+          // ALWAYS call original handler
+          if (handler) {
+            handler.call(socket, event);
+          }
+        };
+        
+        // Actually set it on the underlying WebSocket using the original descriptor
+        if (descriptor && descriptor.set) {
+          descriptor.set.call(socket, realOnMessage);
+        }
+      },
+      configurable: true
+    });
+  }
+  
+  return socket;
+};
+
+// Preserve the original constructor properties
+window.WebSocket.prototype = OriginalWebSocket.prototype;
+window.WebSocket.CONNECTING = OriginalWebSocket.CONNECTING;
+window.WebSocket.OPEN = OriginalWebSocket.OPEN;
+window.WebSocket.CLOSING = OriginalWebSocket.CLOSING;
+window.WebSocket.CLOSED = OriginalWebSocket.CLOSED;
+