opencode-mailbox 0.0.6 → 0.0.8

package/dist/index.js

@@ -12682,6 +12682,21 @@ import { Database } from "bun:sqlite";
 var dbFile = null;
 var db = null;
 var activeWatches = new Map;
+var watchesBySession = new Map;
+function resetDatabaseConnection() {
+  if (db) {
+    try {
+      db.close();
+    } catch {}
+    db = null;
+  }
+}
+function isDatabaseFileError(error45) {
+  if (!(error45 instanceof Error))
+    return false;
+  const message = error45.message.toLowerCase();
+  return message.includes("database disk image is malformed") || message.includes("no such table") || message.includes("unable to open database file") || message.includes("database is locked") || message.includes("disk i/o error") || message.includes("no more rows available") || message.includes("unable to close") || message.includes("bad parameter or other api misuse");
+}
 async function getDbFile(client) {
   if (!dbFile) {
     const result = await client.path.get();
@@ -12741,7 +12756,12 @@ async function markMessageAsRead(client, recipient, timestamp) {
   stmt.run(recipient.toLowerCase(), timestamp);
 }
 function startMailWatch(client, recipient, sessionId, instructions) {
-  if (activeWatches.has(recipient)) {
+  const existingWatch = activeWatches.get(recipient);
+  if (existingWatch) {
+    existingWatch.refCount++;
+    const sessionWatches2 = watchesBySession.get(sessionId) ?? new Set;
+    sessionWatches2.add(recipient);
+    watchesBySession.set(sessionId, sessionWatches2);
     return;
   }
   const interval = setInterval(async () => {
@@ -12759,10 +12779,18 @@ function startMailWatch(client, recipient, sessionId, instructions) {
         await injectMailMessage(client, sessionId, recipient, message, instructions);
       }
     } catch (error45) {
+      if (isDatabaseFileError(error45)) {
+        console.error(`[Mailbox] Database file error for ${recipient}, resetting connection...`);
+        resetDatabaseConnection();
+        return;
+      }
       console.error(`[Mailbox] Error watching mail for ${recipient}:`, error45);
     }
   }, 5000);
-  activeWatches.set(recipient, { interval, instructions });
+  activeWatches.set(recipient, { interval, instructions, refCount: 1 });
+  const sessionWatches = watchesBySession.get(sessionId) ?? new Set;
+  sessionWatches.add(recipient);
+  watchesBySession.set(sessionId, sessionWatches);
 }
 function stopMailWatch(recipient) {
   const watch = activeWatches.get(recipient);
@@ -12799,7 +12827,12 @@ ${message.message}
         await client.session.prompt({
           path: { id: sessionId },
           body: {
-            parts: [{ type: "text", text: "You have new mail. Please review the injected message above and respond accordingly." }]
+            parts: [
+              {
+                type: "text",
+                text: "You have new mail. Please review the injected message above and respond accordingly."
+              }
+            ]
           }
         });
       }
@@ -12825,7 +12858,17 @@ var mailboxPlugin = async (ctx) => {
       const to = args.to.toLowerCase();
       const from = args.from.toLowerCase();
       const timestamp = Date.now();
-      await addMessage(client, to, from, args.message, timestamp);
+      try {
+        await addMessage(client, to, from, args.message, timestamp);
+      } catch (error45) {
+        if (isDatabaseFileError(error45)) {
+          console.error(`[Mailbox] Database file error while sending mail, resetting connection...`);
+          resetDatabaseConnection();
+          await addMessage(client, to, from, args.message, timestamp);
+        } else {
+          throw error45;
+        }
+      }
       return `Mail sent to "${args.to}" from "${args.from}" at ${new Date(timestamp).toISOString()}`;
     }
   });
@@ -12843,16 +12886,27 @@ var mailboxPlugin = async (ctx) => {
     }
   });
   const stopWatchingMailTool = tool3({
-    description: "Stop all mail watching",
+    description: "Stop all mail watching for this session",
     args: {},
-    async execute() {
+    async execute(_args, toolCtx) {
+      const sessionId = toolCtx.sessionID;
       const stoppedWatches = [];
-      for (const recipient of activeWatches.keys()) {
-        stopMailWatch(recipient);
-        stoppedWatches.push(recipient);
+      const sessionWatches = watchesBySession.get(sessionId);
+      if (sessionWatches) {
+        for (const recipient of sessionWatches) {
+          const watch = activeWatches.get(recipient);
+          if (watch) {
+            watch.refCount--;
+            if (watch.refCount <= 0) {
+              stopMailWatch(recipient);
+            }
+            stoppedWatches.push(recipient);
+          }
+        }
+        watchesBySession.delete(sessionId);
       }
       if (stoppedWatches.length === 0) {
-        return "No active mail watches found.";
+        return "No active mail watches found for this session.";
       }
       return `Stopped watching mail for: ${stoppedWatches.join(", ")}`;
     }
@@ -12869,10 +12923,21 @@ var mailboxPlugin = async (ctx) => {
       input.experimental.primary_tools.push("send_mail", "watch_unread_mail", "stop_watching_mail");
     },
     hooks: {
-      "session.end": async () => {
-        for (const recipient of activeWatches.keys()) {
-          stopMailWatch(recipient);
+      "session.end": async (input) => {
+        const sessionId = input.sessionID;
+        const sessionWatches = watchesBySession.get(sessionId);
+        if (!sessionWatches)
+          return;
+        for (const recipient of sessionWatches) {
+          const watch = activeWatches.get(recipient);
+          if (watch) {
+            watch.refCount--;
+            if (watch.refCount <= 0) {
+              stopMailWatch(recipient);
+            }
+          }
         }
+        watchesBySession.delete(sessionId);
       }
     }
   };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-mailbox",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "A simple mailbox system for sending and receiving messages between sessions",
   "type": "module",
   "main": "./dist/index.js",