SnowLuma v2.4.2 source snapshot

2026-11-13 09:02:00 +08:00
commit fa9c73363d
@@ -0,0 +1,150 @@
+import { describe, expect, it, vi } from 'vitest';
+import { protobuf_decode, protobuf_encode } from '@snowluma/proton';
+import type { OidbBase } from '@snowluma/proto-defs/oidb';
+import type { Oidb0x112eReq, Oidb0x112eResp } from '@snowluma/proto-defs/oidb-actions/base';
+import type { SendPacketResult } from '@snowluma/common/packet-sender';
+
+import { ClickInlineKeyboardButton } from '../../../src/oidb-services/misc/click-inline-keyboard-button';
+
+function makeSender(resp?: Buffer) {
+    const r: SendPacketResult = {
+        success: true,
+        gotResponse: true,
+        errorCode: 0,
+        errorMessage: '',
+        responseData: resp ?? Buffer.alloc(0),
+    };
+    return { sendRawPacket: vi.fn(async () => r) };
+}
+
+describe('ClickInlineKeyboardButton namespace', () => {
+    it('declares 0x112E_1', () => {
+        expect(ClickInlineKeyboardButton.command).toBe(0x112e);
+        expect(ClickInlineKeyboardButton.subCommand).toBe(1);
+    });
+
+    describe('serialize', () => {
+        it('widens numeric ids to BigInt for uint_64 fields', () => {
+            const out = ClickInlineKeyboardButton.serialize({} as any, {
+                groupId: 12345,
+                botAppid: 67890,
+                buttonId: 'btn',
+                callbackData: 'data',
+                msgSeq: 100,
+            });
+            expect(out.groupId).toBe(12345n);
+            expect(out.botAppid).toBe(67890n);
+            expect(out.msgSeq).toBe(100n);
+        });
+
+        it('always sends unknown7=0 / unknown9=1 magic values', () => {
+            const out = ClickInlineKeyboardButton.serialize({} as any, {
+                groupId: 1,
+                botAppid: 1,
+                buttonId: 'x',
+                callbackData: 'y',
+                msgSeq: 1,
+            });
+            expect(out.unknown7).toBe(0);
+            expect(out.unknown9).toBe(1);
+        });
+
+        it('coerces buttonId / callbackData to strings (defensive)', () => {
+            const out = ClickInlineKeyboardButton.serialize({} as any, {
+                groupId: 1,
+                botAppid: 1,
+                buttonId: 123 as any,
+                callbackData: null as any,
+                msgSeq: 1,
+            });
+            expect(out.buttonId).toBe('123');
+            expect(out.callbackData).toBe('');
+        });
+    });
+
+    describe('deserialize', () => {
+        it('shapes the response with status/promptType/promptIcon = 0', () => {
+            expect(
+                ClickInlineKeyboardButton.deserialize(
+                    {} as any,
+                    {
+                        result: 1,
+                        errMsg: 'ok',
+                        promptText: 'hi',
+                    } as Oidb0x112eResp,
+                ),
+            ).toEqual({
+                result: 1,
+                errMsg: 'ok',
+                status: 0,
+                promptText: 'hi',
+                promptType: 0,
+                promptIcon: 0,
+            });
+        });
+
+        it('defaults result/errMsg/promptText when omitted', () => {
+            expect(ClickInlineKeyboardButton.deserialize({} as any, {} as Oidb0x112eResp)).toEqual({
+                result: 0,
+                errMsg: '',
+                status: 0,
+                promptText: '',
+                promptType: 0,
+                promptIcon: 0,
+            });
+        });
+    });
+
+    describe('invoke (e2e)', () => {
+        it('routes to OidbSvcTrpcTcp.0x112e_1', async () => {
+            const sender = makeSender();
+            await ClickInlineKeyboardButton.invoke(sender, {
+                groupId: 1,
+                botAppid: 1,
+                buttonId: 'x',
+                callbackData: 'y',
+                msgSeq: 1,
+            });
+            expect(sender.sendRawPacket.mock.calls[0]![0]).toBe('OidbSvcTrpcTcp.0x112e_1');
+        });
+
+        it('round-trips a real envelope', async () => {
+            const sender = makeSender(
+                Buffer.from(
+                    protobuf_encode<OidbBase<Oidb0x112eResp>>({
+                        body: { result: 7, errMsg: 'denied', promptText: 'try again' } as any,
+                    }),
+                ),
+            );
+            const out = await ClickInlineKeyboardButton.invoke(sender, {
+                groupId: 1,
+                botAppid: 1,
+                buttonId: 'x',
+                callbackData: 'y',
+                msgSeq: 1,
+            });
+            expect(out).toMatchObject({ result: 7, errMsg: 'denied', promptText: 'try again' });
+        });
+
+        it('encodes envelope body with all fields', async () => {
+            const sender = makeSender();
+            await ClickInlineKeyboardButton.invoke(sender, {
+                groupId: 12345,
+                botAppid: 67890,
+                buttonId: 'b',
+                callbackData: 'c',
+                msgSeq: 999,
+            });
+            const [, bytes] = sender.sendRawPacket.mock.calls[0]!;
+            const env = protobuf_decode<OidbBase<Oidb0x112eReq>>(bytes);
+            expect(env.body).toMatchObject({
+                botAppid: 67890n,
+                msgSeq: 999n,
+                buttonId: 'b',
+                callbackData: 'c',
+                groupId: 12345n,
+                unknown9: 1,
+            });
+        });
+    });
+});
@@ -0,0 +1,107 @@
+import { describe, expect, it, vi } from 'vitest';
+import { protobuf_decode, protobuf_encode } from '@snowluma/proton';
+import type { OidbBase } from '@snowluma/proto-defs/oidb';
+import type { ImageOcrReq, ImageOcrResp } from '@snowluma/proto-defs/oidb-actions/ocr';
+import type { SendPacketResult } from '@snowluma/common/packet-sender';
+
+import { ImageOcr } from '../../../src/oidb-services/misc/image-ocr';
+
+// A server OCR response: retCode 0, one text detection with a 2-vertex box.
+function ocrResponse(): Buffer {
+    const env: OidbBase<ImageOcrResp> = {
+        command: 0xe07,
+        subCommand: 0,
+        body: {
+            retCode: 0,
+            ocrRspBody: {
+                language: 'zh',
+                textDetections: [
+                    {
+                        detectedText: '你好',
+                        confidence: 95,
+                        polygon: {
+                            coordinates: [
+                                { x: 1, y: 2 },
+                                { x: 3, y: 4 },
+                            ],
+                        },
+                    },
+                ],
+            },
+        },
+    };
+    return Buffer.from(protobuf_encode<OidbBase<ImageOcrResp>>(env));
+}
+
+function makeSender(resp = ocrResponse()) {
+    const r: SendPacketResult = {
+        success: true,
+        gotResponse: true,
+        errorCode: 0,
+        errorMessage: '',
+        responseData: resp,
+    };
+    return { sendRawPacket: vi.fn(async () => r) };
+}
+
+describe('ImageOcr namespace', () => {
+    it('declares command 0xE07 sub 0 (no uin form)', () => {
+        expect(ImageOcr.command).toBe(0xe07);
+        expect(ImageOcr.subCommand).toBe(0);
+        expect(ImageOcr.uinForm).toBeUndefined(); // defaults to false in invokeOidb
+    });
+
+    it('serializes the image url into ocrReqBody with the fixed version/client/entrance header', () => {
+        const out = ImageOcr.serialize({} as any, { imageUrl: 'https://x/a.jpg' });
+        expect(out).toMatchObject({ version: 1, client: 0, entrance: 1 });
+        expect(out.ocrReqBody).toMatchObject({ imageUrl: 'https://x/a.jpg', isCut: false });
+    });
+
+    it('routes to OidbSvcTrpcTcp.0xe07_0 and encodes the exact wire bytes (fixed byte-oracle locks pb tags)', async () => {
+        const sender = makeSender();
+        await ImageOcr.invoke(sender, { imageUrl: 'https://x/a.jpg' });
+        const [cmd, bytes] = sender.sendRawPacket.mock.calls[0]!;
+        expect(cmd).toBe('OidbSvcTrpcTcp.0xe07_0');
+        // Independent byte-oracle: a fixed hex constant, NOT re-derived from the
+        // proto, so a tag change actually goes red (a symmetric encode→decode
+        // round-trip would not). Hand-decoded: 08 871c = field1 command varint
+        // 0xE07; 22 17 = field4 body (23B); body: 08 01 version=1, 18 01
+        // entrance=1 (client=0 omitted), 52 11 = field10 ocrReqBody (17B):
+        // 0a 0f = field1 imageUrl + "https://x/a.jpg".
+        expect(Buffer.from(bytes).toString('hex')).toBe(
+            '08871c22170801180152110a0f68747470733a2f2f782f612e6a7067',
+        );
+        // And the structure decodes back as expected.
+        const env = protobuf_decode<OidbBase<ImageOcrReq>>(bytes);
+        expect(env.command).toBe(0xe07);
+        expect(env.body.ocrReqBody?.imageUrl).toBe('https://x/a.jpg');
+    });
+
+    it('deserializes an empty ocr body to {texts:[], language:""}', () => {
+        expect(ImageOcr.deserialize({} as any, { retCode: 0 })).toEqual({
+            texts: [],
+            language: '',
+        });
+    });
+
+    it('deserializes text detections + coordinates + language', async () => {
+        const result = await ImageOcr.invoke(makeSender(), { imageUrl: 'https://x/a.jpg' });
+        expect(result.language).toBe('zh');
+        expect(result.texts).toEqual([
+            {
+                text: '你好',
+                confidence: 95,
+                coordinates: [
+                    { x: 1, y: 2 },
+                    { x: 3, y: 4 },
+                ],
+            },
+        ]);
+    });
+
+    it('throws on a non-zero retCode', () => {
+        expect(() =>
+            ImageOcr.deserialize({} as any, { retCode: 5, errMsg: 'bad', wording: 'nope' }),
+        ).toThrow();
+    });
+});
@@ -0,0 +1,48 @@
+import { describe, expect, it, vi } from 'vitest';
+import { protobuf_decode } from '@snowluma/proton';
+import type { OidbBase } from '@snowluma/proto-defs/oidb';
+import type { Oidb0xeb7Req } from '@snowluma/proto-defs/oidb-actions/base';
+import type { SendPacketResult } from '@snowluma/common/packet-sender';
+
+import { SendGroupSign } from '../../../src/oidb-services/misc/send-group-sign';
+
+function makeDeps() {
+    const r: SendPacketResult = {
+        success: true,
+        gotResponse: true,
+        errorCode: 0,
+        errorMessage: '',
+        responseData: Buffer.alloc(0),
+    };
+    return {
+        sendRawPacket: vi.fn(async () => r),
+        identity: { uin: '10001' } as any,
+    };
+}
+
+describe('SendGroupSign namespace', () => {
+    it('declares 0xEB7_1', () => {
+        expect(SendGroupSign.command).toBe(0xeb7);
+        expect(SendGroupSign.subCommand).toBe(1);
+    });
+
+    it('overrides wire name to preserve the historic UPPERCASE EB7 byte-equality', async () => {
+        const deps = makeDeps();
+        await SendGroupSign.invoke(deps, { groupId: 12345 });
+        expect(deps.sendRawPacket.mock.calls[0]![0]).toBe('OidbSvcTrpcTcp.0xEB7_1');
+    });
+
+    describe('serialize', () => {
+        it('packages uin / groupId as strings + hardcodes version "9.0.90"', async () => {
+            const deps = makeDeps();
+            await SendGroupSign.invoke(deps, { groupId: 12345 });
+            const [, bytes] = deps.sendRawPacket.mock.calls[0]!;
+            const env = protobuf_decode<OidbBase<Oidb0xeb7Req>>(bytes);
+            expect(env.body?.signInInfo).toEqual({
+                uin: '10001',
+                groupId: '12345',
+                version: '9.0.90',
+            });
+        });
+    });
+});
@@ -0,0 +1,95 @@
+import { describe, expect, it, vi } from 'vitest';
+import { protobuf_decode, protobuf_encode } from '@snowluma/proton';
+import type { OidbBase } from '@snowluma/proto-defs/oidb';
+import type { Oidb0x990Req, Oidb0x990Resp } from '@snowluma/proto-defs/oidb-actions/base';
+import type { SendPacketResult } from '@snowluma/common/packet-sender';
+
+import { TranslateEnToZh } from '../../../src/oidb-services/misc/translate-en-to-zh';
+
+function makeSender(resp?: Buffer) {
+    const r: SendPacketResult = {
+        success: true,
+        gotResponse: true,
+        errorCode: 0,
+        errorMessage: '',
+        responseData: resp ?? Buffer.alloc(0),
+    };
+    return { sendRawPacket: vi.fn(async () => r) };
+}
+
+describe('TranslateEnToZh namespace', () => {
+    it('declares 0x990_2', () => {
+        expect(TranslateEnToZh.command).toBe(0x990);
+        expect(TranslateEnToZh.subCommand).toBe(2);
+    });
+
+    describe('serialize', () => {
+        it('hardcodes en → zh and tag10/12 = 1', () => {
+            const out = TranslateEnToZh.serialize({} as any, { words: ['hello'] });
+            expect(out).toEqual({
+                translateReq: { srcLang: 'en', dstLang: 'zh', words: ['hello'] },
+                tag10: 1,
+                tag12: 1,
+            });
+        });
+
+        it('passes the words array verbatim', () => {
+            const out = TranslateEnToZh.serialize({} as any, { words: ['a', 'b', 'c'] });
+            expect(out.translateReq?.words).toEqual(['a', 'b', 'c']);
+        });
+    });
+
+    describe('deserialize', () => {
+        it('returns dstWords when translateResp is present', () => {
+            expect(
+                TranslateEnToZh.deserialize(
+                    {} as any,
+                    {
+                        translateResp: { dstWords: ['你好', '世界'] },
+                    } as Oidb0x990Resp,
+                ),
+            ).toEqual(['你好', '世界']);
+        });
+
+        it('returns [] when dstWords is omitted but translateResp exists', () => {
+            expect(
+                TranslateEnToZh.deserialize({} as any, { translateResp: {} } as Oidb0x990Resp),
+            ).toEqual([]);
+        });
+
+        it('throws when translateResp is missing', () => {
+            expect(() => TranslateEnToZh.deserialize({} as any, {} as Oidb0x990Resp)).toThrow(
+                'translate response empty',
+            );
+        });
+    });
+
+    describe('invoke (e2e)', () => {
+        it('routes to OidbSvcTrpcTcp.0x990_2', async () => {
+            const sender = makeSender(
+                Buffer.from(
+                    protobuf_encode<OidbBase<Oidb0x990Resp>>({
+                        body: { translateResp: { dstWords: ['x'] } } as any,
+                    }),
+                ),
+            );
+            await TranslateEnToZh.invoke(sender, { words: ['hello'] });
+            expect(sender.sendRawPacket.mock.calls[0]![0]).toBe('OidbSvcTrpcTcp.0x990_2');
+        });
+
+        it('encodes envelope body with the request payload', async () => {
+            const sender = makeSender(
+                Buffer.from(
+                    protobuf_encode<OidbBase<Oidb0x990Resp>>({
+                        body: { translateResp: { dstWords: ['你好'] } } as any,
+                    }),
+                ),
+            );
+            const out = await TranslateEnToZh.invoke(sender, { words: ['hello'] });
+            expect(out).toEqual(['你好']);
+            const [, bytes] = sender.sendRawPacket.mock.calls[0]!;
+            const env = protobuf_decode<OidbBase<Oidb0x990Req>>(bytes);
+            expect(env.body?.translateReq?.words).toEqual(['hello']);
+        });
+    });
+});