Vencord/src/webpack/patchWebpack.ts
Vendicated 3a9f692644
Fix Desktop thinking it's web
I forgor to test host after adding web support
2022-10-04 12:57:39 +02:00

149 lines
6.5 KiB
TypeScript

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
});
}