InsanusMokrassar/BooruGrabberTelegramBot
1
0
Fork 0
mirror of https://github.com/InsanusMokrassar/BooruGrabberTelegramBot.git synced 2025-11-11 09:30:12 +00:00
Code Issues Projects Releases Wiki Activity

now requests do not require any query

Browse Source
This commit is contained in:
InsanusMokrassar
2022-10-07 22:15:06 +06:00
parent 96ec7bbb0c
commit 1eb5f75c78
3 changed files with 52 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
src/main/kotlin/EnableArgsParser.kt
View File

@@ -3,27 +3,50 @@ import com.github.ajalt.clikt.parameters.arguments.*
import com.github.ajalt.clikt.parameters.options.*
import com.github.ajalt.clikt.parameters.types.enum
import com.github.ajalt.clikt.parameters.types.int
import dev.inmo.micro_utils.coroutines.launchSafelyWithoutExceptions
import dev.inmo.micro_utils.repos.KeyValueRepo
import dev.inmo.micro_utils.repos.set
import dev.inmo.tgbotapi.types.ChatId
import kotlinx.coroutines.CoroutineScope
import net.kodehawa.lib.imageboards.boards.DefaultBoards
import net.kodehawa.lib.imageboards.entities.Rating
class EnableArgsParser: CliktCommand(name = "enable") {
val count by option("-n").int().help("Amount of pictures to grab each trigger time").default(1).check("Count should be in range 1-10") {
it in 1 .. 10
class EnableArgsParser(
onlyQueryIsRequired: Boolean,
private val base: ChatSettings = ChatSettings("", null, DefaultBoards.SAFEBOORU)
) : CliktCommand(name = "enable") {
private fun <EachT : Any, ValueT> NullableOption<EachT, ValueT>.default(
value: EachT?,
defaultForHelp: String = value.toString(),
): NullableOption<EachT, ValueT> {
return if (value != null) {
transformAll(defaultForHelp) { it.lastOrNull() ?: value }
} else {
this
}
}
val query by argument().multiple(required = true).help("Your query to booru. Use syntax \"-- -sometag\" to add excluding of some tag in query")
val krontab by option("-k", "--krontab").transformValues(5) {
it.joinToString(" ")
}.help("Krontab in format * * * * *. See https://bookstack.inmo.dev/books/krontab/page/string-format")
val board by option("-b", "--board").convert {
ChatSettings.BoardSerializer.types.getValue(it)
}.required().help("Board type. Possible values: ${ChatSettings.BoardSerializer.types.keys.joinToString { it }}")
val gallery by option("-g", "--gallery").flag(default = false).help("Effective only when count passed > 1. Will send chosen images as gallery instead of separated images")
val rating by option("-r", "--rating").enum<Rating> { it.name.lowercase() }
val attachUrls by option("-a", "--attach_urls").flag(default = false)
val count by option("-n")
.int()
.help("Amount of pictures to grab each trigger time")
.check("Count should be in range 1-10") { it in 1 .. 10 }
val query by argument()
.multiple(required = true)
.help("Your query to booru. Use syntax \"-- -sometag\" to add excluding of some tag in query")
val krontab by option("-k", "--krontab")
.transformValues(5) { it.joinToString(" ") }
.help("Krontab in format * * * * *. See https://bookstack.inmo.dev/books/krontab/page/string-format")
val board by option("-b", "--board")
.convert { ChatSettings.BoardSerializer.types.getValue(it) }
.run {
if (onlyQueryIsRequired) {
this
} else {
required()
}
}
.help("Board type. Possible values: ${ChatSettings.BoardSerializer.types.keys.joinToString { it }}")
val gallery by option("-g", "--gallery")
.flag(default = base.gallery)
.help("Effective only when count passed > 1. Will send chosen images as gallery instead of separated images")
val rating by option("-r", "--rating")
.enum<Rating> { it.name.lowercase() }
val attachUrls by option("-a", "--attach_urls")
.flag(default = base.attachUrls)
var resultSettings: ChatSettings? = null
private set
@@ -32,8 +55,8 @@ class EnableArgsParser: CliktCommand(name = "enable") {
resultSettings = ChatSettings(
query.filterNot { it.isEmpty() }.joinToString(" ").trim(),
krontab,
board,
count,
board ?: base.board.boardType as DefaultBoards,
count ?: base.count,
gallery,
rating,
attachUrls