@iletai/nzb 1.5.2 → 1.5.4

package/dist/daemon.js

@@ -1,8 +1,10 @@
 import { spawn } from "child_process";
+import { existsSync, readFileSync, unlinkSync, writeFileSync } from "fs";
 import { broadcastToSSE, startApiServer } from "./api/server.js";
 import { config } from "./config.js";
 import { getClient, stopClient } from "./copilot/client.js";
 import { getWorkers, initOrchestrator, setMessageLogger, setProactiveNotify, setWorkerNotify, stopHealthCheck, } from "./copilot/orchestrator.js";
+import { PID_FILE_PATH } from "./paths.js";
 import { closeDb, getDb } from "./store/db.js";
 import { createBot, sendProactiveMessage, sendWorkerNotification, startBot, stopBot } from "./telegram/bot.js";
 import { checkForUpdate } from "./update.js";
@@ -14,8 +16,64 @@ function truncate(text, max = 200) {
     const oneLine = text.replace(/\n/g, " ").trim();
     return oneLine.length > max ? oneLine.slice(0, max) + "…" : oneLine;
 }
+/**
+ * Check if a process with the given PID is alive.
+ * Sends signal 0 which doesn't kill but checks existence.
+ */
+function isProcessAlive(pid) {
+    try {
+        process.kill(pid, 0);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * Acquire a PID lock file. Prevents multiple daemon instances.
+ * Returns true if lock acquired, false if another instance is running.
+ */
+function acquirePidLock() {
+    if (existsSync(PID_FILE_PATH)) {
+        try {
+            const existingPid = parseInt(readFileSync(PID_FILE_PATH, "utf-8").trim(), 10);
+            if (!isNaN(existingPid) && isProcessAlive(existingPid)) {
+                console.error(`[nzb] Another NZB instance is already running (PID ${existingPid}).`);
+                console.error(`[nzb] Stop it first, or remove ${PID_FILE_PATH} if the process is stale.`);
+                return false;
+            }
+            // Stale PID file — process is dead, remove it
+            console.log(`[nzb] Removed stale PID file (old PID ${existingPid} is no longer running).`);
+        }
+        catch {
+            // Corrupt PID file — remove it
+        }
+        unlinkSync(PID_FILE_PATH);
+    }
+    writeFileSync(PID_FILE_PATH, String(process.pid), { mode: 0o644 });
+    return true;
+}
+/** Release the PID lock file. */
+function releasePidLock() {
+    try {
+        if (existsSync(PID_FILE_PATH)) {
+            const pid = parseInt(readFileSync(PID_FILE_PATH, "utf-8").trim(), 10);
+            // Only remove if it's our PID (in case a new instance took over)
+            if (pid === process.pid) {
+                unlinkSync(PID_FILE_PATH);
+            }
+        }
+    }
+    catch {
+        /* best effort */
+    }
+}
 async function main() {
     console.log("[nzb] Starting NZB daemon...");
+    // Single-instance guard
+    if (!acquirePidLock()) {
+        process.exit(1);
+    }
     if (config.selfEditEnabled) {
         console.log("[nzb] Warning: Self-edit mode enabled — NZB can modify his own source code");
     }
@@ -153,6 +211,7 @@ async function shutdown() {
         /* best effort */
     }
     closeDb();
+    releasePidLock();
     console.log("[nzb] Goodbye.");
     process.exit(0);
 }
@@ -184,6 +243,7 @@ export async function restartDaemon() {
         /* best effort */
     }
     closeDb();
+    releasePidLock();
     // Spawn a detached replacement process with the same args (include execArgv for tsx/loaders)
     const child = spawn(process.execPath, [...process.execArgv, ...process.argv.slice(1)], {
         detached: true,

package/dist/paths.js

@@ -17,6 +17,8 @@ export const HISTORY_PATH = join(NZB_HOME, "tui_history");
 export const TUI_DEBUG_LOG_PATH = join(NZB_HOME, "tui-debug.log");
 /** Path to the API bearer token file */
 export const API_TOKEN_PATH = join(NZB_HOME, "api-token");
+/** Path to the PID lock file for single-instance enforcement */
+export const PID_FILE_PATH = join(NZB_HOME, "nzb.pid");
 /** Ensure ~/.nzb/ exists */
 export function ensureNZBHome() {
     mkdirSync(NZB_HOME, { recursive: true });

package/dist/telegram/bot.js

@@ -33,11 +33,29 @@ const TIMEOUT_PRESETS = [
     { ms: 3_600_000, label: "60min" },
     { ms: 7_200_000, label: "120min" },
 ];
-const MODEL_PRESETS = [
-    "claude-sonnet-4-20250514",
-    "claude-haiku-4-20250414",
-    "claude-opus-4-20250115",
-];
+// Dynamic model list — fetched from Copilot SDK, cached for 5 minutes
+let cachedModels;
+let cachedModelsAt = 0;
+const MODEL_CACHE_TTL = 5 * 60_000;
+async function getAvailableModels() {
+    if (cachedModels && Date.now() - cachedModelsAt < MODEL_CACHE_TTL) {
+        return cachedModels;
+    }
+    try {
+        const { getClient } = await import("../copilot/client.js");
+        const client = await getClient();
+        const models = await client.listModels();
+        if (models.length > 0) {
+            cachedModels = models.map((m) => m.id);
+            cachedModelsAt = Date.now();
+            return cachedModels;
+        }
+    }
+    catch {
+        /* fall through to fallback */
+    }
+    return cachedModels ?? [config.copilotModel];
+}
 function getTimeoutLabel() {
     const preset = TIMEOUT_PRESETS.find((p) => p.ms === config.workerTimeoutMs);
     return preset ? preset.label : `${Math.round(config.workerTimeoutMs / 60_000)}min`;
@@ -62,8 +80,13 @@ const settingsMenu = new Menu("settings-menu")
 })
     .row()
     .text(() => `🤖 ${config.copilotModel}`, async (ctx) => {
-    const idx = MODEL_PRESETS.indexOf(config.copilotModel);
-    const next = MODEL_PRESETS[(idx + 1) % MODEL_PRESETS.length];
+    const models = await getAvailableModels();
+    if (models.length === 0) {
+        await ctx.answerCallbackQuery("No models available");
+        return;
+    }
+    const idx = models.indexOf(config.copilotModel);
+    const next = models[(idx + 1) % models.length];
     config.copilotModel = next;
     persistModel(next);
     ctx.menu.update();
@@ -244,7 +267,8 @@ export function createBot() {
         "⚡ Breakthrough Features:\n" +
         "• @bot query — Use me inline in any chat!\n" +
         "• React to any message to trigger AI:\n" +
-        getReactionHelpText() + "\n" +
+        getReactionHelpText() +
+        "\n" +
         "• Smart suggestions appear after each response", { reply_markup: mainMenu }));
     bot.command("cancel", async (ctx) => {
         const cancelled = await cancelCurrentMessage();
@@ -771,15 +795,22 @@ export async function startBot() {
             void logInfo(`🚀 NZB v${process.env.npm_package_version || "?"} started (model: ${config.copilotModel})`);
         },
     })
-        .catch((err) => {
+        .catch(async (err) => {
         if (err?.error_code === 401) {
             console.error("[nzb] Warning: Telegram bot token is invalid or expired. Run 'nzb setup' and re-enter your bot token from @BotFather.");
+            return; // Unrecoverable — don't retry
         }
-        else if (err?.error_code === 409) {
-            console.error("[nzb] Warning: Another bot instance is already running with this token. Stop the other instance first.");
+        if (err?.error_code === 409) {
+            console.error("[nzb] Warning: Telegram polling conflict (409). Restarting polling in 5 seconds...");
         }
         else {
-            console.error("[nzb] Error: Telegram bot failed to start:", err?.message || err);
+            console.error("[nzb] Error: Telegram polling stopped:", err?.message || err, "— restarting in 5 seconds...");
+        }
+        // Auto-restart polling after a delay
+        await new Promise((r) => setTimeout(r, 5000));
+        if (bot) {
+            console.log("[nzb] Re-starting Telegram polling...");
+            startBot().catch((e) => console.error("[nzb] Failed to re-start Telegram polling:", e));
         }
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@iletai/nzb",
-    "version": "1.5.2",
+    "version": "1.5.4",
     "description": "NZB — a personal AI assistant for developers, built on the GitHub Copilot SDK",
     "bin": {
         "nzb": "dist/cli.js"
@@ -67,4 +67,4 @@
         "typescript": "^5.9.3",
         "vitest": "^4.1.0"
     }
-}
+}