From ddebaf47900dfab41590c97c202984142ae5b9f6 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Wed, 25 Dec 2024 16:15:20 +0100
Subject: WIP: Reforge Recipes

---
 src/main/kotlin/repo/Reforge.kt | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'src/main/kotlin/repo/Reforge.kt')

diff --git a/src/main/kotlin/repo/Reforge.kt b/src/main/kotlin/repo/Reforge.kt
index ea01818..b52adc6 100644
--- a/src/main/kotlin/repo/Reforge.kt
+++ b/src/main/kotlin/repo/Reforge.kt
@@ -108,6 +108,8 @@ data class Reforge(
 
 	@Serializable(with = RarityMapped.Serializer::class)
 	sealed interface RarityMapped<T> {
+		fun get(rarity: Rarity): T?
+
 		class Serializer<T>(
 			val values: KSerializer<T>
 		) : KSerializer<RarityMapped<T>> {
@@ -137,10 +139,18 @@ data class Reforge(
 		}
 
 		@Serializable
-		data class Direct<T>(val value: T) : RarityMapped<T>
+		data class Direct<T>(val value: T) : RarityMapped<T> {
+			override fun get(rarity: Rarity): T {
+				return value
+			}
+		}
 
 		@Serializable
-		data class PerRarity<T>(val values: Map<Rarity, T>) : RarityMapped<T>
+		data class PerRarity<T>(val values: Map<Rarity, T>) : RarityMapped<T> {
+			override fun get(rarity: Rarity): T? {
+				return values[rarity]
+			}
+		}
 	}
 
 }
-- 
cgit