kokkoro-plugin-animetrace 1.0.1 → 1.0.2

package/README.md

@@ -8,13 +8,13 @@
 bun add kokkoro-plugin-animetrace
 ```
 
-发送「/搜角色」时附带图片，插件会识别消息中的第一张图片，并通过 QQ Markdown 展示结果。发送带图的「搜角色」也可以触发识别。
+发送「/搜角色」指令时附带图片，或引用包含图片的消息回复该指令。插件识别首张图片中的角色，并通过 QQ Markdown 返回结果。「搜角色」与「/搜角色」的用法相同。
 
 完整使用说明见 [AnimeTrace 角色识别](https://kokkoro.js.org/plugin/animetrace)。
 
 ## 环境变量
 
-在项目根目录的 `.env` 中设置每个人物展示的候选角色数量上限：
+在项目根目录的 `.env` 中配置每个人物最多显示的候选角色数量：
 
 ```ini
 ANIMETRACE_LIMIT=3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kokkoro-plugin-animetrace",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "AnimeTrace 角色识别，查找图片中的动漫、游戏角色及所属作品。",
   "keywords": [
     "animetrace",

package/src/index.ts

@@ -7,19 +7,23 @@ 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 fetchCharacters(image.url, logger);
     const { data, trace_id } = result;
+    const notice = images.length > 1 ? '检测到消息中包含多张图片，已选取首张图片进行识别。' : undefined;
 
     logger.info('已识别图片中的角色', { count: data.length, trace_id });
 
     await context.reply({
       msg_type: 2,
-      markdown: { content: createMarkdown(data) },
+      markdown: { content: createMarkdown(data, notice) },
     });
   }).shortcut('搜角色');
 };

package/src/service.ts

@@ -53,7 +53,7 @@ export async function fetchCharacters(url: string, logger?: Logger): Promise<Ani
   const response = await fetch(ANIMETRACE_API, { method: 'POST', body: form });
 
   if (!response.ok) {
-    throw new Error(`接口请求失败，状态码 ${response.status}`);
+    throw new Error(`AnimeTrace 请求失败，HTTP 状态码 ${response.status}`);
   }
   const result = <AnimeTrace>await response.json();
 
@@ -64,7 +64,7 @@ export async function fetchCharacters(url: string, logger?: Logger): Promise<Ani
   }
 
   if (result.data.length === 0) {
-    throw new Error('没有识别到角色');
+    throw new Error('未识别到角色');
   }
   return result;
 }

package/src/util.ts

@@ -1,18 +1,22 @@
 import { type CharacterResult } from './service';
 
-const { ANIMETRACE_LIMIT = 3 } = import.meta.env;
+const { ANIMETRACE_LIMIT = '3' } = import.meta.env;
 const limit = Number(ANIMETRACE_LIMIT);
 
 if (!Number.isInteger(limit) || limit <= 0) {
   throw new Error('ANIMETRACE_LIMIT 必须为正整数');
 }
 
-/** 将 AnimeTrace 识别结果转换为 QQ Markdown。 */
-export function createMarkdown(results: readonly CharacterResult[]): string {
-  const lines = ['## AnimeTrace 搜索结果'];
+/** 将 AnimeTrace 识别结果转换为 QQ Markdown，可在标题下方展示提示。 */
+export function createMarkdown(results: readonly CharacterResult[], notice?: string): string {
+  const lines = ['## AnimeTrace 识别结果'];
+
+  if (notice) {
+    lines.push('', `> ${notice}`);
+  }
 
   if (results.some(({ not_confident: isNotConfident }) => isNotConfident)) {
-    lines.push('', '> 带 ＊ 的人物可能识别不准确。');
+    lines.push('', '> 标有 ＊ 的识别结果可能不准确，请注意甄别。');
   }
 
   for (const [index, { character: candidates, not_confident: isNotConfident }] of results.entries()) {