Better authors field (#18)

1 files changed, 148 insertions, 0 deletions

diff --git a/src/webpack/patchWebpack.ts b/src/webpack/patchWebpack.ts
new file mode 100644
index 0000000..469b930
--- /dev/null
+++ b/src/webpack/patchWebpack.ts
@@ -0,0 +1,148 @@
+import { WEBPACK_CHUNK } from '../utils/constants';
+import Logger from "../utils/logger";
+import { _initWebpack } from ".";
+
+let webpackChunk: any[];
+
+const logger = new Logger("WebpackInterceptor", "#8caaee");
+
+Object.defineProperty(window, WEBPACK_CHUNK, {
+    get: () => webpackChunk,
+    set: (v) => {
+        if (v?.push !== Array.prototype.push) {
+            logger.info(`Patching ${WEBPACK_CHUNK}.push`);
+            _initWebpack(v);
+            patchPush();
+            // @ts-ignore
+            delete window[WEBPACK_CHUNK];
+            window[WEBPACK_CHUNK] = v;
+        }
+        webpackChunk = v;
+    },
+    configurable: true
+});
+
+function patchPush() {
+    function handlePush(chunk) {
+        try {
+            const modules = chunk[1];
+            const { subscriptions, listeners } = Vencord.Webpack;
+            const { patches } = Vencord.Plugins;
+
+            for (const id in modules) {
+                let mod = modules[id];
+                // Discords Webpack chunks for some ungodly reason contain random
+                // newlines. Cyn recommended this workaround and it seems to work fine,
+                // however this could potentially break code, so if anything goes weird,
+                // this is probably why.
+                // Additionally, `[actual newline]` is one less char than "\n", so if Discord
+                // ever targets newer browsers, the minifier could potentially use this trick and
+                // cause issues.
+                let code = mod.toString().replaceAll("\n", "");
+                const originalMod = mod;
+                const patchedBy = new Set();
+
+                modules[id] = function (module, exports, require) {
+                    try {
+                        mod(module, exports, require);
+                    } catch (err) {
+                        // Just rethrow discord errors
+                        if (mod === originalMod) throw err;
+
+                        logger.error("Error in patched chunk", err);
+                        return originalMod(module, exports, require);
+                    }
+
+                    // There are (at the time of writing) 11 modules exporting the window
+                    // Make these non enumerable to improve webpack search performance
+                    if (module.exports === window) {
+                        Object.defineProperty(require.c, id, {
+                            value: require.c[id],
+                            enumerable: false,
+                            configurable: true,
+                            writable: true
+                        });
+                        return;
+                    }
+
+                    for (const callback of listeners) {
+                        try {
+                            callback(exports);
+                        } catch (err) {
+                            logger.error("Error in webpack listener", err);
+                        }
+                    }
+                    for (const [filter, callback] of subscriptions) {
+                        try {
+                            if (filter(exports)) {
+                                subscriptions.delete(filter);
+                                callback(exports);
+                            } else if (typeof exports === "object") {
+                                if (exports.default && filter(exports.default)) {
+                                    subscriptions.delete(filter);
+                                    callback(exports.default);
+                                }
+
+                                for (const nested in exports) if (nested.length < 3) {
+                                    if (exports[nested] && filter(exports[nested])) {
+                                        subscriptions.delete(filter);
+                                        callback(exports[nested]);
+                                    }
+                                }
+                            }
+                        } catch (err) {
+                            logger.error("Error while firing callback for webpack chunk", err);
+                        }
+                    }
+                };
+                modules[id].toString = () => mod.toString();
+                modules[id].original = originalMod;
+
+                for (let i = 0; i < patches.length; i++) {
+                    const patch = patches[i];
+                    if (code.includes(patch.find)) {
+                        patchedBy.add(patch.plugin);
+                        // @ts-ignore we change all patch.replacement to array in plugins/index
+                        for (const replacement of patch.replacement) {
+                            const lastMod = mod;
+                            const lastCode = code;
+                            try {
+                                const newCode = code.replace(replacement.match, replacement.replace);
+                                if (newCode === code) {
+                                    logger.warn(`Patch by ${patch.plugin} had no effect: ${replacement.match}`);
+                                    logger.debug("Function Source:\n", code);
+                                } else {
+                                    code = newCode;
+                                    mod = (0, eval)(`// Webpack Module ${id} - Patched by ${[...patchedBy].join(", ")}\n${newCode}\n//# sourceURL=WebpackModule${id}`);
+                                }
+                            } catch (err) {
+                                // TODO - More meaningful errors. This probably means moving away from string.replace
+                                // in favour of manual matching. Then cut out the context and log some sort of
+                                // diff
+                                logger.error("Failed to apply patch of", patch.plugin, err);
+                                logger.debug("Original Source\n", lastCode);
+                                logger.debug("Patched Source\n", code);
+                                code = lastCode;
+                                mod = lastMod;
+                                patchedBy.delete(patch.plugin);
+                            }
+                        }
+                        patches.splice(i--, 1);
+                    }
+                }
+            }
+        } catch (err) {
+            logger.error("oopsie", err);
+        }
+
+        return handlePush.original.call(window[WEBPACK_CHUNK], chunk);
+    }
+
+    handlePush.original = window[WEBPACK_CHUNK].push;
+    Object.defineProperty(window[WEBPACK_CHUNK], "push", {
+        get: () => handlePush,
+        set: (v) => (handlePush.original = v),
+        configurable: true
+    });
+}
+