trunk/src/App.js

const { Client, Intents } = require("discord.js");
const { config } = require("dotenv");
const CommandManager = require("./CommandManager");
const path = require("path");
const fs = require("fs");
const Config = require("./Config");
const Database = require("./Database");
const Logger = require("./Logger");
const SpamFilter = require("./SpamFilter");
const server = require("./server");
const AntiRaid = require("./AntiRaid");
const MessageFilter = require("./MessageFilter");
const { random } = require("../commands/pixabay");
const AFKEngine = require("./AFKEngine");
const Starboard = require("./Starboard");

class App {
    constructor(rootdir) {
        global.app = App.app = this;
        this.rootdir = rootdir;
        this.loadConfig();
        this.env = process.env;

        this.client = new Client({
            partials: ["CHANNEL"],
            intents: [
                Intents.FLAGS.GUILDS,
                Intents.FLAGS.GUILD_MESSAGES,
                Intents.FLAGS.DIRECT_MESSAGES, 
                Intents.FLAGS.DIRECT_MESSAGE_TYPING,
                Intents.FLAGS.GUILD_PRESENCES,
                Intents.FLAGS.GUILD_MEMBERS,
                Intents.FLAGS.GUILD_BANS,
                Intents.FLAGS.GUILD_MESSAGE_REACTIONS,
                Intents.FLAGS.GUILD_EMOJIS_AND_STICKERS,
            ]
        });

        this.config = new Config();
        this.db = new Database(path.resolve(__dirname, '..', 'database.db'));
        this.commandManager = new CommandManager(path.resolve(__dirname, rootdir, "commands"));
        this.logger = new Logger();
        this.spamFilter = new SpamFilter();
        this.antiRaid = new AntiRaid();
        this.messageFilter = new MessageFilter();
        this.afkEngine = new AFKEngine();
        this.starboard = new Starboard();
        this.boot();
    }

    boot() {
        const events = {
            MESSAGE_REACTION_ADD: 'messageReactionAdd',
        };
        
        this.on('ready', () => {
            console.log("Logged in as " + this.client.user.tag);

            this.client.user.setStatus(random(['dnd', 'idle']));
            this.client.user.setActivity("over the server", { type: "WATCHING" });

            server();
        });

        this.on('raw', async event => {
            if (!events.hasOwnProperty(event.t))
                return;
        
            const { d: data } = event;
            const user = this.client.users.cache.find(i => i.id === data.user_id);
            const channel = this.client.channels.cache.find(i => i.id === data.channel_id) || await user.createDM();
        
            if (channel.messages.cache.has(data.message_id)) 
                return;
        
            const message = await channel.messages.fetch(data.message_id);
        
            const emojiKey = (data.emoji.id) ? `${data.emoji.name}:${data.emoji.id}` : data.emoji.name;
            const reaction = message.reactions.cache.get(emojiKey);
        
            this.client.emit(events[event.t], reaction, user);
        });

        this.on('messageCreate', async (message) => {
            if (message.author.bot || !message.guild || message.channel.type == 'dm') {
                return;
            }

            await (this.msg = message);
            
            await this.spamFilter.start(message);
            await this.messageFilter.start(message, this.commandManager);

            await this.commandManager.setMessage(message);

            const valid = await this.commandManager.valid();
            const has = await this.commandManager.has();
            const snippet = await this.commandManager.snippet();
            const allowed = await this.commandManager.verify();
            
            if (valid && has && allowed) {
                await this.exec();
            }
            else if (valid && snippet !== undefined) {
                await message.channel.send({
                    content: snippet.content,
                    files: snippet.files.map(f => {
                        return {
                            attachment: path.resolve(__dirname, '..', 'storage', f)
                        }
                    })
                });
            }
            else if (valid && !has) {
                await this.commandManager.notFound();
            }
            else if (valid && has && !allowed) {
                await this.commandManager.notAllowed();
            }
            else if(!valid) {
                await this.afkEngine.start(message);
            }
        });

        this.on("messageUpdate", async (oldMessage, newMessage) => {
            if (oldMessage.author.bot || !oldMessage.guild || oldMessage.channel.type == 'dm' || oldMessage.content === newMessage.content)
                return;

            let msg = await this.msg;
            await (this.msg = newMessage);
        
            await this.spamFilter.basic(newMessage);
            await this.messageFilter.start(newMessage, this.commandManager);

            await this.logger.logEdit(oldMessage, newMessage);
            await (this.msg = msg);
        });

        this.on("messageReactionAdd", async (reaction, message) => {
            console.log('inside');

            if (!reaction || !reaction.message) {
                return;
            }
            
            await (this.msg = reaction.message);
            await this.starboard.handle(reaction, message);
        });

        this.on('guildBanAdd', async (ban) => {
            console.log('test');
            await this.logger.logBanned(ban);
        });

        this.on('guildBanRemove', async (ban) => {
            console.log('test');
            await this.logger.logUnbanned(ban);
        });

        this.on("messageDelete", async (message) => {
            if (message.author.bot || !message.guild || message.channel.type == 'dm')
                return;

            await this.logger.logDelete(message);
        });

        this.on("guildCreate", guild => {
            console.log("Joined a new guild: " + guild.name);

            this.config.props[guild.id] = {
                prefix: "-",
                debug: false,
            };

            this.config.write();
        })
        
        this.on("guildDelete", guild => {
            console.log("Left a guild: " + guild.name);
            delete this.config.props[guild.id];
            this.config.write();
        })

        this.on('guildMemberAdd', async (member) => {
            console.log('Joined');
            await this.antiRaid.start(member);
            await this.logger.logJoined(member);
        });

        this.on('guildMemberRemove', async (member) => {
            await this.logger.logLeft(member);
        });
    }

    loadConfig() {
        if (fs.existsSync(path.join(__dirname, this.rootdir, '.env'))) {
            console.log("Loading .env file");
            config();
        }
    }

    on(event, handler) {
        this.client.on(event, handler);
    }

    exec() {
        this.commandManager.exec();
    }

    run() {
        this.client.login(process.env.TOKEN);
    }

    tempFileCreate(name) {
        const fullname = path.join(__dirname, '..', 'tmp', name);
        const file = fs.createWriteStream(fullname);

        return {
            name: fullname,
            file
        };
    }
}

module.exports = App;
1	const { Client, Intents } = require("discord.js");
2	const { config } = require("dotenv");
3	const CommandManager = require("./CommandManager");
4	const path = require("path");
5	const fs = require("fs");
6	const Config = require("./Config");
7	const Database = require("./Database");
8	const Logger = require("./Logger");
9	const SpamFilter = require("./SpamFilter");
10	const server = require("./server");
11	const AntiRaid = require("./AntiRaid");
12	const MessageFilter = require("./MessageFilter");
13	const { random } = require("../commands/pixabay");
14	const AFKEngine = require("./AFKEngine");
15	const Starboard = require("./Starboard");
16
17	class App {
18	constructor(rootdir) {
19	global.app = App.app = this;
20	this.rootdir = rootdir;
21	this.loadConfig();
22	this.env = process.env;
23
24	this.client = new Client({
25	partials: ["CHANNEL"],
26	intents: [
27	Intents.FLAGS.GUILDS,
28	Intents.FLAGS.GUILD_MESSAGES,
29	Intents.FLAGS.DIRECT_MESSAGES,
30	Intents.FLAGS.DIRECT_MESSAGE_TYPING,
31	Intents.FLAGS.GUILD_PRESENCES,
32	Intents.FLAGS.GUILD_MEMBERS,
33	Intents.FLAGS.GUILD_BANS,
34	Intents.FLAGS.GUILD_MESSAGE_REACTIONS,
35	Intents.FLAGS.GUILD_EMOJIS_AND_STICKERS,
36	]
37	});
38
39	this.config = new Config();
40	this.db = new Database(path.resolve(__dirname, '..', 'database.db'));
41	this.commandManager = new CommandManager(path.resolve(__dirname, rootdir, "commands"));
42	this.logger = new Logger();
43	this.spamFilter = new SpamFilter();
44	this.antiRaid = new AntiRaid();
45	this.messageFilter = new MessageFilter();
46	this.afkEngine = new AFKEngine();
47	this.starboard = new Starboard();
48	this.boot();
49	}
50
51	boot() {
52	const events = {
53	MESSAGE_REACTION_ADD: 'messageReactionAdd',
54	};
55
56	this.on('ready', () => {
57	console.log("Logged in as " + this.client.user.tag);
58
59	this.client.user.setStatus(random(['dnd', 'idle']));
60	this.client.user.setActivity("over the server", { type: "WATCHING" });
61
62	server();
63	});
64
65	this.on('raw', async event => {
66	if (!events.hasOwnProperty(event.t))
67	return;
68
69	const { d: data } = event;
70	const user = this.client.users.cache.find(i => i.id === data.user_id);
71	const channel = this.client.channels.cache.find(i => i.id === data.channel_id) \|\| await user.createDM();
72
73	if (channel.messages.cache.has(data.message_id))
74	return;
75
76	const message = await channel.messages.fetch(data.message_id);
77
78	const emojiKey = (data.emoji.id) ? `${data.emoji.name}:${data.emoji.id}` : data.emoji.name;
79	const reaction = message.reactions.cache.get(emojiKey);
80
81	this.client.emit(events[event.t], reaction, user);
82	});
83
84	this.on('messageCreate', async (message) => {
85	if (message.author.bot \|\| !message.guild \|\| message.channel.type == 'dm') {
86	return;
87	}
88
89	await (this.msg = message);
90
91	await this.spamFilter.start(message);
92	await this.messageFilter.start(message, this.commandManager);
93
94	await this.commandManager.setMessage(message);
95
96	const valid = await this.commandManager.valid();
97	const has = await this.commandManager.has();
98	const snippet = await this.commandManager.snippet();
99	const allowed = await this.commandManager.verify();
100
101	if (valid && has && allowed) {
102	await this.exec();
103	}
104	else if (valid && snippet !== undefined) {
105	await message.channel.send({
106	content: snippet.content,
107	files: snippet.files.map(f => {
108	return {
109	attachment: path.resolve(__dirname, '..', 'storage', f)
110	}
111	})
112	});
113	}
114	else if (valid && !has) {
115	await this.commandManager.notFound();
116	}
117	else if (valid && has && !allowed) {
118	await this.commandManager.notAllowed();
119	}
120	else if(!valid) {
121	await this.afkEngine.start(message);
122	}
123	});
124
125	this.on("messageUpdate", async (oldMessage, newMessage) => {
126	if (oldMessage.author.bot \|\| !oldMessage.guild \|\| oldMessage.channel.type == 'dm' \|\| oldMessage.content === newMessage.content)
127	return;
128
129	let msg = await this.msg;
130	await (this.msg = newMessage);
131
132	await this.spamFilter.basic(newMessage);
133	await this.messageFilter.start(newMessage, this.commandManager);
134
135	await this.logger.logEdit(oldMessage, newMessage);
136	await (this.msg = msg);
137	});
138
139	this.on("messageReactionAdd", async (reaction, message) => {
140	console.log('inside');
141
142	if (!reaction \|\| !reaction.message) {
143	return;
144	}
145
146	await (this.msg = reaction.message);
147	await this.starboard.handle(reaction, message);
148	});
149
150	this.on('guildBanAdd', async (ban) => {
151	console.log('test');
152	await this.logger.logBanned(ban);
153	});
154
155	this.on('guildBanRemove', async (ban) => {
156	console.log('test');
157	await this.logger.logUnbanned(ban);
158	});
159
160	this.on("messageDelete", async (message) => {
161	if (message.author.bot \|\| !message.guild \|\| message.channel.type == 'dm')
162	return;
163
164	await this.logger.logDelete(message);
165	});
166
167	this.on("guildCreate", guild => {
168	console.log("Joined a new guild: " + guild.name);
169
170	this.config.props[guild.id] = {
171	prefix: "-",
172	debug: false,
173	};
174
175	this.config.write();
176	})
177
178	this.on("guildDelete", guild => {
179	console.log("Left a guild: " + guild.name);
180	delete this.config.props[guild.id];
181	this.config.write();
182	})
183
184	this.on('guildMemberAdd', async (member) => {
185	console.log('Joined');
186	await this.antiRaid.start(member);
187	await this.logger.logJoined(member);
188	});
189
190	this.on('guildMemberRemove', async (member) => {
191	await this.logger.logLeft(member);
192	});
193	}
194
195	loadConfig() {
196	if (fs.existsSync(path.join(__dirname, this.rootdir, '.env'))) {
197	console.log("Loading .env file");
198	config();
199	}
200	}
201
202	on(event, handler) {
203	this.client.on(event, handler);
204	}
205
206	exec() {
207	this.commandManager.exec();
208	}
209
210	run() {
211	this.client.login(process.env.TOKEN);
212	}
213
214	tempFileCreate(name) {
215	const fullname = path.join(__dirname, '..', 'tmp', name);
216	const file = fs.createWriteStream(fullname);
217
218	return {
219	name: fullname,
220	file
221	};
222	}
223	}
224
225	module.exports = App;