lansenger-cli 1.0.1 → 1.0.2

package/dist/commands/chat.js

@@ -2,6 +2,25 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerChatCommands = registerChatCommands;
 const utils_1 = require("../utils");
+function splitMonths(startUs, endUs) {
+    const results = [];
+    const startDt = startUs ? new Date(startUs / 1000000) : new Date(2020, 0, 1);
+    const endDt = new Date(endUs / 1000000);
+    let year = startDt.getFullYear();
+    let month = startDt.getMonth();
+    while (new Date(year, month, 1) <= endDt) {
+        const lastDay = new Date(year, month + 1, 0).getDate();
+        const msStart = Math.floor(new Date(year, month, 1).getTime() * 1000);
+        const msEnd = Math.floor(new Date(year, month, lastDay, 23, 59, 59).getTime() * 1000);
+        results.push([msStart, msEnd]);
+        month++;
+        if (month > 11) {
+            month = 0;
+            year++;
+        }
+    }
+    return results;
+}
 function registerChatCommands(program) {
     const cmd = program.command("chat").description("Chat list and message history");
     cmd
@@ -22,7 +41,22 @@ function registerChatCommands(program) {
             user_token: opts.userToken || undefined,
         });
         (0, utils_1.checkError)(result);
-        (0, utils_1.outputResult)(result);
+        if (result.success) {
+            (0, utils_1.outputResult)(result);
+            if (result.staff_infos) {
+                (0, utils_1.outputList)(result.staff_infos, ["Staff ID", "Name", "Sectors"], (s) => [
+                    s.staff_id || "", s.staff_name || "", String(s.sector_names || ""),
+                ]);
+            }
+            if (result.group_infos) {
+                (0, utils_1.outputList)(result.group_infos, ["Group ID", "Name"], (g) => [
+                    g.group_id || "", g.group_name || "",
+                ]);
+            }
+        }
+        else {
+            (0, utils_1.outputResult)(result);
+        }
     });
     cmd
         .command("messages")
@@ -34,20 +68,83 @@ function registerChatCommands(program) {
         .option("--start <start>", "Start time in microseconds", "0")
         .option("--end <end>", "End time in microseconds", "0")
         .option("--sender-id <senderId>", "Filter by sender staffId", "")
+        .option("--split-month", "Auto-split query by month when range > 1 month", false)
+        .option("--progress", "Show pagination progress (pages/messages fetched)", false)
         .option("--user-token <token>", "User token", "")
         .action(async (opts) => {
         const client = (0, utils_1.getClient)();
-        const result = await client.fetchChatMessages({
-            staff_id: opts.staffId || undefined,
-            group_id: opts.groupId || undefined,
-            page_size: parseInt(opts.size),
-            base_version: opts.version,
-            start_time: parseInt(opts.start),
-            end_time: parseInt(opts.end),
-            sender_id: opts.senderId || undefined,
-            user_token: opts.userToken || undefined,
-        });
-        (0, utils_1.checkError)(result);
-        (0, utils_1.outputResult)(result);
+        if (!opts.splitMonth) {
+            const result = await client.fetchChatMessages({
+                staff_id: opts.staffId || undefined,
+                group_id: opts.groupId || undefined,
+                page_size: parseInt(opts.size),
+                base_version: opts.version,
+                start_time: parseInt(opts.start),
+                end_time: parseInt(opts.end),
+                sender_id: opts.senderId || undefined,
+                user_token: opts.userToken || undefined,
+            });
+            (0, utils_1.checkError)(result);
+            if (result.success) {
+                (0, utils_1.outputResult)(result);
+                if (result.messages) {
+                    (0, utils_1.outputList)(result.messages, ["Time", "Sender", "Type"], (m) => [
+                        m.send_time || "", m.sender || "", m.message_type || "",
+                    ]);
+                }
+            }
+            else {
+                (0, utils_1.outputResult)(result);
+            }
+            return;
+        }
+        const allMessages = [];
+        let pageCount = 0;
+        let msgCount = 0;
+        const startUs = parseInt(opts.start) || 0;
+        const endUs = parseInt(opts.end) || Math.floor(Date.now() * 1000);
+        const months = splitMonths(startUs, endUs);
+        for (let i = 0; i < months.length; i++) {
+            const [msStart, msEnd] = months[i];
+            let cursor = "0";
+            while (true) {
+                const result = await client.fetchChatMessages({
+                    staff_id: opts.staffId || undefined,
+                    group_id: opts.groupId || undefined,
+                    page_size: parseInt(opts.size),
+                    base_version: cursor,
+                    start_time: msStart,
+                    end_time: msEnd,
+                    sender_id: opts.senderId || undefined,
+                    user_token: opts.userToken || undefined,
+                });
+                if (!result.success) {
+                    (0, utils_1.outputResult)(result);
+                    return;
+                }
+                pageCount++;
+                msgCount += result.messages ? result.messages.length : 0;
+                if (result.messages)
+                    allMessages.push(...result.messages);
+                if (opts.progress && !utils_1.jsonOutput) {
+                    console.log(`Month ${i + 1}/${months.length} | Page ${pageCount} | ${msgCount} messages total`);
+                }
+                if (!(result.messages && result.has_more))
+                    break;
+                cursor = String(result.last_version || "");
+            }
+        }
+        if (utils_1.jsonOutput) {
+            console.log(JSON.stringify(allMessages.map(m => m.toDict ? m.toDict() : m), null, 2));
+            return;
+        }
+        if (opts.progress) {
+            console.log(`Done: ${pageCount} pages, ${msgCount} messages across ${months.length} months`);
+        }
+        if (allMessages.length) {
+            (0, utils_1.outputList)(allMessages, ["Time", "Sender", "Type"], (m) => [
+                m.send_time || "", m.sender || "", m.message_type || "",
+            ]);
+        }
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lansenger-cli",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "CLI for Lansenger (蓝信) — send messages, manage groups, staff, departments, calendars, todos, and more",
   "main": "dist/main.js",
   "bin": {