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