feat: ImageLimit

cssxsh · Aug 28, 2021 · 1072269 · 1072269
1 parent 63c014d
commit 1072269
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/main/kotlin/xyz/cssxsh/mirai/plugin/BiliUtils.kt b/src/main/kotlin/xyz/cssxsh/mirai/plugin/BiliUtils.kt
@@ -69,6 +69,8 @@ internal fun BiliClient.save() {
 
 internal val ImageCache by lazy { File(BiliHelperSettings.cache) }
 
+internal val ImageLimit by lazy { BiliHelperSettings.limit }
+
 internal val SetupSelenium by BiliHelperPlugin::selenium
 
 internal val RemoteWebDriver by BiliHelperPlugin::driver
@@ -229,6 +231,7 @@ internal suspend fun UserInfo.getFace(contact: Contact): Message {
 }
 
 internal suspend fun DynamicInfo.getImages(contact: Contact) = images.mapIndexed { index, picture ->
+    if (ImageLimit >= 0 && index < ImageLimit) return@mapIndexed "图片[${index + 1}]省略".toPlainText()
     Url(picture).runCatching {
         cache(
             type = CacheType.DYNAMIC,

diff --git a/src/main/kotlin/xyz/cssxsh/mirai/plugin/data/BiliHelperSettings.kt b/src/main/kotlin/xyz/cssxsh/mirai/plugin/data/BiliHelperSettings.kt
@@ -6,6 +6,9 @@ object BiliHelperSettings : ReadOnlyPluginConfig("BiliHelperSettings") {
     @ValueDescription("图片缓存位置")
     val cache: String by value("ImageCache")
 
+    @ValueDescription("动态 订阅 输出图片数量，负数表示输出全部")
+    val limit: Int by value(-1)
+
     @ValueDescription("API 访问间隔时间，单位秒")
     val api: Long by value(10L)