src/services/Utilities.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 { userMention } from "@discordjs/builders";
import { Collection, Message } from "discord.js";
import DiscordClient from "../client/Client";
import StaffAway, { IStaffAway } from "../models/StaffAway";
import { emoji } from "../utils/Emoji";
import Service from "../utils/structures/Service";

export default class Utilities extends Service {
    typingInterval: NodeJS.Timer | null = null;
    typingTimeOut: NodeJS.Timeout | null = null;
    lastDeletedMessage?: Message;
    staffAwayList: IStaffAway[] = [];

    constructor(client: DiscordClient) {
        super(client);

        StaffAway.find({ enabled: true }).then(list => {
            this.staffAwayList = list;
        }).catch(console.error);
    }

    findStaffAway(guild: string, id: string, del = false) {
        let i = 0;

        for (const entry of this.staffAwayList) {
            if (entry.user === id && entry.guild_id === guild) {
                if (del) {
                    this.staffAwayList.splice(i, 1);
                }

                return entry;
            }

            i++;
        }

        return null;
    }

    async checkStaffAway(message: Message) {
        if (!message.author.bot && 
            this.client.config.props[message.guildId!].staffchannel_group && 
            message.channel.type === 'GUILD_TEXT' && 
            message.channel.parent?.id && 
            message.channel.parent?.id === this.client.config.props[message.guildId!].staffchannel_group && 
            message.member?.roles.cache.has(this.client.config.props[message.guildId!].mod_role)) {
            const user = message.mentions.repliedUser ?? message.mentions.users.first();

            const entry = user ? this.findStaffAway(message.guildId!, user.id) : null;
            const selfentry = this.findStaffAway(message.guildId!, message.author.id);

            if (selfentry) {
                this.client.utils.findStaffAway(message.guildId!, message.author.id, true);
                await selfentry.delete();
                await message.reply(`${emoji('check')} Welcome back, we've removed your away status.`);
            }
            else if (entry) {
                await message.reply({
                    content: `${userMention(entry.user)} has taken a break from moderating.`,
                    allowedMentions: {
                        users: []
                    }
                });
            }
        }
    }
}
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 { userMention } from "@discordjs/builders";
21			import { Collection, Message } from "discord.js";
22			import DiscordClient from "../client/Client";
23			import StaffAway, { IStaffAway } from "../models/StaffAway";
24			import { emoji } from "../utils/Emoji";
25			import Service from "../utils/structures/Service";
26
27			export default class Utilities extends Service {
28			typingInterval: NodeJS.Timer \| null = null;
29			typingTimeOut: NodeJS.Timeout \| null = null;
30			lastDeletedMessage?: Message;
31			staffAwayList: IStaffAway[] = [];
32
33			constructor(client: DiscordClient) {
34			super(client);
35
36			StaffAway.find({ enabled: true }).then(list => {
37			this.staffAwayList = list;
38			}).catch(console.error);
39			}
40
41			findStaffAway(guild: string, id: string, del = false) {
42			let i = 0;
43
44			for (const entry of this.staffAwayList) {
45			if (entry.user === id && entry.guild_id === guild) {
46			if (del) {
47			this.staffAwayList.splice(i, 1);
48			}
49
50			return entry;
51			}
52
53			i++;
54			}
55
56			return null;
57			}
58
59			async checkStaffAway(message: Message) {
60			if (!message.author.bot &&
61			this.client.config.props[message.guildId!].staffchannel_group &&
62			message.channel.type === 'GUILD_TEXT' &&
63			message.channel.parent?.id &&
64			message.channel.parent?.id === this.client.config.props[message.guildId!].staffchannel_group &&
65			message.member?.roles.cache.has(this.client.config.props[message.guildId!].mod_role)) {
66			const user = message.mentions.repliedUser ?? message.mentions.users.first();
67
68			const entry = user ? this.findStaffAway(message.guildId!, user.id) : null;
69			const selfentry = this.findStaffAway(message.guildId!, message.author.id);
70
71			if (selfentry) {
72			this.client.utils.findStaffAway(message.guildId!, message.author.id, true);
73			await selfentry.delete();
74			await message.reply(`${emoji('check')} Welcome back, we've removed your away status.`);
75			}
76			else if (entry) {
77			await message.reply({
78			content: `${userMention(entry.user)} has taken a break from moderating.`,
79			allowedMentions: {
80			users: []
81			}
82			});
83			}
84			}
85			}
86			}