From b7b01f1c6fbf889ae9bfcdb5b34c5ccfa48d5ba0 Mon Sep 17 00:00:00 2001
From: nea <nea@nea.moe>
Date: Thu, 31 Aug 2023 18:19:06 +0200
Subject: Fix profile viewer for missing uuid pets

---
 src/main/kotlin/moe/nea/firmament/apis/Profiles.kt | 25 +++++++++++-----------
 .../kotlin/moe/nea/firmament/apis/UrsaManager.kt   | 11 ++++++----
 2 files changed, 20 insertions(+), 16 deletions(-)

(limited to 'src/main/kotlin/moe/nea/firmament/apis')

diff --git a/src/main/kotlin/moe/nea/firmament/apis/Profiles.kt b/src/main/kotlin/moe/nea/firmament/apis/Profiles.kt
index 06fe31f..a7922c3 100644
--- a/src/main/kotlin/moe/nea/firmament/apis/Profiles.kt
+++ b/src/main/kotlin/moe/nea/firmament/apis/Profiles.kt
@@ -10,19 +10,20 @@ package moe.nea.firmament.apis
 
 import io.github.moulberry.repo.constants.Leveling
 import io.github.moulberry.repo.data.Rarity
-import java.util.*
 import kotlinx.datetime.Instant
 import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 import kotlinx.serialization.UseSerializers
-import kotlin.reflect.KProperty1
-import net.minecraft.util.DyeColor
-import net.minecraft.util.Formatting
 import moe.nea.firmament.repo.RepoManager
 import moe.nea.firmament.util.LegacyFormattingCode
 import moe.nea.firmament.util.SkyblockId
+import moe.nea.firmament.util.assertNotNullOr
 import moe.nea.firmament.util.json.DashlessUUIDSerializer
 import moe.nea.firmament.util.json.InstantAsLongSerializer
+import net.minecraft.util.DyeColor
+import net.minecraft.util.Formatting
+import java.util.*
+import kotlin.reflect.KProperty1
 
 
 @Serializable
@@ -81,7 +82,7 @@ enum class Skill(val accessor: KProperty1<Member, Double>, val color: DyeColor,
     ENCHANTING(Member::experienceSkillEnchanting, DyeColor.MAGENTA, SkyblockId("ENCHANTMENT_TABLE")),
     ;
 
-    fun getMaximumLevel(leveling: Leveling) = leveling.maximumLevels[name.lowercase()] ?: TODO("Repo error")
+    fun getMaximumLevel(leveling: Leveling) = assertNotNullOr(leveling.maximumLevels[name.lowercase()]) { 50 }
 
     fun getLadder(leveling: Leveling): List<Int> {
         if (this == SOCIAL) return leveling.socialExperienceRequiredPerLevel
@@ -149,14 +150,14 @@ value class PetType(val name: String)
 
 @Serializable
 data class Pet(
-    val uuid: UUID?,
+    val uuid: UUID? = null,
     val type: PetType,
-    val exp: Double,
-    val active: Boolean,
+    val exp: Double = 0.0,
+    val active: Boolean = false,
     val tier: Rarity,
-    val candyUsed: Int,
-    val heldItem: String?,
-    val skin: String?,
+    val candyUsed: Int = 0,
+    val heldItem: String? = null,
+    val skin: String? = null,
 ) {
     val itemId get() = SkyblockId("${type.name};${tier.ordinal}")
 }
@@ -184,7 +185,7 @@ data class PlayerData(
     val rankData get() = RepoManager.neuRepo.constants.misc.ranks[if (monthlyPackageRank == "NONE" || monthlyPackageRank == null) packageRank else monthlyPackageRank]
     fun getDisplayName(name: String = playerName) = rankData?.let {
         ("§${it.color}[${it.tag}${rankPlusDyeColor.modern}" +
-            "${it.plus ?: ""}§${it.color}] $name")
+                "${it.plus ?: ""}§${it.color}] $name")
     } ?: "${Formatting.GRAY}$name"
 
 
diff --git a/src/main/kotlin/moe/nea/firmament/apis/UrsaManager.kt b/src/main/kotlin/moe/nea/firmament/apis/UrsaManager.kt
index a72d633..8288283 100644
--- a/src/main/kotlin/moe/nea/firmament/apis/UrsaManager.kt
+++ b/src/main/kotlin/moe/nea/firmament/apis/UrsaManager.kt
@@ -9,14 +9,14 @@ package moe.nea.firmament.apis
 import io.ktor.client.request.*
 import io.ktor.client.statement.*
 import io.ktor.http.*
-import java.time.Duration
-import java.time.Instant
-import java.util.*
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.sync.Mutex
 import kotlinx.coroutines.withContext
-import net.minecraft.client.MinecraftClient
 import moe.nea.firmament.Firmament
+import net.minecraft.client.MinecraftClient
+import java.time.Duration
+import java.time.Instant
+import java.util.*
 
 object UrsaManager {
     private data class Token(
@@ -64,6 +64,9 @@ object UrsaManager {
                     ?: (Instant.now() + Duration.ofMinutes(55))
                 currentToken = Token(validUntil, savedToken, host)
             }
+            if (response.status.value != 200) {
+                Firmament.logger.error("Failed to contact ursa minor: ${response.bodyAsText()}")
+            }
             return response
         } finally {
             if (didLock)
-- 
cgit