feat(installer): Implement cross-platform patcher. (#39)

* megu cute

1 files changed, 111 insertions, 0 deletions

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