Include WebExtension, automatically updating config to match active sessions

Author: pimterry

overrides/webextension/build/background.js

@@ -0,0 +1,32 @@
+{
+    "name": "HTTP Toolkit",
+    "description": " A browser extension used in HTTP Toolkit",
+    "icons": {
+        "16": "icon-16.png",
+        "48": "icon-48.png",
+        "128": "icon-128.png"
+    },
+    "key": "jggocihfdfgogbofcmdlfbakdkgdkdoe",
+    "version": "1.1.0",
+    "manifest_version": 3,
+    "background": {
+        "service_worker": "build/background.js"
+    },
+    "permissions": [
+        "scripting",
+        "proxy"
+    ],
+    "host_permissions": [
+        "<all_urls>"
+    ],
+    "web_accessible_resources": [
+        {
+            "resources": [
+                "build/injected-script.js"
+            ],
+            "matches": [
+                "<all_urls>"
+            ]
+        }
+    ]
+}

overrides/webextension/build/content-script.js

@@ -32,6 +32,7 @@ import {
     startDockerInterceptionServices,
     stopDockerInterceptionServices
 } from './interceptors/docker/docker-interception-services';
+import { clearWebExtensionConfig, updateWebExtensionConfig } from './webextension';
 
 const APP_NAME = "HTTP Toolkit";
 
@@ -82,18 +83,29 @@ function manageBackgroundServices(
     }>,
     httpsConfig: { certPath: string, certContent: string }
 ) {
-    standalone.on('mock-session-started', async ({ http }) => {
-        startDockerInterceptionServices(http.getMockServer().port, httpsConfig, ruleParameters)
+    standalone.on('mock-session-started', async ({ http, webrtc }, sessionId) => {
+        const httpProxyPort = http.getMockServer().port;
+
+        startDockerInterceptionServices(httpProxyPort, httpsConfig, ruleParameters)
         .catch((error) => {
             console.log("Could not start Docker components:", error);
         });
+
+        updateWebExtensionConfig(sessionId, httpProxyPort, !!webrtc)
+        .catch((error) => {
+            console.log("Could not update WebRTC config:", error);
+        });
     });
 
     standalone.on('mock-session-stopping', ({ http }) => {
-        stopDockerInterceptionServices(http.getMockServer().port, ruleParameters)
+        const httpProxyPort = http.getMockServer().port;
+
+        stopDockerInterceptionServices(httpProxyPort, ruleParameters)
         .catch((error) => {
             console.log("Could not stop Docker components:", error);
         });
+
+        clearWebExtensionConfig(httpProxyPort);
     });
 }
 

overrides/webextension/build/injected-script.js

@@ -0,0 +1,46 @@
+import * as path from 'path';
+import { OVERRIDES_DIR } from './interceptors/terminal/terminal-env-overrides';
+
+import { deleteFile, writeFile } from "./util/fs";
+
+export const WEBEXTENSION_PATH = path.join(OVERRIDES_DIR, 'webextension');
+
+interface WebExtensionConfig { // Should match config in the WebExtension itself
+    mockRtc: {
+        peerId: string;
+        adminBaseUrl: string;
+    } | false;
+}
+
+const getConfigKey = (proxyPort: number) =>
+    `127_0_0_1.${proxyPort}`; // Filename-safe proxy address
+
+const getConfigPath = (proxyPort: number) =>
+    path.join(WEBEXTENSION_PATH, 'config', getConfigKey(proxyPort));
+
+export function clearWebExtensionConfig(httpProxyPort: number) {
+    return deleteFile(getConfigPath(httpProxyPort))
+        .catch(() => {}); // We ignore errors - nothing we can do, not very important.
+}
+
+export async function updateWebExtensionConfig(
+    sessionId: string,
+    httpProxyPort: number,
+    webRTCEnabled: boolean
+) {
+    if (webRTCEnabled) {
+        const adminBaseUrl = `http://internal.httptoolkit.localhost:45456/session/${sessionId}`;
+        await writeConfig(httpProxyPort, {
+            mockRtc: {
+                peerId: 'matching-peer',
+                adminBaseUrl
+            }
+        });
+    } else {
+        await writeConfig(httpProxyPort, { mockRtc: false });
+    }
+}
+
+async function writeConfig(proxyPort: number, config: WebExtensionConfig) {
+    return writeFile(getConfigPath(proxyPort), JSON.stringify(config));
+}