package games.coob.laserturrets.util;

import lombok.Getter;
import org.bukkit.Location;
import org.bukkit.entity.ArmorStand;
import org.bukkit.entity.Entity;
import org.bukkit.inventory.ItemStack;
import org.bukkit.util.Consumer;
import org.mineacademy.fo.menu.model.ItemCreator;
import org.mineacademy.fo.remain.CompMaterial;
import org.mineacademy.fo.remain.CompProperty;

/**
 *
 */
@Getter
public class SimpleHologramStand extends SimpleHologram {

    /**
     * The item or material this hologram will have
     */
    private final Object itemOrMaterial;

    /**
     * Is this item stand small?
     */
    private boolean small;

    /**
     * Is this item stand glowing?
     */
    private boolean glowing;

    /**
     * Create a new simple hologram using armor stand showing the given itemstack
     *
     * @param spawnLocation
     * @param item
     */
    public SimpleHologramStand(final Location spawnLocation, final ItemStack item) {
        super(spawnLocation);

        this.itemOrMaterial = item;
    }

    /**
     * Create a new simple hologram using armor stand showing the given material
     *
     * @param spawnLocation
     * @param material
     */
    public SimpleHologramStand(final Location spawnLocation, final CompMaterial material) {
        super(spawnLocation);

        this.itemOrMaterial = material;
    }

    @Override
    protected final Entity createEntity() {

        final ItemCreator item;

        if (this.itemOrMaterial instanceof ItemStack)
            item = ItemCreator.of((ItemStack) this.itemOrMaterial);
        else
            item = ItemCreator.of((CompMaterial) this.itemOrMaterial);

        final Consumer<ArmorStand> consumer = armorStand -> {
            CompProperty.GRAVITY.apply(armorStand, false);
            armorStand.setHelmet(item.glow(this.glowing).make());
            armorStand.setVisible(false);
            armorStand.setSmall(this.small);
        };

        return getLastTeleportLocation().getWorld().spawn(getLastTeleportLocation(), ArmorStand.class, consumer);
    }


    /**
     * @param glowing
     * @return
     */
    public final SimpleHologram setGlowing(final boolean glowing) {
        this.glowing = glowing;

        return this;
    }

    /**
     * @param small the small to set
     * @return
     */
    public final SimpleHologram setSmall(final boolean small) {
        this.small = small;

        return this;
    }
}