zalo-agent-cli 1.2.0 → 1.4.0

package/src/mcp/thread-name-cache.test.js

@@ -0,0 +1,139 @@
+import { describe, it, beforeEach } from "node:test";
+import assert from "node:assert/strict";
+import { ThreadNameCache } from "./thread-name-cache.js";
+
+/** Create a mock API that returns predictable group/friend data */
+function createMockApi(groups = {}, friends = []) {
+    return {
+        getAllGroups: async () => ({
+            gridVerMap: Object.fromEntries(Object.keys(groups).map((id) => [id, 1])),
+        }),
+        getGroupInfo: async (ids) => ({
+            gridInfoMap: Object.fromEntries(ids.filter((id) => groups[id]).map((id) => [id, groups[id]])),
+        }),
+        getAllFriends: async () => friends,
+    };
+}
+
+describe("ThreadNameCache", () => {
+    let cache;
+
+    beforeEach(() => {
+        cache = new ThreadNameCache();
+    });
+
+    it("starts empty and not ready", () => {
+        assert.equal(cache.ready, false);
+        assert.equal(cache.size, 0);
+        assert.equal(cache.get("any"), null);
+        assert.equal(cache.getName("any"), null);
+    });
+
+    it("loads groups and friends on init", async () => {
+        const api = createMockApi(
+            {
+                g1: { name: "Nhóm Chờ Báo Giá", totalMember: 20 },
+                g2: { name: "Soạn hàng Q. Vũ", totalMember: 15 },
+            },
+            [
+                { userId: "u1", displayName: "Viet Anh", zaloName: "VA" },
+                { userId: "u2", zaloName: "Bob" },
+            ],
+        );
+
+        await cache.init(api);
+
+        assert.equal(cache.ready, true);
+        assert.equal(cache.size, 4);
+        assert.deepEqual(cache.get("g1"), { name: "Nhóm Chờ Báo Giá", type: "group", memberCount: 20 });
+        assert.equal(cache.getName("g2"), "Soạn hàng Q. Vũ");
+        assert.deepEqual(cache.get("u1"), { name: "Viet Anh", type: "dm" });
+        assert.equal(cache.getName("u2"), "Bob");
+    });
+
+    it("handles API failures gracefully", async () => {
+        const api = {
+            getAllGroups: async () => {
+                throw new Error("network error");
+            },
+            getAllFriends: async () => {
+                throw new Error("network error");
+            },
+        };
+
+        await cache.init(api);
+
+        assert.equal(cache.ready, true);
+        assert.equal(cache.size, 0);
+    });
+
+    describe("search", () => {
+        beforeEach(async () => {
+            const api = createMockApi(
+                {
+                    g1: { name: "Nhóm Chờ Báo Giá", totalMember: 20 },
+                    g2: { name: "Soạn hàng Q. Vũ - QV", totalMember: 15 },
+                    g3: { name: "Soạn hàng kho 2", totalMember: 8 },
+                    g4: { name: "Admin Team", totalMember: 5 },
+                },
+                [{ userId: "u1", displayName: "Soạn Văn", zaloName: "SV" }],
+            );
+            await cache.init(api);
+        });
+
+        it("finds groups by Vietnamese name (accent-insensitive)", () => {
+            const results = cache.search("soan hang");
+            assert.equal(results.length, 2);
+            assert.equal(results[0].name, "Soạn hàng kho 2");
+            assert.equal(results[1].name, "Soạn hàng Q. Vũ - QV");
+        });
+
+        it("finds with exact Vietnamese diacritics", () => {
+            const results = cache.search("Soạn hàng");
+            assert.equal(results.length, 2);
+            assert.ok(results.every((r) => r.type === "group"));
+        });
+
+        it("filters by type", () => {
+            const groups = cache.search("Soạn", "group");
+            assert.equal(groups.length, 2);
+            assert.ok(groups.every((r) => r.type === "group"));
+
+            const dms = cache.search("Soạn", "dm");
+            assert.equal(dms.length, 1);
+            assert.equal(dms[0].name, "Soạn Văn");
+        });
+
+        it("respects limit parameter", () => {
+            const results = cache.search("Soạn", "all", 1);
+            assert.equal(results.length, 1);
+        });
+
+        it("returns empty for no match", () => {
+            const results = cache.search("xyz_nonexistent");
+            assert.equal(results.length, 0);
+        });
+
+        it("prioritizes prefix matches", () => {
+            const results = cache.search("Admin");
+            assert.equal(results[0].name, "Admin Team");
+        });
+    });
+
+    describe("set (update)", () => {
+        it("updates existing entry", async () => {
+            const api = createMockApi({ g1: { name: "Old Name", totalMember: 5 } }, []);
+            await cache.init(api);
+
+            cache.set("g1", { name: "New Name" });
+            assert.equal(cache.getName("g1"), "New Name");
+            assert.equal(cache.get("g1").type, "group");
+            assert.equal(cache.get("g1").memberCount, 5);
+        });
+
+        it("adds new entry", () => {
+            cache.set("new1", { name: "New Group", type: "group", memberCount: 3 });
+            assert.deepEqual(cache.get("new1"), { name: "New Group", type: "group", memberCount: 3 });
+        });
+    });
+});

package/src/utils/qr-display.js

@@ -59,12 +59,14 @@ export function displayQR(event) {
     // JSON mode: structured output for AI agents — no terminal escapes, no noise
     if (jsonMode) {
         if (imageB64) {
-            console.log(JSON.stringify({
-                event: "qr",
-                image: imageB64,
-                file: QR_PATH,
-                dataUrl: `data:image/png;base64,${imageB64}`,
-            }));
+            console.log(
+                JSON.stringify({
+                    event: "qr",
+                    image: imageB64,
+                    file: QR_PATH,
+                    dataUrl: `data:image/png;base64,${imageB64}`,
+                }),
+            );
         }
         return;
     }

package/src/utils/qr-display.test.js

@@ -9,8 +9,7 @@ import { existsSync, unlinkSync, mkdirSync } from "fs";
 import { dirname } from "path";
 
 // Tiny valid 1x1 white PNG as base64 (for testing without real QR)
-const TINY_PNG_B64 =
-    "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8/5+hHgAHggJ/PchI7wAAAABJRU5ErkJggg==";
+const TINY_PNG_B64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8/5+hHgAHggJ/PchI7wAAAABJRU5ErkJggg==";
 
 describe("displayQR", () => {
     let originalLog;

package/src/utils/qr-http-server.js

@@ -99,6 +99,20 @@ setInterval(async()=>{
 },2000);
 </script>
 </body></html>`);
+        } else if (req.url === "/qr.png") {
+            // Raw PNG image endpoint — for agents to send direct image link
+            if (!existsSync(qrImagePath)) {
+                res.writeHead(404, { "Content-Type": "text/plain" });
+                res.end("QR not generated yet");
+                return;
+            }
+            const img = readFileSync(qrImagePath);
+            res.writeHead(200, {
+                "Content-Type": "image/png",
+                "Content-Length": img.length,
+                "Cache-Control": "no-cache, no-store",
+            });
+            res.end(img);
         } else if (req.url === "/status") {
             // Login status endpoint for browser polling
             let loggedIn = false;
@@ -134,12 +148,16 @@ setInterval(async()=>{
         } catch {}
 
         if (jsonMode) {
-            console.log(JSON.stringify({
-                event: "qr_server",
-                port: actualPort,
-                localUrl: `http://localhost:${actualPort}/qr`,
-                publicUrl: publicIp ? `http://${publicIp}:${actualPort}/qr` : null,
-            }));
+            console.log(
+                JSON.stringify({
+                    event: "qr_server",
+                    port: actualPort,
+                    localUrl: `http://localhost:${actualPort}/qr`,
+                    imageUrl: `http://localhost:${actualPort}/qr.png`,
+                    publicUrl: publicIp ? `http://${publicIp}:${actualPort}/qr` : null,
+                    publicImageUrl: publicIp ? `http://${publicIp}:${actualPort}/qr.png` : null,
+                }),
+            );
         } else {
             info(`QR available at: http://localhost:${actualPort}/qr`);
             if (publicIp) info(`On VPS, open: http://${publicIp}:${actualPort}/qr`);