From ab32fcc7dc7423c5e8a2ebdc950e7aa8ce6eca06 Mon Sep 17 00:00:00 2001
From: dalbodeule <11470513+dalbodeule@users.noreply.github.com>
Date: Wed, 14 Aug 2024 20:40:20 +0900
Subject: [PATCH] asdf12

---
 .../mori/chzzk_bot/webserver/utils/DiscordGuildCache.kt     | 6 +++---
 .../mori/chzzk_bot/webserver/utils/DiscordRatelimits.kt     | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordGuildCache.kt b/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordGuildCache.kt
index e166d2d..cdd5098 100644
--- a/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordGuildCache.kt
+++ b/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordGuildCache.kt
@@ -29,7 +29,7 @@ object DiscordGuildCache {
         return guildId.mapNotNull { getCachedGuilds(it) }
     }
 
-    private suspend fun fetchGuilds(beforeGuildId: String? = null, limit: Int = 100): List<DiscordGuildListAPI> {
+    private suspend fun fetchGuilds(beforeGuildId: String? = null): List<DiscordGuildListAPI> {
         if(DiscordRatelimits.isLimited()) {
             delay(DiscordRatelimits.getRateReset().takeIf { it > 1000L } ?: 3000L)
         }
@@ -37,7 +37,7 @@ object DiscordGuildCache {
             headers {
                 append(HttpHeaders.Authorization, "Bot ${dotenv["DISCORD_TOKEN"]}")
             }
-            parameter("limit", limit)
+            parameter("limit", 200)
             if (beforeGuildId != null) {
                 parameter("before", beforeGuildId)
             }
@@ -57,7 +57,7 @@ object DiscordGuildCache {
         while (true) {
             try {
                 val guilds = fetchGuilds(lastGuildId)
-                if (guilds.isEmpty()) {
+                if (guilds.isEmpty() || guilds.size <= 200) {
                     break
                 }
 
diff --git a/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordRatelimits.kt b/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordRatelimits.kt
index 3a4f795..90a40f8 100644
--- a/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordRatelimits.kt
+++ b/webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordRatelimits.kt
@@ -7,7 +7,7 @@ object DiscordRatelimits {
         return rateLimit.remainin == 0
     }
 
-    fun getRateReset() = rateLimit.resetAfter * 1000L + 300L
+    fun getRateReset() = (rateLimit.resetAfter * 1000) + 300L
 
     private fun setRateLimit(rateLimit: RateLimit) {
         this.rateLimit = rateLimit