basic messages (hello, makeRequest, prepareStream)

src/background/messages/prepareStream.ts

@@ -0,0 +1,76 @@
+import type { PlasmoMessaging } from '@plasmohq/messaging';
+
+import { validateDomainWhiteList } from '~utils/storage';
+
+interface RequestBody {
+  ruleId: number;
+  hostDomain: string;
+  targetDomains: [string, ...string[]];
+  requestHeaders?: Record<string, string>;
+  responseHeaders?: Record<string, string>;
+}
+
+const mapHeadersToDeclarativeNetRequestHeaders = (
+  headers: Record<string, string>,
+): chrome.declarativeNetRequest.ModifyHeaderInfo[] => {
+  return Object.entries(headers).map(([name, value]) => ({
+    header: name,
+    operation: chrome.declarativeNetRequest.HeaderOperation.SET,
+    value,
+  }));
+};
+
+const handler: PlasmoMessaging.MessageHandler<RequestBody> = async (req, res) => {
+  try {
+    await validateDomainWhiteList(req.body.hostDomain);
+
+    await chrome.declarativeNetRequest.updateDynamicRules({
+      removeRuleIds: [req.body.ruleId],
+      addRules: [
+        {
+          id: req.body.ruleId,
+          condition: {
+            requestDomains: req.body.targetDomains,
+          },
+          action: {
+            type: chrome.declarativeNetRequest.RuleActionType.MODIFY_HEADERS,
+            ...(req.body.requestHeaders && {
+              requestHeaders: mapHeadersToDeclarativeNetRequestHeaders(req.body.requestHeaders),
+            }),
+            responseHeaders: [
+              {
+                header: 'Access-Control-Allow-Origin',
+                operation: chrome.declarativeNetRequest.HeaderOperation.SET,
+                value: '*',
+              },
+              {
+                header: 'Access-Control-Allow-Methods',
+                operation: chrome.declarativeNetRequest.HeaderOperation.SET,
+                value: 'GET, POST, PUT, DELETE, PATCH, OPTIONS',
+              },
+              {
+                header: 'Access-Control-Allow-Headers',
+                operation: chrome.declarativeNetRequest.HeaderOperation.SET,
+                value: '*',
+              },
+              ...mapHeadersToDeclarativeNetRequestHeaders(req.body.responseHeaders ?? {}),
+            ],
+          },
+        },
+      ],
+    });
+
+    if (chrome.runtime.lastError?.message) throw new Error(chrome.runtime.lastError.message);
+
+    res.send({
+      success: true,
+    });
+  } catch (err) {
+    res.send({
+      success: false,
+      error: err.message,
+    });
+  }
+};
+
+export default handler;