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

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