jansel/Feixiao
1
0
Fork 0
Code Issues Actions

start feature

Browse Source
This commit is contained in:
Jannik Reimers 2024-11-23 23:52:24 +01:00
parent 4a313832fa
commit 978b0fd65b
Signed by: jansel
GPG Key ID: 39C62D7D5233CFD0
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/kotlin/dev/jansel/feixiao/database/collections/StreamerCollection.kt
View File

@ -20,26 +20,26 @@ class StreamerCollection : KordExKoinComponent {
suspend fun getData(channelName: String): StreamerData? = suspend fun getData(channelName: String): StreamerData? =
collection.findOne(StreamerData::name eq channelName) 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) val coll = collection.findOne(StreamerData::name eq streamerName)
if (coll != null) { if (coll != null) {
collection.updateOne( collection.updateOne(
StreamerData::name eq streamerName, 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 { } else {
collection.insertOne( 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) val coll = collection.findOne(StreamerData::name eq streamerName)
if (coll != null) { if (coll != null) {
collection.updateOne( collection.updateOne(
StreamerData::name eq streamerName, StreamerData::name eq streamerName,
setValue(StreamerData::servers, coll.servers - Server(guildId, channelId, roleId)) setValue(StreamerData::servers, coll.servers - Server(guildId, channelId, roleId, liveMessage))
) )
} }
} }

3
src/main/kotlin/dev/jansel/feixiao/database/entities/StreamerData.kt
View File

@ -13,5 +13,6 @@ data class StreamerData(
data class Server( data class Server(
val guildId: Snowflake, val guildId: Snowflake,
val channelId: Snowflake, val channelId: Snowflake,
val roleId: Snowflake? val roleId: Snowflake?,
val liveMessage: String?
) )