src/services/AFKEngine.ts

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