Recently Written · git

test/helpers/installed.js (2413 bytes)

 "use strict";
 // Loads code from the installed BetterDiscord and BDFDB, so the tests check the versions that Discord runs.
 const fs = require("node:fs");
 const path = require("node:path");
 
 const BD_FOLDER = path.join(process.env.APPDATA || "", "BetterDiscord");
 const BD_ASAR = path.join(BD_FOLDER, "data", "betterdiscord.asar");
 const BDFDB_FILE = path.join(BD_FOLDER, "plugins", "0BDFDB.plugin.js");
 
 /** Reads one top-level file from an asar archive. */
 function readAsarFile(archive, name) {
     const buffer = fs.readFileSync(archive);
     const headerPickleSize = buffer.readUInt32LE(4);
     const jsonLength = buffer.readUInt32LE(12);
     const header = JSON.parse(buffer.toString("utf8", 16, 16 + jsonLength));
     const entry = header.files[name];
     const start = 8 + headerPickleSize + Number(entry.offset);
     return buffer.toString("utf8", start, start + entry.size);
 }
 
 /** Returns the index of the brace that closes the brace at openIndex. */
 function closingBrace(text, openIndex) {
     let depth = 0;
     for (let i = openIndex; i < text.length; i++) {
         if (text[i] === "{") depth++;
         else if (text[i] === "}" && --depth === 0) return i;
     }
     throw new Error("no closing brace");
 }
 
 /** Returns the internal Patcher class of the installed BetterDiscord. */
 function loadBdPatcher(logger) {
     const source = readAsarFile(BD_ASAR, "betterdiscord.js");
     const marker = source.indexOf("static getPatchesByCaller(");
     if (marker < 0) throw new Error("BetterDiscord Patcher not found");
     const classStart = source.lastIndexOf("=class{", marker) + 1;
     const body = source.slice(classStart, closingBrace(source, classStart + "class".length) + 1);
     // The minified class uses g (logger), w (module cache), and ae (module search) from its bundle.
     return new Function("g", "w", "ae", `return ${body};`)(logger, {}, () => null);
 }
 
 /** Returns BDFDB.TimeUtils.suppress from the installed BDFDB library. */
 function loadBdfdbSuppress(logUtils) {
     const source = fs.readFileSync(BDFDB_FILE, "utf8");
     const match = /BDFDB\.TimeUtils\.suppress = (function \(callback, strings, config\) \{[\s\S]*?\n\t*\}\};)/.exec(source);
     if (!match) throw new Error("BDFDB TimeUtils.suppress not found");
     return new Function("BDFDB", `return ${match[1].slice(0, -1)}`)({LogUtils: logUtils});
 }
 
 module.exports = {BD_ASAR, BDFDB_FILE, loadBdPatcher, loadBdfdbSuppress};