@iletai/nzb 1.5.3 → 1.6.0

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 });