From 3dbb602ee3dbb3b257d2bb732533c1c26d00d1b5 Mon Sep 17 00:00:00 2001
From: Auxilor <will@willfp.com>
Date: Fri, 13 Nov 2020 19:10:49 +0000
Subject: [PATCH] Added works-on to transfuse

---
 .../enchantments/ecoenchants/normal/Transfuse.java           | 5 +++++
 Plugin/src/main/resources/enchants/normal/transfuse.yml      | 2 ++
 2 files changed, 7 insertions(+)

diff --git a/Plugin/src/main/java/com/willfp/ecoenchants/enchantments/ecoenchants/normal/Transfuse.java b/Plugin/src/main/java/com/willfp/ecoenchants/enchantments/ecoenchants/normal/Transfuse.java
index 96200298..e0c17941 100644
--- a/Plugin/src/main/java/com/willfp/ecoenchants/enchantments/ecoenchants/normal/Transfuse.java
+++ b/Plugin/src/main/java/com/willfp/ecoenchants/enchantments/ecoenchants/normal/Transfuse.java
@@ -12,6 +12,8 @@ import org.bukkit.entity.Player;
 import org.bukkit.event.block.BlockBreakEvent;
 import org.bukkit.inventory.ItemStack;
 
+import java.util.stream.Collectors;
+
 public final class Transfuse extends EcoEnchant {
     public Transfuse() {
         super(
@@ -32,6 +34,9 @@ public final class Transfuse extends EcoEnchant {
 
         event.setDropItems(false);
 
+        if(!this.getConfig().getStrings(EcoEnchants.CONFIG_LOCATION + "works-on").stream().map(string -> Material.getMaterial(string.toUpperCase())).collect(Collectors.toList()).contains(block.getType()))
+            return;
+
         Material material;
         double random = NumberUtils.randFloat(0, 1);
         double band = 1/(double) this.getConfig().getStrings(EcoEnchants.CONFIG_LOCATION + "blocks").size();
diff --git a/Plugin/src/main/resources/enchants/normal/transfuse.yml b/Plugin/src/main/resources/enchants/normal/transfuse.yml
index ad12123d..ef0ef0eb 100644
--- a/Plugin/src/main/resources/enchants/normal/transfuse.yml
+++ b/Plugin/src/main/resources/enchants/normal/transfuse.yml
@@ -29,3 +29,5 @@ config:
     - redstone_ore
     - iron_ore
   chance-per-level: 0.03 # Chance for drop to be switched
+  works-on:
+    - stone