Clean up main
This commit is contained in:
Vendicated
parent
acd24cdc4f
commit
19801321cc
136
src/ipcMain.ts
136
src/ipcMain.ts
@ -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));
|
|
||||||
});
|
|
||||||
}
|
|
||||||
54
src/ipcMain/index.ts
Normal file
54
src/ipcMain/index.ts
Normal file
@ -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));
|
||||||
|
});
|
||||||
|
}
|
||||||
94
src/ipcMain/updater.ts
Normal file
94
src/ipcMain/updater.ts
Normal file
@ -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));
|
||||||
@ -1,4 +1,3 @@
|
|||||||
|
|
||||||
import electron, { app, BrowserWindowConstructorOptions } from "electron";
|
import electron, { app, BrowserWindowConstructorOptions } from "electron";
|
||||||
import installExt, { REACT_DEVELOPER_TOOLS } from "electron-devtools-installer";
|
import installExt, { REACT_DEVELOPER_TOOLS } from "electron-devtools-installer";
|
||||||
import { join } from "path";
|
import { join } from "path";
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user