package net.babamod.mineclass.classes;

import net.babamod.mineclass.utils.Pair;
import org.bukkit.Material;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.potion.PotionEffectType;

import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class FireDwarfClass extends MineClassImpl {

  private static final Set<Material> forbiddenItems =
      new HashSet<Material>() {
        {
          add(Material.DIAMOND_SWORD);
          add(Material.GOLDEN_SWORD);
          add(Material.IRON_SWORD);
          add(Material.NETHERITE_SWORD);
          add(Material.DIAMOND_HOE);
          add(Material.GOLDEN_HOE);
          add(Material.IRON_HOE);
          add(Material.NETHERITE_HOE);
          add(Material.DIAMOND_SHOVEL);
          add(Material.GOLDEN_SHOVEL);
          add(Material.IRON_SHOVEL);
          add(Material.NETHERITE_SHOVEL);
          add(Material.BOW);
          add(Material.TRIDENT);
        }
      };

  private static final Map<PotionEffectType, Integer> potionEffects =
      Stream.of(
              new Object[][] {
                {PotionEffectType.FIRE_RESISTANCE, 1},
                {PotionEffectType.FAST_DIGGING, 1},
                {PotionEffectType.JUMP, 2},
                {PotionEffectType.NIGHT_VISION, 1},
                {PotionEffectType.HEALTH_BOOST, 2},
              })
          .collect(Collectors.toMap(data -> (PotionEffectType) data[0], data -> (Integer) data[1]));

  private static final Map<Material, List<Pair<Enchantment, Integer>>> classEnchantments =
      Stream.of(
              new AbstractMap.SimpleEntry<>(
                  Material.NETHERITE_AXE,
                  Collections.singletonList(new Pair<>(Enchantment.FIRE_ASPECT, 2))),
              new AbstractMap.SimpleEntry<>(
                  Material.DIAMOND_AXE,
                  Collections.singletonList(new Pair<>(Enchantment.FIRE_ASPECT, 2))),
              new AbstractMap.SimpleEntry<>(
                  Material.IRON_AXE,
                  Collections.singletonList(new Pair<>(Enchantment.FIRE_ASPECT, 2))),
              new AbstractMap.SimpleEntry<>(
                  Material.GOLDEN_AXE,
                  Collections.singletonList(new Pair<>(Enchantment.FIRE_ASPECT, 2))),
              new AbstractMap.SimpleEntry<>(
                  Material.STONE_AXE,
                  Collections.singletonList(new Pair<>(Enchantment.FIRE_ASPECT, 2))),
              new AbstractMap.SimpleEntry<>(
                  Material.WOODEN_AXE,
                  Collections.singletonList(new Pair<>(Enchantment.FIRE_ASPECT, 2))),
              new AbstractMap.SimpleEntry<>(
                  Material.NETHERITE_PICKAXE,
                  Collections.singletonList(new Pair<>(Enchantment.DIG_SPEED, 5))),
              new AbstractMap.SimpleEntry<>(
                  Material.DIAMOND_PICKAXE,
                  Collections.singletonList(new Pair<>(Enchantment.DIG_SPEED, 5))),
              new AbstractMap.SimpleEntry<>(
                  Material.IRON_PICKAXE,
                  Collections.singletonList(new Pair<>(Enchantment.DIG_SPEED, 5))),
              new AbstractMap.SimpleEntry<>(
                  Material.GOLDEN_PICKAXE,
                  Collections.singletonList(new Pair<>(Enchantment.DIG_SPEED, 5))),
              new AbstractMap.SimpleEntry<>(
                  Material.STONE_PICKAXE,
                  Collections.singletonList(new Pair<>(Enchantment.DIG_SPEED, 5))),
              new AbstractMap.SimpleEntry<>(
                  Material.WOODEN_PICKAXE,
                  Collections.singletonList(new Pair<>(Enchantment.DIG_SPEED, 5))),
              new AbstractMap.SimpleEntry<>(
                  Material.CROSSBOW,
                  Collections.singletonList(new Pair<>(Enchantment.ARROW_INFINITE, 1))),
              new AbstractMap.SimpleEntry<>(
                  Material.FLINT_AND_STEEL, new ArrayList<Pair<Enchantment, Integer>>())
              // See to make infinity working on crossbow
              // Inventory auto smelt
              )
          .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

  @Override
  public Set<Material> getForbiddenItems() {
    return forbiddenItems;
  }

  @Override
  public Map<PotionEffectType, Integer> getPotionEffects() {
    return potionEffects;
  }

  @Override
  public Map<Material, List<Pair<Enchantment, Integer>>> getClassEnchantments() {
    return classEnchantments;
  }

  @Override
  public String getCode() {
    return "fire_dwarf";
  }
}