init

Author: maltoze

.gitignore

@@ -0,0 +1 @@
+node_modules

.prettierrc

@@ -0,0 +1,5 @@
+{
+  "tabWidth": 2,
+  "semi": false,
+  "singleQuote": true
+}

package.json

@@ -0,0 +1,5 @@
+{
+  "devDependencies": {
+    "@types/chrome": "^0.0.241"
+  }
+}

pnpm-lock.yaml

@@ -0,0 +1,14 @@
+let es
+
+chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
+  switch (request.type) {
+    case 'close':
+      es.close()
+      break
+    case 'create':
+      es = new EventSource(request.url)
+      break
+    default:
+      break
+  }
+})

src/content-script.js

@@ -0,0 +1,17 @@
+{
+  "manifest_version": 3,
+  "name": "SSE Viewer",
+  "description": "",
+  "version": "0.1.0",
+  "permissions": ["debugger", "tabs"],
+  "background": {
+    "service_worker": "service-worker.js"
+  },
+  "action": {},
+  "content_scripts": [
+    {
+      "matches": ["<all_urls>"],
+      "js": ["content-script.js"]
+    }
+  ]
+}

src/manifest.json

@@ -0,0 +1,77 @@
+function sendCommand(target, method, params) {
+  return new Promise((resolve, reject) => {
+    chrome.debugger.sendCommand(target, method, params, function (result) {
+      if (chrome.runtime.lastError) {
+        reject(chrome.runtime.lastError)
+      } else {
+        resolve(result)
+      }
+    })
+  })
+}
+
+chrome.action.onClicked.addListener(function (tab) {
+  if (tab.url.startsWith('http')) {
+    chrome.debugger.attach({ tabId: tab.id }, '1.2', function () {
+      sendCommand({ tabId: tab.id }, 'Fetch.enable', {
+        patterns: [{ requestStage: 'Request' }, { requestStage: 'Response' }],
+      })
+    })
+  } else {
+    console.log('Debugger can only be attached to HTTP/HTTPS pages.')
+  }
+})
+
+const esPathPrefix = '/__maltoze-sse-viewer'
+let esUrl
+let sseRequestId
+const bodyQueue = []
+
+chrome.debugger.onEvent.addListener(async function (source, method, params) {
+  const [tab] = await chrome.tabs.query({
+    active: true,
+    lastFocusedWindow: true,
+  })
+
+  if (
+    params.requestId === sseRequestId &&
+    (method === 'Network.loadingFinished' || method === 'Network.loadingFailed')
+  ) {
+    chrome.tabs.sendMessage(tab.id, { type: 'close' })
+  }
+
+  if (method === 'Fetch.requestPaused') {
+    const reqUrl = new URL(params.request.url)
+    if (reqUrl.pathname === esUrl) {
+      await sendCommand(source, 'Fetch.fulfillRequest', {
+        requestId: params.requestId,
+        responseCode: 200,
+        responseHeaders: [
+          { name: 'Content-Type', value: 'text/event-stream' },
+          { name: 'Cache-Control', value: 'no-cache' },
+        ],
+        body: bodyQueue.shift(),
+      })
+      chrome.tabs.sendMessage(tab.id, { type: 'close' })
+    } else {
+      if (params.request.headers.accept === 'text/event-stream') {
+        sseRequestId = params.requestId
+        esUrl = `${esPathPrefix}${reqUrl.pathname}`
+        if (params.responseHeaders) {
+          const resp = await sendCommand(source, 'Fetch.getResponseBody', {
+            requestId: params.requestId,
+          })
+          bodyQueue.push(resp.body)
+          chrome.tabs.sendMessage(tab.id, { type: 'create', url: esUrl })
+        }
+        sendCommand(source, 'Fetch.continueRequest', {
+          requestId: params.requestId,
+        })
+      } else {
+        sendCommand(source, 'Fetch.continueRequest', {
+          requestId: params.requestId,
+        })
+      }
+    }
+  }
+})