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	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 { 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			};