src/client/Config.ts

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* SudoBot is free software; you can redistribute it and/or modify it
* under the terms of the GNU Affero General Public License as published by 
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* 
* SudoBot 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License 
* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
*/

import DiscordClient from "./Client";
import path from "path";
import fs from "fs";
import { z as zod } from 'zod';

export type config = {
    [key: string]: any;
};

export type configContainer = {
    [guildID: string | number]: config;  
};

export class Config {
    props: configContainer = {};
    client: DiscordClient;
    configPath: string;

    constructor(client: DiscordClient) {
        this.client = client;
        console.log(`ENV: ${process.env.SUDO_PREFIX}`);
        this.configPath = path.resolve(process.env.SUDO_PREFIX ?? this.client.rootdir, "config", "config.json");
        this.load();
    }

    load() {
        fs.readFile(this.configPath, (err, data) => {
            if (err) {
                console.log(err);
            }

            this.props = JSON.parse(data.toString());
        });
    }

    write() {
        fs.writeFile(this.configPath, JSON.stringify(this.props, undefined, ' '), () => null);
    }

    get(key: string) {
        return typeof this.props[this.client.msg!.guild!.id] === 'object' ? this.props[this.client.msg!.guild!.id][key] : null;
    }

    set(key: string, value: any) {
        this.props[this.client.msg!.guild!.id][key] = value;
    }

    schema(_config: string | config) {
        const config = typeof _config === 'string' ? this.props[_config] : _config;
        const snowflake = zod.string().regex(/\d+/, { message: "The given value is not a Snowflake" });

        return zod.object({
            "prefix": zod.string().optional(),
            "debug": zod.boolean().optional(),
            "mute_role": snowflake.optional(),
            "gen_role": snowflake.optional(),
            "logging_channel": snowflake.optional(),
            "logging_channel_join_leave": snowflake.optional(),
            "mod_role": snowflake.optional(),
            "announcement_channel": snowflake.optional(),
            "admin": snowflake.optional(),
            "lockall": zod.array(zod.string()).optional(),
            "warn_notallowed": zod.boolean().optional(),
            "ai_mod": zod.object({
                "enabled": zod.boolean().optional(),
                "toxicity": zod.number().optional(),
                "severe_toxicity": zod.number().optional(),
                "threat": zod.number().optional(),
            }).optional(),
            "role_commands": zod.record(
                snowflake,
                zod.array(zod.string().min(1))
            ).optional().default({}),
            "autoclear": zod.object({
                "enabled": zod.boolean().optional(),
                "channels": zod.array(snowflake).optional().default(config.autoclear.channels)
            }).optional(),
            "verification": zod.object({
                "enabled": zod.boolean().optional(),
                "role": snowflake.optional()
            }).optional(),
            "welcomer": zod.object({
                "enabled": zod.boolean().optional(),
                "channel": snowflake.optional(),
                "message": zod.string().min(1).or(zod.null()).optional(),
                "randomize": zod.boolean().optional()
            }).optional(),
            "cooldown": zod.object({
                "enabled": zod.boolean().optional(),
                "global": zod.any().optional(),
                "cmds": zod.object({}).optional()
            }).optional(),
            "profile_filter": zod.object({
                "enabled": zod.boolean().optional(),
                "inherit_from_words": zod.boolean().optional(),
                "inherit_from_tokens": zod.boolean().optional(),
                "inherit_from_regex": zod.boolean().optional(),
                "blocked_words": zod.array(zod.string()).optional().default(config.profile_filter.blocked_words),
                "blocked_tokens": zod.array(zod.string()).optional().default(config.profile_filter.blocked_tokens),
                "blocked_regex_patterns": zod.array(zod.string()).optional().default(config.profile_filter.blocked_regex_patterns),
                "components": zod.object({
                    "tag": zod.boolean().optional(),
                    "nickname": zod.boolean().optional(),
                    "about": zod.boolean().optional(),
                    "status": zod.boolean().optional()
                }).optional(),
                "actions": zod.object({
                    "tag": zod.string().optional(),
                    "nickname": zod.string().optional(),
                    "about": zod.string().optional(),
                    "status": zod.string().optional()
                }).optional()
            }).optional(),
            "starboard": zod.object({
                "enabled": zod.boolean().optional(),
                "reactions": zod.number().int().optional(),
                "channel": snowflake.optional()
            }).optional(),
            "autorole": zod.object({
                "enabled": zod.boolean().optional(),
                "roles": zod.array(snowflake).optional().default(config.autorole.roles)
            }).optional(),
            "spam_filter": zod.object({
                "enabled": zod.boolean().optional(),
                "limit": zod.number().int().optional(),
                "time": zod.number().optional(),
                "diff": zod.number().optional(),
                "exclude": zod.array(snowflake).optional().default(config.spam_filter.exclude),
                "samelimit": zod.number().int().optional(),
                "unmute_in": zod.number().optional()
            }).optional(),
            "raid": zod.object({
                "enabled": zod.boolean().optional(),
                "max_joins": zod.number().int().optional(),
                "time": zod.number().optional(),
                "channels": zod.array(snowflake).optional().default(config.raid.channels),
                "exclude": zod.boolean().optional()
            }).optional(),
            "global_commands": zod.array(zod.string()).optional().default(config.global_commands),
            "filters": zod.object({
                "ignore_staff": zod.boolean().optional(),
                "chars_repeated": zod.number().int().optional(),
                "words_repeated": zod.number().int().optional(),
                "words": zod.array(zod.string()).optional().default(config.filters.words),
                "tokens": zod.array(zod.string()).optional().default(config.filters.tokens),
                "invite_message": zod.string().optional(),
                "words_excluded": zod.array(snowflake).optional().default(config.filters.words_excluded),
                "domain_excluded": zod.array(snowflake).optional().default(config.filters.domain_excluded),
                "invite_excluded": zod.array(snowflake).optional().default(config.filters.invite_excluded),
                "words_enabled": zod.boolean().optional(),
                "invite_enabled": zod.boolean().optional(),
                "domain_enabled": zod.boolean().optional(),
                "regex": zod.boolean().optional(),
                "file_mimes_excluded": zod.array(zod.string()).optional().default(config.filters.file_mimes_excluded),
                "file_types_excluded": zod.array(zod.string()).optional().default(config.filters.file_types_excluded),
                "domains": zod.array(zod.string()).optional().default(config.filters.domains),
                "regex_patterns": zod.array(zod.string()).optional().default(config.filters.regex_patterns),
                "rickrolls_enabled": zod.boolean().optional(),
                "pings": zod.number().int().optional()
            }).optional()
        });
    }
}
1	rakinar2	577	/**
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2022 OSN Inc.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import DiscordClient from "./Client";
21			import path from "path";
22			import fs from "fs";
23			import { z as zod } from 'zod';
24
25			export type config = {
26			[key: string]: any;
27			};
28
29			export type configContainer = {
30			[guildID: string \| number]: config;
31			};
32
33			export class Config {
34			props: configContainer = {};
35			client: DiscordClient;
36			configPath: string;
37
38			constructor(client: DiscordClient) {
39			this.client = client;
40			console.log(`ENV: ${process.env.SUDO_PREFIX}`);
41			this.configPath = path.resolve(process.env.SUDO_PREFIX ?? this.client.rootdir, "config", "config.json");
42			this.load();
43			}
44
45			load() {
46			fs.readFile(this.configPath, (err, data) => {
47			if (err) {
48			console.log(err);
49			}
50
51			this.props = JSON.parse(data.toString());
52			});
53			}
54
55			write() {
56			fs.writeFile(this.configPath, JSON.stringify(this.props, undefined, ' '), () => null);
57			}
58
59			get(key: string) {
60			return typeof this.props[this.client.msg!.guild!.id] === 'object' ? this.props[this.client.msg!.guild!.id][key] : null;
61			}
62
63			set(key: string, value: any) {
64			this.props[this.client.msg!.guild!.id][key] = value;
65			}
66
67			schema(_config: string \| config) {
68			const config = typeof _config === 'string' ? this.props[_config] : _config;
69			const snowflake = zod.string().regex(/\d+/, { message: "The given value is not a Snowflake" });
70
71			return zod.object({
72			"prefix": zod.string().optional(),
73			"debug": zod.boolean().optional(),
74			"mute_role": snowflake.optional(),
75			"gen_role": snowflake.optional(),
76			"logging_channel": snowflake.optional(),
77			"logging_channel_join_leave": snowflake.optional(),
78			"mod_role": snowflake.optional(),
79			"announcement_channel": snowflake.optional(),
80			"admin": snowflake.optional(),
81			"lockall": zod.array(zod.string()).optional(),
82			"warn_notallowed": zod.boolean().optional(),
83			"ai_mod": zod.object({
84			"enabled": zod.boolean().optional(),
85			"toxicity": zod.number().optional(),
86			"severe_toxicity": zod.number().optional(),
87			"threat": zod.number().optional(),
88			}).optional(),
89			"role_commands": zod.record(
90			snowflake,
91			zod.array(zod.string().min(1))
92			).optional().default({}),
93			"autoclear": zod.object({
94			"enabled": zod.boolean().optional(),
95			"channels": zod.array(snowflake).optional().default(config.autoclear.channels)
96			}).optional(),
97			"verification": zod.object({
98			"enabled": zod.boolean().optional(),
99			"role": snowflake.optional()
100			}).optional(),
101			"welcomer": zod.object({
102			"enabled": zod.boolean().optional(),
103			"channel": snowflake.optional(),
104			"message": zod.string().min(1).or(zod.null()).optional(),
105			"randomize": zod.boolean().optional()
106			}).optional(),
107			"cooldown": zod.object({
108			"enabled": zod.boolean().optional(),
109			"global": zod.any().optional(),
110			"cmds": zod.object({}).optional()
111			}).optional(),
112			"profile_filter": zod.object({
113			"enabled": zod.boolean().optional(),
114			"inherit_from_words": zod.boolean().optional(),
115			"inherit_from_tokens": zod.boolean().optional(),
116			"inherit_from_regex": zod.boolean().optional(),
117			"blocked_words": zod.array(zod.string()).optional().default(config.profile_filter.blocked_words),
118			"blocked_tokens": zod.array(zod.string()).optional().default(config.profile_filter.blocked_tokens),
119			"blocked_regex_patterns": zod.array(zod.string()).optional().default(config.profile_filter.blocked_regex_patterns),
120			"components": zod.object({
121			"tag": zod.boolean().optional(),
122			"nickname": zod.boolean().optional(),
123			"about": zod.boolean().optional(),
124			"status": zod.boolean().optional()
125			}).optional(),
126			"actions": zod.object({
127			"tag": zod.string().optional(),
128			"nickname": zod.string().optional(),
129			"about": zod.string().optional(),
130			"status": zod.string().optional()
131			}).optional()
132			}).optional(),
133			"starboard": zod.object({
134			"enabled": zod.boolean().optional(),
135			"reactions": zod.number().int().optional(),
136			"channel": snowflake.optional()
137			}).optional(),
138			"autorole": zod.object({
139			"enabled": zod.boolean().optional(),
140			"roles": zod.array(snowflake).optional().default(config.autorole.roles)
141			}).optional(),
142			"spam_filter": zod.object({
143			"enabled": zod.boolean().optional(),
144			"limit": zod.number().int().optional(),
145			"time": zod.number().optional(),
146			"diff": zod.number().optional(),
147			"exclude": zod.array(snowflake).optional().default(config.spam_filter.exclude),
148			"samelimit": zod.number().int().optional(),
149			"unmute_in": zod.number().optional()
150			}).optional(),
151			"raid": zod.object({
152			"enabled": zod.boolean().optional(),
153			"max_joins": zod.number().int().optional(),
154			"time": zod.number().optional(),
155			"channels": zod.array(snowflake).optional().default(config.raid.channels),
156			"exclude": zod.boolean().optional()
157			}).optional(),
158			"global_commands": zod.array(zod.string()).optional().default(config.global_commands),
159			"filters": zod.object({
160			"ignore_staff": zod.boolean().optional(),
161			"chars_repeated": zod.number().int().optional(),
162			"words_repeated": zod.number().int().optional(),
163			"words": zod.array(zod.string()).optional().default(config.filters.words),
164			"tokens": zod.array(zod.string()).optional().default(config.filters.tokens),
165			"invite_message": zod.string().optional(),
166			"words_excluded": zod.array(snowflake).optional().default(config.filters.words_excluded),
167			"domain_excluded": zod.array(snowflake).optional().default(config.filters.domain_excluded),
168			"invite_excluded": zod.array(snowflake).optional().default(config.filters.invite_excluded),
169			"words_enabled": zod.boolean().optional(),
170			"invite_enabled": zod.boolean().optional(),
171			"domain_enabled": zod.boolean().optional(),
172			"regex": zod.boolean().optional(),
173			"file_mimes_excluded": zod.array(zod.string()).optional().default(config.filters.file_mimes_excluded),
174			"file_types_excluded": zod.array(zod.string()).optional().default(config.filters.file_types_excluded),
175			"domains": zod.array(zod.string()).optional().default(config.filters.domains),
176			"regex_patterns": zod.array(zod.string()).optional().default(config.filters.regex_patterns),
177			"rickrolls_enabled": zod.boolean().optional(),
178			"pings": zod.number().int().optional()
179			}).optional()
180			});
181			}
182			}