59 lines
2.1 KiB
Kotlin
59 lines
2.1 KiB
Kotlin
import dev.inmo.krontab.*
|
|
import kotlinx.coroutines.Dispatchers
|
|
import kotlinx.coroutines.withContext
|
|
import kotlinx.serialization.*
|
|
import kotlinx.serialization.builtins.serializer
|
|
import kotlinx.serialization.descriptors.SerialDescriptor
|
|
import kotlinx.serialization.encoding.Decoder
|
|
import kotlinx.serialization.encoding.Encoder
|
|
import net.kodehawa.lib.imageboards.DefaultImageBoards
|
|
import net.kodehawa.lib.imageboards.ImageBoard
|
|
import net.kodehawa.lib.imageboards.boards.DefaultBoards
|
|
import net.kodehawa.lib.imageboards.entities.BoardImage
|
|
|
|
@Serializable
|
|
data class ChatSettings(
|
|
val query: String,
|
|
val krontabTemplate: KrontabTemplate,
|
|
@Serializable(BoardSerializer::class)
|
|
private val boardBase: DefaultBoards,
|
|
val count: Int = 1,
|
|
val gallery: Boolean = false
|
|
) {
|
|
val scheduler by lazy {
|
|
krontabTemplate.toSchedule()
|
|
}
|
|
|
|
val board: ImageBoard<*>
|
|
get() = when (boardBase) {
|
|
DefaultBoards.R34 -> DefaultImageBoards.RULE34
|
|
DefaultBoards.E621 -> DefaultImageBoards.E621
|
|
DefaultBoards.KONACHAN -> DefaultImageBoards.KONACHAN
|
|
DefaultBoards.YANDERE -> DefaultImageBoards.YANDERE
|
|
DefaultBoards.DANBOORU -> DefaultImageBoards.DANBOORU
|
|
DefaultBoards.SAFEBOORU -> DefaultImageBoards.SAFEBOORU
|
|
DefaultBoards.GELBOORU -> DefaultImageBoards.GELBOORU
|
|
DefaultBoards.E926 -> DefaultImageBoards.E926
|
|
}
|
|
|
|
suspend fun makeRequest(): List<BoardImage> {
|
|
return withContext(Dispatchers.IO) {
|
|
board.search(count, query).blocking()
|
|
}
|
|
}
|
|
|
|
@Serializer(DefaultBoards::class)
|
|
object BoardSerializer : KSerializer<DefaultBoards> {
|
|
override val descriptor: SerialDescriptor = String.serializer().descriptor
|
|
val types = DefaultBoards.values().associateBy { it.name.lowercase() }
|
|
override fun deserialize(decoder: Decoder): DefaultBoards {
|
|
val type = decoder.decodeString()
|
|
return types.getValue(type)
|
|
}
|
|
|
|
override fun serialize(encoder: Encoder, value: DefaultBoards) {
|
|
encoder.encodeString(value.name.lowercase())
|
|
}
|
|
}
|
|
}
|