1 files changed, 26 insertions, 10 deletions
diff --git a/src/main/kotlin/moe/nea/firmament/recipes/SBForgeRecipe.kt b/src/main/kotlin/moe/nea/firmament/recipes/SBForgeRecipe.kt
index 9f13cca..d70ebcd 100644
--- a/src/main/kotlin/moe/nea/firmament/recipes/SBForgeRecipe.kt
+++ b/src/main/kotlin/moe/nea/firmament/recipes/SBForgeRecipe.kt
@@ -27,10 +27,14 @@ import me.shedaniel.rei.api.client.gui.widgets.Widgets
 import me.shedaniel.rei.api.client.registry.display.DisplayCategory
 import me.shedaniel.rei.api.common.category.CategoryIdentifier
 import me.shedaniel.rei.api.common.util.EntryStacks
+import kotlin.math.cos
+import kotlin.math.sin
+import kotlin.time.Duration.Companion.seconds
 import net.minecraft.block.Blocks
 import net.minecraft.text.Text
 import moe.nea.firmament.Firmament
 import moe.nea.firmament.rei.SBItemEntryDefinition
+import moe.nea.firmament.rei.plus
 
 class SBForgeRecipe(override val neuRecipe: NEUForgeRecipe) : SBRecipe() {
     override fun getCategoryIdentifier(): CategoryIdentifier<*> = Category.categoryIdentifier
@@ -41,27 +45,39 @@ class SBForgeRecipe(override val neuRecipe: NEUForgeRecipe) : SBRecipe() {
 
         override fun getTitle(): Text = Text.literal("Forge Recipes")
         override fun getDisplayHeight(): Int {
-            return super.getDisplayHeight()
+            return 104
         }
 
         override fun getIcon(): Renderer = EntryStacks.of(Blocks.ANVIL)
         override fun setupDisplay(display: SBForgeRecipe, bounds: Rectangle): List<Widget> {
             return buildList {
-                // TODO: proper gui for this (possibly inspired by the old circular gui)
                 add(Widgets.createRecipeBase(bounds))
-                val resultSlot = Point(bounds.centerX, bounds.centerY + 5)
-                add(Widgets.createResultSlotBackground(resultSlot))
-                val ingredientsCenter = Point(bounds.centerX, bounds.centerY - 20)
+                add(Widgets.createResultSlotBackground(Point(bounds.minX + 124, bounds.minY + 46)))
+                val arrow = Widgets.createArrow(Point(bounds.minX + 90, bounds.minY + 54 - 18 / 2))
+                add(arrow)
+                add(Widgets.createTooltip(arrow.bounds, Text.translatable("firmament.recipe.forge.time", display.neuRecipe.duration.seconds)))
+                val ingredientsCenter = Point(bounds.minX + 49 - 8, bounds.minY + 54 - 8)
                 val count = display.neuRecipe.inputs.size
-                display.neuRecipe.inputs.forEachIndexed { idx, ingredient ->
+                if (count == 1) {
                     add(
-                        Widgets.createSlot(
-                            Point(ingredientsCenter.x + 12 - count * 24 / 2 + idx * 24, ingredientsCenter.y)
-                        ).markInput().entry(SBItemEntryDefinition.getEntry(ingredient))
+                        Widgets.createSlot(Point(ingredientsCenter.x, ingredientsCenter.y)).markInput()
+                            .entry(SBItemEntryDefinition.getEntry(display.neuRecipe.inputs.single()))
                     )
+                } else {
+                    display.neuRecipe.inputs.forEachIndexed { idx, ingredient ->
+                        val rad = Math.PI * 2 * idx / count
+                        add(
+                            Widgets.createSlot(
+                                Point(
+                                    cos(rad) * 30,
+                                    sin(rad) * 30,
+                                ) + ingredientsCenter
+                            ).markInput().entry(SBItemEntryDefinition.getEntry(ingredient))
+                        )
+                    }
                 }
                 add(
-                    Widgets.createSlot(resultSlot).markOutput().disableBackground()
+                    Widgets.createSlot(Point(bounds.minX + 124, bounds.minY + 46)).markOutput().disableBackground()
                         .entry(SBItemEntryDefinition.getEntry(display.neuRecipe.outputStack))
                 )
             }