@zaclinley/dsh-mailharbor 0.12.1 → 0.12.2

package/lib/index.js

@@ -1101,6 +1101,23 @@ var MailDatabase = class {
       where.push("e.date >= ?");
       params.push(query.since);
     }
+    if (query.unread !== void 0) {
+      const mids = query.unread.mids.slice(0, 500);
+      const ids = query.unread.ids.slice(0, 500);
+      if (mids.length === 0 && ids.length === 0) {
+        return { items: [], total: 0, offset, limit };
+      }
+      const parts = [];
+      if (mids.length > 0) {
+        parts.push(`e.message_id IN (${mids.map(() => "?").join(",")})`);
+        params.push(...mids);
+      }
+      if (ids.length > 0) {
+        parts.push(`e.id IN (${ids.map(() => "?").join(",")})`);
+        params.push(...ids);
+      }
+      where.push(`(${parts.join(" OR ")})`);
+    }
     const text2 = String(query.query ?? "").trim();
     if (text2 !== "") {
       const tokens = text2.split(/\s+/).filter(Boolean);
@@ -1187,6 +1204,28 @@ var MailDatabase = class {
     const rows = db.prepare(`${columns} FROM emails e ${clause} ORDER BY e.date DESC, e.id DESC LIMIT ?`).all(...params, limit);
     return { items: rows.map((row) => rowToSummary(row)), total, offset: 0, limit };
   }
+  /** Number of unread keys that still exist in this backup. */
+  countUnread(keys) {
+    const mids = keys.mids.slice(0, 500);
+    const ids = keys.ids.slice(0, 500);
+    if (mids.length === 0 && ids.length === 0) return 0;
+    const parts = [];
+    const params = [];
+    if (mids.length > 0) {
+      parts.push(`message_id IN (${mids.map(() => "?").join(",")})`);
+      params.push(...mids);
+    }
+    if (ids.length > 0) {
+      parts.push(`id IN (${ids.map(() => "?").join(",")})`);
+      params.push(...ids);
+    }
+    try {
+      const row = this.open().prepare(`SELECT COUNT(*) AS count FROM emails WHERE ${parts.join(" OR ")}`).get(...params);
+      return Number(row?.count ?? 0);
+    } catch {
+      return 0;
+    }
+  }
   get(localId) {
     const db = this.open();
     const row = db.prepare(
@@ -1366,10 +1405,57 @@ var BackupEngine = class {
   runtime;
   credentials;
   batchSize;
+  onFlags;
   constructor(runtime, credentials, options = {}) {
     this.runtime = runtime;
     this.credentials = credentials;
     this.batchSize = options.batchSize ?? BATCH;
+    this.onFlags = options.onFlags;
+  }
+  /** Mirror IMAP \Seen flags into the local unread store. */
+  async reconcileReadState(client) {
+    if (this.onFlags === void 0) return;
+    const exists = Number(client.mailbox?.exists ?? 0);
+    if (exists === 0) return;
+    const entries = /* @__PURE__ */ new Map();
+    const add = (messageId, unread) => {
+      if (messageId !== "") entries.set(`mid:${messageId}`, unread);
+    };
+    const flagsOf2 = (flags) => [...flags ?? []].map((flag) => String(flag).toLowerCase());
+    const start = Math.max(1, exists - 299);
+    try {
+      for await (const message of client.fetch(`${String(start)}:${String(exists)}`, {
+        uid: true,
+        envelope: true,
+        flags: true
+      })) {
+        const messageId = normalizeId(message.envelope?.messageId);
+        add(messageId, !flagsOf2(message.flags).includes("\\seen"));
+      }
+    } catch {
+    }
+    try {
+      const unseen = await client.search({ seen: false }, { uid: true });
+      const uids = (Array.isArray(unseen) ? unseen : []).map(Number).filter((uid) => Number.isFinite(uid));
+      for (let index = 0; index < uids.length; index += 200) {
+        const range = uids.slice(index, index + 200).join(",");
+        if (range === "") continue;
+        for await (const message of client.fetch(
+          range,
+          { uid: true, envelope: true, flags: true },
+          { uid: true }
+        )) {
+          const messageId = normalizeId(
+            message.envelope?.messageId
+          );
+          add(messageId, true);
+        }
+      }
+    } catch {
+    }
+    if (entries.size > 0) {
+      this.onFlags([...entries.entries()].map(([key, unread]) => ({ key, unread })));
+    }
   }
   attachmentsDir() {
     return this.runtime.attachments;
@@ -1518,6 +1604,7 @@ var BackupEngine = class {
             abortIfNeeded(signal);
             const batch = uids.slice(offset, offset + this.batchSize);
             db.exec("BEGIN");
+            const flagEntries = [];
             try {
               let batchLastUid = committedUid;
               for await (const message of client.fetch(
@@ -1538,6 +1625,13 @@ var BackupEngine = class {
                   boxAdded += added;
                   newCount += added;
                   batchLastUid = Math.max(batchLastUid, Number(message.uid));
+                  const messageId = normalizeId(parsed.messageId ?? "");
+                  if (messageId !== "") {
+                    const flags = [...message.flags ?? []].map(
+                      (flag) => String(flag).toLowerCase()
+                    );
+                    flagEntries.push({ key: `mid:${messageId}`, unread: !flags.includes("\\seen") });
+                  }
                 } catch (error) {
                   failCount += 1;
                   emit({
@@ -1548,6 +1642,7 @@ var BackupEngine = class {
               }
               this.saveState(db, path, uidValidity, batchLastUid);
               db.exec("COMMIT");
+              if (flagEntries.length > 0) this.onFlags?.(flagEntries);
               committedUid = batchLastUid;
               emit({ mailbox: path, added: boxAdded, message: `${path}: \u5DF2\u65B0\u589E ${String(boxAdded)} \u5C01` });
             } catch (error) {
@@ -1555,6 +1650,7 @@ var BackupEngine = class {
               throw error;
             }
           }
+          if (path.toUpperCase() === "INBOX") await this.reconcileReadState(client);
           mailboxes.push({ name: path, added: boxAdded });
           emit({ mailbox: path, added: boxAdded, message: `${path}: \u65B0\u589E ${String(boxAdded)} \u5C01` });
         } finally {
@@ -2494,7 +2590,7 @@ ${conversation.trim()}
 }
 
 // src/host/version.ts
-var VERSION = "0.12.1";
+var VERSION = "0.12.2";
 
 // src/host/api.ts
 init_smtp();
@@ -2597,7 +2693,14 @@ function buildRouter(runtime) {
       const items = await listServer(runtime, account, mailbox, query, limit, offset);
       return { ...items, offset, limit };
     }
-    const page = runtime.database(account).list({ mailbox: mailbox === "" ? "all" : mailbox, query, offset, limit });
+    const unreadOnly = url.searchParams.get("unread") === "1";
+    const page = runtime.database(account).list({
+      mailbox: mailbox === "" ? "all" : mailbox,
+      query,
+      offset,
+      limit,
+      ...unreadOnly ? { unread: unreadKeys(runtime, account) } : {}
+    });
     return { ...page, items: page.items.map((item) => localSummary(runtime, account, item)) };
   });
   router.get("/messages/lookup", ({ url }) => {
@@ -2693,21 +2796,13 @@ function buildRouter(runtime) {
   router.get("/unread", async ({ url }) => {
     const requested = url.searchParams.get("accountId");
     const accounts = requested === null || requested === "" || requested === "all" ? runtime.current.accounts.filter((item) => item.enabled !== false) : [accountById(runtime, requested)];
-    let server = 0;
-    let answered = 0;
-    await Promise.all(
-      accounts.map(async (account) => {
-        if (!runtime.hasCredentials(account)) return;
-        try {
-          const result = await runtime.provider(account).unseen();
-          server += result.count;
-          answered += 1;
-        } catch {
-        }
-      })
-    );
-    if (answered > 0) return { unread: server, source: "server" };
-    const local = accounts.reduce((sum, account) => sum + runtime.readState(account).count(), 0);
+    const local = accounts.reduce((sum, account) => {
+      try {
+        return sum + runtime.database(account).countUnread(unreadKeys(runtime, account));
+      } catch {
+        return sum;
+      }
+    }, 0);
     return { unread: local, source: "local" };
   });
   router.get("/accounts", () => ({ accounts: runtime.current.accounts }));
@@ -3400,6 +3495,18 @@ function projectView(project) {
     updatedAt: project.updatedAt
   };
 }
+function unreadKeys(runtime, account) {
+  const mids = [];
+  const ids = [];
+  for (const key of runtime.readState(account).list()) {
+    if (key.startsWith("mid:")) mids.push(key.slice(4));
+    else if (key.startsWith("local:")) {
+      const id = Number(key.slice(6));
+      if (Number.isFinite(id)) ids.push(id);
+    }
+  }
+  return { mids, ids };
+}
 function localSummary(runtime, account, row) {
   const store = runtime.readState(account);
   const key = row.messageId !== void 0 && row.messageId !== "" ? `mid:${row.messageId}` : `local:${row.id}`;
@@ -4542,7 +4649,11 @@ var MailHarborRuntime = class {
       exitCode: null
     };
     const job = { status, controller, promise: Promise.resolve() };
-    const engine = new BackupEngine(accountRuntime(account), this.credentialsFor(account));
+    const engine = new BackupEngine(accountRuntime(account), this.credentialsFor(account), {
+      onFlags: (entries) => {
+        this.readState(account).sync(entries);
+      }
+    });
     job.promise = (async () => {
       try {
         const result = await engine.run({