src/client/Client.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 { Client, ClientOptions, Collection, Interaction, Message } from 'discord.js';
import BaseEvent from '../utils/structures/BaseEvent';
import BaseCommand from '../utils/structures/BaseCommand';
import { Config } from './Config';
import Database from './Database';
import path from 'path';
import { appendFile } from "fs/promises";
import Logger from '../automod/Logger';
import SnippetManager from '../services/SnippetManager';
import AFKEngine from '../services/AFKEngine';
import Auth from '../services/Auth';
import MessageFilter from '../automod/MessageFilter';
import AntiRaid from '../automod/AntiRaid';
import Starboard from '../services/Starboard';
import Server from '../api/Server';
import StartupManager from '../services/StartupManager';
import AutoClear from '../automod/AutoClear';
import RandomStatus from '../services/RandomStatus';
import DebugLogger from '../services/DebugLogger';
import BaseCLICommand from '../utils/structures/BaseCLICommand';
import discordModals from 'discord-modals';
import SpamFilter from '../automod/SpamFilter';
import Verification from '../services/Verification';
import Welcomer from '../services/Welcomer';
import Antijoin from '../automod/Antijoin';
import Automute from '../automod/Automute';
import ServiceManager from './ServiceManager';
import ChannelLockManager from '../services/ChannelLockManager';
import Cooldown from '../services/Cooldown';
import ProfileFilter from '../automod/ProfileFilter';
import QueueManager from '../services/QueueManager';

export default class DiscordClient extends Client {
    private _commands = new Collection<string, BaseCommand>();
    private _cliCommands = new Collection<string, BaseCLICommand>();
    private _events = new Collection<string, BaseEvent>();

    rootdir: string;
    msg: Message | Interaction | null = null;

    config: Config;
    db: Database;
    server: Server;
    serviceManager: ServiceManager;

    logger: Logger = {} as Logger;
    snippetManager: SnippetManager = {} as SnippetManager;
    afkEngine: AFKEngine = {} as AFKEngine;
    auth: Auth = {} as Auth;
    spamFilter: SpamFilter = {} as SpamFilter;
    messageFilter: MessageFilter = {} as MessageFilter;
    antiraid: AntiRaid = {} as AntiRaid;
    starboard: Starboard = {} as Starboard;
    startupManager: StartupManager = {} as StartupManager;
    autoClear: AutoClear = {} as AutoClear;
    randomStatus: RandomStatus = {} as RandomStatus;
    debugLogger: DebugLogger = {} as DebugLogger;
    verification: Verification = {} as Verification;
    welcomer: Welcomer = {} as Welcomer;
    antijoin: Antijoin = {} as Antijoin;
    automute: Automute = {} as Automute;
    channelLock: ChannelLockManager = {} as ChannelLockManager;
    cooldown: Cooldown = {} as Cooldown;
    profileFilter: ProfileFilter = {} as ProfileFilter;
    queueManager: QueueManager = {} as QueueManager;

    aliases = {
        automod: path.resolve(__dirname, '..', 'automod'),
        services: path.resolve(__dirname, '..', 'services'),
    };

    services = {
        "@services/DebugLogger": "debugLogger",
        "@automod/Logger": "logger",
        "@services/SnippetManager": "snippetManager",
        "@services/AFKEngine": "afkEngine",
        "@services/Auth": "auth",
        "@automod/SpamFilter": "spamFilter",
        "@automod/MessageFilter": "messageFilter",
        "@automod/AntiRaid": "antiraid",
        "@services/Starboard": "starboard",
        "@services/StartupManager": "startupManager",
        "@automod/AutoClear": "autoClear",
        "@services/RandomStatus": "randomStatus",
        "@services/Verification": "verification",
        "@services/Welcomer": "welcomer",
        "@services/ChannelLockManager": "channelLock",
        "@automod/Antijoin": "antijoin",
        "@automod/Automute": "automute",
        "@services/Cooldown": "cooldown",
        "@automod/ProfileFilter": "profileFilter",
        "@services/QueueManager": "queueManager",
    };

    static client: DiscordClient;

    constructor(options: ClientOptions, rootdir: string = __dirname) {
        super({
            ws: {
                properties: {
                    browser: "Discord iOS"
                }
            },
            ...options
        });

        process.on('uncaughtException', (error, origin) => {
            console.log('Uncaught', error);
            this.handleCrash(error, origin).then(() => process.exit(-1)).catch(err => {
                console.log(err);
                process.exit(-1);
            });
        });
        
        console.log('init');        

        this.rootdir = rootdir;
        
        DiscordClient.client = this;

        this.config = new Config(this);
        this.db = new Database(this);
        this.serviceManager = new ServiceManager(this, this.aliases);
        this.serviceManager.load(this.services);

        // this.logger = new Logger(this);
        // this.snippetManager = new SnippetManager(this);
        // this.afkEngine = new AFKEngine(this);
        // this.auth = new Auth(this);
        // this.spamFilter = new SpamFilter(this);
        // this.messageFilter = new MessageFilter(this);
        // this.antiraid = new AntiRaid(this);
        // this.starboard = new Starboard(this);
        // this.cooldown = new Cooldown(this);
        // this.startupManager = new StartupManager(this);
        // this.autoClear = new AutoClear(this);
        // this.randomStatus = new RandomStatus(this);
        // this.debugLogger = new DebugLogger(this);
        // this.verification = new Verification(this);
        // this.welcomer = new Welcomer(this);
        // this.antijoin = new Antijoin(this);
        // this.automute = new Automute(this);

        this.server = new Server(this);
        
        discordModals(this);        
    }

    get commands(): Collection<string, BaseCommand> {
        return this._commands; 
    }

    get cliCommands(): Collection<string, BaseCLICommand> {
        return this._cliCommands; 
    }

    get events(): Collection<string, BaseEvent> {
        return this._events; 
    }

    setMessage(msg: Message | Interaction) {
        this.msg = msg;
    }

    async handleCrash(error: Error, origin: NodeJS.UncaughtExceptionOrigin) {
        console.log('here');
        await appendFile(path.join(process.env.SUDO_PREFIX ?? (__dirname + "/../../"), "logs", "error.log"), `Uncaught ${error.name}: ${error.message}\n+ ${error.stack}`);
        await this.debugLogger.logToHomeServer(`Uncaught ${error.name}: ${error.message}\n${error.stack}`);
    }
}
1	rakin	393	/**
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	rakin	51	import { Client, ClientOptions, Collection, Interaction, Message } from 'discord.js';
21			import BaseEvent from '../utils/structures/BaseEvent';
22			import BaseCommand from '../utils/structures/BaseCommand';
23			import { Config } from './Config';
24			import Database from './Database';
25			import path from 'path';
26	rakin	318	import { appendFile } from "fs/promises";
27	rakin	51	import Logger from '../automod/Logger';
28			import SnippetManager from '../services/SnippetManager';
29			import AFKEngine from '../services/AFKEngine';
30			import Auth from '../services/Auth';
31			import MessageFilter from '../automod/MessageFilter';
32			import AntiRaid from '../automod/AntiRaid';
33			import Starboard from '../services/Starboard';
34			import Server from '../api/Server';
35	rakin	58	import StartupManager from '../services/StartupManager';
36	rakin	65	import AutoClear from '../automod/AutoClear';
37	rakin	71	import RandomStatus from '../services/RandomStatus';
38	rakin	77	import DebugLogger from '../services/DebugLogger';
39			import BaseCLICommand from '../utils/structures/BaseCLICommand';
40	rakin	81	import discordModals from 'discord-modals';
41	rakin	85	import SpamFilter from '../automod/SpamFilter';
42	rakin	127	import Verification from '../services/Verification';
43	rakin	199	import Welcomer from '../services/Welcomer';
44	rakin	225	import Antijoin from '../automod/Antijoin';
45	rakin	227	import Automute from '../automod/Automute';
46	rakin	234	import ServiceManager from './ServiceManager';
47	rakin	241	import ChannelLockManager from '../services/ChannelLockManager';
48	rakin	397	import Cooldown from '../services/Cooldown';
49	rakin	400	import ProfileFilter from '../automod/ProfileFilter';
50	rakin	421	import QueueManager from '../services/QueueManager';
51	rakin	51
52			export default class DiscordClient extends Client {
53			private _commands = new Collection<string, BaseCommand>();
54	rakin	77	private _cliCommands = new Collection<string, BaseCLICommand>();
55	rakin	51	private _events = new Collection<string, BaseEvent>();
56
57			rootdir: string;
58			msg: Message \| Interaction \| null = null;
59
60			config: Config;
61			db: Database;
62			server: Server;
63	rakin	234	serviceManager: ServiceManager;
64	rakin	51
65	rakin	234	logger: Logger = {} as Logger;
66			snippetManager: SnippetManager = {} as SnippetManager;
67			afkEngine: AFKEngine = {} as AFKEngine;
68			auth: Auth = {} as Auth;
69			spamFilter: SpamFilter = {} as SpamFilter;
70			messageFilter: MessageFilter = {} as MessageFilter;
71			antiraid: AntiRaid = {} as AntiRaid;
72			starboard: Starboard = {} as Starboard;
73			startupManager: StartupManager = {} as StartupManager;
74			autoClear: AutoClear = {} as AutoClear;
75			randomStatus: RandomStatus = {} as RandomStatus;
76			debugLogger: DebugLogger = {} as DebugLogger;
77			verification: Verification = {} as Verification;
78			welcomer: Welcomer = {} as Welcomer;
79			antijoin: Antijoin = {} as Antijoin;
80			automute: Automute = {} as Automute;
81	rakin	241	channelLock: ChannelLockManager = {} as ChannelLockManager;
82	rakin	397	cooldown: Cooldown = {} as Cooldown;
83	rakin	400	profileFilter: ProfileFilter = {} as ProfileFilter;
84	rakin	421	queueManager: QueueManager = {} as QueueManager;
85	rakin	234
86			aliases = {
87			automod: path.resolve(__dirname, '..', 'automod'),
88			services: path.resolve(__dirname, '..', 'services'),
89			};
90
91			services = {
92	rakin	237	"@services/DebugLogger": "debugLogger",
93	rakin	234	"@automod/Logger": "logger",
94			"@services/SnippetManager": "snippetManager",
95			"@services/AFKEngine": "afkEngine",
96			"@services/Auth": "auth",
97			"@automod/SpamFilter": "spamFilter",
98			"@automod/MessageFilter": "messageFilter",
99			"@automod/AntiRaid": "antiraid",
100			"@services/Starboard": "starboard",
101			"@services/StartupManager": "startupManager",
102			"@automod/AutoClear": "autoClear",
103			"@services/RandomStatus": "randomStatus",
104			"@services/Verification": "verification",
105			"@services/Welcomer": "welcomer",
106	rakin	241	"@services/ChannelLockManager": "channelLock",
107	rakin	234	"@automod/Antijoin": "antijoin",
108			"@automod/Automute": "automute",
109	rakin	397	"@services/Cooldown": "cooldown",
110	rakin	400	"@automod/ProfileFilter": "profileFilter",
111	rakin	421	"@services/QueueManager": "queueManager",
112	rakin	234	};
113
114	rakin	51	static client: DiscordClient;
115
116			constructor(options: ClientOptions, rootdir: string = __dirname) {
117			super({
118			ws: {
119	rakin	85	properties: {
120	rakin	225	browser: "Discord iOS"
121	rakin	51	}
122			},
123			...options
124			});
125	rakin	237
126			process.on('uncaughtException', (error, origin) => {
127			console.log('Uncaught', error);
128			this.handleCrash(error, origin).then(() => process.exit(-1)).catch(err => {
129			console.log(err);
130			process.exit(-1);
131			});
132			});
133	rakin	85
134			console.log('init');
135	rakin	51
136			this.rootdir = rootdir;
137	rakin	85
138			DiscordClient.client = this;
139	rakin	51
140			this.config = new Config(this);
141	rakin	346	this.db = new Database(this);
142	rakin	234	this.serviceManager = new ServiceManager(this, this.aliases);
143			this.serviceManager.load(this.services);
144
145			// this.logger = new Logger(this);
146			// this.snippetManager = new SnippetManager(this);
147			// this.afkEngine = new AFKEngine(this);
148			// this.auth = new Auth(this);
149			// this.spamFilter = new SpamFilter(this);
150			// this.messageFilter = new MessageFilter(this);
151			// this.antiraid = new AntiRaid(this);
152			// this.starboard = new Starboard(this);
153			// this.cooldown = new Cooldown(this);
154			// this.startupManager = new StartupManager(this);
155			// this.autoClear = new AutoClear(this);
156			// this.randomStatus = new RandomStatus(this);
157			// this.debugLogger = new DebugLogger(this);
158			// this.verification = new Verification(this);
159			// this.welcomer = new Welcomer(this);
160			// this.antijoin = new Antijoin(this);
161			// this.automute = new Automute(this);
162
163	rakin	51	this.server = new Server(this);
164
165	rakin	85	discordModals(this);
166	rakin	51	}
167
168			get commands(): Collection<string, BaseCommand> {
169			return this._commands;
170			}
171
172	rakin	77	get cliCommands(): Collection<string, BaseCLICommand> {
173			return this._cliCommands;
174			}
175
176	rakin	51	get events(): Collection<string, BaseEvent> {
177			return this._events;
178			}
179
180			setMessage(msg: Message \| Interaction) {
181			this.msg = msg;
182			}
183	rakin	237
184			async handleCrash(error: Error, origin: NodeJS.UncaughtExceptionOrigin) {
185	rakin	241	console.log('here');
186	rakin	372	await appendFile(path.join(process.env.SUDO_PREFIX ?? (__dirname + "/../../"), "logs", "error.log"), `Uncaught ${error.name}: ${error.message}\n+ ${error.stack}`);
187	rakin	237	await this.debugLogger.logToHomeServer(`Uncaught ${error.name}: ${error.message}\n${error.stack}`);
188			}
189	rakin	51	}