Dekescordes/spacebar
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rewrite client cacher (#962)

Browse Source
This commit is contained in:
Madeline 2023-01-22 13:41:31 +11:00 committed by GitHub
parent 084dc0be08
commit 4df7bbcfe5
2 changed files with 48 additions and 136 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
assets/cacheMisses
View File

184
scripts/client.js
View File

@ -17,7 +17,7 @@
*/ */
/* /*
This file downloads a ( mostly ) complete discord.com web client for testing, This file downloads a complete discord.com web client for testing,
and performs some basic patching: and performs some basic patching:
* Replaces all mentions of "Server" -> "Guild" * Replaces all mentions of "Server" -> "Guild"
* Replaces "Discord" -> `INSTANCE_NAME` variable * Replaces "Discord" -> `INSTANCE_NAME` variable
@ -25,15 +25,11 @@
* Prevents `localStorage` deletion ( for `plugins`/`preload-plugins` ) * Prevents `localStorage` deletion ( for `plugins`/`preload-plugins` )
* Adds `fast-identify` support ( TODO: add documentation ) * Adds `fast-identify` support ( TODO: add documentation )
This script can only download javascript client files.
It cannot download images, sounds, video, etc.
For that, a `cacheMisses` file in `fosscord-server/assets` is used.
After running the server for a while, uncached assets will be appended to that file
and downloaded after the next run of this script.
TODO: Make this configurable easily. TODO: Make this configurable easily.
*/ */
/*eslint-env node*/
require("dotenv/config"); require("dotenv/config");
const path = require("path"); const path = require("path");
const fetch = require("node-fetch"); const fetch = require("node-fetch");
@ -51,7 +47,7 @@ const agent = (_parsedURL) =>
const CACHE_PATH = path.join(__dirname, "..", "assets", "cache"); const CACHE_PATH = path.join(__dirname, "..", "assets", "cache");
const BASE_URL = "https://discord.com"; const BASE_URL = "https://discord.com";
const INSTANCE_NAME = "Fosscord"; const INSTANCE_NAME = process.env.CLIENT_PATCH_INSTANCE_NAME ?? "Fosscord";
const ONLY_CACHE_JS = process.env.ONLY_CACHE_JS ? true : false; const ONLY_CACHE_JS = process.env.ONLY_CACHE_JS ? true : false;
// Manual for now // Manual for now
@ -61,11 +57,12 @@ const INDEX_SCRIPTS = [
"f98a039261c37f892cbf", // 0? "f98a039261c37f892cbf", // 0?
"4470c87bb13810847db0", // ~4500. "4470c87bb13810847db0", // ~4500.
// also fetch other assets from index, as they aren't cached // also fetch other assets from index, as we don't have a way to dl old indexes
"40532.f4ff6c4a39fa78f07880.css", "40532.f4ff6c4a39fa78f07880.css",
"b21a783b953e52485dcb.worker.js", "b21a783b953e52485dcb.worker.js",
"2bbea887c6d07e427a1d.worker.js", "2bbea887c6d07e427a1d.worker.js",
"0ec5df6d78ff7a5cc7c8.worker.js", "0ec5df6d78ff7a5cc7c8.worker.js",
"625ccb6efce655a7d928.worker.js",
"05422eb499ddf5616e44a52c4f1063ae.woff2", "05422eb499ddf5616e44a52c4f1063ae.woff2",
"77f603cc7860fcb784e6ef9320a4a9c2.woff2", "77f603cc7860fcb784e6ef9320a4a9c2.woff2",
"e689380400b1f2d2c6320a823a1ab079.svg", "e689380400b1f2d2c6320a823a1ab079.svg",
@ -79,7 +76,7 @@ const doPatch = (content) => {
content = content.replaceAll(/ Nitro/g, " Premium"); content = content.replaceAll(/ Nitro/g, " Premium");
content = content.replaceAll(/\[Nitro\]/g, "[Premium]"); content = content.replaceAll(/\[Nitro\]/g, "[Premium]");
content = content.replaceAll(/\*Nitro\*/g, "*Premium*"); content = content.replaceAll(/\*Nitro\*/g, "*Premium*");
content = content.replaceAll(/\"Nitro \. /g, '"Premium. '); content = content.replaceAll(/"Nitro \. /g, '"Premium. ');
//remove discord references //remove discord references
content = content.replaceAll(/ Discord /g, ` ${INSTANCE_NAME} `); content = content.replaceAll(/ Discord /g, ` ${INSTANCE_NAME} `);
@ -184,162 +181,77 @@ const doPatch = (content) => {
// disable qr code login // disable qr code login
content = content.replaceAll( content = content.replaceAll(
/\w\?\(\d,\w\.jsx\)\(\w*\,{authTokenCallback:this\.handleAuthToken}\):null/g, /\w\?\(\d,\w\.jsx\)\(\w*,{authTokenCallback:this\.handleAuthToken}\):null/g,
"null", "null",
); );
return content; return content;
}; };
const processFile = async (name) => { const print = (x, printover = true) => {
const url = `${BASE_URL}/assets/${name}${name.includes(".") ? "" : ".js"}`; var repeat = process.stdout.columns - x.length;
if (ONLY_CACHE_JS && !url.endsWith(".js")) return []; process.stdout.write(
`${x}${" ".repeat(Math.max(0, repeat))}${printover ? "\r" : "\n"}`,
);
};
const processFile = async (asset) => {
// The asset name may not include the file extension. Usually if it doesn't, it's js though.
asset = `${asset}${asset.includes(".") ? "" : ".js"}`;
if (ONLY_CACHE_JS && !asset.endsWith(".js")) return [];
const url = `${BASE_URL}/assets/${asset}`;
const res = await fetch(url, { agent }); const res = await fetch(url, { agent });
if (res.status !== 200) { if (res.status !== 200) {
print(`${res.status} on ${asset}`, false);
return []; return [];
} }
if (name.includes(".") && !name.includes(".js") && !name.includes(".css")) { if (
await fs.writeFile(path.join(CACHE_PATH, name), await res.buffer()); asset.includes(".") &&
!asset.includes(".js") &&
!asset.includes(".css")
) {
await fs.writeFile(path.join(CACHE_PATH, asset), await res.buffer());
return []; return [];
} }
let text = await res.text(); let text = await res.text();
text = doPatch(text); text = doPatch(text);
await fs.writeFile( await fs.writeFile(path.join(CACHE_PATH, asset), text);
path.join(CACHE_PATH, `${name}${name.includes(".") ? "" : ".js"}`),
text,
);
var additional = []; let ret = new Set([
additional.push(...new Set(text.match(/\"[A-Fa-f0-9]{20}\"/g))); ...(text.match(/"[A-Fa-f0-9]{20}"/g) ?? []), // These are generally JS assets
additional.push( ...[...text.matchAll(/\.exports=.\..\+"(.*?\..{0,5})"/g)].map(
...[ (x) => x[1],
...new Set(text.matchAll(/\.exports=.\..\+\"(.*?\..{0,5})\"/g)), ), // anything that looks like e.exports="filename.ext"
].map((x) => x[1]), ...[...text.matchAll(/\/assets\/(.*?\.[a-z]{0,5})/g)].map((x) => x[1]), // commonly matches `background: url(/assets/blah.svg)`
); ]);
return additional.map((x) => x.replaceAll('"', "")); return [...ret].map((x) => x.replaceAll('"', ""));
};
const print = (x) => {
var repeat = process.stdout.columns - x.length;
process.stdout.write(`${x}${" ".repeat(Math.max(0, repeat))}\r`);
}; };
(async () => { (async () => {
const start = Date.now(); if (!existsSync(CACHE_PATH))
await fs.mkdir(CACHE_PATH, { recursive: true });
// console.log("Deleting previous cache");
// await fs.rm(CACHE_PATH, { recursive: true });
if (!existsSync(CACHE_PATH)) await fs.mkdir(CACHE_PATH);
const assets = [];
while (INDEX_SCRIPTS.length > 0) {
const asset = INDEX_SCRIPTS.shift();
print(`Scraping asset ${asset}. Remaining: ${INDEX_SCRIPTS.length}`);
const newAssets = await processFile(asset);
assets.push(...newAssets);
}
console.log();
const CACHE_MISSES = (
await fs.readFile(path.join(CACHE_PATH, "..", "cacheMisses"))
)
.toString()
.split("\r")
.join("")
.split("\n");
while (CACHE_MISSES.length > 0) {
const asset = CACHE_MISSES.shift();
print(
`Scraping cache misses ${asset}. Remaining: ${CACHE_MISSES.length}`,
);
if (existsSync(path.join(CACHE_PATH, `${asset}`))) {
continue;
}
const newAssets = await processFile(asset);
assets.push(...newAssets);
}
console.log();
var existing = await fs.readdir(CACHE_PATH);
while (existing.length > 0) {
var file = existing.shift();
print(`Patching existing ${file}. Remaining: ${existing.length}.`);
var text = await fs.readFile(path.join(CACHE_PATH, file));
if (file.includes(".js") || file.includes(".css")) {
text = doPatch(text.toString());
await fs.writeFile(path.join(CACHE_PATH, file), text.toString());
var additional = [];
additional.push(...new Set(text.match(/\"[A-Fa-f0-9]{20}\"/g)));
additional.push(
...[
...new Set(
text.matchAll(/\.exports=.\..\+\"(.*?\..{0,5})\"/g),
),
].map((x) => x[1]),
);
assets.push(additional.map((x) => x.replaceAll('"', "")));
}
}
console.log();
let rates = [];
let lastFinished = Date.now();
let previousFinish = Date.now();
// Use a set to remove dupes for us
const assets = new Set(INDEX_SCRIPTS);
let promises = []; let promises = [];
for (var i = 0; i < assets.length; i++) { let index = 0;
const asset = assets[i]; for (let asset of assets) {
index += 1;
if (existsSync(path.join(CACHE_PATH, `${asset}.js`))) { print(`Scraping asset ${asset}. Remaining: ${assets.size - index}`);
continue;
}
while (rates.length > 50) rates.shift();
const averageRate = rates.length
? rates.reduce((prev, curr) => prev + curr) / rates.length
: 1;
const finishTime = averageRate * (assets.length - i);
print(
`Caching asset ${asset}. ` +
`${i}/${assets.length - 1} = ${Math.floor(
(i / (assets.length - 1)) * 100,
)}% `,
// + `Finish at: ${new Date(
// Date.now() + finishTime,
// ).toLocaleTimeString()}`,
);
promises.push(processFile(asset)); promises.push(processFile(asset));
// await processFile(asset); if (promises.length > 100 || index == assets.size) {
if (promises.length > 100) {
const values = await Promise.all(promises); const values = await Promise.all(promises);
assets.push(...values.flat());
promises = []; promises = [];
lastFinished = Date.now(); values.flat().forEach((x) => assets.add(x));
rates.push(lastFinished - previousFinish);
previousFinish = lastFinished;
} }
} }
console.log(`\nDone`); console.log("done");
})(); })();