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	import { formatDistanceToNowStrict } from "date-fns";
2	import { CommandInteraction, GuildMember, Message, User, Util } from "discord.js";
3	import DiscordClient from "../client/Client";
4	import MessageEmbed from "../client/MessageEmbed";
5	import AFK, { IAFK } from "../models/AFK";
6	import Service from "../utils/structures/Service";
7
8	export interface MentionSchema {
9	date: number;
10	user: string;
11	}
12
13	export default class AFKEngine extends Service {
14	list: IAFK[] = [];
15
16	constructor(client: DiscordClient) {
17	super(client);
18	AFK.find().then(models => this.list = models).catch(console.error);
19	}
20
21	findUsers(ids: string[], guild: string) {
22	return this.list.filter(afk => ids.includes(afk.user) && afk.guild_id === guild);
23	}
24
25	async removeUser(id: string, guild: string) {
26	let index = 0;
27
28	for await (const afk of this.list) {
29	if (afk.user === id && afk.guild_id === guild) {
30	await afk.delete();
31	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	}
49
50	let member: GuildMember \| undefined;
51
52	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	}
67
68	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	reason: status ?? undefined,
84	createdAt: new Date()
85	}));
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	}
96
97	async start(msg: Message) {
98	if (msg.author.bot)
99	return;
100
101	const selfAFK = this.findUsers([msg.author.id], msg.guild!.id);
102
103	if (selfAFK.length > 0) {
104	this.toggle(msg, false);
105	}
106
107	const mention = msg.mentions.members?.first();
108
109	if (mention) {
110	const afkRecords: Array<IAFK> = this.findUsers([...msg.mentions.members!.keys()].slice(0, 3), msg.guild!.id).filter(afk => afk.user !== msg.author.id);
111
112	if (!afkRecords \|\| afkRecords.length < 1) {
113	return;
114	}
115
116	for (const record of afkRecords) {
117	const mentions = record.mentions as MentionSchema[];
118
119	mentions.push({
120	date: Date.now(),
121	user: msg.author.id
122	});
123
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	}
133	}
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	});
145	}
146	}
147	};