Refactor/input helper improvements

src/main/java/com/github/mkram17/bazaarutils/features/gui/buttons/inputhelper/amount/BuyOrderAmountHelper.java

@@ -6,6 +6,9 @@
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarScreenMatcher;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarScreenType;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarSlots;
+import com.github.mkram17.bazaarutils.utils.bazaar.gui.layouts.TransactionPageLayout;
+import com.github.mkram17.bazaarutils.utils.bazaar.market.order.OrderUtil;
+import com.github.mkram17.bazaarutils.utils.bazaar.market.price.PricingPosition;
 import com.github.mkram17.bazaarutils.utils.minecraft.components.CustomDataComponents;
 import com.github.mkram17.bazaarutils.utils.bazaar.market.order.TransactionType;
 import com.github.mkram17.bazaarutils.utils.minecraft.gui.ScreenMatcher;
@@ -18,6 +21,8 @@
 import lombok.Getter;
 import net.minecraft.network.chat.Component;
 
+import java.util.OptionalDouble;
+import java.util.OptionalInt;
 import java.util.stream.IntStream;
 
 @Getter
@@ -100,6 +105,20 @@ protected int computeFixedValue(TransactionState state) {
         return getFixedAmount();
     }
 
+    @Override
+    protected OptionalInt computeMaxValue(TransactionAmount.TransactionState state) {
+        OptionalDouble price = OrderUtil.getPriceForPositionOptional(state.productId(), PricingPosition.COMPETITIVE, getTransactionType());
+
+        // A missing or non-positive price divides into an amount that is nonsense rather than large.
+        if (price.isEmpty() || price.getAsDouble() <= 0) return OptionalInt.empty();
+
+        int amountCanAfford = (int) (state.purse() / price.getAsDouble());
+
+        return OptionalInt.of(TransactionPageLayout.findBuyOrderAmountLimit(state.inputSign().itemStack())
+                            .map(limit -> Math.min(amountCanAfford, limit))
+                            .orElse(amountCanAfford));
+    }
+
     @Override
     protected Component getButtonItemText(TransactionState state) {
         return Component.nullToEmpty("Order " + getButtonItemStackSize(state) + " items.");

src/main/java/com/github/mkram17/bazaarutils/features/gui/buttons/inputhelper/amount/InstantBuyAmountHelper.java

@@ -3,10 +3,14 @@
 import com.github.mkram17.bazaarutils.config.util.api.SlotProviders;
 import com.github.mkram17.bazaarutils.config.util.api.annotations.ContainerSlot;
 import com.github.mkram17.bazaarutils.utils.bazaar.SignInputHelper;
+import com.github.mkram17.bazaarutils.utils.bazaar.data.BazaarDataUtil;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarScreenMatcher;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarScreenType;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarSlots;
+import com.github.mkram17.bazaarutils.utils.bazaar.gui.layouts.TransactionPageLayout;
 import com.github.mkram17.bazaarutils.utils.bazaar.market.order.TransactionType;
+import com.github.mkram17.bazaarutils.utils.minecraft.ItemInfo;
+import com.github.mkram17.bazaarutils.utils.minecraft.SlotLookup;
 import com.github.mkram17.bazaarutils.utils.minecraft.components.CustomDataComponents;
 import com.github.mkram17.bazaarutils.utils.minecraft.gui.ScreenMatcher;
 import com.github.mkram17.bazaarutils.utils.minecraft.item.ItemRef;
@@ -18,6 +22,8 @@
 import lombok.Getter;
 import net.minecraft.network.chat.Component;
 
+import java.util.Optional;
+import java.util.OptionalInt;
 import java.util.stream.IntStream;
 
 @Getter
@@ -100,9 +106,34 @@ protected int computeFixedValue(TransactionState state) {
         return getFixedAmount();
     }
 
+    @Override
+    protected OptionalInt computeMaxValue(TransactionAmount.TransactionState state) {
+        return OptionalInt.of(SlotLookup.getInventoryItem(state.container(), BazaarSlots.INSTANT_BUY.INPUT_FILLING_AMOUNT.slot)
+                .map(ItemInfo::itemStack)
+                .flatMap(TransactionPageLayout::findOptionAmount)
+                .map(value -> (int) Math.floor(value))
+                .orElse(state.playerInventory()
+                        .getNonEquipmentItems()
+                        .stream()
+                        .mapToInt(stack -> {
+                            int maxStackSize = state.productItem().itemStack().getMaxStackSize();
+                            if (stack.isEmpty()) return maxStackSize;
+
+                            boolean isSameItem = Optional.ofNullable(stack.getCustomName())
+                                    .map(Component::getString)
+                                    .flatMap(BazaarDataUtil::findProductIdOptional)
+                                    .map(id -> id.equals(state.productId()))
+                                    .orElse(false);
+
+                            return isSameItem ? maxStackSize - stack.getCount() : 0;
+                        })
+                        .sum()
+                ));
+    }
+
     @Override
     protected Component getButtonItemText(TransactionState state) {
-        return Component.nullToEmpty("Offer " + getButtonItemStackSize(state) + " items.");
+        return Component.nullToEmpty("Purchase " + getButtonItemStackSize(state) + " items.");
     }
 
     @Override

src/main/java/com/github/mkram17/bazaarutils/features/gui/buttons/inputhelper/amount/SellOfferAmountHelper.java

@@ -3,6 +3,7 @@
 import com.github.mkram17.bazaarutils.config.util.api.SlotProviders;
 import com.github.mkram17.bazaarutils.config.util.api.annotations.ContainerSlot;
 import com.github.mkram17.bazaarutils.utils.bazaar.SignInputHelper;
+import com.github.mkram17.bazaarutils.utils.bazaar.data.BazaarDataUtil;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarScreenMatcher;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarScreenType;
 import com.github.mkram17.bazaarutils.utils.bazaar.gui.BazaarSlots;
@@ -17,7 +18,10 @@
 import com.teamresourceful.resourcefulconfig.api.types.info.ListEntryInfoProvider;
 import lombok.Getter;
 import net.minecraft.network.chat.Component;
+import net.minecraft.world.item.ItemStack;
 
+import java.util.Optional;
+import java.util.OptionalInt;
 import java.util.stream.IntStream;
 
 @Getter
@@ -100,6 +104,19 @@ protected int computeFixedValue(TransactionState state) {
         return getFixedAmount();
     }
 
+    @Override
+    protected OptionalInt computeMaxValue(TransactionState state) {
+        return OptionalInt.of(state.playerInventory().getNonEquipmentItems().stream()
+                .filter(stack -> !stack.isEmpty())
+                .filter(stack -> Optional.ofNullable(stack.getCustomName())
+                        .map(Component::getString)
+                        .flatMap(BazaarDataUtil::findProductIdOptional)
+                        .map(productId -> productId.equals(state.productId()))
+                        .orElse(false))
+                .mapToInt(ItemStack::getCount)
+                .sum());
+    }
+
     @Override
     protected Component getButtonItemText(TransactionState state) {
         return Component.nullToEmpty("Offer " + getButtonItemStackSize(state) + " items.");

src/main/java/com/github/mkram17/bazaarutils/utils/bazaar/InputHelper.java

@@ -80,8 +80,7 @@ public Result onButtonClicked(int button) {
 
         SoundUtil.playSound(BUTTON_SOUND, BUTTON_VOLUME);
 
-        handleAction(state.get());
-        resetState();
+        handleAction(state.get(), this::resetState);
 
         return Result.CONSUMED;
     }
@@ -94,5 +93,5 @@ public Result onButtonClicked(int button) {
 
     //    Action stuff
 
-    protected abstract void handleAction(T state);
+    protected abstract void handleAction(T state, Runnable callback);
 }