From f37dd2c59a88bfb11b7c8ddd5d551a95cdc9119f Mon Sep 17 00:00:00 2001
From: dalbodeule <11470513+dalbodeule@users.noreply.github.com>
Date: Sun, 8 Jun 2025 16:02:22 +0900
Subject: [PATCH] [hotfix] add category command.

---
 .../chzzk_bot/chatbot/chzzk/MessageHandler.kt | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/chzzk/MessageHandler.kt b/chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/chzzk/MessageHandler.kt
index 8081257..b72a10e 100644
--- a/chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/chzzk/MessageHandler.kt
+++ b/chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/chzzk/MessageHandler.kt
@@ -15,6 +15,7 @@ import xyz.r2turntrue.chzzk4j.chat.ChatMessage
 import xyz.r2turntrue.chzzk4j.chat.ChzzkChat
 import xyz.r2turntrue.chzzk4j.session.ChzzkUserSession
 import xyz.r2turntrue.chzzk4j.session.message.SessionChatMessage
+import xyz.r2turntrue.chzzk4j.types.channel.live.ChzzkLiveSettings
 import java.time.LocalDateTime
 import java.time.format.DateTimeFormatter
 import java.time.temporal.ChronoUnit
@@ -62,6 +63,7 @@ class MessageHandler(
             "!신청곡" to this::songAddCommand,
             "!노래목록" to this::songListCommand,
             "!노래시작" to this::songStartCommand,
+            "!카테고리" to this::categoryChangeCommand,
         )
 
         manageCommands.forEach { (commandName, command) ->
@@ -316,6 +318,26 @@ class MessageHandler(
         }
     }
 
+    private fun categoryChangeCommand(msg: SessionChatMessage, user: User) {
+        val parts = msg.content.split(" ", limit = 2)
+        if(parts.size <= 1) {
+            handler.sendChat("카테고리가 없습니다.")
+            return
+        }
+        val category = parts[1]
+        handler.client.searchCategories(category).handle { result, _ ->
+            if(result.size == 0) {
+                handler.sendChat("$category 카테고리는 없습니다.")
+                return@handle
+            }
+            val settings = ChzzkLiveSettings()
+            settings.category = result[0]
+
+            handler.client.modifyLiveSettings(settings)
+            handler.sendChat("$category 로 수정했어요!")
+        }
+    }
+
     internal fun handle(msg: SessionChatMessage, user: User) {
         if(msg.senderChannelId == ChzzkHandler.botUid) return