src/services/AFKEngine.ts

import { CommandInteraction, GuildMember, Message, User } from "discord.js";
import DiscordClient from "../client/Client";
import MessageEmbed from "../client/MessageEmbed";
import { notAFK } from "../commands/utils/AFKCommand";
import Service from "../utils/structures/Service";

export default class AFKEngine extends Service {
    mention(msg: Message, user: GuildMember, cb: (data: any) => void, msg1?: any) {
        this.client.db.get('SELECT * FROM afk WHERE user_id = ?', [user.id], (err: any, data: any) => {
            if (data) {
                if (msg1 === undefined) {
                    msg.channel!.send({
                        embeds: [
                            new MessageEmbed()
                            .setDescription(`**${user.user.tag}** is AFK right now${data.reason.trim() == '' ? '.' : (' for reason: **' + data.reason.replace(/\*/g, '\\*') + '**')}`)
                        ]
                    });
                }

                this.client.db.get('UPDATE afk SET mentions = ? WHERE id = ?', [parseInt(data.mentions) + 1, data.id], (err: any) => {});

                cb(data);
            }
        });

    }

    start(msg: Message) {
        if (msg.author.bot)
            return;
        
        const mention = msg.mentions.members!.first();

        if (mention) {
            this.mention(msg, msg.member!, data => {
                if (msg.author.id === data.user_id) {
                    notAFK(this.client, msg, data);
                }
            }, true);
            
            msg.mentions.members!.forEach((member) => {
                this.mention(msg, member, data => {
                    if (msg.author.id === data.user_id) {
                        notAFK(this.client, msg, data);
                    }
                });
            });
        }
        else {
            this.client.db.get('SELECT * FROM afk WHERE user_id = ?', [msg.author.id], (err: any, data: any) => {
                if (data) {
                    notAFK(this.client, msg, data);
                }
            });
        }
    }
};
1	import { CommandInteraction, GuildMember, Message, User } from "discord.js";
2	import DiscordClient from "../client/Client";
3	import MessageEmbed from "../client/MessageEmbed";
4	import { notAFK } from "../commands/utils/AFKCommand";
5	import Service from "../utils/structures/Service";
6
7	export default class AFKEngine extends Service {
8	mention(msg: Message, user: GuildMember, cb: (data: any) => void, msg1?: any) {
9	this.client.db.get('SELECT * FROM afk WHERE user_id = ?', [user.id], (err: any, data: any) => {
10	if (data) {
11	if (msg1 === undefined) {
12	msg.channel!.send({
13	embeds: [
14	new MessageEmbed()
15	.setDescription(`${user.user.tag} is AFK right now${data.reason.trim() == '' ? '.' : (' for reason: *' + data.reason.replace(/\/g, '\\') + '*')}`)
16	]
17	});
18	}
19
20	this.client.db.get('UPDATE afk SET mentions = ? WHERE id = ?', [parseInt(data.mentions) + 1, data.id], (err: any) => {});
21
22	cb(data);
23	}
24	});
25
26	}
27
28	start(msg: Message) {
29	if (msg.author.bot)
30	return;
31
32	const mention = msg.mentions.members!.first();
33
34	if (mention) {
35	this.mention(msg, msg.member!, data => {
36	if (msg.author.id === data.user_id) {
37	notAFK(this.client, msg, data);
38	}
39	}, true);
40
41	msg.mentions.members!.forEach((member) => {
42	this.mention(msg, member, data => {
43	if (msg.author.id === data.user_id) {
44	notAFK(this.client, msg, data);
45	}
46	});
47	});
48	}
49	else {
50	this.client.db.get('SELECT * FROM afk WHERE user_id = ?', [msg.author.id], (err: any, data: any) => {
51	if (data) {
52	notAFK(this.client, msg, data);
53	}
54	});
55	}
56	}
57	};