Add option for disabling selling / buying individually.

1 files changed, 12 insertions, 4 deletions

diff --git a/src/main/java/org/chrisoft/trashyaddon/modules/AutoTrade.java b/src/main/java/org/chrisoft/trashyaddon/modules/AutoTrade.java
index 07c6995..c5df1b9 100644
--- a/src/main/java/org/chrisoft/trashyaddon/modules/AutoTrade.java
+++ b/src/main/java/org/chrisoft/trashyaddon/modules/AutoTrade.java
@@ -109,14 +109,22 @@ public class AutoTrade extends Module {
         Items.PAINTING
     );
 
+    private final Setting<Boolean> sellingEnabled = sgGeneral.add(new BoolSetting.Builder()
+        .name("Enable Selling")
+        .build()
+    );
     private final Setting<List<Item>> sellsSetting = sgGeneral.add(new ItemListSetting.Builder()
-        .name("sells")
+        .name("Sells")
         .description("Items to automatically SELL TO villagers.")
         .filter((item) -> allSellItems.contains(item))
         .build()
     );
+    private final Setting<Boolean> buyingEnabled = sgGeneral.add(new BoolSetting.Builder()
+        .name("Enable Buying")
+        .build()
+    );
     private final Setting<List<Item>> buysSetting = sgGeneral.add(new ItemListSetting.Builder()
-        .name("buys")
+        .name("Buys")
         .description("Items to automatically BUY FROM villagers.")
         .filter((item) -> allBuyItems.contains(item))
         .build()
@@ -180,8 +188,8 @@ public class AutoTrade extends Module {
             return false;
         ItemStack mbuy = o.getAdjustedFirstBuyItem();
         ItemStack msell = o.getSellItem();
-        List<Item> sells = this.sellsSetting.get();
-        List<Item> buys = this.buysSetting.get();
+        List<Item> sells = sellingEnabled.get() ? this.sellsSetting.get() : List.of();
+        List<Item> buys = buyingEnabled.get() ? this.buysSetting.get() : List.of();
         if (!sells.contains(mbuy.getItem()) && !buys.contains(msell.getItem()))
             return false;
         FindItemResult rs = InvUtils.find((stack) -> stack.getItem().equals(mbuy.getItem()) && stack.getCount() >= mbuy.getCount());