package dev.plex.command.impl;

import dev.plex.command.PlexCommand;
import dev.plex.command.annotation.CommandParameters;
import dev.plex.command.annotation.CommandPermissions;
import dev.plex.command.source.RequiredCommandSource;
import dev.plex.util.PlexLog;
import dev.plex.util.PlexUtils;
import net.kyori.adventure.text.Component;
import org.apache.commons.lang3.text.WordUtils;
import org.bukkit.Bukkit;
import org.bukkit.World;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Entity;
import org.bukkit.entity.EntityType;
import org.bukkit.entity.LivingEntity;
import org.bukkit.entity.Player;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

@CommandPermissions(permission = "plex.mobpurge", source = RequiredCommandSource.ANY)
@CommandParameters(name = "mobpurge", description = "Purge all mobs.", usage = "/<command> <mob>", aliases = "mp")
public class MobPurgeCMD extends PlexCommand {

    public static final List<EntityType> MOB_TYPES = new ArrayList<>();

    @Override
    protected Component execute(@NotNull CommandSender sender, @Nullable Player playerSender, @NotNull String[] args) {
        EntityType type = null;
        String mobName = null;
        if (args.length > 0) {
            try {
                type = EntityType.valueOf(args[0].toUpperCase());
            } catch (Exception e) {
                PlexLog.debug("A genius tried and failed removing the following invalid mob: " + args[0].toUpperCase());
                send(sender, messageComponent("notAValidMob"));
                return null;
            }
            if (!MOB_TYPES.contains(type)) {
                PlexLog.debug(Arrays.deepToString(MOB_TYPES.toArray()));
                PlexLog.debug("A genius tried to remove a mob that doesn't exist: " + args[0].toUpperCase());
                sender.sendMessage(messageComponent("notAValidMobButValidEntity"));
                return null;
            }
        }
        if (type != null) {
            mobName = WordUtils.capitalizeFully(type.name().replace("_", " "));
            PlexLog.debug("The args aren't null so the mob is: " + mobName);
        }
        int count = purgeMobs(type);
        if (type != null) {
            PlexUtils.broadcast(messageComponent("removedEntitiesOfTypes", sender.getName(), count, mobName));
            PlexLog.debug("All " + count + " valid mobs were removed");
        } else {
            PlexUtils.broadcast(messageComponent("removedMobs", sender.getName(), count));
            PlexLog.debug("All " + count + " valid mobs were removed");
        }
        return null;
    }

    public int purgeMobs(EntityType type) {
        int removed = 0;
        for (World world : Bukkit.getWorlds()) {
            for (Entity entity : world.getLivingEntities()) {
                if (entity instanceof LivingEntity && !(entity instanceof Player)) {
                    if (type != null && !entity.getType().equals(type)) {
                        continue;
                    }
                    entity.remove();
                    removed++;
                }
            }
        }
        return removed;
    }

    // Adds a tab completion for /mp so players stop complaining we (mostly me) nuked all their mobs because a filter for some reason was never added by the REAL plex devs. Go figure. -Alco_Rs11

    public static List<String> getAllMobs() {
        List<String> mobs = new ArrayList<>();
        Arrays.stream(EntityType.values()).filter(EntityType::isAlive).filter(EntityType::isSpawnable).forEach(MOB_TYPES::add);
        for (EntityType entityType : MOB_TYPES) {
            mobs.add(entityType.name());
        }
        return mobs;
    }

    public @NotNull List<String> tabComplete(@NotNull CommandSender sender, @NotNull String alias, @NotNull String[] args) throws IllegalArgumentException {
        if (args.length == 1) {
            return getAllMobs();
        }
        return Collections.emptyList();
    }
}