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())
        }
    }
}