package io.github.hsyyid.essentialcmds.cmdexecutors;

import org.spongepowered.api.block.BlockTypes;
import org.spongepowered.api.data.key.Keys;
import org.spongepowered.api.entity.EntityType;
import org.spongepowered.api.entity.EntityTypes;
import org.spongepowered.api.entity.living.player.Player;
import org.spongepowered.api.text.Texts;
import org.spongepowered.api.text.format.TextColors;
import org.spongepowered.api.util.blockray.BlockRay;
import org.spongepowered.api.util.blockray.BlockRayHit;
import org.spongepowered.api.util.command.CommandException;
import org.spongepowered.api.util.command.CommandResult;
import org.spongepowered.api.util.command.CommandSource;
import org.spongepowered.api.util.command.args.CommandContext;
import org.spongepowered.api.util.command.source.CommandBlockSource;
import org.spongepowered.api.util.command.source.ConsoleSource;
import org.spongepowered.api.util.command.spec.CommandExecutor;
import org.spongepowered.api.world.World;

public class MobSpawnerExecutor implements CommandExecutor
{
    public CommandResult execute(CommandSource src, CommandContext ctx) throws CommandException
    {
        String entityType = ctx.<String> getOne("mob name").get();

        if (src instanceof Player)
        {
            Player player = (Player) src;
            EntityType type = null;

            switch (entityType.toLowerCase())
            {
                case "bat":
                    type = EntityTypes.BAT;
                    break;
                case "blaze":
                    type = EntityTypes.BLAZE;
                    break;
                case "cave spider":
                    type = EntityTypes.CAVE_SPIDER;
                    break;
                case "chicken":
                    type = EntityTypes.CHICKEN;
                    break;
                case "cow":
                    type = EntityTypes.COW;
                    break;
                case "creeper":
                    type = EntityTypes.CREEPER;
                    break;
                case "ender dragon":
                    type = EntityTypes.ENDER_DRAGON;
                    break;
                case "enderman":
                    type = EntityTypes.ENDERMAN;
                    break;
                case "endermite":
                    type = EntityTypes.ENDERMITE;
                    break;
                case "ghast":
                    type = EntityTypes.GHAST;
                    break;
                case "giant":
                    type = EntityTypes.GIANT;
                    break;
                case "guardian":
                    type = EntityTypes.GUARDIAN;
                    break;
                case "horse":
                    type = EntityTypes.HORSE;
                    break;
                case "iron golem":
                    type = EntityTypes.IRON_GOLEM;
                    break;
                case "magma cube":
                    type = EntityTypes.MAGMA_CUBE;
                    break;
                case "mushroom cow":
                    type = EntityTypes.MUSHROOM_COW;
                    break;
                case "ocelot":
                    type = EntityTypes.OCELOT;
                    break;
                case "pig":
                    type = EntityTypes.PIG;
                    break;
                case "pig zombie":
                    type = EntityTypes.PIG_ZOMBIE;
                    break;
                case "rabbit":
                    type = EntityTypes.RABBIT;
                    break;
                case "sheep":
                    type = EntityTypes.SHEEP;
                    break;
                case "silverfish":
                    type = EntityTypes.SILVERFISH;
                    break;
                case "skeleton":
                    type = EntityTypes.SKELETON;
                    break;
                case "slime":
                    type = EntityTypes.SLIME;
                    break;
                case "snowman":
                    type = EntityTypes.SNOWMAN;
                    break;
                case "spider":
                    type = EntityTypes.SPIDER;
                    break;
                case "squid":
                    type = EntityTypes.SQUID;
                    break;
                case "villager":
                    type = EntityTypes.VILLAGER;
                    break;
                case "witch":
                    type = EntityTypes.WITCH;
                    break;
                case "wither":
                    type = EntityTypes.WITHER;
                    break;
                case "wolf":
                    type = EntityTypes.WOLF;
                    break;
                case "zombie":
                    type = EntityTypes.ZOMBIE;
                    break;
                default:
                    src.sendMessage(Texts.of(TextColors.DARK_RED, "Error! ", TextColors.RED, "The mob you inputed was not recognized."));
                    return CommandResult.success();
            }

            BlockRay<World> playerBlockRay = BlockRay.from(player).blockLimit(350).build();
            BlockRayHit<World> finalHitRay = null;

            while (playerBlockRay.hasNext())
            {
                BlockRayHit<World> currentHitRay = playerBlockRay.next();

                if (player.getWorld().getBlockType(currentHitRay.getBlockPosition()).equals(BlockTypes.AIR))
                {
                    continue;
                }
                else
                {
                    finalHitRay = currentHitRay;
                    break;
                }
            }

            if(finalHitRay != null && finalHitRay.getLocation().getBlock().getType() == BlockTypes.MOB_SPAWNER)
            {
                System.out.println(finalHitRay.getLocation().getTileEntity().get().offer(Keys.SPAWNABLE_ENTITY_TYPE, type).getType().toString());
                player.sendMessage(Texts.of(TextColors.GREEN, "Success! ", TextColors.YELLOW, "Updated mob spawner."));
            }
            else
            {
                player.sendMessage(Texts.of(TextColors.DARK_RED, "Error! ", TextColors.RED, "Could not find a mob spawner you're looking at within range!"));
            }
        }
        else if (src instanceof ConsoleSource)
        {
            src.sendMessage(Texts.of(TextColors.DARK_RED, "Error! ", TextColors.RED, "Must be an in-game player to use /mobspawner!"));
        }
        else if (src instanceof CommandBlockSource)
        {
            src.sendMessage(Texts.of(TextColors.DARK_RED, "Error! ", TextColors.RED, "Must be an in-game player to use /mobspawner!"));
        }
        return CommandResult.success();
    }
}