kokkoro-plugin-saucenao 2.0.2 → 2.0.3

package/README.md

@@ -17,11 +17,11 @@ bun add kokkoro-plugin-saucenao
 /搜图
 ```
 
-发送「/搜图」时附带图片，插件会搜索消息中的第一张图片，并以 Markdown 返回搜索结果，默认显示相似度最高的 3 条。
+发送「/搜图」指令时附带图片，或引用包含图片的消息回复该指令。插件搜索首张图片的来源，并通过 QQ Markdown 返回结果，默认显示相似度最高的 3 条。
 
 ## 快捷方式
 
-发送「搜图」时附带图片，也会返回搜索结果。要让普通群消息触发快捷方式，需要开启「获取群内全部消息」权限。
+「搜图」与「/搜图」的用法相同。在普通群消息中使用快捷方式，需要开启「获取群内全部消息」权限。
 
 ```text
 搜图

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kokkoro-plugin-saucenao",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "description": "SauceNAO 图片搜索，查找动漫、漫画、插画等图片的来源。",
   "keywords": [
     "bot",

package/src/index.ts

@@ -7,20 +7,24 @@ const logger = useLogger();
 
 export default () => {
   useCommand('/搜图', async context => {
-    const image = context.attachments?.find(attachment => attachment.content_type?.startsWith('image/'));
+    const [reference] = context.msg_elements ?? [];
+    const attachments = context.message_type === 103 ? reference?.attachments : context.attachments;
+    const images = attachments?.filter(attachment => attachment.content_type?.startsWith('image/')) ?? [];
+    const [image] = images;
 
     if (!image?.url) {
-      throw new Error('请在指令中附带需要搜索的图片');
+      throw new Error('未检测到图片，请在指令消息中附带图片，或引用包含图片的消息后重试。');
     }
     const result = await fetchImageSources(image.url, logger);
     const { results } = result;
+    const notice = images.length > 1 ? '检测到消息中包含多张图片，已选取首张图片进行识别。' : undefined;
 
     logger.info('已找到图片来源', { count: results.length });
 
     await context.reply({
       msg_type: 2,
       markdown: {
-        content: await createMarkdown(results),
+        content: await createMarkdown(results, notice),
       },
     });
   }).shortcut('搜图');

package/src/service.ts

@@ -146,7 +146,7 @@ export async function fetchImageSources(url: string, logger?: Logger): Promise<S
   const response = await fetch(endpoint);
 
   if (!response.ok) {
-    throw new Error(`接口请求失败，状态码 ${response.status}`);
+    throw new Error(`SauceNAO 请求失败，HTTP 状态码 ${response.status}`);
   }
   const result = <SauceNao>await response.json();
   const { header, results } = result;
@@ -158,7 +158,7 @@ export async function fetchImageSources(url: string, logger?: Logger): Promise<S
   }
 
   if (results.length === 0) {
-    throw new Error('没有找到图片来源');
+    throw new Error('未找到图片来源');
   }
   return result;
 }

package/src/util.ts

@@ -2,7 +2,7 @@ import { Image } from 'bun';
 
 import { type ImageSource } from './service';
 
-const { SAUCENAO_SIMILARITY_THRESHOLD = 50 } = import.meta.env;
+const { SAUCENAO_SIMILARITY_THRESHOLD = '50' } = import.meta.env;
 const LOW_SIMILARITY_THUMBNAIL = 'https://kokkoro.js.org/saucenao-low-similarity.jpg';
 const similarityThreshold = Number(SAUCENAO_SIMILARITY_THRESHOLD);
 
@@ -28,8 +28,13 @@ function getUrl(data: ImageSource['data']) {
   }
 }
 
-/** 将 SauceNAO 搜索结果转换为 QQ Markdown。 */
-export async function createMarkdown(results: readonly ImageSource[]): Promise<string> {
+/** 将 SauceNAO 搜索结果转换为 QQ Markdown，可在标题下方展示提示。 */
+export async function createMarkdown(results: readonly ImageSource[], notice?: string): Promise<string> {
+  const lines = ['## SauceNAO 搜图结果'];
+
+  if (notice) {
+    lines.push('', `> ${notice}`);
+  }
   const content = await Promise.all(
     results.map(async ({ data, header }, index) => {
       const { index_name: indexName, similarity, thumbnail } = header;
@@ -45,7 +50,7 @@ export async function createMarkdown(results: readonly ImageSource[]): Promise<s
           const response = await fetch(thumbnail);
 
           if (!response.ok) {
-            throw new Error(`缩略图请求失败，状态码 ${response.status}`);
+            throw new Error(`缩略图下载失败，HTTP 状态码 ${response.status}`);
           }
           const { height, width } = await new Image(await response.blob()).metadata();
 
@@ -60,5 +65,5 @@ export async function createMarkdown(results: readonly ImageSource[]): Promise<s
     }),
   );
 
-  return ['## SauceNAO 搜图结果', ...content].join('\n***\n');
+  return [lines.join('\n'), ...content].join('\n***\n');
 }