aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/ipcMain.ts136
-rw-r--r--src/ipcMain/index.ts54
-rw-r--r--src/ipcMain/updater.ts94
-rw-r--r--src/patcher.ts1
4 files changed, 148 insertions, 137 deletions
diff --git a/src/ipcMain.ts b/src/ipcMain.ts
deleted file mode 100644
index 89c49ab..0000000
--- a/src/ipcMain.ts
+++ /dev/null
@@ -1,136 +0,0 @@
-// TODO: refactor this mess
-
-import { execFile as cpExecFile } from 'child_process';
-import { createHash } from "crypto";
-import { app, BrowserWindow, desktopCapturer, ipcMain, shell } from "electron";
-import { createReadStream, mkdirSync, readFileSync, watch } from "fs";
-import { open, readFile, writeFile } from "fs/promises";
-import { join } from 'path';
-import { promisify } from "util";
-import { debounce } from "./utils/debounce";
-import IpcEvents from './utils/IpcEvents';
-
-const VENCORD_SRC_DIR = join(__dirname, "..");
-const DATA_DIR = join(app.getPath("userData"), "..", "Vencord");
-const SETTINGS_DIR = join(DATA_DIR, "settings");
-const QUICKCSS_PATH = join(SETTINGS_DIR, "quickCss.css");
-const SETTINGS_FILE = join(SETTINGS_DIR, "settings.json");
-
-const execFile = promisify(cpExecFile);
-
-mkdirSync(SETTINGS_DIR, { recursive: true });
-
-async function calculateHashes() {
- const hashes = {} as Record<string, string>;
-
- await Promise.all(
- ["patcher.js", "preload.js", "renderer.js"].map(file => new Promise<void>(r => {
- const fis = createReadStream(join(__dirname, file));
- const hash = createHash("sha1", { encoding: "hex" });
- fis.once("end", () => {
- hash.end();
- hashes[file] = hash.read();
- r();
- });
- fis.pipe(hash);
- }))
- );
-
- return hashes;
-}
-
-function git(...args: string[]) {
- return execFile("git", args, {
- cwd: VENCORD_SRC_DIR
- });
-}
-
-function readCss() {
- return readFile(QUICKCSS_PATH, "utf-8").catch(() => "");
-}
-
-function readSettings() {
- try {
- return readFileSync(SETTINGS_FILE, "utf-8");
- } catch {
- return "{}";
- }
-}
-
-function serializeErrors(func: (...args: any[]) => any) {
- return async function () {
- try {
- return {
- ok: true,
- value: await func(...arguments)
- };
- } catch (e: any) {
- return {
- ok: false,
- error: e instanceof Error ? {
- // prototypes get lost, so turn error into plain object
- ...e
- } : e
- };
- }
- };
-}
-
-ipcMain.handle(IpcEvents.GET_DESKTOP_CAPTURE_SOURCES, (_, opts) => desktopCapturer.getSources(opts));
-ipcMain.handle(IpcEvents.GET_SETTINGS_DIR, () => SETTINGS_DIR);
-ipcMain.handle(IpcEvents.GET_QUICK_CSS, () => readCss());
-ipcMain.handle(IpcEvents.OPEN_PATH, (_, ...pathElements) => shell.openPath(join(...pathElements)));
-ipcMain.handle(IpcEvents.OPEN_EXTERNAL, (_, url) => shell.openExternal(url));
-
-ipcMain.handle(IpcEvents.GET_UPDATES, serializeErrors(async () => {
- await git("fetch");
-
- const res = await git("log", `HEAD...origin/main`, "--pretty=format:%an/%h/%s");
-
- const commits = res.stdout.trim();
- return commits ? commits.split("\n").map(line => {
- const [author, hash, ...rest] = line.split("/");
- return {
- hash, author, message: rest.join("/")
- };
- }) : [];
-}));
-
-ipcMain.handle(IpcEvents.UPDATE, serializeErrors(async () => {
- const res = await git("pull");
- return res.stdout.includes("Fast-forward");
-}));
-
-ipcMain.handle(IpcEvents.BUILD, serializeErrors(async () => {
- const res = await execFile("node", ["build.mjs"], {
- cwd: VENCORD_SRC_DIR
- });
- return !res.stderr.includes("Build failed");
-}));
-
-ipcMain.handle(IpcEvents.GET_HASHES, serializeErrors(calculateHashes));
-
-ipcMain.handle(IpcEvents.GET_REPO, serializeErrors(async () => {
- const res = await git("remote", "get-url", "origin");
- return res.stdout.trim()
- .replace(/git@(.+):/, "https://$1/")
- .replace(/\.git$/, "");
-}));
-
-// .on because we need Settings synchronously (ipcRenderer.sendSync)
-ipcMain.on(IpcEvents.GET_SETTINGS, (e) => e.returnValue = readSettings());
-
-// This is required because otherwise calling SET_SETTINGS in quick succession may lead to concurrent writes
-let settingsWriteQueue = Promise.resolve();
-ipcMain.handle(IpcEvents.SET_SETTINGS, (_, s) => {
- settingsWriteQueue = settingsWriteQueue.then(() => writeFile(SETTINGS_FILE, s));
-});
-
-export function initIpc(mainWindow: BrowserWindow) {
- open(QUICKCSS_PATH, "a+").then(fd => {
- fd.close();
- watch(QUICKCSS_PATH, debounce(async () => {
- mainWindow.webContents.postMessage(IpcEvents.QUICK_CSS_UPDATE, await readCss());
- }, 50));
- });
-}
diff --git a/src/ipcMain/index.ts b/src/ipcMain/index.ts
new file mode 100644
index 0000000..92d25fe
--- /dev/null
+++ b/src/ipcMain/index.ts
@@ -0,0 +1,54 @@
+import { app, BrowserWindow, desktopCapturer, ipcMain, shell } from "electron";
+import { mkdirSync, readFileSync, watch } from "fs";
+import { open, readFile, writeFile } from "fs/promises";
+import { join } from 'path';
+import { debounce } from "../utils/debounce";
+import IpcEvents from '../utils/IpcEvents';
+
+import "./updater";
+
+const DATA_DIR = join(app.getPath("userData"), "..", "Vencord");
+const SETTINGS_DIR = join(DATA_DIR, "settings");
+const QUICKCSS_PATH = join(SETTINGS_DIR, "quickCss.css");
+const SETTINGS_FILE = join(SETTINGS_DIR, "settings.json");
+
+mkdirSync(SETTINGS_DIR, { recursive: true });
+
+function readCss() {
+ return readFile(QUICKCSS_PATH, "utf-8").catch(() => "");
+}
+
+function readSettings() {
+ try {
+ return readFileSync(SETTINGS_FILE, "utf-8");
+ } catch {
+ return "{}";
+ }
+}
+
+// Fix for screensharing in Electron >= 17
+ipcMain.handle(IpcEvents.GET_DESKTOP_CAPTURE_SOURCES, (_, opts) => desktopCapturer.getSources(opts));
+
+ipcMain.handle(IpcEvents.OPEN_PATH, (_, ...pathElements) => shell.openPath(join(...pathElements)));
+ipcMain.handle(IpcEvents.OPEN_EXTERNAL, (_, url) => shell.openExternal(url));
+
+
+ipcMain.handle(IpcEvents.GET_QUICK_CSS, () => readCss());
+
+ipcMain.handle(IpcEvents.GET_SETTINGS_DIR, () => SETTINGS_DIR);
+ipcMain.on(IpcEvents.GET_SETTINGS, (e) => e.returnValue = readSettings());
+
+let settingsWriteQueue = Promise.resolve();
+ipcMain.handle(IpcEvents.SET_SETTINGS, (_, s) => {
+ settingsWriteQueue = settingsWriteQueue.then(() => writeFile(SETTINGS_FILE, s));
+});
+
+
+export function initIpc(mainWindow: BrowserWindow) {
+ open(QUICKCSS_PATH, "a+").then(fd => {
+ fd.close();
+ watch(QUICKCSS_PATH, debounce(async () => {
+ mainWindow.webContents.postMessage(IpcEvents.QUICK_CSS_UPDATE, await readCss());
+ }, 50));
+ });
+}
diff --git a/src/ipcMain/updater.ts b/src/ipcMain/updater.ts
new file mode 100644
index 0000000..cd1fc6c
--- /dev/null
+++ b/src/ipcMain/updater.ts
@@ -0,0 +1,94 @@
+import { ipcMain } from 'electron';
+import { promisify } from "util";
+import IpcEvents from "../utils/IpcEvents";
+import { execFile as cpExecFile } from 'child_process';
+import { join } from 'path';
+import { createReadStream } from 'fs';
+import { createHash } from 'crypto';
+
+const VENCORD_SRC_DIR = join(__dirname, "..");
+
+const execFile = promisify(cpExecFile);
+
+function git(...args: string[]) {
+ return execFile("git", args, {
+ cwd: VENCORD_SRC_DIR
+ });
+}
+
+async function calculateHashes() {
+ const hashes = {} as Record<string, string>;
+
+ await Promise.all(
+ ["patcher.js", "preload.js", "renderer.js"].map(file => new Promise<void>(r => {
+ const fis = createReadStream(join(__dirname, file));
+ const hash = createHash("sha1", { encoding: "hex" });
+ fis.once("end", () => {
+ hash.end();
+ hashes[file] = hash.read();
+ r();
+ });
+ fis.pipe(hash);
+ }))
+ );
+
+ return hashes;
+}
+
+function serializeErrors(func: (...args: any[]) => any) {
+ return async function () {
+ try {
+ return {
+ ok: true,
+ value: await func(...arguments)
+ };
+ } catch (e: any) {
+ return {
+ ok: false,
+ error: e instanceof Error ? {
+ // prototypes get lost, so turn error into plain object
+ ...e
+ } : e
+ };
+ }
+ };
+}
+
+async function getRepo() {
+ const res = await git("remote", "get-url", "origin");
+ return res.stdout.trim()
+ .replace(/git@(.+):/, "https://$1/")
+ .replace(/\.git$/, "");
+}
+
+async function calculateGitChanges() {
+ await git("fetch");
+
+ const res = await git("log", `HEAD...origin/main`, "--pretty=format:%an/%h/%s");
+
+ const commits = res.stdout.trim();
+ return commits ? commits.split("\n").map(line => {
+ const [author, hash, ...rest] = line.split("/");
+ return {
+ hash, author, message: rest.join("/")
+ };
+ }) : [];
+}
+
+async function pull() {
+ const res = await git("pull");
+ return res.stdout.includes("Fast-forward");
+}
+
+async function build() {
+ const res = await execFile("node", ["build.mjs"], {
+ cwd: VENCORD_SRC_DIR
+ });
+ return !res.stderr.includes("Build failed");
+}
+
+ipcMain.handle(IpcEvents.GET_HASHES, serializeErrors(calculateHashes));
+ipcMain.handle(IpcEvents.GET_REPO, serializeErrors(getRepo));
+ipcMain.handle(IpcEvents.GET_UPDATES, serializeErrors(calculateGitChanges));
+ipcMain.handle(IpcEvents.UPDATE, serializeErrors(pull));
+ipcMain.handle(IpcEvents.BUILD, serializeErrors(build));
diff --git a/src/patcher.ts b/src/patcher.ts
index a1aba2d..8ff92d2 100644
--- a/src/patcher.ts
+++ b/src/patcher.ts
@@ -1,4 +1,3 @@
-
import electron, { app, BrowserWindowConstructorOptions } from "electron";
import installExt, { REACT_DEVELOPER_TOOLS } from "electron-devtools-installer";
import { join } from "path";
generated by cgit (git 2.46.0) at 2024-11-25 14:34:57 +0000