/* * 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 . */ // eslint-disable-next-line spaced-comment /// // eslint-disable-next-line spaced-comment /// import { readFileSync } from "fs"; import pup, { JSHandle } from "puppeteer-core"; for (const variable of ["DISCORD_TOKEN", "CHROMIUM_BIN"]) { if (!process.env[variable]) { console.error(`Missing environment variable ${variable}`); process.exit(1); } } const CANARY = process.env.USE_CANARY === "true"; const browser = await pup.launch({ headless: "new", executablePath: process.env.CHROMIUM_BIN }); const page = await browser.newPage(); await page.setUserAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36"); function maybeGetError(handle: JSHandle) { return (handle as JSHandle)?.getProperty("message") .then(m => m.jsonValue()); } const report = { badPatches: [] as { plugin: string; type: string; id: string; match: string; error?: string; }[], badStarts: [] as { plugin: string; error: string; }[], otherErrors: [] as string[], badWebpackFinds: [] as string[] }; const IGNORED_DISCORD_ERRORS = [ "KeybindStore: Looking for callback action", "Unable to process domain list delta: Client revision number is null", "Downloading the full bad domains file", /\[GatewaySocket\].{0,110}Cannot access '/, "search for 'name' in undefined" ] as Array; function toCodeBlock(s: string) { s = s.replace(/```/g, "`\u200B`\u200B`"); return "```" + s + " ```"; } async function printReport() { console.log(); console.log("# Vencord Report" + (CANARY ? " (Canary)" : "")); console.log(); console.log("## Bad Patches"); report.badPatches.forEach(p => { console.log(`- ${p.plugin} (${p.type})`); console.log(` - ID: \`${p.id}\``); console.log(` - Match: ${toCodeBlock(p.match)}`); if (p.error) console.log(` - Error: ${toCodeBlock(p.error)}`); }); console.log(); console.log("## Bad Webpack Finds"); report.badWebpackFinds.forEach(p => console.log("- " + p)); console.log(); console.log("## Bad Starts"); report.badStarts.forEach(p => { console.log(`- ${p.plugin}`); console.log(` - Error: ${toCodeBlock(p.error)}`); }); console.log(); report.otherErrors = report.otherErrors.filter(e => !IGNORED_DISCORD_ERRORS.some(regex => e.match(regex))); console.log("## Discord Errors"); report.otherErrors.forEach(e => { console.log(`- ${toCodeBlock(e)}`); }); console.log(); if (process.env.DISCORD_WEBHOOK) { await fetch(process.env.DISCORD_WEBHOOK, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ description: "Here's the latest Vencord Report!", username: "Vencord Reporter" + (CANARY ? " (Canary)" : ""), avatar_url: "https://cdn.discordapp.com/icons/1015060230222131221/6101cff21e241cebb60c4a01563d0c01.webp?size=512", embeds: [ { title: "Bad Patches", description: report.badPatches.map(p => { const lines = [ `**__${p.plugin} (${p.type}):__**`, `ID: \`${p.id}\``, `Match: ${toCodeBlock(p.match)}` ]; if (p.error) lines.push(`Error: ${toCodeBlock(p.error)}`); return lines.join("\n"); }).join("\n\n") || "None", color: report.badPatches.length ? 0xff0000 : 0x00ff00 }, { title: "Bad Webpack Finds", description: report.badWebpackFinds.map(toCodeBlock).join("\n") || "None", color: report.badWebpackFinds.length ? 0xff0000 : 0x00ff00 }, { title: "Bad Starts", description: report.badStarts.map(p => { const lines = [ `**__${p.plugin}:__**`, toCodeBlock(p.error) ]; return lines.join("\n"); } ).join("\n\n") || "None", color: report.badStarts.length ? 0xff0000 : 0x00ff00 }, { title: "Discord Errors", description: report.otherErrors.length ? toCodeBlock(report.otherErrors.join("\n")) : "None", color: report.otherErrors.length ? 0xff0000 : 0x00ff00 } ] }) }).then(res => { if (!res.ok) console.error(`Webhook failed with status ${res.status}`); else console.error("Posted to Discord Webhook successfully"); }); } } page.on("console", async e => { const level = e.type(); const rawArgs = e.args(); const firstArg = await rawArgs[0]?.jsonValue(); if (firstArg === "[PUPPETEER_TEST_DONE_SIGNAL]") { await browser.close(); await printReport(); process.exit(); } const isVencord = firstArg === "[Vencord]"; const isDebug = firstArg === "[PUP_DEBUG]"; const isWebpackFindFail = firstArg === "[PUP_WEBPACK_FIND_FAIL]"; if (isWebpackFindFail) { process.exitCode = 1; report.badWebpackFinds.push(await rawArgs[1].jsonValue() as string); } if (isVencord) { const args = await Promise.all(e.args().map(a => a.jsonValue())); const [, tag, message] = args as Array; const cause = await maybeGetError(e.args()[3]); switch (tag) { case "WebpackInterceptor:": process.exitCode = 1; const [, plugin, type, id, regex] = message.match(/Patch by (.+?) (had no effect|errored|found no module) \(Module id is (.+?)\): (.+)/)!; report.badPatches.push({ plugin, type, id, match: regex.replace(/\[A-Za-z_\$\]\[\\w\$\]\*/g, "\\i"), error: cause }); break; case "PluginManager:": const failedToStartMatch = message.match(/Failed to start (.+)/); if (!failedToStartMatch) break; process.exitCode = 1; const [, name] = failedToStartMatch; report.badStarts.push({ plugin: name, error: cause }); break; } } if (isDebug) { console.log(e.text()); } else if (level === "error") { const text = await Promise.all( e.args().map(async a => { try { return await maybeGetError(a) || await a.jsonValue(); } catch (e) { return a.toString(); } }) ).then(a => a.join(" ").trim()); if (text.length && !text.startsWith("Failed to load resource: the server responded with a status of") && !text.includes("found no module Filter:")) { console.error("[Unexpected Error]", text); report.otherErrors.push(text); } } }); page.on("error", e => console.error("[Error]", e)); page.on("pageerror", e => console.error("[Page Error]", e)); await page.setBypassCSP(true); function runTime(token: string) { console.log("[PUP_DEBUG]", "Starting test..."); try { // Spoof languages to not be suspicious Object.defineProperty(navigator, "languages", { get: function () { return ["en-US", "en"]; }, }); // Monkey patch Logger to not log with custom css // @ts-ignore Vencord.Util.Logger.prototype._log = function (level, levelColor, args) { if (level === "warn" || level === "error") console[level]("[Vencord]", this.name + ":", ...args); }; // Force enable all plugins and patches Vencord.Plugins.patches.length = 0; Object.values(Vencord.Plugins.plugins).forEach(p => { // Needs native server to run if (p.name === "WebRichPresence (arRPC)") return; Vencord.Settings.plugins[p.name].enabled = true; p.patches?.forEach(patch => { patch.plugin = p.name; delete patch.predicate; if (!Array.isArray(patch.replacement)) patch.replacement = [patch.replacement]; patch.replacement.forEach(r => { delete r.predicate; }); Vencord.Plugins.patches.push(patch); }); }); Vencord.Webpack.waitFor( "loginToken", m => { console.log("[PUP_DEBUG]", "Logging in with token..."); m.loginToken(token); } ); // Force load all chunks Vencord.Webpack.onceReady.then(() => setTimeout(async () => { console.log("[PUP_DEBUG]", "Webpack is ready!"); const { wreq } = Vencord.Webpack; console.log("[PUP_DEBUG]", "Loading all chunks..."); let chunks = null as Record | null; const sym = Symbol("Vencord.chunksExtract"); Object.defineProperty(Object.prototype, sym, { get() { chunks = this; }, set() { }, configurable: true, }); await (wreq as any).el(sym); delete Object.prototype[sym]; const validChunksEntryPoints = [] as string[]; const validChunks = [] as string[]; const invalidChunks = [] as string[]; if (!chunks) throw new Error("Failed to get chunks"); chunksLoop: for (const entryPoint in chunks) { const chunkIds = chunks[entryPoint]; for (const id of chunkIds) { if (!wreq.u(id)) continue; const isWasm = await fetch(wreq.p + wreq.u(id)) .then(r => r.text()) .then(t => t.includes(".module.wasm") || !t.includes("(this.webpackChunkdiscord_app=this.webpackChunkdiscord_app||[]).push")); if (isWasm) { invalidChunks.push(id); continue chunksLoop; } validChunks.push(id); } validChunksEntryPoints.push(entryPoint); } for (const entryPoint of validChunksEntryPoints) { try { // Loads all chunks required for an entry point await (wreq as any).el(entryPoint); } catch (err) { } } const allChunks = Function("return " + (wreq.u.toString().match(/(?<=\()\{.+?\}/s)?.[0] ?? "null"))() as Record | null; if (!allChunks) throw new Error("Failed to get all chunks"); const chunksLeft = Object.keys(allChunks).filter(id => { return !(validChunks.includes(id) || invalidChunks.includes(id)); }); for (const id of chunksLeft) { const isWasm = await fetch(wreq.p + wreq.u(id)) .then(r => r.text()) .then(t => t.includes(".module.wasm") || !t.includes("(this.webpackChunkdiscord_app=this.webpackChunkdiscord_app||[]).push")); // Loads a chunk if (!isWasm) await wreq.e(id as any); } // Make sure every chunk has finished loading await new Promise(r => setTimeout(r, 1000)); for (const entryPoint of validChunksEntryPoints) { try { if (wreq.m[entryPoint]) wreq(entryPoint as any); } catch (err) { console.error(err); } } console.log("[PUP_DEBUG]", "Finished loading all chunks!"); for (const patch of Vencord.Plugins.patches) { if (!patch.all) { new Vencord.Util.Logger("WebpackInterceptor").warn(`Patch by ${patch.plugin} found no module (Module id is -): ${patch.find}`); } } for (const [searchType, args] of Vencord.Webpack.lazyWebpackSearchHistory) { let method = searchType; if (searchType === "findComponent") method = "find"; if (searchType === "findExportedComponent") method = "findByProps"; if (searchType === "waitFor" || searchType === "waitForComponent" || searchType === "waitForStore") { if (typeof args[0] === "string") method = "findByProps"; else method = "find"; } try { let result: any; if (method === "proxyLazyWebpack" || method === "LazyComponentWebpack") { const [factory] = args; result = factory(); } else { // @ts-ignore result = Vencord.Webpack[method](...args); } if (result == null || ("$$get" in result && result.$$get() == null)) throw "a rock at ben shapiro"; } catch (e) { let logMessage = searchType; if (method === "find" || method === "proxyLazyWebpack" || method === "LazyComponentWebpack") logMessage += `(${args[0].toString().slice(0, 147)}...)`; else logMessage += `(${args.map(arg => `"${arg}"`).join(", ")})`; console.log("[PUP_WEBPACK_FIND_FAIL]", logMessage); } } setTimeout(() => console.log("[PUPPETEER_TEST_DONE_SIGNAL]"), 1000); }, 1000)); } catch (e) { console.log("[PUP_DEBUG]", "A fatal error occurred:", e); process.exit(1); } } await page.evaluateOnNewDocument(` ${readFileSync("./dist/browser.js", "utf-8")} ;(${runTime.toString()})(${JSON.stringify(process.env.DISCORD_TOKEN)}); `); await page.goto(CANARY ? "https://canary.discord.com/login" : "https://discord.com/login");