Vendicated
2 years ago
parent
44f6f71c3e
commit
c116d00d03
@ -0,0 +1,35 @@
|
|||||||
|
/*
|
||||||
|
* 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 { join } from "path";
|
||||||
|
import { app } from "electron";
|
||||||
|
|
||||||
|
export 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")
|
||||||
|
);
|
||||||
|
export const SETTINGS_DIR = join(DATA_DIR, "settings");
|
||||||
|
export const QUICKCSS_PATH = join(SETTINGS_DIR, "quickCss.css");
|
||||||
|
export const SETTINGS_FILE = join(SETTINGS_DIR, "settings.json");
|
||||||
|
export const ALLOWED_PROTOCOLS = [
|
||||||
|
"https:",
|
||||||
|
"http:",
|
||||||
|
"steam:",
|
||||||
|
"spotify:"
|
||||||
|
];
|
||||||
@ -0,0 +1,57 @@
|
|||||||
|
/* eslint-disable header/header */
|
||||||
|
|
||||||
|
/*!
|
||||||
|
* crxToZip
|
||||||
|
* Copyright (c) 2013 Rob Wu <rob@robwu.nl>
|
||||||
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
*/
|
||||||
|
|
||||||
|
export function crxToZip(buf: Buffer) {
|
||||||
|
function calcLength(a: number, b: number, c: number, d: number) {
|
||||||
|
let length = 0;
|
||||||
|
|
||||||
|
length += a << 0;
|
||||||
|
length += b << 8;
|
||||||
|
length += c << 16;
|
||||||
|
length += d << 24 >>> 0;
|
||||||
|
return length;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 50 4b 03 04
|
||||||
|
// This is actually a zip file
|
||||||
|
if (buf[0] === 80 && buf[1] === 75 && buf[2] === 3 && buf[3] === 4) {
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 43 72 32 34 (Cr24)
|
||||||
|
if (buf[0] !== 67 || buf[1] !== 114 || buf[2] !== 50 || buf[3] !== 52) {
|
||||||
|
throw new Error("Invalid header: Does not start with Cr24");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 02 00 00 00
|
||||||
|
// or
|
||||||
|
// 03 00 00 00
|
||||||
|
const isV3 = buf[4] === 3;
|
||||||
|
const isV2 = buf[4] === 2;
|
||||||
|
|
||||||
|
if ((!isV2 && !isV3) || buf[5] || buf[6] || buf[7]) {
|
||||||
|
throw new Error("Unexpected crx format version number.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isV2) {
|
||||||
|
const publicKeyLength = calcLength(buf[8], buf[9], buf[10], buf[11]);
|
||||||
|
const signatureLength = calcLength(buf[12], buf[13], buf[14], buf[15]);
|
||||||
|
|
||||||
|
// 16 = Magic number (4), CRX format version (4), lengths (2x4)
|
||||||
|
const zipStartOffset = 16 + publicKeyLength + signatureLength;
|
||||||
|
|
||||||
|
return buf.subarray(zipStartOffset, buf.length);
|
||||||
|
}
|
||||||
|
// v3 format has header size and then header
|
||||||
|
const headerSize = calcLength(buf[8], buf[9], buf[10], buf[11]);
|
||||||
|
const zipStartOffset = 12 + headerSize;
|
||||||
|
|
||||||
|
return buf.subarray(zipStartOffset, buf.length);
|
||||||
|
}
|
||||||
@ -0,0 +1,93 @@
|
|||||||
|
/*
|
||||||
|
* 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 https from "https";
|
||||||
|
import { mkdir, rm, writeFile, access } from "fs/promises";
|
||||||
|
import { constants as fsConstants } from "fs";
|
||||||
|
import { join } from "path";
|
||||||
|
import { DATA_DIR } from "./constants";
|
||||||
|
import { unzip } from "fflate";
|
||||||
|
import { session } from "electron";
|
||||||
|
import { crxToZip } from "./crxToZip";
|
||||||
|
|
||||||
|
const extensionCacheDir = join(DATA_DIR, "ExtensionCache");
|
||||||
|
|
||||||
|
function download(url: string) {
|
||||||
|
return new Promise<Buffer>((resolve, reject) => {
|
||||||
|
https.get(url, res => {
|
||||||
|
const { statusCode, statusMessage, headers } = res;
|
||||||
|
if (statusCode! >= 400)
|
||||||
|
return void reject(`${statusCode}: ${statusMessage} - ${url}`);
|
||||||
|
if (statusCode! >= 300)
|
||||||
|
return void resolve(download(headers.location!));
|
||||||
|
|
||||||
|
const chunks = [] as Buffer[];
|
||||||
|
res.on("error", reject);
|
||||||
|
|
||||||
|
res.on("data", chunk => chunks.push(chunk));
|
||||||
|
res.once("end", () => resolve(Buffer.concat(chunks)));
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function extract(data: Buffer, outDir: string) {
|
||||||
|
await mkdir(outDir, { recursive: true });
|
||||||
|
return new Promise<void>((resolve, reject) => {
|
||||||
|
unzip(data, (err, files) => {
|
||||||
|
if (err) return void reject(err);
|
||||||
|
Promise.all(Object.keys(files).map(async f => {
|
||||||
|
// Signature stuff
|
||||||
|
// 'Cannot load extension with file or directory name
|
||||||
|
// _metadata. Filenames starting with "_" are reserved for use by the system.';
|
||||||
|
if (f.startsWith("_metadata/")) return;
|
||||||
|
|
||||||
|
if (f.endsWith("/")) return void mkdir(join(outDir, f), { recursive: true });
|
||||||
|
|
||||||
|
const pathElements = f.split("/");
|
||||||
|
const name = pathElements.pop()!;
|
||||||
|
const directories = pathElements.join("/");
|
||||||
|
const dir = join(outDir, directories);
|
||||||
|
|
||||||
|
if (directories) {
|
||||||
|
await mkdir(dir, { recursive: true });
|
||||||
|
}
|
||||||
|
|
||||||
|
await writeFile(join(dir, name), files[f]);
|
||||||
|
}))
|
||||||
|
.then(() => resolve())
|
||||||
|
.catch(err => {
|
||||||
|
rm(outDir, { recursive: true, force: true });
|
||||||
|
reject(err);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function installExt(id: string) {
|
||||||
|
const extDir = join(extensionCacheDir, `${id}`);
|
||||||
|
|
||||||
|
try {
|
||||||
|
await access(extDir, fsConstants.F_OK);
|
||||||
|
} catch (err) {
|
||||||
|
const url = `https://clients2.google.com/service/update2/crx?response=redirect&acceptformat=crx2,crx3&x=id%3D${id}%26uc&prodversion=32`;
|
||||||
|
const buf = await download(url);
|
||||||
|
await extract(crxToZip(buf), extDir);
|
||||||
|
}
|
||||||
|
|
||||||
|
session.defaultSession.loadExtension(extDir);
|
||||||
|
}
|
||||||
Loading…
Reference in new issue