NapCatQQ source snapshot

2026-11-13 09:01:47 +08:00
commit f954252a34
@@ -0,0 +1,76 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+import path from 'node:path';
+import { DEFAULT_FLASH_VALIDITY_SECONDS } from '@/napcat-core/data/flash';
+
+const richMediaList = [
+  '.mp4', '.mov', '.avi', '.wmv', '.mpeg', '.mpg', '.flv', '.mkv',
+  '.png', '.gif', '.jpg', '.jpeg', '.webp', '.bmp',
+];
+
+export const CreateFlashTaskPayloadSchema = Type.Object({
+  files: Type.Union([
+    Type.Array(Type.String()),
+    Type.String(),
+  ], { description: '文件列表或单个文件路径' }),
+  name: Type.Optional(Type.String({ description: '任务名称' })),
+  thumb_path: Type.Optional(Type.String({ description: '缩略图路径' })),
+  validitySeconds: Type.Optional(Type.Number({ description: '文件集有效期（秒），可选 14 天 1209600 / 90 天 7776000 / 180 天 15552000，默认 1209600' })),
+});
+export type CreateFlashTaskPayload = Static<typeof CreateFlashTaskPayloadSchema>;
+
+export class CreateFlashTask extends OneBotAction<CreateFlashTaskPayload, any> {
+  override actionName = ActionName.CreateFlashTask;
+  override payloadSchema = CreateFlashTaskPayloadSchema;
+  override returnSchema = Type.Any({ description: '任务创建结果' });
+  override actionSummary = '创建闪传任务';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    files: 'C:\\test.jpg',
+    name: 'test_task',
+    validitySeconds: DEFAULT_FLASH_VALIDITY_SECONDS,
+  };
+
+  override returnExample = {
+    task_id: 'task_123',
+  };
+
+  async _handle (payload: CreateFlashTaskPayload) {
+    const fileList = Array.isArray(payload.files) ? payload.files : [payload.files];
+    let thumbPath: string = '';
+
+    if (fileList.length === 1) {
+      // 我是真没hook到那种合并的缩略图是哪个方法产生的，暂时不实现(怀疑是js直接canvas渲染的！！) // 确认了猜想
+      const filePath = fileList[0];
+      if (filePath === undefined) {
+        return {};
+      }
+      const ext = path.extname(filePath).toLowerCase();
+
+      if (richMediaList.includes(ext)) {
+        try {
+          const res = await this.core.apis.FlashApi.createFileThumbnail(filePath);
+          if (res && typeof res === 'object' && 'result' in res && res.result === 0) {
+            thumbPath = res.targetPath as string;
+          }
+        } catch (_e) {
+        }
+      }
+    }
+
+    function toPlatformPath (inputPath: string) {
+      const unifiedPath = inputPath.replace(/[\\/]/g, path.sep);
+      return path.normalize(unifiedPath);
+    }
+
+    let normalPath: string;
+    if (payload.thumb_path !== undefined) {
+      normalPath = path.normalize(payload.thumb_path);
+    } else {
+      normalPath = toPlatformPath(thumbPath);
+    }
+    const validitySeconds = payload.validitySeconds ?? DEFAULT_FLASH_VALIDITY_SECONDS;
+    return await this.core.apis.FlashApi.createFlashTransferUploadTask(fileList, normalPath, payload.name || '', validitySeconds);
+  }
+}
@@ -0,0 +1,27 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+
+export const DownloadFilesetPayloadSchema = Type.Object({
+  fileset_id: Type.String({ description: '文件集 ID' }),
+});
+
+export type DownloadFilesetPayload = Static<typeof DownloadFilesetPayloadSchema>;
+
+export class DownloadFileset extends OneBotAction<DownloadFilesetPayload, any> {
+  override actionName = ActionName.DownloadFileset;
+  override payloadSchema = DownloadFilesetPayloadSchema;
+  override returnSchema = Type.Any({ description: '下载结果' });
+  override actionSummary = '下载文件集';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    fileset_id: 'set_123',
+  };
+
+  override returnExample = null;
+
+  async _handle (payload: DownloadFilesetPayload) {
+    // 默认路径 / fileset_id /为下载路径
+    return await this.core.apis.FlashApi.downloadFileSetBySetId(payload.fileset_id);
+  }
+}
@@ -0,0 +1,34 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+
+export const GetFilesetIdPayloadSchema = Type.Object({
+  share_code: Type.String({ description: '分享码或分享链接' }),
+});
+
+export type GetFilesetIdPayload = Static<typeof GetFilesetIdPayloadSchema>;
+
+export class GetFilesetId extends OneBotAction<GetFilesetIdPayload, { fileset_id: string }> {
+  override actionName = ActionName.GetFilesetId;
+  override payloadSchema = GetFilesetIdPayloadSchema;
+  override returnSchema = Type.Object({
+    fileset_id: Type.String({ description: '文件集 ID' }),
+  });
+
+  override actionSummary = '获取文件集 ID';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    share_code: '123456',
+  };
+
+  override returnExample = {
+    fileset_id: 'set_123',
+  };
+
+  async _handle (payload: GetFilesetIdPayload) {
+    // 适配share_link 防止被传 Link无法解析
+    const code = payload.share_code.includes('=') ? payload.share_code.split('=').slice(1).join('=') : payload.share_code;
+    const result = await this.core.apis.FlashApi.fromShareLinkFindSetId(code);
+    return { fileset_id: result.fileSetId };
+  }
+}
@@ -0,0 +1,29 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+
+export const GetFilesetInfoPayloadSchema = Type.Object({
+  fileset_id: Type.String({ description: '文件集 ID' }),
+});
+
+export type GetFilesetInfoPayload = Static<typeof GetFilesetInfoPayloadSchema>;
+
+export class GetFilesetInfo extends OneBotAction<GetFilesetInfoPayload, any> {
+  override actionName = ActionName.GetFilesetInfo;
+  override payloadSchema = GetFilesetInfoPayloadSchema;
+  override returnSchema = Type.Any({ description: '文件集信息' });
+  override actionSummary = '获取文件集信息';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    fileset_id: 'set_123',
+  };
+
+  override returnExample = {
+    fileset_id: 'set_123',
+    file_list: [],
+  };
+
+  async _handle (payload: GetFilesetInfoPayload) {
+    return await this.core.apis.FlashApi.getFileSetIndoBySetId(payload.fileset_id);
+  }
+}
@@ -0,0 +1,31 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+
+export const GetFlashFileListPayloadSchema = Type.Object({
+  fileset_id: Type.String({ description: '文件集 ID' }),
+});
+
+export type GetFlashFileListPayload = Static<typeof GetFlashFileListPayloadSchema>;
+
+export class GetFlashFileList extends OneBotAction<GetFlashFileListPayload, any> {
+  override actionName = ActionName.GetFlashFileList;
+  override payloadSchema = GetFlashFileListPayloadSchema;
+  override returnSchema = Type.Any({ description: '文件列表' });
+  override actionSummary = '获取闪传文件列表';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    fileset_id: 'set_123',
+  };
+
+  override returnExample = [
+    {
+      file_name: 'test.jpg',
+      size: 1024,
+    },
+  ];
+
+  async _handle (payload: GetFlashFileListPayload) {
+    return await this.core.apis.FlashApi.getFileListBySetId(payload.fileset_id);
+  }
+}
@@ -0,0 +1,34 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+
+export const GetFlashFileUrlPayloadSchema = Type.Object({
+  fileset_id: Type.String({ description: '文件集 ID' }),
+  file_name: Type.Optional(Type.String({ description: '文件名' })),
+  file_index: Type.Optional(Type.Number({ description: '文件索引' })),
+});
+
+export type GetFlashFileUrlPayload = Static<typeof GetFlashFileUrlPayloadSchema>;
+
+export class GetFlashFileUrl extends OneBotAction<GetFlashFileUrlPayload, any> {
+  override actionName = ActionName.GetFlashFileUrl;
+  override payloadSchema = GetFlashFileUrlPayloadSchema;
+  override returnSchema = Type.Any({ description: '文件下载链接' });
+  override actionSummary = '获取闪传文件链接';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    fileset_id: 'set_123',
+  };
+
+  override returnExample = {
+    url: 'http://example.com/flash.jpg',
+  };
+
+  async _handle (payload: GetFlashFileUrlPayload) {
+    // 文件的索引依旧从0开始
+    return await this.core.apis.FlashApi.getFileTransUrl(payload.fileset_id, {
+      fileName: payload.file_name,
+      fileIndex: payload.file_index,
+    });
+  }
+}
@@ -0,0 +1,26 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+
+export const GetShareLinkPayloadSchema = Type.Object({
+  fileset_id: Type.String({ description: '文件集 ID' }),
+});
+
+export type GetShareLinkPayload = Static<typeof GetShareLinkPayloadSchema>;
+
+export class GetShareLink extends OneBotAction<GetShareLinkPayload, any> {
+  override actionName = ActionName.GetShareLink;
+  override payloadSchema = GetShareLinkPayloadSchema;
+  override returnSchema = Type.Any({ description: '分享链接' });
+  override actionSummary = '获取文件分享链接';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    fileset_id: 'set_123',
+  };
+
+  override returnExample = 'http://example.com/share';
+
+  async _handle (payload: GetShareLinkPayload) {
+    return await this.core.apis.FlashApi.getShareLinkBySetId(payload.fileset_id);
+  }
+}
@@ -0,0 +1,50 @@
+import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
+import { ActionName } from '@/napcat-onebot/action/router';
+import { Static, Type } from '@sinclair/typebox';
+import { ChatType, Peer } from 'napcat-core/types';
+
+export const SendFlashMsgPayloadSchema = Type.Object({
+  fileset_id: Type.String({ description: '文件集 ID' }),
+  user_id: Type.Optional(Type.String({ description: '用户 QQ' })),
+  group_id: Type.Optional(Type.String({ description: '群号' })),
+});
+
+export type SendFlashMsgPayload = Static<typeof SendFlashMsgPayloadSchema>;
+
+export class SendFlashMsg extends OneBotAction<SendFlashMsgPayload, any> {
+  override actionName = ActionName.SendFlashMsg;
+  override payloadSchema = SendFlashMsgPayloadSchema;
+  override returnSchema = Type.Any({ description: '发送结果' });
+  override actionSummary = '发送闪传消息';
+  override actionTags = ['文件扩展'];
+  override payloadExample = {
+    fileset_id: 'set_123',
+    user_id: '123456789',
+  };
+
+  override returnExample = {
+    message_id: 123456,
+  };
+
+  async _handle (payload: SendFlashMsgPayload) {
+    let peer: Peer;
+
+    if (payload.group_id) {
+      peer = { chatType: ChatType.KCHATTYPEGROUP, peerUid: payload.group_id.toString() };
+    } else if (payload.user_id) {
+      const uid = await this.core.apis.UserApi.getUidByUinV2(payload.user_id.toString());
+      if (!uid) throw new Error('User not found');
+
+      // 可能需要更严格的判断
+      const isBuddy = await this.core.apis.FriendApi.isBuddy(uid);
+      peer = {
+        chatType: isBuddy ? ChatType.KCHATTYPEC2C : ChatType.KCHATTYPETEMPC2CFROMGROUP,
+        peerUid: uid,
+      };
+    } else {
+      throw new Error('user_id or group_id is required');
+    }
+
+    return await this.core.apis.FlashApi.sendFlashMessage(payload.fileset_id, peer);
+  }
+}