From 8f4f605094dab97b76649675e40a9ceb9f6cbdb8 Mon Sep 17 00:00:00 2001
From: ty <786220806@qq.com>
Date: Sun, 22 Mar 2026 14:47:30 +0800
Subject: [PATCH] feat(cli): add wait command with client-side polling

---
 packages/cli/src/commands/wait.ts | 62 +++++++++++++++++++++++++++++++
 packages/cli/src/index.ts         |  2 +
 2 files changed, 64 insertions(+)
 create mode 100644 packages/cli/src/commands/wait.ts

diff --git a/packages/cli/src/commands/wait.ts b/packages/cli/src/commands/wait.ts
new file mode 100644
index 0000000..db5d050
--- /dev/null
+++ b/packages/cli/src/commands/wait.ts
@@ -0,0 +1,62 @@
+import { Command } from "commander";
+import { api } from "../api.js";
+import { log, printJson, printText, msToIso } from "../output.js";
+
+function sleep(ms: number): Promise<void> {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+export function registerWaitCommand(program: Command) {
+  program
+    .command("wait")
+    .description("Wait for a new email to arrive")
+    .requiredOption("--email-id <id>", "email ID to watch")
+    .option("--timeout <seconds>", "max wait time in seconds", "120")
+    .option("--interval <seconds>", "poll interval in seconds", "5")
+    .action(async (opts) => {
+      const json = program.opts().json;
+      const timeout = parseInt(opts.timeout, 10);
+      const interval = parseInt(opts.interval, 10);
+      const emailId = opts.emailId;
+
+      try {
+        const initial = (await api.listMessages(emailId)) as any;
+        const knownIds = new Set<string>(initial.messages.map((m: any) => m.id));
+
+        const startTime = Date.now();
+
+        while (true) {
+          const elapsed = Math.floor((Date.now() - startTime) / 1000);
+          if (elapsed >= timeout) {
+            log(`Timeout: no new messages received within ${timeout}s`);
+            process.exit(1);
+          }
+
+          log(`Polling... (${elapsed}/${timeout}s)`);
+          await sleep(interval * 1000);
+
+          const current = (await api.listMessages(emailId)) as any;
+          const newMessages = current.messages.filter((m: any) => !knownIds.has(m.id));
+
+          if (newMessages.length > 0) {
+            const msg = newMessages[0];
+            if (json) {
+              printJson({
+                messageId: msg.id,
+                from: msg.from_address,
+                subject: msg.subject,
+                receivedAt: msg.received_at ? msToIso(msg.received_at) : null,
+              });
+            } else {
+              printText(`New message from ${msg.from_address}: "${msg.subject}"`);
+              printText(`Message ID: ${msg.id}`);
+            }
+            return;
+          }
+        }
+      } catch (e: any) {
+        log(`Error: ${e.message}`);
+        process.exit(1);
+      }
+    });
+}
diff --git a/packages/cli/src/index.ts b/packages/cli/src/index.ts
index e9e1c35..bc73837 100644
--- a/packages/cli/src/index.ts
+++ b/packages/cli/src/index.ts
@@ -3,6 +3,7 @@ import { Command } from "commander";
 import { registerConfigCommand } from "./commands/config.js";
 import { registerCreateCommand } from "./commands/create.js";
 import { registerListCommand } from "./commands/list.js";
+import { registerWaitCommand } from "./commands/wait.js";
 
 const program = new Command();
 
@@ -15,5 +16,6 @@ program
 registerConfigCommand(program);
 registerCreateCommand(program);
 registerListCommand(program);
+registerWaitCommand(program);
 
 program.parse();