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";

export default class AFKEngine {
    client: DiscordClient;

    constructor(client: DiscordClient) {
        this.client = client;
    }

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