mirror of
https://github.com/dalbodeule/chibot-chzzk-bot.git
synced 2025-08-07 21:01:14 +00:00
add webserver, etc...
- add ktor webserver - package is subpackaged.
This commit is contained in:
50
chatbot/build.gradle.kts
Normal file
50
chatbot/build.gradle.kts
Normal file
@@ -0,0 +1,50 @@
|
||||
plugins {
|
||||
kotlin("jvm")
|
||||
}
|
||||
|
||||
group = project.rootProject.group
|
||||
version = project.rootProject.version
|
||||
|
||||
repositories {
|
||||
mavenCentral()
|
||||
}
|
||||
|
||||
dependencies {
|
||||
// https://mvnrepository.com/artifact/net.dv8tion/JDA
|
||||
implementation("net.dv8tion:JDA:5.0.1") {
|
||||
exclude(module = "opus-java")
|
||||
}
|
||||
|
||||
// https://mvnrepository.com/artifact/io.github.R2turnTrue/chzzk4j
|
||||
implementation("io.github.R2turnTrue:chzzk4j:0.0.9")
|
||||
|
||||
// https://mvnrepository.com/artifact/ch.qos.logback/logback-classic
|
||||
implementation("ch.qos.logback:logback-classic:1.5.6")
|
||||
|
||||
// https://mvnrepository.com/artifact/org.jetbrains.kotlinx/kotlinx-coroutines-core
|
||||
implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.9.0-RC")
|
||||
// https://mvnrepository.com/artifact/org.jetbrains.kotlin/kotlin-reflect
|
||||
implementation("org.jetbrains.kotlin:kotlin-reflect:2.0.0")
|
||||
|
||||
// https://mvnrepository.com/artifact/com.google.code.gson/gson
|
||||
implementation("com.google.code.gson:gson:2.11.0")
|
||||
|
||||
// https://mvnrepository.com/artifact/io.github.cdimascio/dotenv-kotlin
|
||||
implementation("io.github.cdimascio:dotenv-kotlin:6.4.1")
|
||||
|
||||
// https://mvnrepository.com/artifact/com.squareup.okhttp3/okhttp
|
||||
implementation("com.squareup.okhttp3:okhttp:4.12.0")
|
||||
|
||||
testImplementation(kotlin("test"))
|
||||
|
||||
listOf(project(":common")).forEach {
|
||||
implementation(it)
|
||||
}
|
||||
}
|
||||
|
||||
tasks.test {
|
||||
useJUnitPlatform()
|
||||
}
|
||||
kotlin {
|
||||
jvmToolchain(21)
|
||||
}
|
@@ -0,0 +1,148 @@
|
||||
package space.mori.chzzk_bot.chatbot.chzzk
|
||||
|
||||
import com.google.gson.Gson
|
||||
import com.google.gson.reflect.TypeToken
|
||||
import okhttp3.OkHttpClient
|
||||
import okhttp3.Request
|
||||
import java.io.IOException
|
||||
|
||||
data class IData<T>(
|
||||
val code: Int = 200,
|
||||
val message: String? = null,
|
||||
val content: T
|
||||
)
|
||||
|
||||
// Follows
|
||||
data class IFollowContent(
|
||||
val userIdHash: String = "",
|
||||
val nickname: String = "",
|
||||
val profileImageUrl: String = "",
|
||||
val userRoleCode: String = "",
|
||||
val badge: Badge? = null,
|
||||
val title: Title? = null,
|
||||
val verifiedMark: Boolean = false,
|
||||
val activityBadges: List<Badge> = emptyList(),
|
||||
val streamingProperty: StreamingProperty = StreamingProperty()
|
||||
)
|
||||
|
||||
data class Badge(
|
||||
val badgeNo: Int?,
|
||||
val badgeId: String?,
|
||||
val imageUrl: String?,
|
||||
val title: String?,
|
||||
val description: String?,
|
||||
val activated: Boolean?
|
||||
)
|
||||
|
||||
data class Title(
|
||||
val name: String = "",
|
||||
val color: String = ""
|
||||
)
|
||||
|
||||
data class StreamingProperty(
|
||||
val following: Following? = Following(),
|
||||
val nicknameColor: NicknameColor = NicknameColor()
|
||||
)
|
||||
|
||||
data class Following(
|
||||
val followDate: String? = null
|
||||
)
|
||||
|
||||
data class NicknameColor(
|
||||
val colorCode: String = ""
|
||||
)
|
||||
|
||||
// Stream info
|
||||
data class IStreamInfo(
|
||||
val liveId: Int = 0,
|
||||
val liveTitle: String = "",
|
||||
val status: String = "",
|
||||
val liveImageUrl: String = "",
|
||||
val defaultThumbnailImageUrl: String? = null,
|
||||
val concurrentUserCount: Int = 0,
|
||||
val accumulateCount: Int = 0,
|
||||
val openDate: String = "",
|
||||
val closeDate: String = "",
|
||||
val adult: Boolean = false,
|
||||
val clipActive: Boolean = false,
|
||||
val tags: List<String> = emptyList(),
|
||||
val chatChannelId: String = "",
|
||||
val categoryType: String = "",
|
||||
val liveCategory: String = "",
|
||||
val liveCategoryValue: String = "",
|
||||
val chatActive: Boolean = true,
|
||||
val chatAvailableGroup: String = "",
|
||||
val paidPromotion: Boolean = false,
|
||||
val chatAvailableCondition: String = "",
|
||||
val minFollowerMinute: Int = 0,
|
||||
val livePlaybackJson: String = "",
|
||||
val p2pQuality: List<Any> = emptyList(),
|
||||
val channel: Channel = Channel(),
|
||||
val livePollingStatusJson: String = "",
|
||||
val userAdultStatus: String? = null,
|
||||
val chatDonationRankingExposure: Boolean = true,
|
||||
val adParameter: AdParameter = AdParameter()
|
||||
)
|
||||
|
||||
data class Channel(
|
||||
val channelId: String = "",
|
||||
val channelName: String = "",
|
||||
val channelImageUrl: String = "",
|
||||
val verifiedMark: Boolean = false
|
||||
)
|
||||
|
||||
data class AdParameter(
|
||||
val tag: String = ""
|
||||
)
|
||||
|
||||
// OkHttpClient에 Interceptor 추가
|
||||
val client = OkHttpClient.Builder()
|
||||
.addNetworkInterceptor { chain ->
|
||||
chain.proceed(
|
||||
chain.request()
|
||||
.newBuilder()
|
||||
.header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
|
||||
.build()
|
||||
)
|
||||
}
|
||||
.build()
|
||||
val gson = Gson()
|
||||
|
||||
fun getFollowDate(chatID: String, userId: String) : IData<IFollowContent> {
|
||||
val url = "https://comm-api.game.naver.com/nng_main/v1/chats/$chatID/users/$userId/profile-card?chatType=STREAMING"
|
||||
val request = Request.Builder()
|
||||
.url(url)
|
||||
.build()
|
||||
|
||||
client.newCall(request).execute().use { response ->
|
||||
try {
|
||||
if(!response.isSuccessful) throw IOException("Unexpected code ${response.code}")
|
||||
val body = response.body?.string()
|
||||
val follow = gson.fromJson(body, object: TypeToken<IData<IFollowContent>>() {})
|
||||
|
||||
return follow
|
||||
} catch(e: Exception) {
|
||||
println(e.stackTrace)
|
||||
throw e
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun getStreamInfo(userId: String) : IData<IStreamInfo> {
|
||||
val url = "https://api.chzzk.naver.com/service/v2/channels/${userId}/live-detail"
|
||||
val request = Request.Builder()
|
||||
.url(url)
|
||||
.build()
|
||||
|
||||
client.newCall(request).execute().use { response ->
|
||||
try {
|
||||
if(!response.isSuccessful) throw IOException("Unexpected code ${response.code}")
|
||||
val body = response.body?.string()
|
||||
val follow = gson.fromJson(body, object: TypeToken<IData<IStreamInfo>>() {})
|
||||
|
||||
return follow
|
||||
} catch(e: Exception) {
|
||||
throw e
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,148 @@
|
||||
package space.mori.chzzk_bot.chatbot.chzzk
|
||||
|
||||
import org.slf4j.Logger
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.Connector.chzzk
|
||||
import space.mori.chzzk_bot.chatbot.discord.Discord
|
||||
import space.mori.chzzk_bot.common.models.User
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
import xyz.r2turntrue.chzzk4j.chat.ChatEventListener
|
||||
import xyz.r2turntrue.chzzk4j.chat.ChatMessage
|
||||
import xyz.r2turntrue.chzzk4j.chat.ChzzkChat
|
||||
import xyz.r2turntrue.chzzk4j.types.channel.ChzzkChannel
|
||||
import java.lang.Exception
|
||||
import java.net.SocketTimeoutException
|
||||
|
||||
object ChzzkHandler {
|
||||
private val handlers = mutableListOf<UserHandler>()
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
@Volatile private var running: Boolean = false
|
||||
|
||||
fun addUser(chzzkChannel: ChzzkChannel, user: User) {
|
||||
handlers.add(UserHandler(chzzkChannel, logger, user))
|
||||
}
|
||||
|
||||
fun enable() {
|
||||
UserService.getAllUsers().map {
|
||||
chzzk.getChannel(it.token)?.let { token -> addUser(token, it) }
|
||||
}
|
||||
}
|
||||
|
||||
fun disable() {
|
||||
handlers.forEach { handler ->
|
||||
handler.disable()
|
||||
}
|
||||
}
|
||||
|
||||
internal fun reloadCommand(chzzkChannel: ChzzkChannel) {
|
||||
val handler = handlers.firstOrNull { it.channel.channelId == chzzkChannel.channelId }
|
||||
if (handler != null)
|
||||
handler.reloadCommand()
|
||||
else
|
||||
throw RuntimeException("${chzzkChannel.channelName} doesn't have handler")
|
||||
}
|
||||
|
||||
internal fun reloadUser(chzzkChannel: ChzzkChannel, user: User) {
|
||||
val handler = handlers.firstOrNull { it.channel.channelId == chzzkChannel.channelId }
|
||||
if (handler != null)
|
||||
handler.reloadUser(user)
|
||||
else
|
||||
throw RuntimeException("${chzzkChannel.channelName} doesn't have handler")
|
||||
}
|
||||
|
||||
fun runStreamInfo() {
|
||||
running = true
|
||||
val thread = Thread({
|
||||
while(running) {
|
||||
handlers.forEach {
|
||||
if (!running) return@forEach
|
||||
try {
|
||||
val streamInfo = getStreamInfo(it.channel.channelId)
|
||||
if (streamInfo.content.status == "OPEN" && !it.isActive) it.isActive(true, streamInfo)
|
||||
if (streamInfo.content.status == "CLOSE" && it.isActive) it.isActive(false, streamInfo)
|
||||
} catch(e: SocketTimeoutException) {
|
||||
logger.info("Timeout: ${it.channel.channelName} / ${e.stackTraceToString()}")
|
||||
} catch (e: Exception) {
|
||||
logger.info("Exception: ${it.channel.channelName} / ${e.stackTraceToString()}")
|
||||
} finally {
|
||||
Thread.sleep(5000)
|
||||
}
|
||||
}
|
||||
Thread.sleep(60000)
|
||||
}
|
||||
}, "Chzzk-StreamInfo")
|
||||
|
||||
thread.start()
|
||||
}
|
||||
|
||||
fun stopStreamInfo() {
|
||||
running = false
|
||||
}
|
||||
}
|
||||
|
||||
class UserHandler(
|
||||
val channel: ChzzkChannel,
|
||||
private val logger: Logger,
|
||||
private var user: User,
|
||||
private var _isActive: Boolean = false
|
||||
) {
|
||||
private lateinit var messageHandler: MessageHandler
|
||||
|
||||
private var listener: ChzzkChat = chzzk.chat(channel.channelId)
|
||||
.withAutoReconnect(true)
|
||||
.withChatListener(object : ChatEventListener {
|
||||
override fun onConnect(chat: ChzzkChat, isReconnecting: Boolean) {
|
||||
logger.info("ChzzkChat connected. ${channel.channelName} - ${channel.channelId} / reconnected: $isReconnecting")
|
||||
messageHandler = MessageHandler(channel, logger, chat)
|
||||
}
|
||||
|
||||
override fun onError(ex: Exception) {
|
||||
logger.info("ChzzkChat error. ${channel.channelName} - ${channel.channelId}")
|
||||
logger.debug(ex.stackTraceToString())
|
||||
}
|
||||
|
||||
override fun onChat(msg: ChatMessage) {
|
||||
if(!_isActive) return
|
||||
messageHandler.handle(msg, user)
|
||||
}
|
||||
|
||||
override fun onConnectionClosed(code: Int, reason: String?, remote: Boolean, tryingToReconnect: Boolean) {
|
||||
logger.info("ChzzkChat closed. ${channel.channelName} - ${channel.channelId}")
|
||||
logger.info("Reason: $reason / $tryingToReconnect")
|
||||
}
|
||||
})
|
||||
.build()
|
||||
|
||||
internal fun disable() {
|
||||
listener.closeAsync()
|
||||
}
|
||||
|
||||
internal fun reloadCommand() {
|
||||
messageHandler.reloadCommand()
|
||||
}
|
||||
|
||||
internal fun reloadUser(user: User) {
|
||||
this.user = user
|
||||
}
|
||||
|
||||
internal val isActive: Boolean
|
||||
get() = _isActive
|
||||
|
||||
internal fun isActive(value: Boolean, status: IData<IStreamInfo>) {
|
||||
_isActive = value
|
||||
if(value) {
|
||||
logger.info("${user.username} is live.")
|
||||
|
||||
logger.info("ChzzkChat connecting... ${channel.channelName} - ${channel.channelId}")
|
||||
listener.connectBlocking()
|
||||
|
||||
Discord.sendDiscord(user, status)
|
||||
|
||||
listener.sendChat("${user.username} 님의 방송이 감지되었습니다.")
|
||||
|
||||
} else {
|
||||
logger.info("${user.username} is offline.")
|
||||
listener.closeAsync()
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,24 @@
|
||||
package space.mori.chzzk_bot.chatbot.chzzk
|
||||
|
||||
import io.github.cdimascio.dotenv.dotenv
|
||||
import org.slf4j.LoggerFactory
|
||||
import xyz.r2turntrue.chzzk4j.Chzzk
|
||||
import xyz.r2turntrue.chzzk4j.ChzzkBuilder
|
||||
import xyz.r2turntrue.chzzk4j.types.channel.ChzzkChannel
|
||||
|
||||
val dotenv = dotenv {
|
||||
ignoreIfMissing = true
|
||||
}
|
||||
|
||||
object Connector {
|
||||
val chzzk: Chzzk = ChzzkBuilder()
|
||||
.withAuthorization(dotenv["NID_AUT"], dotenv["NID_SES"])
|
||||
.build()
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
|
||||
fun getChannel(channelId: String): ChzzkChannel? = chzzk.getChannel(channelId)
|
||||
|
||||
init {
|
||||
logger.info("chzzk logged: ${chzzk.isLoggedIn} / ${chzzk.loggedUser?.nickname ?: "----"}")
|
||||
}
|
||||
}
|
@@ -0,0 +1,193 @@
|
||||
package space.mori.chzzk_bot.chatbot.chzzk
|
||||
|
||||
import org.slf4j.Logger
|
||||
import space.mori.chzzk_bot.common.models.User
|
||||
import space.mori.chzzk_bot.common.services.CommandService
|
||||
import space.mori.chzzk_bot.common.services.CounterService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
import xyz.r2turntrue.chzzk4j.chat.ChatMessage
|
||||
import xyz.r2turntrue.chzzk4j.chat.ChzzkChat
|
||||
import xyz.r2turntrue.chzzk4j.types.channel.ChzzkChannel
|
||||
import java.time.LocalDateTime
|
||||
import java.time.format.DateTimeFormatter
|
||||
import java.time.temporal.ChronoUnit
|
||||
|
||||
|
||||
class MessageHandler(
|
||||
private val channel: ChzzkChannel,
|
||||
private val logger: Logger,
|
||||
private val listener: ChzzkChat
|
||||
) {
|
||||
private val commands = mutableMapOf<String, (msg: ChatMessage, user: User) -> Unit>()
|
||||
|
||||
private val counterPattern = Regex("<counter:([^>]+)>")
|
||||
private val personalCounterPattern = Regex("<counter_personal:([^>]+)>")
|
||||
private val dailyCounterPattern = Regex("<daily_counter:([^>]+)>")
|
||||
private val namePattern = Regex("<name>")
|
||||
private val followPattern = Regex("<following>")
|
||||
private val daysPattern = """<days:(\d{4})-(\d{2})-(\d{2})>""".toRegex()
|
||||
|
||||
init {
|
||||
reloadCommand()
|
||||
}
|
||||
|
||||
internal fun reloadCommand() {
|
||||
val user = UserService.getUser(channel.channelId)
|
||||
?: throw RuntimeException("User not found. it's bug? ${channel.channelName} - ${channel.channelId}")
|
||||
val commands = CommandService.getCommands(user)
|
||||
val manageCommands = mapOf("!명령어추가" to this::manageAddCommand, "!명령어삭제" to this::manageRemoveCommand, "!명령어수정" to this::manageUpdateCommand)
|
||||
|
||||
manageCommands.forEach { (commandName, command) ->
|
||||
this.commands[commandName] = command
|
||||
}
|
||||
|
||||
commands.map {
|
||||
this.commands.put(it.command.lowercase()) { msg, user ->
|
||||
logger.debug("${channel.channelName} - ${it.command} - ${it.content}/${it.failContent}")
|
||||
|
||||
val result = replaceCounters(Pair(it.content, it.failContent), user, msg, listener, msg.profile?.nickname ?: "")
|
||||
listener.sendChat(result)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun manageAddCommand(msg: ChatMessage, user: User) {
|
||||
if (msg.profile?.userRoleCode == "common_user") {
|
||||
listener.sendChat("매니저만 명령어를 추가할 수 있습니다.")
|
||||
return
|
||||
}
|
||||
|
||||
val parts = msg.content.split(" ", limit = 3)
|
||||
if (parts.size < 3) {
|
||||
listener.sendChat("명령어 추가 형식은 '!명령어추가 명령어 내용'입니다.")
|
||||
return
|
||||
}
|
||||
if (commands.containsKey(parts[1])) {
|
||||
listener.sendChat("${parts[1]} 명령어는 이미 있는 명령어입니다.")
|
||||
return
|
||||
}
|
||||
val command = parts[1]
|
||||
val content = parts[2]
|
||||
CommandService.saveCommand(user, command, content, "")
|
||||
listener.sendChat("명령어 '$command' 추가되었습니다.")
|
||||
}
|
||||
|
||||
private fun manageUpdateCommand(msg: ChatMessage, user: User) {
|
||||
if (msg.profile?.userRoleCode == "common_user") {
|
||||
listener.sendChat("매니저만 명령어를 추가할 수 있습니다.")
|
||||
return
|
||||
}
|
||||
|
||||
val parts = msg.content.split(" ", limit = 3)
|
||||
if (parts.size < 3) {
|
||||
listener.sendChat("명령어 수정 형식은 '!명령어수정 명령어 내용'입니다.")
|
||||
return
|
||||
}
|
||||
if (!commands.containsKey(parts[1])) {
|
||||
listener.sendChat("${parts[1]} 명령어는 없는 명령어입니다.")
|
||||
return
|
||||
}
|
||||
val command = parts[1]
|
||||
val content = parts[2]
|
||||
CommandService.updateCommand(user, command, content, "")
|
||||
listener.sendChat("명령어 '$command' 수정되었습니다.")
|
||||
}
|
||||
|
||||
private fun manageRemoveCommand(msg: ChatMessage, user: User) {
|
||||
if (msg.profile?.userRoleCode == "common_user") {
|
||||
listener.sendChat("매니저만 명령어를 삭제할 수 있습니다.")
|
||||
return
|
||||
}
|
||||
|
||||
val parts = msg.content.split(" ", limit = 2)
|
||||
if (parts.size < 2) {
|
||||
listener.sendChat("명령어 삭제 형식은 '!명령어삭제 명령어'입니다.")
|
||||
return
|
||||
}
|
||||
val command = parts[1]
|
||||
CommandService.removeCommand(user, command)
|
||||
listener.sendChat("명령어 '$command' 삭제되었습니다.")
|
||||
}
|
||||
|
||||
internal fun handle(msg: ChatMessage, user: User) {
|
||||
val commandKey = msg.content.split(' ')[0]
|
||||
|
||||
commands[commandKey.lowercase()]?.let { it(msg, user) }
|
||||
}
|
||||
|
||||
private fun replaceCounters(chat: Pair<String, String>, user: User, msg: ChatMessage, listener: ChzzkChat, userName: String): String {
|
||||
var result = chat.first
|
||||
var isFail = false
|
||||
|
||||
// Replace dailyCounterPattern
|
||||
result = dailyCounterPattern.replace(result) { matchResult ->
|
||||
val name = matchResult.groupValues[1]
|
||||
val dailyCounter = CounterService.getDailyCounterValue(name, msg.userId, user)
|
||||
|
||||
if (dailyCounter.second) {
|
||||
CounterService.updateDailyCounterValue(name, msg.userId, 1, user).first.toString()
|
||||
} else {
|
||||
isFail = true
|
||||
dailyCounter.first.toString()
|
||||
}
|
||||
}
|
||||
|
||||
// Handle fail case
|
||||
if (isFail && chat.second.isNotEmpty()) {
|
||||
result = chat.second
|
||||
result = dailyCounterPattern.replace(result) { matchResult ->
|
||||
val name = matchResult.groupValues[1]
|
||||
val dailyCounter = CounterService.getDailyCounterValue(name, msg.userId, user)
|
||||
dailyCounter.first.toString()
|
||||
}
|
||||
}
|
||||
|
||||
// Replace followPattern
|
||||
result = followPattern.replace(result) { matchResult ->
|
||||
try {
|
||||
val followingDate = getFollowDate(listener.chatId, msg.userId)
|
||||
.content.streamingProperty.following?.followDate
|
||||
|
||||
val period = followingDate?.let {
|
||||
val formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")
|
||||
val pastDate = LocalDateTime.parse(it, formatter)
|
||||
val today = LocalDateTime.now()
|
||||
ChronoUnit.DAYS.between(pastDate, today)
|
||||
} ?: 0
|
||||
|
||||
period.toString()
|
||||
} catch (e: Exception) {
|
||||
logger.error(e.message)
|
||||
"0"
|
||||
}
|
||||
}
|
||||
|
||||
// Replace daysPattern
|
||||
result = daysPattern.replace(result) { matchResult ->
|
||||
val (year, month, day) = matchResult.destructured
|
||||
val pastDate = LocalDateTime.of(year.toInt(), month.toInt(), day.toInt(), 0, 0, 0)
|
||||
val today = LocalDateTime.now()
|
||||
|
||||
val daysBetween = ChronoUnit.DAYS.between(pastDate, today)
|
||||
daysBetween.toString()
|
||||
}
|
||||
|
||||
// Replace counterPattern
|
||||
result = counterPattern.replace(result) { matchResult ->
|
||||
val name = matchResult.groupValues[1]
|
||||
CounterService.updateCounterValue(name, 1, user).toString()
|
||||
}
|
||||
|
||||
// Replace personalCounterPattern
|
||||
result = personalCounterPattern.replace(result) { matchResult ->
|
||||
val name = matchResult.groupValues[1]
|
||||
CounterService.updatePersonalCounterValue(name, msg.userId, 1, user).toString()
|
||||
}
|
||||
|
||||
// Replace namePattern
|
||||
result = namePattern.replace(result, userName)
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
}
|
@@ -0,0 +1,11 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.build.CommandData
|
||||
|
||||
interface CommandInterface {
|
||||
val name: String
|
||||
fun run(event: SlashCommandInteractionEvent, bot: JDA): Unit
|
||||
val command: CommandData
|
||||
}
|
@@ -0,0 +1,133 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord
|
||||
|
||||
import io.github.cdimascio.dotenv.dotenv
|
||||
import net.dv8tion.jda.api.EmbedBuilder
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.JDABuilder
|
||||
import net.dv8tion.jda.api.entities.Activity
|
||||
import net.dv8tion.jda.api.entities.Guild
|
||||
import net.dv8tion.jda.api.events.guild.GuildJoinEvent
|
||||
import net.dv8tion.jda.api.events.guild.member.GuildMemberRemoveEvent
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.hooks.ListenerAdapter
|
||||
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.IData
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.IStreamInfo
|
||||
import space.mori.chzzk_bot.chatbot.discord.commands.*
|
||||
import space.mori.chzzk_bot.common.models.User
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import java.time.Instant
|
||||
|
||||
val dotenv = dotenv {
|
||||
ignoreIfMissing = true
|
||||
}
|
||||
|
||||
class Discord: ListenerAdapter() {
|
||||
private var guild: Guild? = null
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
|
||||
companion object {
|
||||
lateinit var bot: JDA
|
||||
|
||||
internal fun getChannel(guildId: Long, channelId: Long) =
|
||||
bot.getGuildById(guildId)?.getTextChannelById(channelId)
|
||||
|
||||
fun sendDiscord(user: User, status: IData<IStreamInfo>) {
|
||||
if(user.liveAlertMessage != "" && user.liveAlertGuild != null && user.liveAlertChannel != null) {
|
||||
val channel = getChannel(user.liveAlertGuild!!, user.liveAlertChannel!!) ?: throw RuntimeException("${user.liveAlertChannel} is not valid.")
|
||||
|
||||
val embed = EmbedBuilder()
|
||||
embed.setTitle(status.content.liveTitle, "https://chzzk.naver.com/live/${user.token}")
|
||||
embed.setDescription("${user.username} 님이 방송을 시작했습니다.")
|
||||
embed.setUrl(status.content.channel.channelImageUrl)
|
||||
embed.setTimestamp(Instant.now())
|
||||
embed.setAuthor(user.username, "https://chzzk.naver.com/live/${user.token}", status.content.channel.channelImageUrl)
|
||||
embed.addField("카테고리", status.content.liveCategoryValue, true)
|
||||
embed.addField("태그", status.content.tags.joinToString(", "), true)
|
||||
embed.setImage(status.content.liveImageUrl.replace("{type}", "1080"))
|
||||
|
||||
channel.sendMessage(
|
||||
MessageCreateBuilder()
|
||||
.setContent(user.liveAlertMessage)
|
||||
.setEmbeds(embed.build())
|
||||
.build()
|
||||
).queue()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private val commands = listOf(
|
||||
AddCommand,
|
||||
AlertCommand,
|
||||
PingCommand,
|
||||
RegisterCommand,
|
||||
RemoveCommand,
|
||||
UpdateCommand,
|
||||
AddManagerCommand,
|
||||
ListManagerCommand,
|
||||
RemoveManagerCommand,
|
||||
)
|
||||
|
||||
override fun onSlashCommandInteraction(event: SlashCommandInteractionEvent) {
|
||||
event.deferReply().queue()
|
||||
val handler = commands.find { it.name == event.name }
|
||||
logger.debug("Handler: ${handler?.name ?: "undefined"} command")
|
||||
handler?.run(event, bot)
|
||||
}
|
||||
|
||||
override fun onGuildMemberRemove(event: GuildMemberRemoveEvent) {
|
||||
event.member?.let { ManagerService.deleteManager(event.guild.idLong, it.idLong) }
|
||||
}
|
||||
|
||||
override fun onGuildJoin(event: GuildJoinEvent) {
|
||||
commandUpdate(event.guild)
|
||||
}
|
||||
|
||||
private fun commandUpdate(guild: Guild) {
|
||||
guild.updateCommands().addCommands(* commands.map { it.command}.toTypedArray())
|
||||
.onSuccess {
|
||||
logger.info("Command update on guild success!")
|
||||
}
|
||||
.queue()
|
||||
}
|
||||
|
||||
private fun commandUpdate(bot: JDA) {
|
||||
bot.updateCommands().addCommands(* commands.map { it.command}.toTypedArray())
|
||||
.onSuccess {
|
||||
logger.info("Command update bot boot success!")
|
||||
}
|
||||
.queue()
|
||||
}
|
||||
|
||||
fun enable() {
|
||||
val thread = Thread {
|
||||
try {
|
||||
bot = JDABuilder.createDefault(dotenv["DISCORD_TOKEN"])
|
||||
.setActivity(Activity.playing("치지직 보는중"))
|
||||
.addEventListeners(this)
|
||||
.build().awaitReady()
|
||||
|
||||
guild = bot.getGuildById(dotenv["GUILD_ID"])
|
||||
|
||||
commandUpdate(bot)
|
||||
bot.guilds.forEach {
|
||||
commandUpdate(it)
|
||||
}
|
||||
} catch (e: Exception) {
|
||||
logger.info("Could not enable Discord!")
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
thread.start()
|
||||
}
|
||||
|
||||
fun disable() {
|
||||
try {
|
||||
bot.shutdown()
|
||||
} catch(e: Exception) {
|
||||
logger.info("Error while shutting down Discord!")
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,65 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.ChzzkHandler
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.Connector
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.CommandService
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object AddCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "add"
|
||||
override val command = Commands.slash(name, "명령어를 추가합니다.")
|
||||
.addOptions(OptionData(OptionType.STRING, "label", "작동할 명령어를 입력하세요.", true))
|
||||
.addOptions(OptionData(OptionType.STRING, "content", "표시될 텍스트를 입력하세요.", true))
|
||||
.addOptions(OptionData(OptionType.STRING, "fail_content", "카운터 업데이트 실패시 표시될 텍스트를 입력하세요.", false))
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val label = event.getOption("label")?.asString
|
||||
val content = event.getOption("content")?.asString
|
||||
val failContent = event.getOption("fail_content")?.asString
|
||||
|
||||
if(label == null || content == null) {
|
||||
event.hook.sendMessage("명령어와 텍스트는 필수 입력입니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
var user = UserService.getUser(event.user.idLong)
|
||||
val manager = event.guild?.idLong?.let { ManagerService.getUser(it, event.user.idLong) }
|
||||
if(user == null && manager == null) {
|
||||
event.hook.sendMessage("당신은 이 명령어를 사용할 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
if (manager != null) {
|
||||
user = manager.user
|
||||
ManagerService.updateManager(user, event.user.idLong, event.user.effectiveName)
|
||||
}
|
||||
|
||||
val commands = CommandService.getCommands(user!!)
|
||||
if (commands.any { it.command == label }) {
|
||||
event.hook.sendMessage("$label 명령어는 이미 있습니다! 업데이트 명령어를 써주세요.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
val chzzkChannel = Connector.getChannel(user.token)
|
||||
|
||||
try {
|
||||
CommandService.saveCommand(user, label, content, failContent ?: "")
|
||||
try {
|
||||
ChzzkHandler.reloadCommand(chzzkChannel!!)
|
||||
} catch (_: Exception) {}
|
||||
event.hook.sendMessage("등록이 완료되었습니다. $label = $content/$failContent").queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,47 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object AddManagerCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "addmanager"
|
||||
override val command = Commands.slash(name, "매니저를 추가합니다.")
|
||||
.addOptions(OptionData(OptionType.USER, "user", "추가할 유저를 선택하세요.", true))
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val manager = event.getOption("user")?.asUser
|
||||
|
||||
if(manager == null) {
|
||||
event.hook.sendMessage("유저는 필수사항입니다.").queue()
|
||||
return
|
||||
}
|
||||
if(manager.idLong == event.user.idLong) {
|
||||
event.hook.sendMessage("자신은 매니저로 설정할 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
val user = UserService.getUser(event.user.idLong)
|
||||
if(user == null) {
|
||||
event.hook.sendMessage("치지직 계정을 찾을 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
try {
|
||||
ManagerService.saveManager(user, manager.idLong, manager.effectiveName)
|
||||
if(user.liveAlertGuild == null)
|
||||
UserService.updateLiveAlert(user.id.value, event.guild!!.idLong, event.channelIdLong, "")
|
||||
event.hook.sendMessage("등록이 완료되었습니다. ${manager.effectiveName}").queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,51 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.ChzzkHandler
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.Connector
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object AlertCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "alert"
|
||||
override val command = Commands.slash(name, "명령어를 추가합니다.")
|
||||
.addOptions(OptionData(OptionType.CHANNEL, "channel", "알림을 보낼 채널을 입력하세요."))
|
||||
.addOptions(OptionData(OptionType.STRING, "content", "표시될 텍스트를 입력하세요. 비워두면 알람이 취소됩니다."))
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val channel = event.getOption("channel")?.asChannel
|
||||
val content = event.getOption("content")?.asString
|
||||
|
||||
var user = UserService.getUser(event.user.idLong)
|
||||
val manager = event.guild?.idLong?.let { ManagerService.getUser(it, event.user.idLong) }
|
||||
if(user == null && manager == null) {
|
||||
event.hook.sendMessage("당신은 이 명령어를 사용할 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
if (manager != null) {
|
||||
user = manager.user
|
||||
ManagerService.updateManager(user, event.user.idLong, event.user.effectiveName)
|
||||
}
|
||||
|
||||
val chzzkChannel = Connector.getChannel(user!!.token)
|
||||
|
||||
try {
|
||||
val newUser = UserService.updateLiveAlert(user.id.value, channel?.guild?.idLong ?: 0L, channel?.idLong ?: 0L, content ?: "")
|
||||
try {
|
||||
ChzzkHandler.reloadUser(chzzkChannel!!, newUser)
|
||||
} catch (_: Exception) {}
|
||||
event.hook.sendMessage("업데이트가 완료되었습니다.").queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,44 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.EmbedBuilder
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object ListManagerCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "listmanager"
|
||||
override val command = Commands.slash(name, "매니저 목록을 확인합니다.")
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
try {
|
||||
val managers = event.guild?.idLong?.let { ManagerService.getAllUsers(it) }
|
||||
|
||||
if(managers == null) {
|
||||
event.channel.sendMessage("여기에서는 사용할 수 없습니다.")
|
||||
return
|
||||
}
|
||||
|
||||
val user = UserService.getUserWithGuildId(event.guild!!.idLong)
|
||||
|
||||
val embed = EmbedBuilder()
|
||||
embed.setTitle("${user!!.username} 매니저 목록")
|
||||
embed.setDescription("매니저 목록입니다.")
|
||||
|
||||
var idx = 1
|
||||
|
||||
managers.forEach {
|
||||
embed.addField("${idx++}", it.lastUserName ?: it.managerId.toString(), true)
|
||||
}
|
||||
|
||||
event.channel.sendMessageEmbeds(embed.build()).queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,17 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
|
||||
object PingCommand: CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name = "ping"
|
||||
override val command = Commands.slash(name, "봇이 살아있을까요?")
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
event.hook.sendMessage("${event.user.asMention} Pong!").queue()
|
||||
}
|
||||
}
|
@@ -0,0 +1,49 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.ChzzkHandler
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.Connector
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object RegisterCommand: CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name = "register"
|
||||
override val command = Commands.slash(name, "치지직 계정을 등록합니다.")
|
||||
.addOptions(
|
||||
OptionData(
|
||||
OptionType.STRING,
|
||||
"chzzk_id",
|
||||
"36da10b7c35800f298e9c565a396bafd 형식으로 입력해주세요.",
|
||||
true
|
||||
)
|
||||
)
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val chzzkID = event.getOption("chzzk_id")?.asString
|
||||
if(chzzkID == null) {
|
||||
event.hook.sendMessage("치지직 계정은 필수 입력입니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
val chzzkChannel = Connector.getChannel(chzzkID)
|
||||
if (chzzkChannel == null) {
|
||||
event.hook.sendMessage("치지직 계정을 찾을 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
try {
|
||||
val user = UserService.saveUser(chzzkChannel.channelName, chzzkChannel.channelId, event.user.idLong)
|
||||
event.hook.sendMessage("등록이 완료되었습니다. `${chzzkChannel.channelId}` - `${chzzkChannel.channelName}`")
|
||||
ChzzkHandler.addUser(chzzkChannel, user)
|
||||
} catch(e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,55 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.ChzzkHandler
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.Connector
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.CommandService
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object RemoveCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "remove"
|
||||
override val command = Commands.slash(name, "명령어를 삭제합니다.")
|
||||
.addOptions(OptionData(OptionType.STRING, "label", "삭제할 명령어를 입력하세요.", true))
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val label = event.getOption("label")?.asString
|
||||
|
||||
if(label == null) {
|
||||
event.hook.sendMessage("명령어는 필수 입력입니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
var user = UserService.getUser(event.user.idLong)
|
||||
val manager = event.guild?.idLong?.let { ManagerService.getUser(it, event.user.idLong) }
|
||||
if(user == null && manager == null) {
|
||||
event.hook.sendMessage("당신은 이 명령어를 사용할 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
if (manager != null) {
|
||||
user = manager.user
|
||||
ManagerService.updateManager(user, event.user.idLong, event.user.effectiveName)
|
||||
}
|
||||
|
||||
val chzzkChannel = Connector.getChannel(user!!.token)
|
||||
|
||||
try {
|
||||
CommandService.removeCommand(user, label)
|
||||
try {
|
||||
ChzzkHandler.reloadCommand(chzzkChannel!!)
|
||||
} catch (_: Exception) {}
|
||||
event.hook.sendMessage("삭제가 완료되었습니다. $label").queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,49 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object RemoveManagerCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "removemanager"
|
||||
override val command = Commands.slash(name, "매니저를 삭제합니다.")
|
||||
.addOptions(OptionData(OptionType.USER, "user", "삭제할 유저를 선택하세요.", true))
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val manager = event.getOption("user")?.asUser
|
||||
|
||||
if(manager == null) {
|
||||
event.hook.sendMessage("유저는 필수사항입니다.").queue()
|
||||
return
|
||||
}
|
||||
if(manager.idLong == event.user.idLong) {
|
||||
event.hook.sendMessage("자신은 매니저로 설정할 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
val user = UserService.getUser(event.user.idLong)
|
||||
if(user == null) {
|
||||
event.hook.sendMessage("치지직 계정을 찾을 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
if(ManagerService.getUser(user.liveAlertGuild ?: 0L, manager.idLong) == null) {
|
||||
event.hook.sendMessage("${manager.name}은 매니저가 아닙니다.")
|
||||
}
|
||||
|
||||
try {
|
||||
ManagerService.deleteManager(user, manager.idLong)
|
||||
event.hook.sendMessage("삭제가 완료되었습니다. ${manager.effectiveName}").queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
@@ -0,0 +1,57 @@
|
||||
package space.mori.chzzk_bot.chatbot.discord.commands
|
||||
|
||||
import net.dv8tion.jda.api.JDA
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionType
|
||||
import net.dv8tion.jda.api.interactions.commands.build.Commands
|
||||
import net.dv8tion.jda.api.interactions.commands.build.OptionData
|
||||
import org.slf4j.LoggerFactory
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.ChzzkHandler
|
||||
import space.mori.chzzk_bot.chatbot.chzzk.Connector
|
||||
import space.mori.chzzk_bot.chatbot.discord.CommandInterface
|
||||
import space.mori.chzzk_bot.common.services.CommandService
|
||||
import space.mori.chzzk_bot.common.services.ManagerService
|
||||
import space.mori.chzzk_bot.common.services.UserService
|
||||
|
||||
object UpdateCommand : CommandInterface {
|
||||
private val logger = LoggerFactory.getLogger(this::class.java)
|
||||
override val name: String = "update"
|
||||
override val command = Commands.slash(name, "명령어를 수정합니다.")
|
||||
.addOptions(OptionData(OptionType.STRING, "label", "수정할 명령어를 입력하세요.", true))
|
||||
.addOptions(OptionData(OptionType.STRING, "content", "표시될 텍스트를 입력하세요.", true))
|
||||
.addOptions(OptionData(OptionType.STRING, "fail_content", "카운터 업데이트 실패시 표시될 텍스트를 입력하세요.", false))
|
||||
|
||||
override fun run(event: SlashCommandInteractionEvent, bot: JDA) {
|
||||
val label = event.getOption("label")?.asString
|
||||
val content = event.getOption("content")?.asString
|
||||
val failContent = event.getOption("fail_content")?.asString
|
||||
|
||||
if(label == null || content == null) {
|
||||
event.hook.sendMessage("명령어와 텍스트는 필수 입력입니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
var user = UserService.getUser(event.user.idLong)
|
||||
val manager = event.guild?.idLong?.let { ManagerService.getUser(it, event.user.idLong) }
|
||||
if(user == null && manager == null) {
|
||||
event.hook.sendMessage("당신은 이 명령어를 사용할 수 없습니다.").queue()
|
||||
return
|
||||
}
|
||||
|
||||
if (manager != null) {
|
||||
user = manager.user
|
||||
ManagerService.updateManager(user, event.user.idLong, event.user.effectiveName)
|
||||
}
|
||||
|
||||
val chzzkChannel = Connector.getChannel(user!!.token)
|
||||
|
||||
try {
|
||||
CommandService.updateCommand(user, label, content, failContent ?: "")
|
||||
chzzkChannel?.let { ChzzkHandler.reloadCommand(it) }
|
||||
event.hook.sendMessage("등록이 완료되었습니다. $label = $content").queue()
|
||||
} catch (e: Exception) {
|
||||
event.hook.sendMessage("에러가 발생했습니다.").queue()
|
||||
logger.debug(e.stackTraceToString())
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user