TotalFreedomMod/commons/src/main/java/me/totalfreedom/totalfreedommod/AntiSpam.java

package me.totalfreedom.totalfreedommod;

import me.totalfreedom.totalfreedommod.config.ConfigEntry;
import me.totalfreedom.totalfreedommod.player.FPlayer;
import net.kyori.adventure.text.Component;
import net.kyori.adventure.text.format.NamedTextColor;
import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.player.AsyncPlayerChatEvent;
import org.bukkit.event.player.PlayerCommandPreprocessEvent;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class AntiSpam extends FreedomService
{
    private ScheduledThreadPoolExecutor cycle;
    public static final int MSG_PER_CYCLE = 8;
    //
    private Map<Player, Integer> muteCounts = new HashMap<>();

    @Override
    public void onStart()
    {
        cycle = new ScheduledThreadPoolExecutor(1);
        cycle.scheduleAtFixedRate(this::cycle, 0, 1, TimeUnit.SECONDS);
    }

    @Override
    public void onStop()
    {
        cycle.shutdownNow();
    }

    private void cycle()
    {
        server.getOnlinePlayers().stream().map(player -> plugin.pl.getPlayer(player)).forEach(fPlayer ->
        {
            // TODO: Move each to their own section
            fPlayer.resetMsgCount();
            fPlayer.resetBlockDestroyCount();
            fPlayer.resetBlockPlaceCount();
        });
    }

    @EventHandler(priority = EventPriority.LOW)
    public void onAsyncPlayerChat(AsyncPlayerChatEvent event)
    {
        final Player player = event.getPlayer();

        if (plugin.al.isAdmin(player))
        {
            return;
        }

        final FPlayer playerdata = plugin.pl.getPlayerSync(player);
        int count = muteCounts.getOrDefault(player, 0);
        int minutes = ConfigEntry.ANTISPAM_MINUTES.getInteger();

        // Check for spam
        if (playerdata.incrementAndGetMsgCount() > MSG_PER_CYCLE && !playerdata.isMuted())
        {
            count++;
            muteCounts.put(player, count);

            int time = count * minutes;
            playerdata.setMuted(true, time);

            server.broadcast(Component.text(player.getName()).append(Component.text(" has been automatically muted for "))
                    .append(Component.text(time)).append(Component.text(" minutes for spamming chat."))
                    .color(NamedTextColor.RED));

            playerdata.resetMsgCount();
            event.setCancelled(true);
        }
        else if (playerdata.incrementAndGetMsgCount() > MSG_PER_CYCLE / 2)
        {
            player.sendMessage(Component.text("Please refrain from spamming chat.", NamedTextColor.GRAY));
            event.setCancelled(true);
        }

    }

    @EventHandler(priority = EventPriority.LOW)
    public void onPlayerCommandPreprocess(PlayerCommandPreprocessEvent event)
    {
        String command = event.getMessage();
        final Player player = event.getPlayer();
        final FPlayer fPlayer = plugin.pl.getPlayer(player);
        fPlayer.setLastCommand(command);

        if (fPlayer.allCommandsBlocked())
        {
            player.sendMessage(Component.text("Your commands have been blocked by an admin.", NamedTextColor.RED));
            event.setCancelled(true);
            return;
        }

        if (plugin.al.isAdmin(player))
        {
            return;
        }

        if (fPlayer.incrementAndGetMsgCount() > MSG_PER_CYCLE)
        {
            server.broadcast(Component.text(player.getName())
                    .append(Component.text(" was automatically kicked for spamming commands."))
                    .color(NamedTextColor.RED));
            plugin.ae.autoEject(player, "Kicked for spamming commands.");

            fPlayer.resetMsgCount();
            event.setCancelled(true);
        }
    }
}