src/services/AFKEngine.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 { formatDistanceToNowStrict } from "date-fns";
import { CommandInteraction, GuildMember, Message, Util } from "discord.js";
import DiscordClient from "../client/Client";
import MessageEmbed from "../client/MessageEmbed";
import AFK, { IAFK } from "../models/AFK";
import Service from "../utils/structures/Service";

export interface MentionSchema {
    date: number;
    user: string; 
}

export default class AFKEngine extends Service {
    list: IAFK[] = [];

    constructor(client: DiscordClient) {
        super(client);
        AFK.find().then(models => this.list = models).catch(console.error);
    }

    findUsers(ids: string[], guild: string) {
        return this.list.filter(afk => ids.includes(afk.user) && afk.guild_id === guild);
    }
 
    async removeUser(id: string, guild: string) {
        let index = 0;

        for await (const afk of this.list) {
            if (afk.user === id && afk.guild_id === guild) {
                await afk.delete();
                this.list.splice(index, 1);
            }

            index++;
        }
    }
    
    async toggle(message: Message | CommandInteraction, enable: boolean = false, status?: string) {
        const afk = this.findUsers([message.member!.user.id], message.guild!.id);

        if (afk.length > 0) {
            const mentions = afk[0].get("mentions")! as Array<MentionSchema>;
            let count = 0, text = '';

            for await (const m of mentions) {
                if (count >= 3) {
                    break;
                }

                let member: GuildMember | undefined;

                try {
                    member = await message.guild!.members.fetch(m.user);

                    if (!member) {
                        throw new Error("user not found");
                    }
                }
                catch (e) {
                    console.log(e);
                    continue;                    
                }
                
                text += `\nFrom ${member.toString()}, ${formatDistanceToNowStrict(m.date, { addSuffix: true })}`;
                count++;
            }

            await this.client.afkEngine.removeUser(message.member!.user.id, message.guild!.id);

            await message.reply({
                embeds: [
                    new MessageEmbed({
                        description: `You're no longer AFK. You had ${mentions.length ?? 0} mentions in the server(s) where SudoBot is in.${mentions.length > 0 ? `\n\n**Mentions**:${text}` : ''}`,
                    })
                ]
            });
        }
        else if (enable) {
            this.client.afkEngine.list.push(await AFK.create({
                user: message.member!.user.id,
                guild_id: message.guild!.id,
                mentions: [],
                reason: status ?? undefined,
                createdAt: new Date()
            }));

            await message.reply({
                embeds: [
                    new MessageEmbed({
                        description: `You're AFK now${status ? `, for reason: **${Util.escapeMarkdown(status)}**` : ''}.`
                    })
                ]
            });
        }
    }

    async start(msg: Message) {
        if (msg.author.bot)
            return;

        const selfAFK = this.findUsers([msg.author.id], msg.guild!.id);

        if (selfAFK.length > 0) {
            this.toggle(msg, false);
        }
        
        const mention = msg.mentions.members?.first();

        if (mention) {
            const afkRecords: Array<IAFK> = this.findUsers([...msg.mentions.members!.keys()].slice(0, 3), msg.guild!.id).filter(afk => afk.user !== msg.author.id);

            if (!afkRecords || afkRecords.length < 1) {
                return;
            }

            for (const record of afkRecords) {
                const mentions = record.mentions as MentionSchema[];

                mentions.push({
                    date: Date.now(),
                    user: msg.author.id
                });

                record.set("mentions", mentions).save();
            }

            let text = `The following users are AFK right now:`;

            if (afkRecords.length > 1) {
                for await (const afkRecord of afkRecords) {
                    text += `\n**${msg.mentions.members!.get(afkRecord.get("user") as string)!.user.tag}**${afkRecord.get("reason") as (null | string) ? `\n**Reason**: ${Util.escapeMarkdown(afkRecord.get("reason") as string)}` : ""}`;
                }
            }
            else {
                text = `${msg.mentions.members!.get(afkRecords[0].get("user") as string)!.user.tag} is AFK right now${afkRecords[0].get("reason") as (null | string) ? `, for reason **${Util.escapeMarkdown(afkRecords[0].get("reason") as string)}**` : ""}.`;
            }

            await msg.reply({
                embeds: [
                    new MessageEmbed({
                        description: text
                    })
                ]
            });
        }
    }
};
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	304	import { formatDistanceToNowStrict } from "date-fns";
21	rakin	344	import { CommandInteraction, GuildMember, Message, Util } from "discord.js";
22	rakin	51	import DiscordClient from "../client/Client";
23			import MessageEmbed from "../client/MessageEmbed";
24	rakin	327	import AFK, { IAFK } from "../models/AFK";
25	rakin	226	import Service from "../utils/structures/Service";
26	rakin	51
27	rakin	304	export interface MentionSchema {
28			date: number;
29			user: string;
30			}
31
32	rakin	226	export default class AFKEngine extends Service {
33	rakin	327	list: IAFK[] = [];
34	rakin	304
35			constructor(client: DiscordClient) {
36			super(client);
37	rakin	327	AFK.find().then(models => this.list = models).catch(console.error);
38	rakin	304	}
39
40			findUsers(ids: string[], guild: string) {
41	rakin	327	return this.list.filter(afk => ids.includes(afk.user) && afk.guild_id === guild);
42	rakin	304	}
43
44			async removeUser(id: string, guild: string) {
45			let index = 0;
46
47			for await (const afk of this.list) {
48	rakin	327	if (afk.user === id && afk.guild_id === guild) {
49			await afk.delete();
50	rakin	304	this.list.splice(index, 1);
51			}
52
53			index++;
54			}
55			}
56
57			async toggle(message: Message \| CommandInteraction, enable: boolean = false, status?: string) {
58			const afk = this.findUsers([message.member!.user.id], message.guild!.id);
59
60			if (afk.length > 0) {
61			const mentions = afk[0].get("mentions")! as Array<MentionSchema>;
62			let count = 0, text = '';
63
64			for await (const m of mentions) {
65			if (count >= 3) {
66			break;
67	rakin	51	}
68
69	rakin	304	let member: GuildMember \| undefined;
70	rakin	51
71	rakin	304	try {
72			member = await message.guild!.members.fetch(m.user);
73
74			if (!member) {
75			throw new Error("user not found");
76			}
77			}
78			catch (e) {
79			console.log(e);
80			continue;
81			}
82
83			text += `\nFrom ${member.toString()}, ${formatDistanceToNowStrict(m.date, { addSuffix: true })}`;
84			count++;
85	rakin	51	}
86
87	rakin	304	await this.client.afkEngine.removeUser(message.member!.user.id, message.guild!.id);
88
89			await message.reply({
90			embeds: [
91			new MessageEmbed({
92			description: `You're no longer AFK. You had ${mentions.length ?? 0} mentions in the server(s) where SudoBot is in.${mentions.length > 0 ? `\n\nMentions:${text}` : ''}`,
93			})
94			]
95			});
96			}
97			else if (enable) {
98			this.client.afkEngine.list.push(await AFK.create({
99			user: message.member!.user.id,
100			guild_id: message.guild!.id,
101			mentions: [],
102	rakin	327	reason: status ?? undefined,
103			createdAt: new Date()
104	rakin	304	}));
105
106			await message.reply({
107			embeds: [
108			new MessageEmbed({
109			description: `You're AFK now${status ? `, for reason: ${Util.escapeMarkdown(status)}` : ''}.`
110			})
111			]
112			});
113			}
114	rakin	51	}
115
116	rakin	304	async start(msg: Message) {
117	rakin	69	if (msg.author.bot)
118			return;
119	rakin	304
120			const selfAFK = this.findUsers([msg.author.id], msg.guild!.id);
121
122			if (selfAFK.length > 0) {
123			this.toggle(msg, false);
124			}
125	rakin	69
126	rakin	304	const mention = msg.mentions.members?.first();
127	rakin	51
128			if (mention) {
129	rakin	327	const afkRecords: Array<IAFK> = this.findUsers([...msg.mentions.members!.keys()].slice(0, 3), msg.guild!.id).filter(afk => afk.user !== msg.author.id);
130	rakin	304
131			if (!afkRecords \|\| afkRecords.length < 1) {
132			return;
133			}
134
135			for (const record of afkRecords) {
136	rakin	327	const mentions = record.mentions as MentionSchema[];
137	rakin	304
138			mentions.push({
139			date: Date.now(),
140			user: msg.author.id
141	rakin	51	});
142	rakin	304
143			record.set("mentions", mentions).save();
144			}
145
146			let text = `The following users are AFK right now:`;
147
148			if (afkRecords.length > 1) {
149			for await (const afkRecord of afkRecords) {
150			text += `\n${msg.mentions.members!.get(afkRecord.get("user") as string)!.user.tag}${afkRecord.get("reason") as (null \| string) ? `\nReason: ${Util.escapeMarkdown(afkRecord.get("reason") as string)}` : ""}`;
151	rakin	51	}
152	rakin	304	}
153			else {
154			text = `${msg.mentions.members!.get(afkRecords[0].get("user") as string)!.user.tag} is AFK right now${afkRecords[0].get("reason") as (null \| string) ? `, for reason ${Util.escapeMarkdown(afkRecords[0].get("reason") as string)}` : ""}.`;
155			}
156
157			await msg.reply({
158			embeds: [
159			new MessageEmbed({
160			description: text
161			})
162			]
163	rakin	51	});
164			}
165			}
166			};