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	rakinar2	577	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			}