Logout on exit with DisposableBean.

Author: JustinFreitas

src/main/kotlin/dev/arbjerg/ukulele/jda/JdaConfig.kt

@@ -10,6 +10,10 @@ import net.dv8tion.jda.api.utils.cache.CacheFlag
 import net.dv8tion.jda.api.utils.messages.MessageRequest
 import org.springframework.context.annotation.Bean
 import org.springframework.context.annotation.Configuration
+import org.springframework.beans.factory.DisposableBean
+import org.springframework.stereotype.Component
+import java.io.File
+import java.time.Instant
 import javax.security.auth.login.LoginException
 import kotlin.concurrent.thread
 
@@ -55,25 +59,28 @@ class JdaConfig {
             throw RuntimeException("Failed to log in to Discord! Is your token invalid?", e)
         }
 
-        Runtime.getRuntime().addShutdownHook(thread(start = false) {
-            val logFile = java.io.File("shutdown-debug.log")
-            logFile.appendText("Shutdown hook started at ${java.time.Instant.now()}\n")
-            try {
-                shardManager.setStatus(OnlineStatus.OFFLINE)
-                logFile.appendText("Set status to OFFLINE\n")
-                shardManager.guildCache.forEach {
-                    if (it.audioManager.isConnected) it.audioManager.closeAudioConnection()
-                }
-                logFile.appendText("Closed audio connections\n")
-                shardManager.shutdown()
-                logFile.appendText("Called shardManager.shutdown()\n")
-                shardManager.shards.forEach { it.awaitShutdown() }
-                logFile.appendText("Awaited shutdown. Complete.\n")
-            } catch (e: Exception) {
-                logFile.appendText("Error in shutdown hook: ${e.stackTraceToString()}\n")
-            }
-        })
-
         return shardManager
     }
+}
+
+@Component
+class JdaLifecycleManager(private val shardManager: ShardManager) : DisposableBean {
+    override fun destroy() {
+        val logFile = File("shutdown-debug.log")
+        logFile.appendText("Spring DisposableBean destroy() started at ${Instant.now()}\n")
+        try {
+            shardManager.setStatus(OnlineStatus.OFFLINE)
+            logFile.appendText("Set status to OFFLINE\n")
+            shardManager.guildCache.forEach {
+                if (it.audioManager.isConnected) it.audioManager.closeAudioConnection()
+            }
+            logFile.appendText("Closed audio connections\n")
+            shardManager.shutdown()
+            logFile.appendText("Called shardManager.shutdown()\n")
+            shardManager.shards.forEach { it.awaitShutdown() }
+            logFile.appendText("Awaited shutdown. Complete.\n")
+        } catch (e: Exception) {
+            logFile.appendText("Error in shutdown hook: ${e.stackTraceToString()}\n")
+        }
+    }
 }