src/automod/AntiRaid.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 { Collection, Guild, GuildMember, TextChannel } from 'discord.js';
import DiscordClient from '../client/Client';
import { lockAll } from '../commands/moderation/LockallCommand';
import { hasConfig } from '../utils/util';

export default class AntiRaid {
    constructor(private client: DiscordClient, private joins = 0, private maxJoins = 0, private channels: string[] = [], private exclude = false, private time = 0, private enabled = false) {
        
    }

    load(guild: Guild) {
        this.maxJoins = this.client.config.props[guild.id].raid.max_joins;
        this.channels = this.client.config.props[guild.id].raid.channels;
        this.time = this.client.config.props[guild.id].raid.time;
        this.enabled = this.client.config.props[guild.id].raid.enabled;
        this.exclude = this.client.config.props[guild.id].raid.exclude;
    }

    async start(member: GuildMember) {
        if (!hasConfig(this.client, member.guild.id, "raid"))
            return;
        
        if (member.user.bot) {
            console.log('bot');
            return;
        }

        await this.load(member.guild);

        if (!this.enabled)
            return;

        console.log('Joined');

        setTimeout(() => {
            this.joins = 0;
            console.log('RAID reset');
        }, this.time);

        this.joins++;

        if (this.joins === this.maxJoins) {
            await this.trigger(member.guild);
        }
    }

    async trigger(guild: Guild) {
        let role = guild.roles.everyone;
            
        let channels = <Collection <string, TextChannel>> guild.channels.cache.filter(channel => {
            let cond: boolean;

            if (this.exclude) {
                cond = this.channels.indexOf(channel.id) === -1 && this.channels.indexOf(channel.parent?.id!) === -1;
            }
            else {
                cond = this.channels.indexOf(channel.id) !== -1 || this.channels.indexOf(channel.parent?.id!) !== -1;
            }

            return cond && channel.type === 'GUILD_TEXT';
        });

        await lockAll(this.client, role, channels, this.client.user!, true);
    }
};
1	/**
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 { Collection, Guild, GuildMember, TextChannel } from 'discord.js';
21	import DiscordClient from '../client/Client';
22	import { lockAll } from '../commands/moderation/LockallCommand';
23	import { hasConfig } from '../utils/util';
24
25	export default class AntiRaid {
26	constructor(private client: DiscordClient, private joins = 0, private maxJoins = 0, private channels: string[] = [], private exclude = false, private time = 0, private enabled = false) {
27
28	}
29
30	load(guild: Guild) {
31	this.maxJoins = this.client.config.props[guild.id].raid.max_joins;
32	this.channels = this.client.config.props[guild.id].raid.channels;
33	this.time = this.client.config.props[guild.id].raid.time;
34	this.enabled = this.client.config.props[guild.id].raid.enabled;
35	this.exclude = this.client.config.props[guild.id].raid.exclude;
36	}
37
38	async start(member: GuildMember) {
39	if (!hasConfig(this.client, member.guild.id, "raid"))
40	return;
41
42	if (member.user.bot) {
43	console.log('bot');
44	return;
45	}
46
47	await this.load(member.guild);
48
49	if (!this.enabled)
50	return;
51
52	console.log('Joined');
53
54	setTimeout(() => {
55	this.joins = 0;
56	console.log('RAID reset');
57	}, this.time);
58
59	this.joins++;
60
61	if (this.joins === this.maxJoins) {
62	await this.trigger(member.guild);
63	}
64	}
65
66	async trigger(guild: Guild) {
67	let role = guild.roles.everyone;
68
69	let channels = <Collection <string, TextChannel>> guild.channels.cache.filter(channel => {
70	let cond: boolean;
71
72	if (this.exclude) {
73	cond = this.channels.indexOf(channel.id) === -1 && this.channels.indexOf(channel.parent?.id!) === -1;
74	}
75	else {
76	cond = this.channels.indexOf(channel.id) !== -1 \|\| this.channels.indexOf(channel.parent?.id!) !== -1;
77	}
78
79	return cond && channel.type === 'GUILD_TEXT';
80	});
81
82	await lockAll(this.client, role, channels, this.client.user!, true);
83	}
84	};