94 lines
2.9 KiB
TypeScript
94 lines
2.9 KiB
TypeScript
/*
|
|
* Vencord, a modification for Discord's desktop app
|
|
* Copyright (c) 2022 Vendicated and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
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 = process.env.VENCORD_USER_DATA_DIR ?? (
|
|
process.env.DISCORD_USER_DATA_DIR
|
|
? join(process.env.DISCORD_USER_DATA_DIR, "..", "VencordData")
|
|
: 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 ALLOWED_PROTOCOLS = [
|
|
"https:",
|
|
"http:",
|
|
"steam:",
|
|
"spotify:"
|
|
];
|
|
|
|
mkdirSync(SETTINGS_DIR, { recursive: true });
|
|
|
|
function readCss() {
|
|
return readFile(QUICKCSS_PATH, "utf-8").catch(() => "");
|
|
}
|
|
|
|
export 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_QUICKCSS, () => shell.openPath(QUICKCSS_PATH));
|
|
|
|
ipcMain.handle(IpcEvents.OPEN_EXTERNAL, (_, url) => {
|
|
try {
|
|
var { protocol } = new URL(url);
|
|
} catch {
|
|
throw "Malformed URL";
|
|
}
|
|
if (!ALLOWED_PROTOCOLS.includes(protocol))
|
|
throw "Disallowed protocol.";
|
|
|
|
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));
|
|
});
|
|
}
|