SnowLuma v2.4.2 source snapshot

packages/protocol/tests/msg-push/flash-transfer-decode.test.ts

@@ -0,0 +1,242 @@
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import { describe, expect, it } from 'vitest';
+import { protobuf_decode, protobuf_encode } from '@snowluma/proton';
+import { decodeRichBody } from '../../src/msg-push/rich-body-decoder';
+import type { MessageBody, PushMsg } from '@snowluma/proto-defs/message';
+import type { MarkdownData } from '@snowluma/proto-defs/action';
+
+// Build a svc=45 richui markdown commonElem carrying a FlashTransfer card, the
+// way older QQ clients (≤9.9.30) deliver a 闪传 file (#199/#200).
+function flashBody(json: unknown): MessageBody {
+    const markdown = `[闪传](mqqapi://markdown/node?nodeType=richui&json=${encodeURIComponent(JSON.stringify(json))})`;
+    const pbElem = protobuf_encode<MarkdownData>({ content: markdown });
+    return {
+        richText: {
+            elems: [{ commonElem: { serviceType: 45, businessType: 3, pbElem } } as never],
+        },
+    };
+}
+
+describe('decodeRichBody / 闪传 flash_file (#199/#200)', () => {
+    it('extracts filesetId, title, sceneType from a nested FlashTransfer card', () => {
+        const out = decodeRichBody(
+            flashBody({
+                busId: 'FlashTransfer',
+                templateId: 'flash',
+                version: 2,
+                layout: { viewId: 'flash_file', width: -2, height: -2 },
+                data: { fileSetId: 'fs-abc-123', title: 'report.pdf', sceneType: 2 },
+            }),
+            true,
+        );
+        expect(out).toEqual([
+            { type: 'flash_file', filesetId: 'fs-abc-123', fileName: 'report.pdf', sceneType: 2 },
+        ]);
+    });
+
+    it('recursive search finds the fields at any depth + accepts the filesetId alias', () => {
+        const out = decodeRichBody(
+            flashBody({
+                busId: 'FlashTransfer',
+                templateId: 'flash',
+                layout: { deep: { nested: { filesetId: 'fs-deep', name: 'a.zip' } } },
+            }),
+            true,
+        );
+        expect(out).toEqual([
+            { type: 'flash_file', filesetId: 'fs-deep', fileName: 'a.zip', sceneType: 0 },
+        ]);
+    });
+
+    it('drops a malformed card that has no fileset identity', () => {
+        const out = decodeRichBody(
+            flashBody({ busId: 'FlashTransfer', templateId: 'flash' }),
+            true,
+        );
+        expect(out).toEqual([]);
+    });
+
+    it('does not misclassify a non-FlashTransfer markdown card as flash_file', () => {
+        const out = decodeRichBody(
+            flashBody({ busId: 'SomethingElse', templateId: 'button' }),
+            true,
+        );
+        expect(out.some((e) => e.type === 'flash_file')).toBe(false);
+        expect(out[0]?.type).toBe('markdown');
+    });
+});
+
+// Current NT cards (#358) put fileset identity on MarkdownData.extInfo
+// (extType=1 / DecodeMdExtInfoFileTransfer) and in the click scheme. The
+// richui JSON no longer has `data.fileSetId`.
+function currentNtFlashBody(
+    json: unknown,
+    ext?: MarkdownData['extInfo'],
+    summary?: string,
+): MessageBody {
+    const markdown = `[闪传](mqqapi://markdown/node?nodeType=richui&json=${encodeURIComponent(JSON.stringify(json))})`;
+    const pbElem = protobuf_encode<MarkdownData>({
+        content: markdown,
+        summary,
+        extType: ext ? 1 : undefined,
+        extInfo: ext,
+    });
+    return {
+        richText: {
+            elems: [{ commonElem: { serviceType: 45, businessType: 3, pbElem } } as never],
+        },
+    };
+}
+
+const currentNtCardJson = {
+    busId: 'FlashTransfer',
+    templateId: 'flash',
+    version: 2,
+    layout: { viewId: 'flash_file', width: -2, height: -2, direction: 'horizontal' },
+    attributes: {
+        viewId: 'flash_file',
+        attributes: [
+            {
+                viewId: 'file',
+                scheme: 'mqqrouter://flash_transfer/open_fileset?fileset_id=fs-nt-358&version=1&channel_id=1&src_type=internal&scene_type=1',
+            },
+        ],
+    },
+};
+
+describe('decodeRichBody / 闪传 flash_file (#358 current NT card)', () => {
+    it('reads filesetId and name from markdown extInfo when the JSON has no data.fileSetId', () => {
+        const out = decodeRichBody(
+            currentNtFlashBody(
+                currentNtCardJson,
+                {
+                    filesetId: 'fs-nt-358',
+                    name: 'cpwd截图.zip',
+                    fileSize: 9196055,
+                    expireTime: 1787815533,
+                },
+                '[QQ闪传] cpwd截图.zip',
+            ),
+            true,
+        );
+        expect(out).toEqual([
+            {
+                type: 'flash_file',
+                filesetId: 'fs-nt-358',
+                fileName: 'cpwd截图.zip',
+                sceneType: 1,
+            },
+        ]);
+    });
+
+    it('reads the cover URL from extInfo.thumbnail', () => {
+        const out = decodeRichBody(
+            currentNtFlashBody(currentNtCardJson, {
+                filesetId: 'fs-nt-358',
+                name: 'a.zip',
+                thumbnail: {
+                    download: { field1: 2, downloadUrl: 'https://example.test/cover.jpg' },
+                },
+            }),
+            true,
+        );
+        expect(out).toEqual([
+            {
+                type: 'flash_file',
+                filesetId: 'fs-nt-358',
+                fileName: 'a.zip',
+                sceneType: 1,
+                thumbUrl: 'https://example.test/cover.jpg',
+            },
+        ]);
+    });
+
+    it('falls back to the open_fileset scheme when extInfo is absent', () => {
+        const out = decodeRichBody(currentNtFlashBody(currentNtCardJson), true);
+        expect(out).toEqual([
+            {
+                type: 'flash_file',
+                filesetId: 'fs-nt-358',
+                fileName: '',
+                sceneType: 1,
+            },
+        ]);
+    });
+
+    it('still decodes from extInfo when the richui JSON is truncated', () => {
+        const markdown =
+            '[闪传](mqqapi://markdown/node?nodeType=richui&json=%7B%22busId%22%3A%22FlashTransfer%22%2C%22src%22%3A%22';
+        const pbElem = protobuf_encode<MarkdownData>({
+            content: markdown,
+            summary: '[QQ闪传] report.pdf',
+            extType: 1,
+            extInfo: { filesetId: 'fs-from-ext', name: 'report.pdf' },
+        });
+        const out = decodeRichBody(
+            {
+                richText: {
+                    elems: [{ commonElem: { serviceType: 45, businessType: 3, pbElem } } as never],
+                },
+            },
+            true,
+        );
+        expect(out).toEqual([
+            {
+                type: 'flash_file',
+                filesetId: 'fs-from-ext',
+                fileName: 'report.pdf',
+                sceneType: 0,
+            },
+        ]);
+    });
+
+    it('drops the compatibility text sibling after a successful flash_file decode', () => {
+        const markdown = `[闪传](mqqapi://markdown/node?nodeType=richui&json=${encodeURIComponent(JSON.stringify(currentNtCardJson))})`;
+        const pbElem = protobuf_encode<MarkdownData>({
+            content: markdown,
+            extType: 1,
+            extInfo: { filesetId: 'fs-nt-358', name: 'a.zip' },
+        });
+        const out = decodeRichBody(
+            {
+                richText: {
+                    elems: [
+                        { commonElem: { serviceType: 45, businessType: 3, pbElem } } as never,
+                        { text: { str: '[闪传]' } } as never,
+                    ],
+                },
+            },
+            true,
+        );
+        expect(out).toEqual([
+            {
+                type: 'flash_file',
+                filesetId: 'fs-nt-358',
+                fileName: 'a.zip',
+                sceneType: 1,
+            },
+        ]);
+    });
+
+    it('decodes the sanitized #358 C2C push (extInfo + open_fileset scheme)', () => {
+        const hex = readFileSync(
+            fileURLToPath(new URL('./fixtures/issue-358-flash-push.hex', import.meta.url)),
+            'utf8',
+        ).trim();
+        const push = protobuf_decode<PushMsg>(Buffer.from(hex, 'hex'));
+        const out = decodeRichBody(push.message?.body, false);
+        // Reporter redacted the richui JSON (unterminated src), so scene_type
+        // cannot be recovered from the scheme. extInfo still supplies fileset + name
+        // and the cover URL (placeholder).
+        expect(out).toEqual([
+            {
+                type: 'flash_file',
+                filesetId: '[UUID]',
+                fileName: 'cpwd截图.zip',
+                sceneType: 0,
+                thumbUrl: '[FILE_URL]',
+            },
+        ]);
+    });
+});