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; 
    messageLink: 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[]) {
        return this.list.filter(afk => ids.includes(afk.user));
    }
 
    async removeUser(id: string) {
        let index = 0;

        for await (const afk of this.list) {
            if (afk.user === id) {
                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]);

        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 })}, [Navigate](${m.messageLink ?? '*Unavailable*'})`;
                count++;
            }

            await this.client.afkEngine.removeUser(message.member!.user.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]);

        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)).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,
                    messageLink: msg.url
                });

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