NapCatQQ source snapshot

packages/napcat-onebot/action/extends/OCRImage.ts

@@ -0,0 +1,57 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { checkFileExist, uriToLocalFile } from 'napcat-common/src/file';
+import fs from 'fs';
+import { Static, Type } from '@sinclair/typebox';
+
+import { ExtendsActionsExamples } from '../example/ExtendsActionsExamples';
+
+const PayloadSchema = Type.Object({
+  image: Type.String({ description: '图片路径、URL或Base64' }),
+});
+
+type PayloadType = Static<typeof PayloadSchema>;
+
+const ReturnSchema = Type.Any({ description: 'OCR结果' });
+
+type ReturnType = Static<typeof ReturnSchema>;
+
+class OCRImageBase extends OneBotAction<PayloadType, ReturnType> {
+  override payloadSchema = PayloadSchema;
+  override returnSchema = ReturnSchema;
+  override actionSummary = '图片 OCR 识别';
+  override actionDescription = '识别图片中的文字内容(仅Windows端支持)';
+  override actionTags = ['扩展接口'];
+  override payloadExample = ExtendsActionsExamples.OCRImage.payload;
+  override returnExample = ExtendsActionsExamples.OCRImage.response;
+
+  async _handle (payload: PayloadType): Promise<ReturnType> {
+    const { path, success, isLocal } = await uriToLocalFile(this.core.NapCatTempPath, payload.image);
+    if (!success) {
+      throw new Error(`OCR ${payload.image}失败, image字段可能格式不正确`);
+    }
+    if (path) {
+      try {
+        await checkFileExist(path, 5000); // 避免崩溃
+        const ret = await this.core.apis.SystemApi.ocrImage(path);
+        if (!ret) {
+          throw new Error(`OCR ${payload.image}失败`);
+        }
+        return ret.result;
+      } finally {
+        if (!isLocal) fs.unlink(path, () => { });
+      }
+    }
+    throw new Error(`OCR ${payload.image}失败, 文件可能不存在`);
+  }
+}
+
+export class OCRImage extends OCRImageBase {
+  override actionName = ActionName.OCRImage;
+  override actionSummary = '图片 OCR 识别';
+}
+
+export class IOCRImage extends OCRImageBase {
+  override actionName = ActionName.IOCRImage;
+  override actionSummary = '图片 OCR 识别 (内部)';
+}