await item instead of failing

Author: beanbag44

common/src/main/kotlin/com/lambda/interaction/construction/result/BreakResult.kt

@@ -99,8 +99,14 @@ sealed class BreakResult : BuildResult() {
         override fun resolve() =
             selectStack {
                 isItem(badItem).not()
-            }.transfer(MainHandContainer, inventory)
-                ?: MaterialContainer.FailureTask("Couldn't find a tool for ${blockState.block.name.string} with $badItem in main hand.")
+            }.let { selection ->
+                selection.transfer(MainHandContainer, inventory)
+                    ?: MaterialContainer.AwaitItemTask(
+                        "Couldn't find a tool for ${blockState.block.name.string} with $badItem in main hand.",
+                        selection,
+                        inventory
+                    )
+            }
 
         override fun SafeContext.buildRenderer() {
             withPos(blockPos, color)

common/src/main/kotlin/com/lambda/interaction/construction/result/BuildResult.kt

@@ -205,8 +205,15 @@ abstract class BuildResult : ComparableResult<Rank>, Nameable {
 
         override val pausesParent get() = true
 
-        override fun resolve() = neededSelection
-            .transfer(MainHandContainer, inventory) ?: MaterialContainer.FailureTask("Couldn't find $neededSelection anywhere.")
+        override fun resolve() =
+            neededSelection.let { selection ->
+                selection.transfer(MainHandContainer, inventory)
+                    ?: MaterialContainer.AwaitItemTask(
+                        "Couldn't find $neededSelection anywhere.",
+                        selection,
+                        inventory
+                    )
+            }
 
         override fun SafeContext.buildRenderer() {
             if (blockState(blockPos).isAir) {
@@ -242,8 +249,14 @@ abstract class BuildResult : ComparableResult<Rank>, Nameable {
         override val pausesParent get() = true
 
         override fun resolve() =
-            neededStack.select()
-                .transfer(MainHandContainer, inventory) ?: MaterialContainer.FailureTask("Couldn't find ${neededStack.item.name.string} anywhere.")
+            neededStack.select().let { selection ->
+                selection.transfer(MainHandContainer, inventory)
+                    ?: MaterialContainer.AwaitItemTask(
+                        "Couldn't find ${neededStack.item.name.string} anywhere.",
+                        selection,
+                        inventory
+                    )
+            }
 
         override fun SafeContext.buildRenderer() {
             if (blockState(blockPos).isAir) {

common/src/main/kotlin/com/lambda/interaction/material/container/MaterialContainer.kt

@@ -17,19 +17,29 @@
 
 package com.lambda.interaction.material.container
 
+import com.lambda.config.groups.InventoryConfig
 import com.lambda.context.SafeContext
+import com.lambda.event.events.TickEvent
+import com.lambda.event.listener.SafeListener.Companion.listen
 import com.lambda.interaction.material.StackSelection
+import com.lambda.interaction.material.container.ContainerManager.findContainerWithMaterial
 import com.lambda.interaction.material.container.containers.ShulkerBoxContainer
 import com.lambda.interaction.material.transfer.TransferResult
 import com.lambda.task.Task
+import com.lambda.util.Communication.logError
 import com.lambda.util.Nameable
 import com.lambda.util.item.ItemStackUtils.count
 import com.lambda.util.item.ItemStackUtils.empty
 import com.lambda.util.item.ItemStackUtils.shulkerBoxContents
 import com.lambda.util.item.ItemStackUtils.spaceLeft
 import com.lambda.util.item.ItemUtils
 import com.lambda.util.item.ItemUtils.toItemCount
-import com.lambda.util.text.*
+import com.lambda.util.text.TextBuilder
+import com.lambda.util.text.TextDsl
+import com.lambda.util.text.buildText
+import com.lambda.util.text.highlighted
+import com.lambda.util.text.literal
+import com.lambda.util.text.text
 import net.minecraft.item.ItemStack
 import net.minecraft.text.Text
 
@@ -87,6 +97,20 @@ abstract class MaterialContainer(
         }
     }
 
+    class AwaitItemTask(override val name: String, val selection: StackSelection, inventory: InventoryConfig) : Task<Unit>() {
+        init {
+            listen<TickEvent.Post> {
+                if (selection.findContainerWithMaterial(inventory) != null) {
+                    success()
+                }
+            }
+        }
+
+        override fun SafeContext.onStart() {
+            logError(name)
+        }
+    }
+
     /**
      * Withdraws items from the container to the player's inventory.
      */