feat: Load mining islands from repo

7 files changed, 118 insertions, 44 deletions

diff --git a/src/compat/jade/java/moe/nea/firmament/compat/jade/utils.kt b/src/compat/jade/java/moe/nea/firmament/compat/jade/utils.kt
index 520190b..0d949d8 100644
--- a/src/compat/jade/java/moe/nea/firmament/compat/jade/utils.kt
+++ b/src/compat/jade/java/moe/nea/firmament/compat/jade/utils.kt
@@ -1,13 +1,11 @@
 package moe.nea.firmament.compat.jade
 
-import snownee.jade.api.IWailaClientRegistration
 import snownee.jade.api.ui.IElement
 import snownee.jade.api.ui.IElementHelper
 import net.minecraft.item.ItemStack
 import net.minecraft.item.Items
 import net.minecraft.util.Identifier
 import moe.nea.firmament.util.SBData
-import moe.nea.firmament.util.SkyBlockIsland
 import moe.nea.firmament.util.SkyblockId
 import moe.nea.firmament.util.setSkyBlockId
 
@@ -17,16 +15,6 @@ fun String.jadeId(): Identifier = Identifier.of("firmament", this)
 // This drill icon should work for CIT resource packs
 val drillItem: ItemStack = Items.PRISMARINE_SHARD.defaultStack.setSkyBlockId(SkyblockId("TITANIUM_DRILL_1"))
 val drillIcon: IElement = IElementHelper.get().item(drillItem, 0.5f).message(null)
-fun IWailaClientRegistration.registerGemstone(type: String) {
 
-}
-
-fun isOnMiningIsland(): Boolean {
-	if (!SBData.isOnSkyblock) return false
-	// how does a when loop work
-	if (SBData.skyblockLocation == SkyBlockIsland.forMode("dwarven_mines")) return true
-	if (SBData.skyblockLocation == SkyBlockIsland.MINESHAFT) return true
-	if (SBData.skyblockLocation == SkyBlockIsland.forMode("crystal_hollows")) return true
-	if (SBData.skyblockLocation == SkyBlockIsland.forMode("crimson_isle")) return true
-	return false
-}
+fun isOnMiningIsland(): Boolean =
+	SBData.skyblockLocation?.hasCustomMining ?: false
diff --git a/src/main/kotlin/commands/rome.kt b/src/main/kotlin/commands/rome.kt
index 5412792..6fc01e5 100644
--- a/src/main/kotlin/commands/rome.kt
+++ b/src/main/kotlin/commands/rome.kt
@@ -246,6 +246,7 @@ fun firmamentCommand() = literal("firmament") {
 					source.sendFeedback(Text.stringifiedTranslatable("firmament.sbinfo.gametype", locrawInfo.gametype))
 					source.sendFeedback(Text.stringifiedTranslatable("firmament.sbinfo.mode", locrawInfo.mode))
 					source.sendFeedback(Text.stringifiedTranslatable("firmament.sbinfo.map", locrawInfo.map))
+					source.sendFeedback(tr("firmament.sbinfo.custommining", "Custom Mining: ${formatBool(locrawInfo.skyblockLocation?.hasCustomMining ?: false)}"))
 				}
 			}
 		}
diff --git a/src/main/kotlin/repo/ItemCache.kt b/src/main/kotlin/repo/ItemCache.kt
index c672ac1..0967ad1 100644
--- a/src/main/kotlin/repo/ItemCache.kt
+++ b/src/main/kotlin/repo/ItemCache.kt
@@ -61,11 +61,12 @@ object ItemCache : IReloadable {
 		putShort("Damage", damage.toShort())
 	}
 
-	private fun NbtCompound.transformFrom10809ToModern(): NbtCompound? =
+	private fun NbtCompound.transformFrom10809ToModern() = convert189ToModern(this@transformFrom10809ToModern)
+	fun convert189ToModern(nbtComponent: NbtCompound): NbtCompound? =
 		try {
 			df.update(
 				TypeReferences.ITEM_STACK,
-				Dynamic(NbtOps.INSTANCE, this),
+				Dynamic(NbtOps.INSTANCE, nbtComponent),
 				-1,
 				SharedConstants.getGameVersion().saveVersion.id
 			).value as NbtCompound
diff --git a/src/main/kotlin/repo/MiningRepoData.kt b/src/main/kotlin/repo/MiningRepoData.kt
new file mode 100644
index 0000000..bbaeea3
--- /dev/null
+++ b/src/main/kotlin/repo/MiningRepoData.kt
@@ -0,0 +1,72 @@
+package moe.nea.firmament.repo
+
+import io.github.moulberry.repo.IReloadable
+import io.github.moulberry.repo.NEURepository
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.Transient
+import kotlinx.serialization.serializer
+import kotlin.jvm.optionals.getOrNull
+import kotlin.streams.asSequence
+import net.minecraft.block.Block
+import net.minecraft.item.BlockItem
+import net.minecraft.item.ItemStack
+import net.minecraft.nbt.NbtCompound
+import moe.nea.firmament.repo.ReforgeStore.kJson
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.SkyBlockIsland
+import moe.nea.firmament.util.SkyblockId
+
+class MiningRepoData : IReloadable {
+	var customMiningAreas: Map<SkyBlockIsland, CustomMiningArea> = mapOf()
+		private set
+	var customMiningBlocks: List<CustomMiningBlock> = listOf()
+		private set
+
+
+	override fun reload(repo: NEURepository) {
+		customMiningAreas = repo.file("mining/custom_mining_areas.json")
+			?.kJson(serializer()) ?: mapOf()
+		customMiningBlocks = repo.tree("mining/blocks")
+			.asSequence()
+			.filter { it.path.endsWith(".json") }
+			.map { it.kJson(serializer<CustomMiningBlock>()) }
+			.toList()
+	}
+
+	@Serializable
+	data class CustomMiningBlock(
+		val breakingPower: Int = 0,
+		val blockStrength: Int = 0,
+		val name: String? = null,
+		val baseDrop: SkyblockId? = null,
+		val blocks189: List<Block189> = emptyList()
+	)
+
+	@Serializable
+	data class Block189(
+		val itemId: String,
+		val damage: Short = 0,
+		val onlyIn: List<SkyBlockIsland>? = null,
+	) {
+		@Transient
+		val block = convertToModernBlock()
+
+		private fun convertToModernBlock(): Block? {
+			// TODO: this should be in a shared util, really
+			val newCompound = ItemCache.convert189ToModern(NbtCompound().apply {
+				putString("id", itemId)
+				putShort("Damage", damage)
+			}) ?: return null
+			val itemStack = ItemStack.fromNbt(MC.defaultRegistries, newCompound).getOrNull() ?: return null
+			val blockItem = itemStack.item as? BlockItem ?: return null
+			return blockItem.block
+		}
+	}
+
+	@Serializable
+	data class CustomMiningArea(
+		val isSpecialMining: Boolean = true
+	)
+
+
+}
diff --git a/src/main/kotlin/repo/RepoManager.kt b/src/main/kotlin/repo/RepoManager.kt
index 84b9a41..5bcf163 100644
--- a/src/main/kotlin/repo/RepoManager.kt
+++ b/src/main/kotlin/repo/RepoManager.kt
@@ -54,6 +54,7 @@ object RepoManager {
 
 	val essenceRecipeProvider = EssenceRecipeProvider()
 	val recipeCache = BetterRepoRecipeCache(essenceRecipeProvider, ReforgeStore)
+	val miningData = MiningRepoData()
 
 	fun makeNEURepository(path: Path): NEURepository {
 		return NEURepository.of(path).apply {
@@ -63,6 +64,7 @@ object RepoManager {
 			registerReloadListener(ItemNameLookup)
 			registerReloadListener(ReforgeStore)
 			registerReloadListener(essenceRecipeProvider)
+			registerReloadListener(miningData)
 			ReloadRegistrationEvent.publish(ReloadRegistrationEvent(this))
 			registerReloadListener {
 				if (TestUtil.isInTest) return@registerReloadListener
diff --git a/src/main/kotlin/util/SBData.kt b/src/main/kotlin/util/SBData.kt
index b2f9449..1a4734c 100644
--- a/src/main/kotlin/util/SBData.kt
+++ b/src/main/kotlin/util/SBData.kt
@@ -31,6 +31,10 @@ object SBData {
 	val hypixelTimeZone = ZoneId.of("US/Eastern")
 	private var hasReceivedProfile = false
 	var locraw: Locraw? = null
+
+	/**
+	 * The current server location the player is in. This will be null outside of SkyBlock.
+	 */
 	val skyblockLocation: SkyBlockIsland? get() = locraw?.skyblockLocation
 	val hasValidLocraw get() = locraw?.server !in listOf("limbo", null)
 	val isOnSkyblock get() = locraw?.gametype == "SKYBLOCK"
diff --git a/src/main/kotlin/util/SkyBlockIsland.kt b/src/main/kotlin/util/SkyBlockIsland.kt
index a86543c..1f00915 100644
--- a/src/main/kotlin/util/SkyBlockIsland.kt
+++ b/src/main/kotlin/util/SkyBlockIsland.kt
@@ -1,4 +1,3 @@
-
 package moe.nea.firmament.util
 
 import kotlinx.serialization.KSerializer
@@ -13,33 +12,40 @@ import moe.nea.firmament.repo.RepoManager
 @Serializable(with = SkyBlockIsland.Serializer::class)
 class SkyBlockIsland
 private constructor(
-    val locrawMode: String,
+	val locrawMode: String,
 ) {
 
-    object Serializer : KSerializer<SkyBlockIsland> {
-        override val descriptor: SerialDescriptor
-            get() = PrimitiveSerialDescriptor("SkyBlockIsland", PrimitiveKind.STRING)
-
-        override fun deserialize(decoder: Decoder): SkyBlockIsland {
-            return forMode(decoder.decodeString())
-        }
-
-        override fun serialize(encoder: Encoder, value: SkyBlockIsland) {
-            encoder.encodeString(value.locrawMode)
-        }
-    }
-    companion object {
-        private val allIslands = mutableMapOf<String, SkyBlockIsland>()
-        fun forMode(mode: String): SkyBlockIsland = allIslands.computeIfAbsent(mode, ::SkyBlockIsland)
-        val HUB = forMode("hub")
-        val PRIVATE_ISLAND = forMode("dynamic")
-        val RIFT = forMode("rift")
-        val MINESHAFT = forMode("mineshaft")
-	    val GARDEN = forMode("garden")
-	    val DUNGEON = forMode("dungeon")
-    }
-
-    val userFriendlyName
-        get() = RepoManager.neuRepo.constants.islands.areaNames
-            .getOrDefault(locrawMode, locrawMode)
+	object Serializer : KSerializer<SkyBlockIsland> {
+		override val descriptor: SerialDescriptor
+			get() = PrimitiveSerialDescriptor("SkyBlockIsland", PrimitiveKind.STRING)
+
+		override fun deserialize(decoder: Decoder): SkyBlockIsland {
+			return forMode(decoder.decodeString())
+		}
+
+		override fun serialize(encoder: Encoder, value: SkyBlockIsland) {
+			encoder.encodeString(value.locrawMode)
+		}
+	}
+
+	companion object {
+		private val allIslands = mutableMapOf<String, SkyBlockIsland>()
+		fun forMode(mode: String): SkyBlockIsland = allIslands.computeIfAbsent(mode, ::SkyBlockIsland)
+		val HUB = forMode("hub")
+		val DWARVEN_MINES = forMode("dwarven_mines")
+		val CRYSTAL_HOLLOWS = forMode("crystal_hollows")
+		val CRIMSON_ISLE = forMode("crimson_isle")
+		val PRIVATE_ISLAND = forMode("dynamic")
+		val RIFT = forMode("rift")
+		val MINESHAFT = forMode("mineshaft")
+		val GARDEN = forMode("garden")
+		val DUNGEON = forMode("dungeon")
+	}
+
+	val hasCustomMining
+		get() = RepoManager.miningData.customMiningAreas[this]?.isSpecialMining ?: false
+
+	val userFriendlyName
+		get() = RepoManager.neuRepo.constants.islands.areaNames
+			.getOrDefault(locrawMode, locrawMode)
 }