62 lines
2.4 KiB
Kotlin
62 lines
2.4 KiB
Kotlin
package moe.kageru.kagebot.features
|
|
|
|
import moe.kageru.kagebot.Log
|
|
import moe.kageru.kagebot.Util.findRole
|
|
import moe.kageru.kagebot.Util.findUser
|
|
import moe.kageru.kagebot.Util.ifNotEmpty
|
|
import moe.kageru.kagebot.config.Config
|
|
import moe.kageru.kagebot.config.RawTimeoutFeature
|
|
import moe.kageru.kagebot.persistence.Dao
|
|
import org.javacord.api.entity.permission.Role
|
|
import org.javacord.api.event.message.MessageCreateEvent
|
|
import java.lang.IllegalArgumentException
|
|
import java.time.Duration
|
|
import java.time.Instant
|
|
|
|
class TimeoutFeature(raw: RawTimeoutFeature) : MessageFeature {
|
|
private val timeoutRole: Role = raw.role?.let(::findRole)
|
|
?: throw IllegalArgumentException("No timeout role defined")
|
|
|
|
override fun handle(message: MessageCreateEvent) {
|
|
val (_, target, time) = message.readableMessageContent.split(' ', limit = 3)
|
|
findUser(target)?.let { user ->
|
|
val oldRoles = user.getRoles(Config.server).map { role ->
|
|
user.removeRole(role)
|
|
role.id
|
|
}
|
|
user.addRole(timeoutRole)
|
|
val releaseTime = Instant.now().plus(Duration.ofMinutes(time.toLong())).epochSecond
|
|
Dao.saveTimeout(releaseTime, listOf(user.id) + oldRoles)
|
|
} ?: message.channel.sendMessage("Could not find user $target. Consider using the user ID.")
|
|
}
|
|
|
|
fun checkAndRelease() {
|
|
val now = Instant.now().epochSecond
|
|
Dao.getAllTimeouts()
|
|
.filter { releaseTime -> now > releaseTime }
|
|
.map {
|
|
Dao.deleteTimeout(it).let { rawIds ->
|
|
UserInTimeout.ofLongs(rawIds).toPair()
|
|
}
|
|
}.forEach { (userId, roleIds) ->
|
|
Config.server.getMemberById(userId).ifNotEmpty { user ->
|
|
roleIds.forEach { roleId ->
|
|
user.addRole(findRole("$roleId"))
|
|
}
|
|
user.removeRole(timeoutRole)
|
|
} ?: Log.warn("Tried to free user $userId, but couldn’t find them on the server anymore")
|
|
}
|
|
}
|
|
}
|
|
|
|
class UserInTimeout(private val id: Long, private val roles: List<Long>) {
|
|
fun toPair() = Pair(id, roles)
|
|
|
|
companion object {
|
|
fun ofLongs(longs: LongArray): UserInTimeout = longs.run {
|
|
val userId = first()
|
|
val roles = if (size > 1) slice(1 until size) else emptyList()
|
|
return UserInTimeout(userId, roles)
|
|
}
|
|
}
|
|
} |