From f0285e44ed113bcf2a11b5d6635f6124deb7fc3b Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Mon, 23 Dec 2024 02:22:06 +0100
Subject: feat: Add in game updater

---
 src/main/kotlin/moe/nea/ledger/utils/di/DI.kt | 37 ++++++++++++++++-----------
 1 file changed, 22 insertions(+), 15 deletions(-)

(limited to 'src/main/kotlin/moe/nea/ledger/utils/di/DI.kt')

diff --git a/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt b/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt
index 133637a..a9061d7 100644
--- a/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt
+++ b/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt
@@ -9,25 +9,32 @@ class DI {
 	private fun formatInjectionStack() =
 		injectionStack.joinToString(" -> ")
 
+	fun <T : Any> getProvider(type: Class<T>): BaseDIProvider<T, *> {
+		val provider = providers[type] as BaseDIProvider<T, *>?
+			?: error("Could not find provider for type $type")
+		return provider
+	}
+
 	private fun <T : Any, C> internalProvide(type: Class<T>, element: AnnotatedElement? = null): T {
-		val provider = providers[type] as BaseDIProvider<T, C>
-		val context = if (element == null) provider.createEmptyContext() else provider.createContext(element)
-		val key = Pair(type, context)
-		val existingValue = values[key]
-		if (existingValue != null) return existingValue as T
-		if (type in injectionStack) {
-			error("Found injection cycle: ${formatInjectionStack()} -> $type")
-		}
-		injectionStack.push(type)
-		val value = try {
-			provider.provideWithContext(this, context)
+		try {
+			val provider = getProvider(type) as BaseDIProvider<T, C>
+			val context = if (element == null) provider.createEmptyContext() else provider.createContext(element)
+			val key = Pair(type, context)
+			val existingValue = values[key]
+			if (existingValue != null) return existingValue as T
+			if (type in injectionStack) {
+				error("Found injection cycle: ${formatInjectionStack()} -> $type")
+			}
+			injectionStack.push(type)
+			val value =
+				provider.provideWithContext(this, context)
+			val cycleCheckCookie = injectionStack.pop()
+			require(cycleCheckCookie == type) { "Unbalanced stack cookie: $cycleCheckCookie != $type" }
+			values[key] = value
+			return value
 		} catch (ex: Exception) {
 			throw RuntimeException("Could not create instance for type $type (in stack ${formatInjectionStack()})", ex)
 		}
-		val cycleCheckCookie = injectionStack.pop()
-		require(cycleCheckCookie == type) { "Unbalanced stack cookie: $cycleCheckCookie != $type" }
-		values[key] = value
-		return value
 	}
 
 	fun <T : Any> provide(type: Class<T>, element: AnnotatedElement? = null): T {
-- 
cgit