@marlinjai/email-mcp 1.2.7 → 1.3.0

package/dist/setup/wizard.js

@@ -363,6 +363,7 @@ var FolderType = {
 };
 
 // src/providers/gmail/adapter.ts
+import { Readable } from "node:stream";
 import { google } from "googleapis";
 import { OAuth2Client as OAuth2Client2 } from "google-auth-library";
 
@@ -567,20 +568,32 @@ var GmailAdapter = class {
   async search(query) {
     const gmail = this.ensureConnected();
     const q = buildGmailQuery(query);
-    const params = {
-      userId: "me",
-      maxResults: query.limit || 20
-    };
-    if (q) params.q = q;
-    if (query.folder) params.labelIds = [query.folder];
-    const res = await gmail.users.messages.list(params);
-    const messages = res.data.messages || [];
-    if (messages.length === 0) return [];
+    const limit = query.limit || 20;
+    const offset = query.offset || 0;
+    const needed = offset + limit;
+    const ids = [];
+    let pageToken;
+    do {
+      const params = {
+        userId: "me",
+        maxResults: Math.min(500, needed - ids.length)
+      };
+      if (q) params.q = q;
+      if (query.folder) params.labelIds = [query.folder];
+      if (pageToken) params.pageToken = pageToken;
+      const res = await gmail.users.messages.list(params);
+      for (const msg of res.data.messages || []) {
+        if (msg.id) ids.push(msg.id);
+      }
+      pageToken = res.data.nextPageToken || void 0;
+    } while (pageToken && ids.length < needed);
+    const windowIds = ids.slice(offset, offset + limit);
+    if (windowIds.length === 0) return [];
     const emails = [];
-    for (const msg of messages) {
+    for (const id of windowIds) {
       const full = await gmail.users.messages.get({
         userId: "me",
-        id: msg.id,
+        id,
         format: "full"
       });
       emails.push(mapGmailMessage(full.data, this.accountId));
@@ -828,6 +841,40 @@ var GmailAdapter = class {
     }
     return result;
   }
+  async getRawMessage(emailId, _sourceFolder) {
+    const gmail = this.ensureConnected();
+    const res = await gmail.users.messages.get({
+      userId: "me",
+      id: emailId,
+      format: "raw"
+    });
+    return Buffer.from(res.data.raw || "", "base64url");
+  }
+  async appendRawMessage(raw, targetFolder, flags) {
+    const gmail = this.ensureConnected();
+    const labelIds = [];
+    labelIds.push(targetFolder ? await this.resolveLabelId(targetFolder) : "INBOX");
+    if (flags?.read === false) labelIds.push("UNREAD");
+    if (flags?.starred === true) labelIds.push("STARRED");
+    const res = await gmail.users.messages.insert({
+      userId: "me",
+      internalDateSource: "dateHeader",
+      requestBody: { labelIds },
+      media: { mimeType: "message/rfc822", body: Readable.from(raw) }
+    });
+    return { id: res.data.id || "" };
+  }
+  /** Resolve a Gmail label name or id to a label id (accepts either). */
+  async resolveLabelId(nameOrId) {
+    const gmail = this.ensureConnected();
+    const res = await gmail.users.labels.list({ userId: "me" });
+    const labels = res.data.labels || [];
+    const byId = labels.find((l) => l.id === nameOrId);
+    if (byId?.id) return byId.id;
+    const byName = labels.find((l) => (l.name || "").toLowerCase() === nameOrId.toLowerCase());
+    if (byName?.id) return byName.id;
+    return nameOrId;
+  }
   async addLabels(emailId, labels) {
     const gmail = this.ensureConnected();
     await gmail.users.messages.modify({
@@ -1209,6 +1256,41 @@ var OutlookAdapter = class {
       destinationId
     });
   }
+  async getRawMessage(emailId, _sourceFolder) {
+    this.ensureClient();
+    const res = await fetch(
+      `https://graph.microsoft.com/v1.0/me/messages/${encodeURIComponent(emailId)}/$value`,
+      { headers: { Authorization: `Bearer ${this.accessToken}` } }
+    );
+    if (!res.ok) {
+      throw new Error(`Graph $value fetch failed for ${emailId}: HTTP ${res.status}`);
+    }
+    return Buffer.from(await res.arrayBuffer());
+  }
+  async appendRawMessage(raw, targetFolder, flags) {
+    this.ensureClient();
+    const endpoint = targetFolder ? `https://graph.microsoft.com/v1.0/me/mailFolders/${encodeURIComponent(await this.resolveFolder(targetFolder))}/messages` : "https://graph.microsoft.com/v1.0/me/messages";
+    const res = await fetch(endpoint, {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${this.accessToken}`,
+        "Content-Type": "text/plain"
+      },
+      body: raw.toString("base64")
+    });
+    if (!res.ok) {
+      const detail = await res.text().catch(() => "");
+      throw new Error(`Graph MIME import failed: HTTP ${res.status} ${detail.slice(0, 200)}`);
+    }
+    const data = await res.json().catch(() => ({}));
+    if (flags && data.id && (flags.read !== void 0 || flags.starred !== void 0)) {
+      try {
+        await this.markEmail(data.id, flags);
+      } catch {
+      }
+    }
+    return { id: data.id || "" };
+  }
   async deleteEmail(emailId, permanent, _sourceFolder) {
     const client = this.ensureClient();
     if (permanent) {
@@ -1623,14 +1705,7 @@ var ImapAdapter = class {
     }
     try {
       const mailboxExists = this.client.mailbox?.exists ?? 0;
-      let statusCount = 0;
-      try {
-        const status = await this.client.status(folder, { messages: true });
-        statusCount = status.messages ?? 0;
-      } catch {
-      }
-      const effectiveCount = Math.max(mailboxExists, statusCount);
-      if (effectiveCount === 0) {
+      if (mailboxExists === 0) {
         return [];
       }
       const criteria = this.buildSearchCriteria(query);
@@ -1643,7 +1718,11 @@ var ImapAdapter = class {
         );
         allUids = Array.isArray(searchResult) ? searchResult : [];
       } catch {
-        allUids = await this.collectUidsViaFetch(query);
+        try {
+          allUids = await this.collectUidsViaFetch(query);
+        } catch {
+          return [];
+        }
       }
       const offset = query.offset || 0;
       const slicedUids = query.limit ? allUids.slice(offset, offset + query.limit) : allUids.slice(offset);
@@ -1705,19 +1784,27 @@ var ImapAdapter = class {
   async fetchEmails(uids, folder, returnBody) {
     if (!this.client) return [];
     const emails = [];
-    if (returnBody) {
-      for await (const msg of this.client.fetch(uids, { source: true, uid: true, flags: true })) {
+    const fetchOpts = returnBody ? { source: true, uid: true, flags: true } : { envelope: true, uid: true, flags: true, bodyStructure: true };
+    const uidRange = uids.join(",");
+    let messages;
+    try {
+      messages = await this.client.fetchAll(uidRange, fetchOpts, { uid: true });
+    } catch {
+      messages = [];
+      for (const uid of uids) {
+        try {
+          const batch = await this.client.fetchAll(String(uid), fetchOpts, { uid: true });
+          messages.push(...batch);
+        } catch {
+        }
+      }
+    }
+    for (const msg of messages) {
+      if (returnBody) {
         const parsed = await simpleParser(msg.source);
         parsed.flags = msg.flags;
         emails.push(mapParsedEmail(parsed, folder, this.accountId, msg.uid));
-      }
-    } else {
-      for await (const msg of this.client.fetch(uids, {
-        envelope: true,
-        uid: true,
-        flags: true,
-        bodyStructure: true
-      })) {
+      } else {
         const env = msg.envelope;
         emails.push({
           id: String(msg.uid),
@@ -1926,6 +2013,32 @@ var ImapAdapter = class {
       lock.release();
     }
   }
+  async getRawMessage(emailId, sourceFolder) {
+    if (!this.client) throw new Error("Not connected");
+    const folder = sourceFolder ? await this.resolveFolder(sourceFolder) : "INBOX";
+    let lock;
+    try {
+      lock = await this.client.getMailboxLock(folder);
+    } catch (error) {
+      throw formatImapError(error, `Failed to open folder "${folder}"`);
+    }
+    try {
+      const msg = await this.client.fetchOne(String(emailId), { source: true, uid: true }, { uid: true });
+      if (!msg || !msg.source) throw new Error(`Email ${emailId} not found in "${folder}"`);
+      return msg.source;
+    } finally {
+      lock.release();
+    }
+  }
+  async appendRawMessage(raw, targetFolder, flags) {
+    if (!this.client) throw new Error("Not connected");
+    const folder = targetFolder ? await this.resolveFolder(targetFolder) : "INBOX";
+    const imapFlags = [];
+    if (flags?.read) imapFlags.push("\\Seen");
+    if (flags?.starred) imapFlags.push("\\Flagged");
+    const result = await this.client.append(folder, raw, imapFlags);
+    return { id: String(result?.uid ?? "") };
+  }
   async batchDelete(emailIds, permanent, sourceFolder) {
     if (!this.client) throw new Error("Not connected");
     const result = { succeeded: [], failed: [] };