6.x/tests/clientsetup.ts

import fs from "fs";
import fsPromises from "fs/promises";
import { randomSnowflake } from "./utils";

export const GUILD_ID = randomSnowflake();

const config = {
    guild: `
        {
            "${GUILD_ID}": {}
        }
    `,
    system: `
        {
            "system_admins": ["${randomSnowflake()}"]
        }
    `
};

export function setCustomGuildConfig(configJSON: string) {
    config.guild = configJSON;
}

export function setCustomSystemConfig(configJSON: string) {
    config.system = configJSON;
}

export function fileHandler(path: any) {
    if (path === configPath) {
        return config.guild;
    } else if (path === systemConfigPath) {
        return config.system;
    }
}

export const readFileSync = fs.readFileSync;
export const readFile = fsPromises.readFile;

export function registerFileHandler() {
    (fs as any).readFileSync = (path: string, ...args: [any]) =>
        path === configPath || path === systemConfigPath ? fileHandler(path) : readFileSync(path, ...args);
    (fsPromises as any).readFile = (path: string, ...args: [any]) =>
        path === configPath || path === systemConfigPath ? fileHandler(path) : readFile(path, ...args);
}

export function unregisterFileHandler() {
    (fs as any).readFileSync = readFileSync;
    (fsPromises as any).readFile = readFile;
}
1	import fs from "fs";
2	import fsPromises from "fs/promises";
3	import { randomSnowflake } from "./utils";
4
5	export const GUILD_ID = randomSnowflake();
6
7	const config = {
8	guild: `
9	{
10	"${GUILD_ID}": {}
11	}
12	`,
13	system: `
14	{
15	"system_admins": ["${randomSnowflake()}"]
16	}
17	`
18	};
19
20	export function setCustomGuildConfig(configJSON: string) {
21	config.guild = configJSON;
22	}
23
24	export function setCustomSystemConfig(configJSON: string) {
25	config.system = configJSON;
26	}
27
28	export function fileHandler(path: any) {
29	if (path === configPath) {
30	return config.guild;
31	} else if (path === systemConfigPath) {
32	return config.system;
33	}
34	}
35
36	export const readFileSync = fs.readFileSync;
37	export const readFile = fsPromises.readFile;
38
39	export function registerFileHandler() {
40	(fs as any).readFileSync = (path: string, ...args: [any]) =>
41	path === configPath \|\| path === systemConfigPath ? fileHandler(path) : readFileSync(path, ...args);
42	(fsPromises as any).readFile = (path: string, ...args: [any]) =>
43	path === configPath \|\| path === systemConfigPath ? fileHandler(path) : readFile(path, ...args);
44	}
45
46	export function unregisterFileHandler() {
47	(fs as any).readFileSync = readFileSync;
48	(fsPromises as any).readFile = readFile;
49	}