src/automod/Cooldown.ts

import { CommandInteraction, GuildMember, Message, User } from "discord.js";
import DiscordClient from "../client/Client";
import MessageEmbed from "../client/MessageEmbed";
import CommandOptions from "../types/CommandOptions";
import InteractionOptions from "../types/InteractionOptions";
import { timeProcess, timeSince } from "../utils/util";

export type CooldownConfig = {
    enabled: boolean;
    global: number;
    cmds: {
        [cmd: string]: number;
    }
};

export type RegistryData = {
    user: User;
    guilds: {
        [id: string]: {
            date: Date;
            timeout: NodeJS.Timeout;
        }
    }
};

export default class Cooldown {
    config: CooldownConfig;
    cmdRegistry: Map <string, RegistryData>;

    constructor(protected client: DiscordClient) {
        this.config = {} as CooldownConfig;
        this.cmdRegistry = new Map();
    }

    async start(msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions): Promise <boolean> {
        this.config = this.client.config.get('cooldown');

        if (!this.config.enabled) {
            return true;
        }

        if ((msg.member as GuildMember).roles.cache.has(this.client.config.get('mod_role')))
            return true;

        const { cmdName } = options;
        let time = null;

        if (this.config.cmds[cmdName]) {
            time = this.config.cmds[cmdName];
        }
        else {
            time = this.config.global;
        }

        if (time === null) {
            return true;
        }

        console.log(time);
        

        if (!this.cmdRegistry.has(msg.member!.user.id) || this.cmdRegistry.get(msg.member!.user.id)?.guilds[msg.guild!.id] === undefined) {
            let prevData: any = {};

            if (this.cmdRegistry.has(msg.member!.user.id)) {
                prevData = this.cmdRegistry.get(msg.member!.user.id)?.guilds;
            }
            
            this.cmdRegistry.set(msg.member!.user.id, {
                user: msg.member!.user as User,
                guilds: {
                    ...prevData,
                    [msg.guild!.id]: {
                        date: new Date(),
                        timeout: setTimeout(() => {
                            console.log('Cleared');
                            this.cmdRegistry.delete(msg.member!.user.id);
                        }, time)
                    }
                }
            });
        }
        else {
            const registry = this.cmdRegistry.get(msg.member!.user.id);

            console.log(new Date(registry?.guilds[msg.guild!.id].date.getTime()! + time), new Date);
            

            if ((registry?.guilds[msg.guild!.id].date.getTime()! + time)! > Date.now()) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(":clock: Please try again in " + timeSince(registry?.guilds[msg.guild!.id].date.getTime()!).replace(/ ago$/g, ''))
                    ]
                });

                return false;
            }
        }

        return true;
    }
};
1	import { CommandInteraction, GuildMember, Message, User } from "discord.js";
2	import DiscordClient from "../client/Client";
3	import MessageEmbed from "../client/MessageEmbed";
4	import CommandOptions from "../types/CommandOptions";
5	import InteractionOptions from "../types/InteractionOptions";
6	import { timeProcess, timeSince } from "../utils/util";
7
8	export type CooldownConfig = {
9	enabled: boolean;
10	global: number;
11	cmds: {
12	[cmd: string]: number;
13	}
14	};
15
16	export type RegistryData = {
17	user: User;
18	guilds: {
19	[id: string]: {
20	date: Date;
21	timeout: NodeJS.Timeout;
22	}
23	}
24	};
25
26	export default class Cooldown {
27	config: CooldownConfig;
28	cmdRegistry: Map <string, RegistryData>;
29
30	constructor(protected client: DiscordClient) {
31	this.config = {} as CooldownConfig;
32	this.cmdRegistry = new Map();
33	}
34
35	async start(msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions): Promise <boolean> {
36	this.config = this.client.config.get('cooldown');
37
38	if (!this.config.enabled) {
39	return true;
40	}
41
42	if ((msg.member as GuildMember).roles.cache.has(this.client.config.get('mod_role')))
43	return true;
44
45	const { cmdName } = options;
46	let time = null;
47
48	if (this.config.cmds[cmdName]) {
49	time = this.config.cmds[cmdName];
50	}
51	else {
52	time = this.config.global;
53	}
54
55	if (time === null) {
56	return true;
57	}
58
59	console.log(time);
60
61
62	if (!this.cmdRegistry.has(msg.member!.user.id) \|\| this.cmdRegistry.get(msg.member!.user.id)?.guilds[msg.guild!.id] === undefined) {
63	let prevData: any = {};
64
65	if (this.cmdRegistry.has(msg.member!.user.id)) {
66	prevData = this.cmdRegistry.get(msg.member!.user.id)?.guilds;
67	}
68
69	this.cmdRegistry.set(msg.member!.user.id, {
70	user: msg.member!.user as User,
71	guilds: {
72	...prevData,
73	[msg.guild!.id]: {
74	date: new Date(),
75	timeout: setTimeout(() => {
76	console.log('Cleared');
77	this.cmdRegistry.delete(msg.member!.user.id);
78	}, time)
79	}
80	}
81	});
82	}
83	else {
84	const registry = this.cmdRegistry.get(msg.member!.user.id);
85
86	console.log(new Date(registry?.guilds[msg.guild!.id].date.getTime()! + time), new Date);
87
88
89	if ((registry?.guilds[msg.guild!.id].date.getTime()! + time)! > Date.now()) {
90	await msg.reply({
91	embeds: [
92	new MessageEmbed()
93	.setColor('#f14a60')
94	.setDescription(":clock: Please try again in " + timeSince(registry?.guilds[msg.guild!.id].date.getTime()!).replace(/ ago$/g, ''))
95	]
96	});
97
98	return false;
99	}
100	}
101
102	return true;
103	}
104	};