feat: Add reforge ability to reforge recipes

2 files changed, 24 insertions, 1 deletions

diff --git a/src/main/kotlin/repo/ItemCache.kt b/src/main/kotlin/repo/ItemCache.kt
index e140dd8..9fa0083 100644
--- a/src/main/kotlin/repo/ItemCache.kt
+++ b/src/main/kotlin/repo/ItemCache.kt
@@ -24,6 +24,7 @@ import net.minecraft.nbt.NbtCompound
 import net.minecraft.nbt.NbtElement
 import net.minecraft.nbt.NbtOps
 import net.minecraft.nbt.NbtString
+import net.minecraft.text.MutableText
 import net.minecraft.text.Style
 import net.minecraft.text.Text
 import moe.nea.firmament.Firmament
@@ -100,7 +101,7 @@ object ItemCache : IReloadable {
 		}
 	}
 
-	fun un189Lore(lore: String): Text {
+	fun un189Lore(lore: String): MutableText {
 		val base = Text.literal("")
 		base.setStyle(Style.EMPTY.withItalic(false))
 		var lastColorCode = Style.EMPTY
diff --git a/src/main/kotlin/repo/SBItemStack.kt b/src/main/kotlin/repo/SBItemStack.kt
index 4d07801..060e9b3 100644
--- a/src/main/kotlin/repo/SBItemStack.kt
+++ b/src/main/kotlin/repo/SBItemStack.kt
@@ -5,6 +5,7 @@ import com.mojang.serialization.codecs.RecordCodecBuilder
 import io.github.moulberry.repo.constants.PetNumbers
 import io.github.moulberry.repo.data.NEUIngredient
 import io.github.moulberry.repo.data.NEUItem
+import net.minecraft.client.util.ChatMessages
 import net.minecraft.item.ItemStack
 import net.minecraft.network.RegistryByteBuf
 import net.minecraft.network.codec.PacketCodec
@@ -17,8 +18,10 @@ import moe.nea.firmament.repo.ItemCache.asItemStack
 import moe.nea.firmament.repo.ItemCache.withFallback
 import moe.nea.firmament.util.FirmFormatters
 import moe.nea.firmament.util.LegacyFormattingCode
+import moe.nea.firmament.util.MC
 import moe.nea.firmament.util.ReforgeId
 import moe.nea.firmament.util.SkyblockId
+import moe.nea.firmament.util.blue
 import moe.nea.firmament.util.directLiteralStringContent
 import moe.nea.firmament.util.extraAttributes
 import moe.nea.firmament.util.getReforgeId
@@ -27,12 +30,15 @@ import moe.nea.firmament.util.grey
 import moe.nea.firmament.util.mc.appendLore
 import moe.nea.firmament.util.mc.displayNameAccordingToNbt
 import moe.nea.firmament.util.mc.loreAccordingToNbt
+import moe.nea.firmament.util.mc.modifyLore
 import moe.nea.firmament.util.petData
 import moe.nea.firmament.util.prepend
+import moe.nea.firmament.util.reconstitute
 import moe.nea.firmament.util.skyBlockId
 import moe.nea.firmament.util.skyblock.ItemType
 import moe.nea.firmament.util.skyblock.Rarity
 import moe.nea.firmament.util.skyblockId
+import moe.nea.firmament.util.unformattedString
 import moe.nea.firmament.util.useMatch
 import moe.nea.firmament.util.withColor
 
@@ -308,6 +314,22 @@ data class SBItemStack constructor(
 		data.putString("modifier", reforgeId.id)
 		itemStack.extraAttributes = data
 		appendEnhancedStats(itemStack, reforgeStats, BuffKind.REFORGE)
+		reforge.reforgeAbility?.get(rarity)?.let { reforgeAbility ->
+			val formattedReforgeAbility = ItemCache.un189Lore(reforgeAbility)
+				.grey()
+			itemStack.modifyLore {
+				val lastBlank = it.indexOfLast { it.unformattedString.isBlank() }
+				val newList = mutableListOf<Text>()
+				newList.addAll(it.subList(0, lastBlank))
+				newList.add(Text.literal(""))
+				newList.add(Text.literal("${reforge.reforgeName} Bonus").blue())
+				MC.font.textHandler.wrapLines(formattedReforgeAbility, 180, Style.EMPTY).mapTo(newList) {
+					it.reconstitute()
+				}
+				newList.addAll(it.subList(lastBlank, it.size))
+				return@modifyLore newList
+			}
+		}
 	}
 
 	// TODO: avoid instantiating the item stack here