1 files changed, 32 insertions, 18 deletions

diff --git a/src/main/kotlin/repo/RepoManager.kt b/src/main/kotlin/repo/RepoManager.kt
index 667ab73..df89092 100644
--- a/src/main/kotlin/repo/RepoManager.kt
+++ b/src/main/kotlin/repo/RepoManager.kt
@@ -11,6 +11,7 @@ import kotlinx.coroutines.launch
 import net.minecraft.client.MinecraftClient
 import net.minecraft.network.packet.s2c.play.SynchronizeRecipesS2CPacket
 import net.minecraft.recipe.display.CuttingRecipeDisplay
+import net.minecraft.util.StringIdentifiable
 import moe.nea.firmament.Firmament
 import moe.nea.firmament.Firmament.logger
 import moe.nea.firmament.events.ReloadRegistrationEvent
@@ -46,6 +47,16 @@ object RepoManager {
 		}
 		val alwaysSuperCraft by toggle("enable-super-craft") { true }
 		var warnForMissingItemListMod by toggle("warn-for-missing-item-list-mod") { true }
+		val perfectRenders by choice("perfect-renders") { PerfectRender.RENDER }
+	}
+
+	enum class PerfectRender(val label: String) : StringIdentifiable {
+		NOTHING("nothing"),
+		RENDER("render"),
+		RENDER_AND_TEXT("text"),
+		;
+
+		override fun asString(): String? = label
 	}
 
 	val currentDownloadedSha by RepoDownloadManager::latestSavedVersionHash
@@ -53,13 +64,21 @@ object RepoManager {
 	var recentlyFailedToUpdateItemList = false
 
 	val essenceRecipeProvider = EssenceRecipeProvider()
-	val recipeCache = BetterRepoRecipeCache(essenceRecipeProvider)
+	val recipeCache = BetterRepoRecipeCache(essenceRecipeProvider, ReforgeStore)
+	val miningData = MiningRepoData()
+	val overlayData = ModernOverlaysData()
 
 	fun makeNEURepository(path: Path): NEURepository {
 		return NEURepository.of(path).apply {
+			registerReloadListener(overlayData)
 			registerReloadListener(ItemCache)
+			registerReloadListener(RepoItemTypeCache)
 			registerReloadListener(ExpLadders)
 			registerReloadListener(ItemNameLookup)
+			registerReloadListener(ReforgeStore)
+			registerReloadListener(essenceRecipeProvider)
+			registerReloadListener(recipeCache)
+			registerReloadListener(miningData)
 			ReloadRegistrationEvent.publish(ReloadRegistrationEvent(this))
 			registerReloadListener {
 				if (TestUtil.isInTest) return@registerReloadListener
@@ -70,8 +89,6 @@ object RepoManager {
 					}
 				}
 			}
-			registerReloadListener(essenceRecipeProvider)
-			registerReloadListener(recipeCache)
 		}
 	}
 
@@ -98,16 +115,16 @@ object RepoManager {
 
 	fun getNEUItem(skyblockId: SkyblockId): NEUItem? = neuRepo.items.getItemBySkyblockId(skyblockId.neuItem)
 
+	fun downloadOverridenBranch(branch: String) {
+		Firmament.coroutineScope.launch {
+			RepoDownloadManager.downloadUpdate(true, branch)
+			reload()
+		}
+	}
+
 	fun launchAsyncUpdate(force: Boolean = false) {
 		Firmament.coroutineScope.launch {
-			ItemCache.ReloadProgressHud.reportProgress("Downloading", 0, -1) // TODO: replace with a proper bouncy bar
-			ItemCache.ReloadProgressHud.isEnabled = true
-			try {
-				RepoDownloadManager.downloadUpdate(force)
-				ItemCache.ReloadProgressHud.reportProgress("Download complete", 1, 1)
-			} finally {
-				ItemCache.ReloadProgressHud.isEnabled = false
-			}
+			RepoDownloadManager.downloadUpdate(force)
 			reload()
 		}
 	}
@@ -125,20 +142,17 @@ object RepoManager {
 			return
 		}
 		try {
-			ItemCache.ReloadProgressHud.reportProgress("Reloading from Disk",
-			                                           0,
-			                                           -1) // TODO: replace with a proper bouncy bar
-			ItemCache.ReloadProgressHud.isEnabled = true
 			logger.info("Repo reload started.")
 			neuRepo.reload()
 			logger.info("Repo reload completed.")
 		} catch (exc: NEURepositoryException) {
 			ErrorUtil.softError("Failed to reload repository", exc)
 			MC.sendChat(
-				tr("firmament.repo.reloadfail",
-				   "Failed to reload repository. This will result in some mod features not working.")
+				tr(
+					"firmament.repo.reloadfail",
+					"Failed to reload repository. This will result in some mod features not working."
+				)
 			)
-			ItemCache.ReloadProgressHud.isEnabled = false
 		}
 	}