@iletai/nzb 1.7.0 → 1.7.3

package/dist/daemon.js

@@ -26,6 +26,7 @@ function isProcessAlive(pid) {
         return true;
     }
     catch {
+        // Expected: process.kill(0) throws when process doesn't exist
         return false;
     }
 }
@@ -64,8 +65,8 @@ function releasePidLock() {
             }
         }
     }
-    catch {
-        /* best effort */
+    catch (err) {
+        console.error("[nzb] PID lock cleanup:", err instanceof Error ? err.message : err);
     }
 }
 async function main() {
@@ -197,8 +198,8 @@ async function shutdown() {
         try {
             await stopBot();
         }
-        catch {
-            /* best effort */
+        catch (err) {
+            console.error("[nzb] stopBot during shutdown:", err instanceof Error ? err.message : err);
         }
     }
     // Destroy all active worker sessions to free memory
@@ -207,8 +208,8 @@ async function shutdown() {
     try {
         await stopClient();
     }
-    catch {
-        /* best effort */
+    catch (err) {
+        console.error("[nzb] stopClient during shutdown:", err instanceof Error ? err.message : err);
     }
     closeDb();
     releasePidLock();
@@ -229,8 +230,8 @@ export async function restartDaemon() {
         try {
             await stopBot();
         }
-        catch {
-            /* best effort */
+        catch (err) {
+            console.error("[nzb] stopBot during restart:", err instanceof Error ? err.message : err);
         }
     }
     // Destroy all active worker sessions to free memory
@@ -239,8 +240,8 @@ export async function restartDaemon() {
     try {
         await stopClient();
     }
-    catch {
-        /* best effort */
+    catch (err) {
+        console.error("[nzb] stopClient during restart:", err instanceof Error ? err.message : err);
     }
     closeDb();
     releasePidLock();

package/dist/setup.js

@@ -28,14 +28,15 @@ async function fetchModels() {
         });
     }
     catch {
+        // Expected: Copilot CLI may not be authenticated yet
         return [];
     }
     finally {
         try {
             await client?.stop();
         }
-        catch {
-            /* best-effort */
+        catch (err) {
+            console.error("[nzb] CopilotClient stop:", err instanceof Error ? err.message : err);
         }
     }
 }

package/dist/store/conversation.js

@@ -0,0 +1,96 @@
+import { getDb } from "./db.js";
+// Lazy per-connection prepared statement cache
+let cachedDb;
+let stmtCache;
+function ensureStmtCache() {
+    const db = getDb();
+    if (db !== cachedDb) {
+        cachedDb = db;
+        stmtCache = {
+            logConversation: db.prepare(`INSERT INTO conversation_log (role, content, source, telegram_msg_id) VALUES (?, ?, ?, ?)`),
+            pruneConversation: db.prepare(`DELETE FROM conversation_log WHERE id NOT IN (SELECT id FROM conversation_log ORDER BY id DESC LIMIT 200)`),
+            getConversationByMsgId: db.prepare(`SELECT id FROM conversation_log WHERE telegram_msg_id = ? LIMIT 1`),
+        };
+    }
+    return stmtCache;
+}
+/** Log a conversation turn (user, assistant, or system) with optional Telegram message ID. Returns the row ID. */
+export function logConversation(role, content, source, telegramMsgId) {
+    const cache = ensureStmtCache();
+    const result = cache.logConversation.run(role, content, source, telegramMsgId ?? null);
+    // Prune every ~50 inserts using rowid (crash-safe, no in-memory counter)
+    const rowId = result.lastInsertRowid;
+    if (rowId % 50 === 0) {
+        try {
+            cache.pruneConversation.run();
+        }
+        catch (err) {
+            console.error("[nzb] Conversation prune failed:", err instanceof Error ? err.message : err);
+        }
+    }
+    return rowId;
+}
+/** Get conversation context around a Telegram message ID (±4 rows using proper subquery). */
+export function getConversationContext(telegramMsgId) {
+    const db = getDb();
+    const cache = ensureStmtCache();
+    const row = cache.getConversationByMsgId.get(telegramMsgId);
+    if (!row)
+        return undefined;
+    // Fetch 4 rows before + the target + 4 rows after (handles ID gaps from pruning)
+    const rows = db
+        .prepare(`
+		SELECT role, content, source, ts FROM (
+			SELECT * FROM conversation_log WHERE id < ? ORDER BY id DESC LIMIT 4
+		)
+		UNION ALL
+		SELECT role, content, source, ts FROM conversation_log WHERE id = ?
+		UNION ALL
+		SELECT role, content, source, ts FROM (
+			SELECT * FROM conversation_log WHERE id > ? ORDER BY id ASC LIMIT 4
+		)
+	`)
+        .all(row.id, row.id, row.id);
+    if (rows.length === 0)
+        return undefined;
+    return rows
+        .map((r) => {
+        const tag = r.role === "user" ? "You" : r.role === "assistant" ? "NZB" : "System";
+        const content = r.content.length > 400 ? r.content.slice(0, 400) + "…" : r.content;
+        return `${tag}: ${content}`;
+    })
+        .join("\n");
+}
+/** Set Telegram message ID on a specific conversation_log row (race-free). */
+export function setConversationTelegramMsgId(rowId, telegramMsgId) {
+    const db = getDb();
+    db.prepare(`UPDATE conversation_log SET telegram_msg_id = ? WHERE id = ?`).run(telegramMsgId, rowId);
+}
+/** Look up conversation content by Telegram message ID. Returns the message content or undefined. */
+export function getConversationByTelegramMsgId(telegramMsgId) {
+    const db = getDb();
+    const row = db
+        .prepare(`SELECT content FROM conversation_log WHERE telegram_msg_id = ? LIMIT 1`)
+        .get(telegramMsgId);
+    return row?.content;
+}
+/** Get recent conversation history formatted for injection into system message. */
+export function getRecentConversation(limit = 20) {
+    const db = getDb();
+    const rows = db
+        .prepare(`SELECT role, content, source, ts FROM conversation_log ORDER BY id DESC LIMIT ?`)
+        .all(limit);
+    if (rows.length === 0)
+        return "";
+    // Reverse so oldest is first (chronological order)
+    rows.reverse();
+    return rows
+        .map((r) => {
+        const tag = r.role === "user" ? `[${r.source}] User` : r.role === "system" ? `[${r.source}] System` : "NZB";
+        // Truncate long messages to keep context manageable
+        const content = r.content.length > 500 ? r.content.slice(0, 500) + "…" : r.content;
+        return `${tag}: ${content}`;
+    })
+        .join("\n\n");
+}
+//# sourceMappingURL=conversation.js.map

package/dist/store/db.js

@@ -1,8 +1,7 @@
 import Database from "better-sqlite3";
 import { DB_PATH, ensureNZBHome } from "../paths.js";
 let db;
-let logInsertCount = 0;
-// Cached prepared statements for hot-path queries (created lazily after DB init)
+// Cached prepared statements for state operations (created lazily after DB init)
 let stmtCache;
 export function getDb() {
     if (!db) {
@@ -115,17 +114,11 @@ export function getDb() {
             // FTS5 may not be available — will fall back to LIKE
             console.log("[nzb] FTS5 not available, using LIKE fallback for memory search");
         }
-        // Initialize cached prepared statements for hot-path operations
+        // Initialize cached prepared statements for state operations
         stmtCache = {
             getState: db.prepare(`SELECT value FROM nzb_state WHERE key = ?`),
             setState: db.prepare(`INSERT OR REPLACE INTO nzb_state (key, value) VALUES (?, ?)`),
             deleteState: db.prepare(`DELETE FROM nzb_state WHERE key = ?`),
-            logConversation: db.prepare(`INSERT INTO conversation_log (role, content, source, telegram_msg_id) VALUES (?, ?, ?, ?)`),
-            pruneConversation: db.prepare(`DELETE FROM conversation_log WHERE id NOT IN (SELECT id FROM conversation_log ORDER BY id DESC LIMIT 200)`),
-            addMemory: db.prepare(`INSERT INTO memories (category, content, source) VALUES (?, ?, ?)`),
-            removeMemory: db.prepare(`DELETE FROM memories WHERE id = ?`),
-            memorySummary: db.prepare(`SELECT id, category, content FROM memories ORDER BY category, last_accessed DESC`),
-            getConversationByMsgId: db.prepare(`SELECT id FROM conversation_log WHERE telegram_msg_id = ? LIMIT 1`),
         };
     }
     return db;
@@ -144,203 +137,6 @@ export function deleteState(key) {
     getDb(); // ensure init
     stmtCache.deleteState.run(key);
 }
-/** Log a conversation turn (user, assistant, or system) with optional Telegram message ID. Returns the row ID. */
-export function logConversation(role, content, source, telegramMsgId) {
-    getDb(); // ensure init
-    const result = stmtCache.logConversation.run(role, content, source, telegramMsgId ?? null);
-    // Keep last 200 entries to support context recovery after session loss
-    logInsertCount++;
-    if (logInsertCount % 50 === 0) {
-        stmtCache.pruneConversation.run();
-    }
-    return result.lastInsertRowid;
-}
-/** Get conversation context around a Telegram message ID (±4 rows using proper subquery). */
-export function getConversationContext(telegramMsgId) {
-    const db = getDb();
-    const row = stmtCache.getConversationByMsgId.get(telegramMsgId);
-    if (!row)
-        return undefined;
-    // Fetch 4 rows before + the target + 4 rows after (handles ID gaps from pruning)
-    const rows = db
-        .prepare(`
-		SELECT role, content, source, ts FROM (
-			SELECT * FROM conversation_log WHERE id < ? ORDER BY id DESC LIMIT 4
-		)
-		UNION ALL
-		SELECT role, content, source, ts FROM conversation_log WHERE id = ?
-		UNION ALL
-		SELECT role, content, source, ts FROM (
-			SELECT * FROM conversation_log WHERE id > ? ORDER BY id ASC LIMIT 4
-		)
-	`)
-        .all(row.id, row.id, row.id);
-    if (rows.length === 0)
-        return undefined;
-    return rows
-        .map((r) => {
-        const tag = r.role === "user" ? "You" : r.role === "assistant" ? "NZB" : "System";
-        const content = r.content.length > 400 ? r.content.slice(0, 400) + "…" : r.content;
-        return `${tag}: ${content}`;
-    })
-        .join("\n");
-}
-/** Set Telegram message ID on a specific conversation_log row (race-free). */
-export function setConversationTelegramMsgId(rowId, telegramMsgId) {
-    const db = getDb();
-    db.prepare(`UPDATE conversation_log SET telegram_msg_id = ? WHERE id = ?`).run(telegramMsgId, rowId);
-}
-/** Look up conversation content by Telegram message ID. Returns the message content or undefined. */
-export function getConversationByTelegramMsgId(telegramMsgId) {
-    const db = getDb();
-    const row = db
-        .prepare(`SELECT content FROM conversation_log WHERE telegram_msg_id = ? LIMIT 1`)
-        .get(telegramMsgId);
-    return row?.content;
-}
-/** Get recent conversation history formatted for injection into system message. */
-export function getRecentConversation(limit = 20) {
-    const db = getDb();
-    const rows = db
-        .prepare(`SELECT role, content, source, ts FROM conversation_log ORDER BY id DESC LIMIT ?`)
-        .all(limit);
-    if (rows.length === 0)
-        return "";
-    // Reverse so oldest is first (chronological order)
-    rows.reverse();
-    return rows
-        .map((r) => {
-        const tag = r.role === "user" ? `[${r.source}] User` : r.role === "system" ? `[${r.source}] System` : "NZB";
-        // Truncate long messages to keep context manageable
-        const content = r.content.length > 500 ? r.content.slice(0, 500) + "…" : r.content;
-        return `${tag}: ${content}`;
-    })
-        .join("\n\n");
-}
-/** Add a memory to long-term storage. */
-export function addMemory(category, content, source = "user") {
-    getDb(); // ensure init
-    const result = stmtCache.addMemory.run(category, content, source);
-    return result.lastInsertRowid;
-}
-/** Search memories by keyword and/or category. Uses FTS5 when available, falls back to LIKE. */
-export function searchMemories(keyword, category, limit = 20) {
-    const db = getDb();
-    // Try FTS5 first for keyword search (much faster than LIKE)
-    if (keyword) {
-        try {
-            const catFilter = category ? `AND m.category = ?` : "";
-            const params = [keyword + "*"];
-            if (category)
-                params.push(category);
-            params.push(limit);
-            const rows = db
-                .prepare(`SELECT m.id, m.category, m.content, m.source, m.created_at
-					 FROM memories_fts f
-					 JOIN memories m ON f.rowid = m.id
-					 WHERE memories_fts MATCH ? ${catFilter}
-					 ORDER BY rank LIMIT ?`)
-                .all(...params);
-            return rows;
-        }
-        catch {
-            // FTS5 not available — fall through to LIKE
-        }
-    }
-    // Fallback: LIKE-based search
-    const conditions = [];
-    const params = [];
-    if (keyword) {
-        conditions.push(`content LIKE ?`);
-        params.push(`%${keyword}%`);
-    }
-    if (category) {
-        conditions.push(`category = ?`);
-        params.push(category);
-    }
-    const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
-    params.push(limit);
-    const rows = db
-        .prepare(`SELECT id, category, content, source, created_at FROM memories ${where} ORDER BY last_accessed DESC LIMIT ?`)
-        .all(...params);
-    // Update last_accessed only when explicitly requested, not on every search
-    // (removed automatic last_accessed update to avoid write side effects on reads)
-    return rows;
-}
-/** Remove a memory by ID. */
-export function removeMemory(id) {
-    getDb(); // ensure init
-    const result = stmtCache.removeMemory.run(id);
-    return result.changes > 0;
-}
-/** Get a compact summary of all memories for injection into system message. */
-export function getMemorySummary() {
-    getDb(); // ensure init
-    const rows = stmtCache.memorySummary.all();
-    if (rows.length === 0)
-        return "";
-    // Group by category
-    const grouped = {};
-    for (const r of rows) {
-        if (!grouped[r.category])
-            grouped[r.category] = [];
-        grouped[r.category].push({ id: r.id, content: r.content });
-    }
-    const sections = Object.entries(grouped).map(([cat, items]) => {
-        const lines = items.map((i) => `  - [#${i.id}] ${i.content}`).join("\n");
-        return `**${cat}**:\n${lines}`;
-    });
-    return sections.join("\n");
-}
-// ── Agent Teams CRUD ──────────────────────────────────────────
-export function createTeam(id, taskDescription, originChannel) {
-    const db = getDb();
-    db.prepare(`INSERT INTO agent_teams (id, task_description, origin_channel) VALUES (?, ?, ?)`).run(id, taskDescription, originChannel ?? null);
-}
-export function addTeamMember(teamId, workerName, role) {
-    const db = getDb();
-    db.prepare(`INSERT INTO team_members (team_id, worker_name, role, status) VALUES (?, ?, ?, 'pending')`).run(teamId, workerName, role);
-    db.prepare(`UPDATE agent_teams SET member_count = member_count + 1 WHERE id = ?`).run(teamId);
-}
-export function updateTeamMemberResult(teamId, workerName, result, status) {
-    const db = getDb();
-    db.prepare(`UPDATE team_members SET result = ?, status = ?, completed_at = CURRENT_TIMESTAMP WHERE team_id = ? AND worker_name = ?`).run(result, status, teamId, workerName);
-    if (status === "completed" || status === "error") {
-        db.prepare(`UPDATE agent_teams SET completed_count = completed_count + 1 WHERE id = ?`).run(teamId);
-    }
-}
-export function getTeam(id) {
-    const db = getDb();
-    return db.prepare(`SELECT * FROM agent_teams WHERE id = ?`).get(id);
-}
-export function getTeamMembers(teamId) {
-    const db = getDb();
-    return db
-        .prepare(`SELECT worker_name, role, status, result FROM team_members WHERE team_id = ? ORDER BY id`)
-        .all(teamId);
-}
-export function completeTeam(teamId, aggregatedResult, status = "completed") {
-    const db = getDb();
-    db.prepare(`UPDATE agent_teams SET status = ?, aggregated_result = ?, completed_at = CURRENT_TIMESTAMP WHERE id = ?`).run(status, aggregatedResult, teamId);
-}
-export function getActiveTeams() {
-    const db = getDb();
-    return db
-        .prepare(`SELECT id, status, task_description, member_count, completed_count, created_at FROM agent_teams WHERE status = 'active' ORDER BY created_at DESC`)
-        .all();
-}
-export function getTeamByWorkerName(workerName) {
-    const db = getDb();
-    const row = db
-        .prepare(`SELECT team_id FROM team_members WHERE worker_name = ? AND status IN ('pending', 'running') LIMIT 1`)
-        .get(workerName);
-    return row?.team_id;
-}
-export function cleanupTeam(teamId) {
-    const db = getDb();
-    db.prepare(`DELETE FROM team_members WHERE team_id = ?`).run(teamId);
-    db.prepare(`DELETE FROM agent_teams WHERE id = ?`).run(teamId);
-}
 export function closeDb() {
     if (db) {
         stmtCache = undefined;
@@ -348,4 +144,8 @@ export function closeDb() {
         db = undefined;
     }
 }
+// Re-export for backward compatibility
+export { logConversation, getConversationContext, setConversationTelegramMsgId, getConversationByTelegramMsgId, getRecentConversation, } from "./conversation.js";
+export { addMemory, searchMemories, removeMemory, getMemorySummary } from "./memory.js";
+export { createTeam, addTeamMember, updateTeamMemberResult, getTeam, getTeamMembers, completeTeam, getActiveTeams, getTeamByWorkerName, cleanupTeam, } from "./team-store.js";
 //# sourceMappingURL=db.js.map

package/dist/store/memory.js

@@ -0,0 +1,90 @@
+import { getDb } from "./db.js";
+// Lazy per-connection prepared statement cache
+let cachedDb;
+let stmtCache;
+function ensureStmtCache() {
+    const db = getDb();
+    if (db !== cachedDb) {
+        cachedDb = db;
+        stmtCache = {
+            addMemory: db.prepare(`INSERT INTO memories (category, content, source) VALUES (?, ?, ?)`),
+            removeMemory: db.prepare(`DELETE FROM memories WHERE id = ?`),
+            memorySummary: db.prepare(`SELECT id, category, content FROM memories ORDER BY category, last_accessed DESC`),
+        };
+    }
+    return stmtCache;
+}
+/** Add a memory to long-term storage. */
+export function addMemory(category, content, source = "user") {
+    const cache = ensureStmtCache();
+    const result = cache.addMemory.run(category, content, source);
+    return result.lastInsertRowid;
+}
+/** Search memories by keyword and/or category. Uses FTS5 when available, falls back to LIKE. */
+export function searchMemories(keyword, category, limit = 20) {
+    const db = getDb();
+    // Try FTS5 first for keyword search (much faster than LIKE)
+    if (keyword) {
+        try {
+            const catFilter = category ? `AND m.category = ?` : "";
+            const params = [keyword + "*"];
+            if (category)
+                params.push(category);
+            params.push(limit);
+            const rows = db
+                .prepare(`SELECT m.id, m.category, m.content, m.source, m.created_at
+					 FROM memories_fts f
+					 JOIN memories m ON f.rowid = m.id
+					 WHERE memories_fts MATCH ? ${catFilter}
+					 ORDER BY rank LIMIT ?`)
+                .all(...params);
+            return rows;
+        }
+        catch {
+            // FTS5 not available — fall through to LIKE
+        }
+    }
+    // Fallback: LIKE-based search
+    const conditions = [];
+    const params = [];
+    if (keyword) {
+        conditions.push(`content LIKE ?`);
+        params.push(`%${keyword}%`);
+    }
+    if (category) {
+        conditions.push(`category = ?`);
+        params.push(category);
+    }
+    const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+    params.push(limit);
+    const rows = db
+        .prepare(`SELECT id, category, content, source, created_at FROM memories ${where} ORDER BY last_accessed DESC LIMIT ?`)
+        .all(...params);
+    return rows;
+}
+/** Remove a memory by ID. */
+export function removeMemory(id) {
+    const cache = ensureStmtCache();
+    const result = cache.removeMemory.run(id);
+    return result.changes > 0;
+}
+/** Get a compact summary of all memories for injection into system message. */
+export function getMemorySummary() {
+    const cache = ensureStmtCache();
+    const rows = cache.memorySummary.all();
+    if (rows.length === 0)
+        return "";
+    // Group by category
+    const grouped = {};
+    for (const r of rows) {
+        if (!grouped[r.category])
+            grouped[r.category] = [];
+        grouped[r.category].push({ id: r.id, content: r.content });
+    }
+    const sections = Object.entries(grouped).map(([cat, items]) => {
+        const lines = items.map((i) => `  - [#${i.id}] ${i.content}`).join("\n");
+        return `**${cat}**:\n${lines}`;
+    });
+    return sections.join("\n");
+}
+//# sourceMappingURL=memory.js.map

package/dist/store/team-store.js

@@ -0,0 +1,51 @@
+import { getDb } from "./db.js";
+// ── Agent Teams CRUD ──────────────────────────────────────────
+export function createTeam(id, taskDescription, originChannel) {
+    const db = getDb();
+    db.prepare(`INSERT INTO agent_teams (id, task_description, origin_channel) VALUES (?, ?, ?)`).run(id, taskDescription, originChannel ?? null);
+}
+export function addTeamMember(teamId, workerName, role) {
+    const db = getDb();
+    db.prepare(`INSERT INTO team_members (team_id, worker_name, role, status) VALUES (?, ?, ?, 'pending')`).run(teamId, workerName, role);
+    db.prepare(`UPDATE agent_teams SET member_count = member_count + 1 WHERE id = ?`).run(teamId);
+}
+export function updateTeamMemberResult(teamId, workerName, result, status) {
+    const db = getDb();
+    db.prepare(`UPDATE team_members SET result = ?, status = ?, completed_at = CURRENT_TIMESTAMP WHERE team_id = ? AND worker_name = ?`).run(result, status, teamId, workerName);
+    if (status === "completed" || status === "error") {
+        db.prepare(`UPDATE agent_teams SET completed_count = completed_count + 1 WHERE id = ?`).run(teamId);
+    }
+}
+export function getTeam(id) {
+    const db = getDb();
+    return db.prepare(`SELECT * FROM agent_teams WHERE id = ?`).get(id);
+}
+export function getTeamMembers(teamId) {
+    const db = getDb();
+    return db
+        .prepare(`SELECT worker_name, role, status, result FROM team_members WHERE team_id = ? ORDER BY id`)
+        .all(teamId);
+}
+export function completeTeam(teamId, aggregatedResult, status = "completed") {
+    const db = getDb();
+    db.prepare(`UPDATE agent_teams SET status = ?, aggregated_result = ?, completed_at = CURRENT_TIMESTAMP WHERE id = ?`).run(status, aggregatedResult, teamId);
+}
+export function getActiveTeams() {
+    const db = getDb();
+    return db
+        .prepare(`SELECT id, status, task_description, member_count, completed_count, created_at FROM agent_teams WHERE status = 'active' ORDER BY created_at DESC`)
+        .all();
+}
+export function getTeamByWorkerName(workerName) {
+    const db = getDb();
+    const row = db
+        .prepare(`SELECT team_id FROM team_members WHERE worker_name = ? AND status IN ('pending', 'running') LIMIT 1`)
+        .get(workerName);
+    return row?.team_id;
+}
+export function cleanupTeam(teamId) {
+    const db = getDb();
+    db.prepare(`DELETE FROM team_members WHERE team_id = ?`).run(teamId);
+    db.prepare(`DELETE FROM agent_teams WHERE id = ?`).run(teamId);
+}
+//# sourceMappingURL=team-store.js.map