@ouro.bot/cli 0.1.0-alpha.730 → 0.1.0-alpha.732

package/dist/mailroom/search-cache.js

@@ -36,6 +36,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.MAIL_SEARCH_TEXT_PROJECTION_VERSION = void 0;
 exports.buildMailSearchCacheDocument = buildMailSearchCacheDocument;
 exports.upsertMailSearchCacheDocument = upsertMailSearchCacheDocument;
+exports.inspectMailSearchCacheFiles = inspectMailSearchCacheFiles;
+exports.reloadMailSearchCache = reloadMailSearchCache;
+exports.removeMailSearchCacheDocument = removeMailSearchCacheDocument;
+exports.removeMailSearchCacheFile = removeMailSearchCacheFile;
 exports.syncMailSearchCacheMetadata = syncMailSearchCacheMetadata;
 exports.searchMailSearchCache = searchMailSearchCache;
 exports.snapshotMailSearchCacheForFilters = snapshotMailSearchCacheForFilters;
@@ -96,6 +100,17 @@ function readJsonDocument(filePath) {
         return null;
     }
 }
+function writeJsonAtomically(filePath, value) {
+    const tempPath = `${filePath}.${process.pid}.${Date.now()}.tmp`;
+    try {
+        fs.writeFileSync(tempPath, `${JSON.stringify(value)}\n`, "utf-8");
+        fs.renameSync(tempPath, filePath);
+    }
+    finally {
+        if (fs.existsSync(tempPath))
+            fs.unlinkSync(tempPath);
+    }
+}
 function cacheState(agentId, options) {
     const key = `${agentId}:${cacheDir(agentId, options)}`;
     let state = cacheStates.get(key);
@@ -123,8 +138,10 @@ function loadCache(agentId, options) {
     }
     return state.docs;
 }
-function buildMailSearchCacheDocument(message, privateEnvelope) {
+function buildMailSearchCacheDocument(message, privateEnvelope, provenance = {}) {
     const readableText = (0, core_1.privateMailEnvelopeReadableText)(privateEnvelope);
+    const decryptionKeyId = provenance.decryptionKeyId
+        ?? (message.bodyForm === "encrypted" ? message.privateEnvelope?.keyId : undefined);
     return {
         schemaVersion: 1,
         messageId: message.id,
@@ -142,13 +159,14 @@ function buildMailSearchCacheDocument(message, privateEnvelope) {
         searchText: normalizeSearchText(privateEnvelope),
         textProjectionVersion: exports.MAIL_SEARCH_TEXT_PROJECTION_VERSION,
         attachmentCount: privateEnvelope.attachments.length,
+        ...(decryptionKeyId ? { decryptionKeyId } : {}),
     };
 }
-function upsertMailSearchCacheDocument(message, privateEnvelope, options) {
-    const document = buildMailSearchCacheDocument(message, privateEnvelope);
+function upsertMailSearchCacheDocument(message, privateEnvelope, options, provenance) {
+    const document = buildMailSearchCacheDocument(message, privateEnvelope, provenance);
     const dir = cacheDir(message.agentId, options);
     fs.mkdirSync(dir, { recursive: true });
-    fs.writeFileSync(cachePath(message.agentId, message.id, options), `${JSON.stringify(document)}\n`, "utf-8");
+    writeJsonAtomically(cachePath(message.agentId, message.id, options), document);
     const state = cacheState(message.agentId, options);
     if (state.loaded)
         state.docs.set(document.messageId, document);
@@ -165,6 +183,58 @@ function upsertMailSearchCacheDocument(message, privateEnvelope, options) {
     });
     return document;
 }
+function inspectMailSearchCacheFiles(agentId, options) {
+    const dir = cacheDir(agentId, options);
+    if (!fs.existsSync(dir))
+        return [];
+    return fs.readdirSync(dir, { withFileTypes: true })
+        .filter((entry) => entry.isFile() && entry.name.endsWith(".json"))
+        .map((entry) => {
+        const filePath = path.join(dir, entry.name);
+        return { fileName: entry.name, filePath, document: readJsonDocument(filePath) };
+    })
+        .sort((left, right) => left.fileName.localeCompare(right.fileName));
+}
+function reloadMailSearchCache(agentId, options) {
+    const state = cacheState(agentId, options);
+    state.loaded = false;
+    state.docs.clear();
+}
+function removeMailSearchCacheDocument(agentId, messageId, options) {
+    const filePath = cachePath(agentId, messageId, options);
+    let removed = false;
+    try {
+        if (fs.lstatSync(filePath).isFile()) {
+            fs.unlinkSync(filePath);
+            removed = true;
+        }
+    }
+    catch (error) {
+        if (!(typeof error === "object" && error !== null && "code" in error && error.code === "ENOENT"))
+            throw error;
+    }
+    const state = cacheState(agentId, options);
+    if (state.loaded)
+        state.docs.delete(messageId);
+    return removed;
+}
+function removeMailSearchCacheFile(agentId, fileName, options) {
+    if (path.basename(fileName) !== fileName || !fileName.endsWith(".json"))
+        return false;
+    const filePath = path.join(cacheDir(agentId, options), fileName);
+    try {
+        if (!fs.lstatSync(filePath).isFile())
+            return false;
+        fs.unlinkSync(filePath);
+        reloadMailSearchCache(agentId, options);
+        return true;
+    }
+    catch (error) {
+        if (!(typeof error === "object" && error !== null && "code" in error && error.code === "ENOENT"))
+            throw error;
+        return false;
+    }
+}
 function syncMailSearchCacheMetadata(message, options) {
     const existing = readJsonDocument(cachePath(message.agentId, message.id, options));
     if (!existing)
@@ -177,7 +247,7 @@ function syncMailSearchCacheMetadata(message, options) {
         ...(message.ownerEmail ? { ownerEmail: message.ownerEmail } : {}),
         ...(message.source ? { source: message.source } : {}),
     };
-    fs.writeFileSync(cachePath(message.agentId, message.id, options), `${JSON.stringify(updated)}\n`, "utf-8");
+    writeJsonAtomically(cachePath(message.agentId, message.id, options), updated);
     const state = cacheState(message.agentId, options);
     if (state.loaded)
         state.docs.set(updated.messageId, updated);
@@ -312,7 +382,7 @@ function writeMailSearchCoverageRecord(record, options) {
         ...(record.newestReceivedAt ? { newestReceivedAt: record.newestReceivedAt } : {}),
     };
     fs.mkdirSync(coverageDir(document.agentId, options), { recursive: true });
-    fs.writeFileSync(coveragePath(document, options), `${JSON.stringify(document)}\n`, "utf-8");
+    writeJsonAtomically(coveragePath(document, options), document);
     (0, runtime_1.emitNervesEvent)({
         component: "senses",
         event: "senses.mail_search_coverage_written",