Make EconomyShopGUI hook compatible with BUY and SELL task types

AuroraNetworkStudios · Aug 7, 2024 · 2c73ef4 · 2c73ef4
1 parent 3fd3e07
commit 2c73ef4
Showing 1 changed file with 38 additions and 2 deletions.
diff --git a/src/main/java/gg/auroramc/quests/hooks/economyshopgui/EconomyShopGUIListener.java b/src/main/java/gg/auroramc/quests/hooks/economyshopgui/EconomyShopGUIListener.java
@@ -1,5 +1,6 @@
 package gg.auroramc.quests.hooks.economyshopgui;
 
+import gg.auroramc.aurora.api.AuroraAPI;
 import gg.auroramc.quests.AuroraQuests;
 import gg.auroramc.quests.api.quest.TaskType;
 import me.gypopo.economyshopgui.api.events.PostTransactionEvent;
@@ -8,6 +9,7 @@
 import org.bukkit.event.EventPriority;
 import org.bukkit.event.Listener;
 
+import java.util.Map;
 import java.util.Set;
 
 public class EconomyShopGUIListener implements Listener {
@@ -25,18 +27,52 @@ public class EconomyShopGUIListener implements Listener {
             Transaction.Type.SHOPSTAND_SELL_SCREEN
     );
 
+
     @EventHandler(priority = EventPriority.MONITOR)
     public void onTransaction(PostTransactionEvent e) {
         if (!successResults.contains(e.getTransactionResult())) {
             return;
         }
 
+        var manager = AuroraQuests.getInstance().getQuestManager();
         var price = e.getPrice();
 
         if (sellTypes.contains(e.getTransactionType())) {
-            AuroraQuests.getInstance().getQuestManager().progress(e.getPlayer(), TaskType.SELL_WORTH, price, null);
+            manager.progress(e.getPlayer(), TaskType.SELL_WORTH, price, null);
+        } else if (buyTypes.contains(e.getTransactionType())) {
+            manager.progress(e.getPlayer(), TaskType.BUY_WORTH, price, null);
+        }
+
+        if (sellTypes.contains(e.getTransactionType()) && !e.getItems().isEmpty()) {
+            for (var entry : e.getItems().entrySet()) {
+                var item = entry.getKey().getItemToGive();
+                var amount = entry.getValue();
+
+                if (item != null) {
+                    var id = AuroraAPI.getItemManager().resolveId(item);
+                    if (id != null) {
+                        manager.progress(e.getPlayer(), TaskType.SELL, amount, Map.of("type", id));
+                    }
+                }
+            }
+        } else if (sellTypes.contains(e.getTransactionType())) {
+            var item = e.getShopItem().getItemToGive();
+            var amount = e.getAmount();
+            if (item != null) {
+                var id = AuroraAPI.getItemManager().resolveId(item);
+                if (id != null) {
+                    manager.progress(e.getPlayer(), TaskType.SELL, amount, Map.of("type", id));
+                }
+            }
         } else if (buyTypes.contains(e.getTransactionType())) {
-            AuroraQuests.getInstance().getQuestManager().progress(e.getPlayer(), TaskType.BUY_WORTH, price, null);
+            var item = e.getShopItem().getItemToGive();
+            var amount = e.getAmount();
+            if (item != null) {
+                var id = AuroraAPI.getItemManager().resolveId(item);
+                if (id != null) {
+                    manager.progress(e.getPlayer(), TaskType.BUY, amount, Map.of("type", id));
+                }
+            }
         }
     }
 }