From 1072269ea582193508bee463d3ad866f04af8dbd Mon Sep 17 00:00:00 2001
From: cssxsh <cssxsh@gmail.com>
Date: Sat, 28 Aug 2021 09:42:55 +0800
Subject: [PATCH] feat: ImageLimit

---
 src/main/kotlin/xyz/cssxsh/mirai/plugin/BiliUtils.kt           | 3 +++
 .../kotlin/xyz/cssxsh/mirai/plugin/data/BiliHelperSettings.kt  | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/src/main/kotlin/xyz/cssxsh/mirai/plugin/BiliUtils.kt b/src/main/kotlin/xyz/cssxsh/mirai/plugin/BiliUtils.kt
index 9d3015ea..38d085aa 100644
--- 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
index 3406a046..c0917e3e 100644
--- 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)