temporal progress

src/main/kotlin/ChatSettings.kt

@@ -0,0 +1,58 @@
+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())
+        }
+    }
+}