cn.nukkit.command.defaults.DamageCommand Maven / Gradle / Ivy

Go to download
package cn.nukkit.command.defaults;

import cn.nukkit.api.PowerNukkitXOnly;
import cn.nukkit.api.Since;
import cn.nukkit.command.CommandSender;
import cn.nukkit.command.data.CommandParamType;
import cn.nukkit.command.data.CommandParameter;
import cn.nukkit.command.tree.ParamList;
import cn.nukkit.command.utils.CommandLogger;
import cn.nukkit.entity.Entity;
import cn.nukkit.entity.item.EntityItem;
import cn.nukkit.event.entity.EntityDamageByEntityEvent;
import cn.nukkit.event.entity.EntityDamageEvent;
import cn.nukkit.lang.TranslationContainer;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@PowerNukkitXOnly
@Since("1.6.0.0-PNX")
public class DamageCommand extends VanillaCommand {

    public DamageCommand(String name) {
        super(name, "commands.damage.description");
        this.setPermission("nukkit.command.damage");
        this.getCommandParameters().clear();
        this.addCommandParameters("default", new CommandParameter[]{
                CommandParameter.newType("target", false, CommandParamType.TARGET),
                CommandParameter.newType("amount", false, CommandParamType.INT),
                CommandParameter.newEnum("cause", true, Arrays.stream(EntityDamageEvent.DamageCause.values()).map(e -> e.name().toLowerCase()).toList().toArray(new String[0]))
        });
        this.addCommandParameters("damager", new CommandParameter[]{
                CommandParameter.newType("target", false, CommandParamType.TARGET),
                CommandParameter.newType("amount", false, CommandParamType.INT),
                CommandParameter.newEnum("cause", false, Arrays.stream(EntityDamageEvent.DamageCause.values()).map(e -> e.name().toLowerCase()).toList().toArray(new String[0])),
                CommandParameter.newEnum("entity", false, new String[]{"entity"}),
                CommandParameter.newType("damager", false, CommandParamType.TARGET)
        });
        this.enableParamTree();
    }

    @Since("1.19.60-r1")
    @Override
    public int execute(CommandSender sender, String commandLabel, Map.Entry result, CommandLogger log) {
        var list = result.getValue();
        List entities = list.getResult(0);
        if (entities.isEmpty()) {
            log.addNoTargetMatch().output();
            return 0;
        }
        String entities_str = entities.stream().map(Entity::getName).collect(Collectors.joining(" "));
        int amount = list.getResult(1);
        if (amount < 0) {
            log.addError("commands.damage.specify.damage").output();
            return 0;
        }
        switch (result.getKey()) {
            case "default" -> {
                EntityDamageEvent.DamageCause cause = EntityDamageEvent.DamageCause.NONE;
                if (list.hasResult(2)) {
                    String str = list.getResult(2);
                    cause = EntityDamageEvent.DamageCause.valueOf(str);
                }
                boolean all_success = true;
                List failed = new ArrayList<>();
                for (Entity entity : entities) {
                    EntityDamageEvent event = new EntityDamageEvent(entity, cause, amount);
                    boolean success = entity.attack(event);
                    if (!success) {
                        if (entity instanceof EntityItem) entity.kill();
                        all_success = false;
                        failed.add(entity);
                    }
                }
                if (all_success) {
                    log.addSuccess("commands.damage.success", entities_str).output();
                    return 1;
                } else {
                    log.addError("commands.damage.failed", failed.stream().map(Entity::getName).collect(Collectors.joining(" "))).output();
                    return 0;
                }
            }
            case "damager" -> {
                String str = list.getResult(2);
                EntityDamageEvent.DamageCause cause = EntityDamageEvent.DamageCause.valueOf(str.toUpperCase());
                List damagers = list.getResult(4);
                if (damagers.isEmpty()) {
                    log.addNoTargetMatch().output();
                    return 0;
                }
                if (damagers.size() > 1) {
                    log.addError("commands.damage.tooManySources").output();
                    return 0;
                }
                Entity damager = damagers.get(0);
                boolean all_success = true;
                List failed = new ArrayList<>();
                for (Entity entity : entities) {
                    EntityDamageByEntityEvent event = new EntityDamageByEntityEvent(damager, entity, cause, amount);
                    boolean success = entity.attack(event);
                    if (!success) {
                        all_success = false;
                        failed.add(entity);
                    }
                }
                if (all_success) {
                    log.addSuccess("commands.damage.success", entities_str).output();
                    sender.sendMessage(new TranslationContainer("", entities_str));
                    return 1;
                } else {
                    log.addError("commands.damage.failed", failed.stream().map(Entity::getName).collect(Collectors.joining(" "))).output();
                    return 0;
                }
            }
            default -> {
                return 0;
            }
        }
    }
}