transforms

leo60228 · leo60228 · commit 78bd1d5c584f · 2021-04-11T21:05:42.000-04:00
diff --git a/addon/manifest.json b/addon/manifest.json
@@ -15,7 +15,10 @@
             "js": ["content_script.js"]
         }
     ],
-    "permissions": ["storage"],
+    "background": {
+        "scripts": ["background_script.js"]
+    },
+    "permissions": ["storage", "webRequest", "webRequestBlocking", "*://*.blaseball.com/*", "https://d35iw2jmbg6ut8.cloudfront.net/*"],
     "options_ui": {
         "browser_style": true,
         "page": "options_page.html"
diff --git a/lib/background.js b/lib/background.js
@@ -0,0 +1,30 @@
+import browser from 'webextension-polyfill';
+import transform from './transform.js';
+
+function jsHandler(details) {
+    let filter = browser.webRequest.filterResponseData(details.requestId);
+
+    let buf = '';
+    let decoder = new TextDecoder('utf-8');
+    const encoder = new TextEncoder();
+
+    filter.ondata = event => {
+        buf += decoder.decode(event.data, { stream: true });
+    };
+
+    filter.onstop = async () => {
+        const script = await transform(buf);
+        filter.write(encoder.encode(script));
+        filter.disconnect();
+    };
+
+    return {};
+}
+
+browser.webRequest.onBeforeRequest.addListener(
+    jsHandler,
+    {
+        urls: ["https://d35iw2jmbg6ut8.cloudfront.net/static/js/*.js"]
+    },
+    ["blocking"]
+);
diff --git a/lib/transform.js b/lib/transform.js
@@ -0,0 +1,16 @@
+import { transformAsync } from '@babel/core';
+import alwaysFk from './transforms/alwaysFk.js';
+
+const options = {
+    plugins: [alwaysFk],
+    generatorOpts: {
+        compact: true
+    }
+};
+
+export default async function transform(source) {
+    console.time('transforming');
+    const { code } = await transformAsync(source, options);
+    console.timeEnd('transforming');
+    return code;
+}
diff --git a/lib/transforms/alwaysFk.js b/lib/transforms/alwaysFk.js
@@ -0,0 +1,27 @@
+export default function({ types: t }) {
+    return {
+        visitor: {
+            ConditionalExpression(path) {
+                if (t.isLogicalExpression(path.node.test)) {
+                    if (path.node.test.operator !== '&&') return;
+                    if (!t.isBinaryExpression(path.node.test.left)) return;
+                    if (!t.isMemberExpression(path.node.test.left.right)) return;
+                    if (path.node.test.left.right.computed) return;
+                    if (path.node.test.left.right.property.name !== 'Forbidden_Knowledge_Access') return;
+
+                    if (t.isBinaryExpression(path.node.test.right) && t.isStringLiteral(path.node.test.right.left)) {
+                        path.node.test = path.node.test.right;
+                    } else {
+                        path.replaceWith(path.node.consequent);
+                    }
+                } else if (t.isBinaryExpression(path.node.test)) {
+                    if (!t.isMemberExpression(path.node.test.right)) return;
+                    if (path.node.test.right.computed) return;
+                    if (path.node.test.right.property.name !== 'Forbidden_Knowledge_Access') return;
+
+                    path.replaceWith(path.node.consequent);
+                }
+            }
+        }
+    }
+}
diff --git a/package.json b/package.json
@@ -30,8 +30,13 @@
         "webpack-cli": "^4.5.0"
     },
     "dependencies": {
+        "@babel/core": "^7.13.15",
         "@popperjs/core": "^2.8.6",
+        "assert": "^2.0.0",
+        "buffer": "^6.0.3",
         "camelcase": "^6.2.0",
+        "path-browserify": "^1.0.1",
+        "process": "^0.11.10",
         "webextension-polyfill": "^0.7.0"
     },
     "type": "module"
diff --git a/webpack.config.mjs b/webpack.config.mjs
@@ -1,6 +1,10 @@
 import { dirname, resolve } from 'path';
 import { fileURLToPath } from 'url';
 import WebExtWebpackPlugin from '@leo60228/web-ext-webpack-plugin';
+import { createRequire } from 'module';
+import webpack from 'webpack';
+
+const require = createRequire(import.meta.url);
 
 const cwd = dirname(fileURLToPath(import.meta.url));
 const addonDir = resolve(cwd, 'addon');
@@ -10,6 +14,7 @@ const config = env => ({
     devtool: env.production ? 'source-map' : 'cheap-source-map',
     entry: {
         content_script: './lib/content.js',
+        background_script: './lib/background.js',
         options_page: './lib/optionsPage.js',
     },
     output: {
@@ -26,6 +31,10 @@ const config = env => ({
             firefoxProfile: resolve(cwd, '.ff-profile'),
             profileCreateIfMissing: true,
             keepProfileChanges: true
+        }),
+        new webpack.ProvidePlugin({
+            process: 'process/browser',
+            Buffer: 'buffer/'
         })
     ],
     node: {
@@ -42,6 +51,13 @@ const config = env => ({
                 type: 'asset/source'
             }
         ]
+    },
+    resolve: {
+        fallback: {
+            path: require.resolve('path-browserify'),
+            assert: require.resolve('assert/'),
+            fs: false
+        }
     }
 });
 
diff --git a/yarn.lock b/yarn.lock
