diff --git a/src/main/kotlin/dev/jansel/feixiao/database/collections/StreamerCollection.kt b/src/main/kotlin/dev/jansel/feixiao/database/collections/StreamerCollection.kt index 8df3691..8304211 100644 --- a/src/main/kotlin/dev/jansel/feixiao/database/collections/StreamerCollection.kt +++ b/src/main/kotlin/dev/jansel/feixiao/database/collections/StreamerCollection.kt @@ -20,26 +20,26 @@ class StreamerCollection : KordExKoinComponent { suspend fun getData(channelName: String): StreamerData? = collection.findOne(StreamerData::name eq channelName) - suspend fun updateData(guildId: Snowflake, channelId: Snowflake, streamerName: String, roleId: Snowflake?) { + suspend fun updateData(guildId: Snowflake, channelId: Snowflake, streamerName: String, roleId: Snowflake?, liveMessage: String?) { val coll = collection.findOne(StreamerData::name eq streamerName) if (coll != null) { collection.updateOne( StreamerData::name eq streamerName, - setValue(StreamerData::servers, coll.servers + listOf(Server(guildId, channelId, roleId))) + setValue(StreamerData::servers, coll.servers + listOf(Server(guildId, channelId, roleId, liveMessage))) ) } else { collection.insertOne( - StreamerData(streamerName, listOf(Server(guildId, channelId, roleId))) + StreamerData(streamerName, listOf(Server(guildId, channelId, roleId, liveMessage))) ) } } - suspend fun removeData(guildId: Snowflake, channelId: Snowflake, streamerName: String, roleId: Snowflake?) { + suspend fun removeData(guildId: Snowflake, channelId: Snowflake, streamerName: String, roleId: Snowflake?, liveMessage: String?) { val coll = collection.findOne(StreamerData::name eq streamerName) if (coll != null) { collection.updateOne( StreamerData::name eq streamerName, - setValue(StreamerData::servers, coll.servers - Server(guildId, channelId, roleId)) + setValue(StreamerData::servers, coll.servers - Server(guildId, channelId, roleId, liveMessage)) ) } } diff --git a/src/main/kotlin/dev/jansel/feixiao/database/entities/StreamerData.kt b/src/main/kotlin/dev/jansel/feixiao/database/entities/StreamerData.kt index 66de7f2..3425532 100644 --- a/src/main/kotlin/dev/jansel/feixiao/database/entities/StreamerData.kt +++ b/src/main/kotlin/dev/jansel/feixiao/database/entities/StreamerData.kt @@ -13,5 +13,6 @@ data class StreamerData( data class Server( val guildId: Snowflake, val channelId: Snowflake, - val roleId: Snowflake? + val roleId: Snowflake?, + val liveMessage: String? )