aboutsummaryrefslogtreecommitdiff
path: root/scripts/patcher
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/patcher')
-rw-r--r--scripts/patcher/common.js297
-rw-r--r--scripts/patcher/install.js111
-rw-r--r--scripts/patcher/uninstall.js59
3 files changed, 467 insertions, 0 deletions
diff --git a/scripts/patcher/common.js b/scripts/patcher/common.js
new file mode 100644
index 0000000..94cb383
--- /dev/null
+++ b/scripts/patcher/common.js
@@ -0,0 +1,297 @@
+const path = require("path");
+const readline = require("readline");
+const fs = require("fs");
+const menu = require("console-menu");
+
+const BRANCH_NAMES = [
+ "Discord",
+ "DiscordPTB",
+ "DiscordCanary",
+ "DiscordDevelopment",
+ "discord",
+ "discordptb",
+ "discordcanary",
+ "discorddevelopment",
+ "discord-ptb",
+ "discord-canary",
+ "discord-development",
+ // Flatpak
+ "com.discordapp.Discord",
+ "com.discordapp.DiscordPTB",
+ "com.discordapp.DiscordCanary",
+ "com.discordapp.DiscordDevelopment",
+];
+
+const MACOS_DISCORD_DIRS = [
+ "Discord.app",
+ "Discord PTB.app",
+ "Discord Canary.app",
+ "Discord Development.app",
+];
+
+if (process.platform === "linux" && process.env.SUDO_USER) {
+ process.env.HOME = fs
+ .readFileSync("/etc/passwd", "utf-8")
+ .match(new RegExp(`^${process.env.SUDO_USER}.+$`, "m"))[0]
+ .split(":")[5];
+}
+
+const LINUX_DISCORD_DIRS = [
+ "/usr/share",
+ "/usr/lib64",
+ "/opt",
+ `${process.env.HOME}/.local/share`,
+ "/var/lib/flatpak/app",
+ `${process.env.HOME}/.local/share/flatpak/app`,
+];
+
+const FLATPAK_NAME_MAPPING = {
+ DiscordCanary: "discord-canary",
+ DiscordPTB: "discord-ptb",
+ DiscordDevelopment: "discord-development",
+ Discord: "discord",
+};
+
+const ENTRYPOINT = path
+ .join(process.cwd(), "dist", "patcher.js")
+ .replace(/\\/g, "/");
+
+function question(question) {
+ const rl = readline.createInterface({
+ input: process.stdin,
+ output: process.stdout,
+ terminal: false,
+ });
+
+ return new Promise((resolve) => {
+ rl.question(question, (answer) => {
+ rl.close();
+ resolve(answer);
+ });
+ });
+}
+
+async function getMenuItem(installations) {
+ let menuItems = installations.map((info) => ({
+ title: info.patched ? "[MODIFIED] " + info.location : info.location,
+ info,
+ }));
+
+ if (menuItems.length === 0) {
+ console.log("No Discord installations found.");
+ process.exit(1);
+ }
+
+ const result = await menu(
+ [...menuItems, { title: "Exit without patching", exit: true }],
+ {
+ header: "Select a Discord installation to patch:",
+ border: true,
+ helpMessage:
+ "Use the up/down arrow keys to select an option. " +
+ "Press ENTER to confirm.",
+ }
+ );
+
+ if (!result || !result.info || result.exit) {
+ console.log("No installation selected.");
+ process.exit(0);
+ }
+
+ if (result.info.patched) {
+ const answer = await question(
+ "This installation has already been modified. Overwrite? [Y/n]: "
+ );
+
+ if (!["y", "yes", "yeah", ""].includes(answer.toLowerCase())) {
+ console.log("Not patching.");
+ process.exit(0);
+ }
+ }
+
+ return result.info;
+}
+
+function getWindowsDirs() {
+ const dirs = [];
+ for (const dir of fs.readdirSync(process.env.LOCALAPPDATA)) {
+ if (!BRANCH_NAMES.includes(dir)) continue;
+
+ const location = path.join(process.env.LOCALAPPDATA, dir);
+ if (!fs.statSync(location).isDirectory()) continue;
+
+ const appDirs = fs
+ .readdirSync(location, { withFileTypes: true })
+ .filter((file) => file.isDirectory())
+ .filter((file) => file.name.startsWith("app-"))
+ .map((file) => path.join(location, file.name));
+
+ let versions = [];
+ let patched = false;
+
+ for (const fqAppDir of appDirs) {
+ const resourceDir = path.join(fqAppDir, "resources");
+ if (!fs.existsSync(path.join(resourceDir, "app.asar"))) {
+ continue;
+ }
+ const appDir = path.join(resourceDir, "app");
+ if (fs.existsSync(appDir)) {
+ patched = true;
+ }
+ versions.push({
+ path: appDir,
+ name: /app-([0-9\.]+)/.exec(fqAppDir)[1],
+ });
+ }
+
+ if (appDirs.length) {
+ dirs.push({
+ branch: dir,
+ patched,
+ location,
+ versions,
+ arch: "win32",
+ flatpak: false,
+ });
+ }
+ }
+ return dirs;
+}
+
+function getDarwinDirs() {
+ const dirs = [];
+ for (const dir of fs.readdirSync("/Applications")) {
+ if (!MACOS_DISCORD_DIRS.includes(dir)) continue;
+
+ const location = path.join("/Applications", dir, "Contents");
+ if (!fs.existsSync(location)) continue;
+ if (!fs.statSync(location).isDirectory()) continue;
+
+ const appDirs = fs
+ .readdirSync(location, { withFileTypes: true })
+ .filter((file) => file.isDirectory())
+ .filter((file) => file.name.startsWith("Resources"))
+ .map((file) => path.join(location, file.name));
+
+ let versions = [];
+ let patched = false;
+
+ for (const resourceDir of appDirs) {
+ if (!fs.existsSync(path.join(resourceDir, "app.asar"))) {
+ continue;
+ }
+ const appDir = path.join(resourceDir, "app");
+ if (fs.existsSync(appDir)) {
+ patched = true;
+ }
+
+ versions.push({
+ path: appDir,
+ name: null, // MacOS installs have no version number
+ });
+ }
+
+ if (appDirs.length) {
+ dirs.push({
+ branch: dir,
+ patched,
+ location,
+ versions,
+ arch: "win32",
+ });
+ }
+ }
+ return dirs;
+}
+
+function getLinuxDirs() {
+ const dirs = [];
+ for (const dir of LINUX_DISCORD_DIRS) {
+ if (!fs.existsSync(dir)) continue;
+ for (const branch of fs.readdirSync(dir)) {
+ if (!BRANCH_NAMES.includes(branch)) continue;
+
+ const location = path.join(dir, branch);
+ if (!fs.statSync(location).isDirectory()) continue;
+
+ const isFlatpak = location.includes("/flatpak/");
+
+ let appDirs = [];
+
+ if (isFlatpak) {
+ const fqDir = path.join(location, "current", "active", "files");
+ if (!/com\.discordapp\.(\w+)\//.test(fqDir)) continue;
+ const branchName = /com\.discordapp\.(\w+)\//.exec(fqDir)[1];
+ if (!Object.keys(FLATPAK_NAME_MAPPING).includes(branchName)) {
+ continue;
+ }
+ const appDir = path.join(
+ fqDir,
+ FLATPAK_NAME_MAPPING[branchName]
+ );
+
+ if (!fs.existsSync(appDir)) continue;
+ if (!fs.statSync(appDir).isDirectory()) continue;
+
+ const resourceDir = path.join(appDir, "resources");
+
+ appDirs.push(resourceDir);
+ } else {
+ appDirs = fs
+ .readdirSync(location, { withFileTypes: true })
+ .filter((file) => file.isDirectory())
+ .filter(
+ (file) =>
+ file.name.startsWith("app-") ||
+ file.name === "resources"
+ )
+ .map((file) => path.join(location, file.name));
+ }
+
+ let versions = [];
+ let patched = false;
+
+ for (const resourceDir of appDirs) {
+ if (!fs.existsSync(path.join(resourceDir, "app.asar"))) {
+ continue;
+ }
+ const appDir = path.join(resourceDir, "app");
+ if (fs.existsSync(appDir)) {
+ patched = true;
+ }
+
+ const version = /app-([0-9\.]+)/.exec(resourceDir);
+
+ versions.push({
+ path: appDir,
+ name: version && version.length > 1 ? version[1] : null,
+ });
+ }
+
+ if (appDirs.length) {
+ dirs.push({
+ branch,
+ patched,
+ location,
+ versions,
+ arch: "linux",
+ isFlatpak,
+ });
+ }
+ }
+ }
+ return dirs;
+}
+
+module.exports = {
+ BRANCH_NAMES,
+ MACOS_DISCORD_DIRS,
+ LINUX_DISCORD_DIRS,
+ FLATPAK_NAME_MAPPING,
+ ENTRYPOINT,
+ question,
+ getMenuItem,
+ getWindowsDirs,
+ getDarwinDirs,
+ getLinuxDirs,
+};
diff --git a/scripts/patcher/install.js b/scripts/patcher/install.js
new file mode 100644
index 0000000..42ba47f
--- /dev/null
+++ b/scripts/patcher/install.js
@@ -0,0 +1,111 @@
+const path = require("path");
+const fs = require("fs");
+const { execSync } = require("child_process");
+
+console.log("\nVencord Installer\n");
+
+if (!fs.existsSync(path.join(process.cwd(), "node_modules"))) {
+ console.log("You need to install dependencies first. Run:", "pnpm install");
+ process.exit(1);
+}
+
+if (!fs.existsSync(path.join(process.cwd(), "dist", "patcher.js"))) {
+ console.log("You need to build the project first. Run:", "pnpm build");
+ process.exit(1);
+}
+
+const {
+ getMenuItem,
+ getWindowsDirs,
+ getDarwinDirs,
+ getLinuxDirs,
+ ENTRYPOINT,
+} = require("./common");
+
+switch (process.platform) {
+ case "win32":
+ install(getWindowsDirs());
+ break;
+ case "darwin":
+ install(getDarwinDirs());
+ break;
+ case "linux":
+ install(getLinuxDirs());
+ break;
+ default:
+ console.log("Unknown OS");
+ break;
+}
+
+async function install(installations) {
+ const selected = await getMenuItem(installations);
+
+ // Attempt to give flatpak perms
+ if (selected.isFlatpak) {
+ try {
+ const branch = selected.branch;
+ const cwd = process.cwd();
+ const globalCmd = `flatpak override ${branch} --filesystem=${cwd}`;
+ const userCmd = `flatpak override --user ${branch} --filesystem=${cwd}`;
+ const cmd = selected.location.startsWith("/home")
+ ? userCmd
+ : globalCmd;
+ execSync(cmd);
+ console.log("Successfully gave write perms to Discord Flatpak.");
+ } catch (e) {
+ console.log("Failed to give write perms to Discord Flatpak.");
+ console.log(
+ "Try running this script as an administrator:",
+ "sudo pnpm patch"
+ );
+ process.exit(1);
+ }
+ }
+
+ for (const version of selected.versions) {
+ const dir = version.path;
+ // Check if we have write perms to the install directory...
+ try {
+ fs.accessSync(selected.location, fs.constants.W_OK);
+ } catch (e) {
+ console.error("No write access to", selected.location);
+ console.error(
+ "Try running this script as an administrator:",
+ "sudo pnpm patch"
+ );
+ process.exit(1);
+ }
+ if (fs.existsSync(dir) && fs.lstatSync(dir).isDirectory()) {
+ fs.rmSync(dir, { recursive: true });
+ }
+ if (!fs.existsSync(dir)) {
+ fs.mkdirSync(dir, { recursive: true });
+ }
+
+ fs.writeFileSync(
+ path.join(dir, "index.js"),
+ `require("${ENTRYPOINT}"); require("../app.asar");`
+ );
+ fs.writeFileSync(
+ path.join(dir, "package.json"),
+ JSON.stringify({
+ name: "discord",
+ main: "index.js",
+ })
+ );
+
+ const requiredFiles = ["index.js", "package.json"];
+
+ if (requiredFiles.every((f) => fs.existsSync(path.join(dir, f)))) {
+ console.log(
+ "Successfully patched",
+ version.name
+ ? `${selected.branch} ${version.name}`
+ : selected.branch
+ );
+ } else {
+ console.log("Failed to patch", dir);
+ console.log("Files in directory:", fs.readdirSync(dir));
+ }
+ }
+}
diff --git a/scripts/patcher/uninstall.js b/scripts/patcher/uninstall.js
new file mode 100644
index 0000000..c470c64
--- /dev/null
+++ b/scripts/patcher/uninstall.js
@@ -0,0 +1,59 @@
+const path = require("path");
+const fs = require("fs");
+
+console.log("\nVencord Uninstaller\n");
+
+if (!fs.existsSync(path.join(process.cwd(), "node_modules"))) {
+ console.log("You need to install dependencies first. Run:", "pnpm install");
+ process.exit(1);
+}
+
+const {
+ getMenuItem,
+ getWindowsDirs,
+ getDarwinDirs,
+ getLinuxDirs,
+} = require("./common");
+
+switch (process.platform) {
+ case "win32":
+ uninstall(getWindowsDirs());
+ break;
+ case "darwin":
+ uninstall(getDarwinDirs());
+ break;
+ case "linux":
+ uninstall(getLinuxDirs());
+ break;
+ default:
+ console.log("Unknown OS");
+ break;
+}
+
+async function uninstall(installations) {
+ const selected = await getMenuItem(installations);
+
+ for (const version of selected.versions) {
+ const dir = version.path;
+ // Check if we have write perms to the install directory...
+ try {
+ fs.accessSync(selected.location, fs.constants.W_OK);
+ } catch (e) {
+ console.error("No write access to", selected.location);
+ console.error(
+ "Try running this script as an administrator:",
+ "sudo pnpm unpatch"
+ );
+ process.exit(1);
+ }
+ if (fs.existsSync(dir)) {
+ fs.rmSync(dir, { recursive: true });
+ }
+ console.log(
+ "Successfully unpatched",
+ version.name
+ ? `${selected.branch} ${version.name}`
+ : selected.branch
+ );
+ }
+}
generated by cgit (git 2.46.0) at 2024-11-28 23:30:33 +0000