From 7c14c7f175e9bd2f2be465756a4981f72dba5526 Mon Sep 17 00:00:00 2001
From: Ellpeck <me@ellpeck.de>
Date: Mon, 24 Oct 2022 13:39:01 +0200
Subject: [PATCH] improved the storage drawer handling from
 239aa499c402728400e3fa56ed4a49dbfc86fb41

---
 build.gradle                                  |  3 +++
 .../prettypipes/pipe/PipeBlockEntity.java     | 23 ++++++++++++++++---
 2 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/build.gradle b/build.gradle
index 7ac4f96..4f04216 100644
--- a/build.gradle
+++ b/build.gradle
@@ -111,6 +111,9 @@ dependencies {
     compileOnly fg.deobf("mezz.jei:jei-1.18.2:9.5.5.174:api")
     runtimeOnly fg.deobf("mezz.jei:jei-1.18.2:9.5.5.174")
 
+    // test storage drawers
+    runtimeOnly fg.deobf("curse.maven:storage-drawers-223852:3807626")
+
     // to test the rf requiring and crafting stuff
 /*    runtimeOnly fg.deobf("curse.maven:powah-352656:3057732")
     runtimeOnly fg.deobf("curse.maven:lollipop-347954:3057731")
diff --git a/src/main/java/de/ellpeck/prettypipes/pipe/PipeBlockEntity.java b/src/main/java/de/ellpeck/prettypipes/pipe/PipeBlockEntity.java
index c4994be..dc7331b 100644
--- a/src/main/java/de/ellpeck/prettypipes/pipe/PipeBlockEntity.java
+++ b/src/main/java/de/ellpeck/prettypipes/pipe/PipeBlockEntity.java
@@ -40,7 +40,6 @@ import net.minecraftforge.common.util.Lazy;
 import net.minecraftforge.common.util.LazyOptional;
 import net.minecraftforge.items.CapabilityItemHandler;
 import net.minecraftforge.items.IItemHandler;
-import net.minecraftforge.items.ItemHandlerHelper;
 import net.minecraftforge.items.ItemStackHandler;
 import org.apache.commons.lang3.tuple.Pair;
 
@@ -199,10 +198,27 @@ public class PipeBlockEntity extends BlockEntity implements MenuProvider, IPipeC
                 continue;
             if (!force && this.streamModules().anyMatch(m -> !m.getRight().canAcceptItem(m.getLeft(), this, stack, dir, handler)))
                 continue;
-            var remain = ItemHandlerHelper.insertItem(handler, stack, true);
+
+            var startSlot = 0;
+            var slotAmount = handler.getSlots();
+            // the 0th slot of a storage drawer is a "catch-all" for any items that can be inserted into the drawer, and all
+            // subsequent slots are the actual slots for the items. this causes a problem because the drawer will seem to have
+            // space for more items that it actually does, so we restrict to only inspecting the 0th slot here.
+            // see https://github.com/Ellpeck/PrettyPipes/issues/131#issuecomment-1288653623 and Discord convo with Quinteger
+            if (handler.getClass().getName().equals("com.jaquadro.minecraft.storagedrawers.capabilities.DrawerItemHandler"))
+                slotAmount = 1;
+
+            // check how many items from the stack we can insert into this destination in total
+            var remain = stack;
+            for (var i = startSlot; i < slotAmount; i++) {
+                remain = handler.insertItem(i, remain, true);
+                if (remain.isEmpty())
+                    break;
+            }
             // did we insert anything?
             if (remain.getCount() == stack.getCount())
                 continue;
+
             var toInsert = stack.copy();
             toInsert.shrink(remain.getCount());
             // limit to the max amount that modules allow us to insert
@@ -224,7 +240,7 @@ public class PipeBlockEntity extends BlockEntity implements MenuProvider, IPipeC
                     }
                     // totalSpace will be the amount of items that fit into the attached container
                     var totalSpace = 0;
-                    for (var i = 0; i < handler.getSlots(); i++) {
+                    for (var i = startSlot; i < slotAmount; i++) {
                         var copy = stack.copy();
                         var maxStackSize = copy.getMaxStackSize();
                         // if the container can store more than 64 items in this slot, then it's likely
@@ -243,6 +259,7 @@ public class PipeBlockEntity extends BlockEntity implements MenuProvider, IPipeC
                         toInsert.setCount(totalSpace - onTheWay);
                 }
             }
+
             // we return the item that can actually be inserted, NOT the remainder!
             if (!toInsert.isEmpty())
                 return Pair.of(offset, toInsert);