Vencord/src/webpack/patchWebpack.ts

212 lines
9.8 KiB
TypeScript
Raw Normal View History

2022-10-21 23:17:06 +00:00
/*
* Vencord, a modification for Discord's desktop app
* Copyright (c) 2022 Vendicated and contributors
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
import { WEBPACK_CHUNK } from "../utils/constants";
2022-10-01 00:27:28 +00:00
import Logger from "../utils/logger";
import { _initWebpack } from ".";
2022-08-29 00:25:27 +00:00
let webpackChunk: any[];
2022-08-29 18:27:47 +00:00
const logger = new Logger("WebpackInterceptor", "#8caaee");
2022-08-29 00:25:27 +00:00
Object.defineProperty(window, WEBPACK_CHUNK, {
get: () => webpackChunk,
set: v => {
2022-09-15 16:17:52 +00:00
if (v?.push !== Array.prototype.push) {
2022-08-29 18:27:47 +00:00
logger.info(`Patching ${WEBPACK_CHUNK}.push`);
_initWebpack(v);
2022-08-29 00:25:27 +00:00
patchPush();
// @ts-ignore
delete window[WEBPACK_CHUNK];
window[WEBPACK_CHUNK] = v;
}
webpackChunk = v;
},
configurable: true
});
function patchPush() {
function handlePush(chunk: any) {
2022-08-29 00:25:27 +00:00
try {
const modules = chunk[1];
2022-08-29 18:27:47 +00:00
const { subscriptions, listeners } = Vencord.Webpack;
const { patches } = Vencord.Plugins;
2022-08-29 00:25:27 +00:00
for (const id in modules) {
let mod = modules[id];
2022-09-15 16:17:52 +00:00
// 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: string = mod.toString().replaceAll("\n", "");
// a very small minority of modules use function() instead of arrow functions,
// but, unnamed toplevel functions aren't valid. However 0, function() makes it a statement
if (code.startsWith("function(")) {
code = "0," + code;
}
2022-08-29 00:25:27 +00:00
const originalMod = mod;
const patchedBy = new Set();
const factory = modules[id] = function (module, exports, require) {
2022-08-29 00:25:27 +00:00
try {
mod(module, exports, require);
} catch (err) {
// Just rethrow discord errors
if (mod === originalMod) throw err;
2022-08-29 18:27:47 +00:00
logger.error("Error in patched chunk", err);
2022-10-08 18:36:57 +00:00
return void originalMod(module, exports, require);
2022-08-29 00:25:27 +00:00
}
// 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;
}
2022-08-29 18:27:47 +00:00
for (const callback of listeners) {
try {
callback(exports);
} catch (err) {
logger.error("Error in webpack listener", err);
}
}
2022-08-29 00:25:27 +00:00
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)) {
2022-09-27 12:34:57 +00:00
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]);
}
2022-09-27 12:34:57 +00:00
}
2022-08-29 00:25:27 +00:00
}
} catch (err) {
2022-08-29 18:27:47 +00:00
logger.error("Error while firing callback for webpack chunk", err);
2022-08-29 00:25:27 +00:00
}
}
} as any as { toString: () => string, original: any, (...args: any[]): void; };
// for some reason throws some error on which calling .toString() leads to infinite recursion
// when you force load all chunks???
try {
factory.toString = () => mod.toString();
factory.original = originalMod;
} catch { }
2022-08-29 00:25:27 +00:00
for (let i = 0; i < patches.length; i++) {
const patch = patches[i];
if (patch.predicate && !patch.predicate()) continue;
2022-08-29 00:25:27 +00:00
if (code.includes(patch.find)) {
patchedBy.add(patch.plugin);
2022-08-29 18:27:47 +00:00
// @ts-ignore we change all patch.replacement to array in plugins/index
for (const replacement of patch.replacement) {
const lastMod = mod;
const lastCode = code;
2022-08-29 18:27:47 +00:00
try {
const newCode = code.replace(replacement.match, replacement.replace);
2022-11-11 15:43:40 +00:00
if (newCode === code && !replacement.noWarn) {
logger.warn(`Patch by ${patch.plugin} had no effect (Module id is ${id}): ${replacement.match}`);
if (IS_DEV) {
logger.debug("Function Source:\n", code);
}
2022-08-31 02:07:16 +00:00
} else {
code = newCode;
2022-09-02 03:11:45 +00:00
mod = (0, eval)(`// Webpack Module ${id} - Patched by ${[...patchedBy].join(", ")}\n${newCode}\n//# sourceURL=WebpackModule${id}`);
2022-08-31 02:07:16 +00:00
}
2022-08-29 18:27:47 +00:00
} catch (err) {
logger.error(`Patch by ${patch.plugin} errored (Module id is ${id}): ${replacement.match}\n`, err);
if (IS_DEV) {
const changeSize = code.length - lastCode.length;
const match = lastCode.match(replacement.match)!;
// Use 200 surrounding characters of context
const start = Math.max(0, match.index! - 200);
const end = Math.min(lastCode.length, match.index! + match[0].length + 200);
// (changeSize may be negative)
const endPatched = end + changeSize;
const context = lastCode.slice(start, end);
const patchedContext = code.slice(start, endPatched);
// inline require to avoid including it in !IS_DEV builds
const diff = (require("diff") as typeof import("diff")).diffWordsWithSpace(context, patchedContext);
let fmt = "%c %s ";
const elements = [] as string[];
for (const d of diff) {
const color = d.removed
? "red"
: d.added
? "lime"
: "grey";
fmt += "%c%s";
elements.push("color:" + color, d.value);
}
logger.errorCustomFmt(...Logger.makeTitle("white", "Before"), context);
logger.errorCustomFmt(...Logger.makeTitle("white", "After"), context);
const [titleFmt, ...titleElements] = Logger.makeTitle("white", "Diff");
logger.errorCustomFmt(titleFmt + fmt, ...titleElements, ...elements);
}
2022-08-29 18:27:47 +00:00
code = lastCode;
mod = lastMod;
patchedBy.delete(patch.plugin);
}
2022-08-29 00:25:27 +00:00
}
if (!patch.all) patches.splice(i--, 1);
2022-08-29 00:25:27 +00:00
}
}
}
} catch (err) {
logger.error("Error in handlePush", err);
2022-08-29 00:25:27 +00:00
}
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),
2022-08-29 00:25:27 +00:00
configurable: true
});
}