codemem 0.30.0-alpha.2 → 0.30.0-alpha.3

package/dist/index.js

@@ -3118,11 +3118,22 @@ cmd$1.action((inputFile, opts) => {
 			`Prompts:        ${payload.user_prompts.length.toLocaleString()}`
 		].join("\n"));
 	}
-	const result = importMemories(payload, {
-		dbPath: resolveDbPath(resolveDbOpt(opts)),
-		remapProject: opts.remapProject,
-		dryRun: opts.dryRun
-	});
+	let result;
+	try {
+		result = importMemories(payload, {
+			dbPath: resolveDbPath(resolveDbOpt(opts)),
+			remapProject: opts.remapProject,
+			dryRun: opts.dryRun
+		});
+	} catch (error) {
+		const message = error instanceof Error ? error.message : "Import failed";
+		if (opts.json) emitJsonError("import_failed", message);
+		else {
+			p.log.error(message);
+			process.exitCode = 1;
+		}
+		return;
+	}
 	if (opts.json) {
 		console.log(JSON.stringify({
 			sessions: result.sessions,
@@ -3343,6 +3354,14 @@ function forgetMemoryAction(idStr, opts) {
 	}
 	const store = new MemoryStore(resolveDbPath(resolveDbOpt(opts)));
 	try {
+		if (!store.get(memoryId)) {
+			if (opts.json) emitJsonError("not_found", `Memory ${memoryId} not found`);
+			else {
+				p.log.error(`Memory ${memoryId} not found`);
+				process.exitCode = 1;
+			}
+			return;
+		}
 		store.forget(memoryId);
 		if (opts.json) console.log(JSON.stringify({
 			id: memoryId,
@@ -3353,6 +3372,19 @@ function forgetMemoryAction(idStr, opts) {
 		store.close();
 	}
 }
+function rollbackManualMemory(store, sessionId, memoryId) {
+	store.db.transaction(() => {
+		const row = store.db.prepare("SELECT import_key FROM memory_items WHERE id = ?").get(memoryId);
+		store.db.prepare("DELETE FROM memory_vectors WHERE memory_id = ?").run(memoryId);
+		store.db.prepare("DELETE FROM memory_file_refs WHERE memory_id = ?").run(memoryId);
+		store.db.prepare("DELETE FROM memory_concept_refs WHERE memory_id = ?").run(memoryId);
+		store.db.prepare("DELETE FROM replication_ops WHERE entity_type = 'memory_item' AND (entity_id = ? OR entity_id = ?)").run(row?.import_key ?? "", String(memoryId));
+		store.db.prepare("DELETE FROM memory_items WHERE id = ?").run(memoryId);
+		store.db.prepare(`DELETE FROM sessions
+				 WHERE id = ?
+				   AND NOT EXISTS (SELECT 1 FROM memory_items WHERE session_id = ?)`).run(sessionId, sessionId);
+	})();
+}
 async function rememberMemoryAction(opts) {
 	const store = new MemoryStore(resolveDbPath(resolveDbOpt(opts)));
 	let sessionId = null;
@@ -3366,8 +3398,14 @@ async function rememberMemoryAction(opts) {
 			metadata: { manual: true }
 		});
 		const memId = store.remember(sessionId, opts.kind, opts.title, opts.body, .5, opts.tags);
-		await store.flushPendingVectorWrites();
+		if (!store.get(memId)) {
+			await store.flushPendingVectorWrites();
+			rollbackManualMemory(store, sessionId, memId);
+			sessionId = null;
+			throw new Error("unauthorized_scope");
+		}
 		store.endSession(sessionId, { manual: true });
+		await store.flushPendingVectorWrites();
 		if (opts.json) console.log(JSON.stringify({ id: memId }));
 		else p.log.success(`Stored memory ${memId}`);
 	} catch (err) {