studio.magemonkey.divinity.stats.items.attributes.AmmoAttribute Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of divinity Show documentation
Custom items, combat, and more!
The newest version!
package studio.magemonkey.divinity.stats.items.attributes;

import org.bukkit.Location;
import org.bukkit.NamespacedKey;
import org.bukkit.entity.*;
import org.bukkit.inventory.ItemStack;
import org.bukkit.persistence.PersistentDataType;
import org.jetbrains.annotations.NotNull;
import studio.magemonkey.divinity.stats.items.ItemTags;
import studio.magemonkey.divinity.stats.items.api.ItemLoreStat;
import studio.magemonkey.divinity.utils.ItemUtils;

public class AmmoAttribute extends ItemLoreStat {

    private AmmoAttribute.Type type;

    public AmmoAttribute(
            @NotNull Type type,
            @NotNull String name,
            @NotNull String format
    ) {
        super(type.name(),
                name,
                format,
                ItemTags.PLACEHOLDER_ITEM_AMMO,
                ItemTags.TAG_ITEM_AMMO,
                PersistentDataType.STRING);
        this.type = type;

        // Legacy keys
        this.keys.add(NamespacedKey.fromString("prorpgitems:item_ammo_" + this.getId()));
        this.keys.add(NamespacedKey.fromString("prorpgitems:item_ammo" + this.getId()));
        this.keys.add(NamespacedKey.fromString("prorpgitems:qrpg_item_ammo" + this.getId()));
        this.keys.add(NamespacedKey.fromString("quantumrpg:qrpg_item_ammo" + this.getId()));
    }

    @Override
    @NotNull
    public Class getParameterClass() {
        return String.class;
    }

    public static enum Type {

        ARROW(Arrow.class, EntityType.ARROW),
        SNOWBALL(Snowball.class, EntityType.SNOWBALL),
        EGG(Egg.class, EntityType.EGG),
        FIREBALL(Fireball.class, EntityType.FIREBALL),
        WITHER_SKULL(WitherSkull.class, EntityType.WITHER_SKULL),
        SHULKER_BULLET(ShulkerBullet.class, EntityType.SHULKER_BULLET),
        LLAMA_SPIT(LlamaSpit.class, EntityType.LLAMA_SPIT),
        ENDER_PEARL(EnderPearl.class, EntityType.ENDER_PEARL),
        EXP_POTION(ThrownExpBottle.class, EntityType.fromName("experience_bottle")), // THROWN_EXP_BOTTLE/EXPERIENCE_BOTTLE
        ;

        private Class clazz;
        private EntityType                  eType;

        private Type(@NotNull Class clazz, @NotNull EntityType eType) {
            this.clazz = clazz;
            this.eType = eType;
        }
    }

    @Override
    protected boolean isSingle() {
        return true;
    }

    @NotNull
    public AmmoAttribute.Type getType() {
        return this.type;
    }

    @Override
    public boolean add(@NotNull ItemStack item, @NotNull String value, int line) {
        return this.add(item, line);
    }

    public boolean add(@NotNull ItemStack item, int line) {
        return super.add(item, this.getType().name(), line);
    }

    @NotNull
    public Projectile getProjectile(@NotNull LivingEntity shooter) {
        switch (this.type) {
            case ARROW:
            case SNOWBALL:
            case EGG:
            case ENDER_PEARL:
            case FIREBALL:
            case WITHER_SKULL: {
                return shooter.launchProjectile(this.getType().clazz);
            }

            case EXP_POTION:
            case LLAMA_SPIT:
            case SHULKER_BULLET: {
                Location eye = shooter.getEyeLocation();
                Projectile sb =
                        (Projectile) shooter.getWorld().spawnEntity(eye.add(eye.getDirection()), this.getType().eType);
                return sb;
            }
        }
        return shooter.launchProjectile(this.type.clazz);
    }

    @Override
    @NotNull
    public String formatValue(@NotNull ItemStack item, @NotNull String value) {
        if (!ItemUtils.isBow(item)) return "";

        return value;
    }
}