@timo972/cc-router 0.7.0

package/dist/daemon/pid.js

@@ -0,0 +1,98 @@
+import { existsSync, readFileSync, writeFileSync, unlinkSync } from "fs";
+import { PID_PATH, PROXY_PORT } from "../config/paths.js";
+import { ensureConfigDir } from "../config/manager.js";
+/** Write the current process PID to the PID file. */
+export function writePid(pid) {
+    try {
+        ensureConfigDir();
+        writeFileSync(PID_PATH, String(pid), "utf-8");
+    }
+    catch (err) {
+        console.warn(`Warning: cannot write PID file ${PID_PATH}: ${err.message}`);
+        console.warn(`  Daemon is running as PID ${pid} but may not be stoppable via cc-router stop`);
+    }
+}
+/** Read PID from file. Returns null if missing or unreadable. */
+export function readPid() {
+    try {
+        if (!existsSync(PID_PATH))
+            return null;
+        const raw = readFileSync(PID_PATH, "utf-8").trim();
+        const pid = parseInt(raw, 10);
+        return Number.isNaN(pid) ? null : pid;
+    }
+    catch (err) {
+        // File exists but can't be read — likely a permissions issue
+        console.warn(`Warning: cannot read PID file ${PID_PATH}: ${err.message}`);
+        return null;
+    }
+}
+/** Remove the PID file. */
+export function removePid() {
+    try {
+        if (existsSync(PID_PATH))
+            unlinkSync(PID_PATH);
+    }
+    catch (err) {
+        console.warn(`Warning: cannot remove PID file ${PID_PATH}: ${err.message}`);
+    }
+}
+/**
+ * Check if a process with the given PID is alive.
+ * Uses signal 0 — doesn't actually kill the process, just checks existence.
+ */
+export function isProcessAlive(pid) {
+    try {
+        process.kill(pid, 0);
+        return true;
+    }
+    catch (err) {
+        // ESRCH = no such process — it's dead
+        // EPERM = exists but no permission — it's alive
+        return err.code === "EPERM";
+    }
+}
+/**
+ * Read PID and verify the process is actually alive.
+ * Cleans up stale PID files where the process has died.
+ */
+export function getRunningPid() {
+    const pid = readPid();
+    if (pid === null)
+        return null;
+    if (isProcessAlive(pid))
+        return pid;
+    // Stale PID — process died without cleanup
+    removePid();
+    return null;
+}
+/**
+ * Double-check: PID is alive AND health endpoint responds.
+ * Prevents false positives from recycled PIDs (a different process
+ * reusing the same PID number).
+ */
+export async function isProxyRunning(port = PROXY_PORT) {
+    const pid = getRunningPid();
+    if (pid !== null) {
+        // PID exists — verify it's actually cc-router via health endpoint
+        try {
+            const res = await fetch(`http://localhost:${port}/cc-router/health`, {
+                signal: AbortSignal.timeout(1_000),
+            });
+            return res.ok;
+        }
+        catch {
+            return false;
+        }
+    }
+    // No PID file — try health endpoint directly (foreground/legacy processes)
+    try {
+        const res = await fetch(`http://localhost:${port}/cc-router/health`, {
+            signal: AbortSignal.timeout(1_000),
+        });
+        return res.ok;
+    }
+    catch {
+        return false;
+    }
+}

package/dist/daemon/service.js

@@ -0,0 +1,260 @@
+import { existsSync, mkdirSync, writeFileSync, unlinkSync } from "fs";
+import { execFile, execFileSync } from "child_process";
+import { promisify } from "util";
+import { fileURLToPath } from "url";
+import { dirname, join } from "path";
+import os from "os";
+import chalk from "chalk";
+import { LOG_PATH } from "../config/paths.js";
+import { detectPlatform } from "../utils/platform.js";
+const execFileAsync = promisify(execFile);
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const CLI_ENTRY = join(__dirname, "..", "cli", "index.js");
+// ─── Platform paths ──────────────────────────────────────────────────────────
+const LAUNCHD_LABEL = "com.cc-router.proxy";
+const LAUNCHD_PLIST = join(os.homedir(), "Library", "LaunchAgents", `${LAUNCHD_LABEL}.plist`);
+const SYSTEMD_DIR = join(os.homedir(), ".config", "systemd", "user");
+const SYSTEMD_SERVICE = join(SYSTEMD_DIR, "cc-router.service");
+const WINDOWS_REG_KEY = "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run";
+const WINDOWS_REG_NAME = "CC-Router";
+// ─── Public API ──────────────────────────────────────────────────────────────
+export async function installService(serverMode) {
+    const platform = detectPlatform();
+    switch (platform) {
+        case "macos": return installMacOS(serverMode);
+        case "linux": return installLinux(serverMode);
+        case "windows": return installWindows(serverMode);
+    }
+}
+export async function uninstallService() {
+    const platform = detectPlatform();
+    switch (platform) {
+        case "macos": return uninstallMacOS();
+        case "linux": return uninstallLinux();
+        case "windows": return uninstallWindows();
+    }
+}
+export function isServiceInstalled() {
+    const platform = detectPlatform();
+    switch (platform) {
+        case "macos": return existsSync(LAUNCHD_PLIST);
+        case "linux": return existsSync(SYSTEMD_SERVICE);
+        case "windows": return isWindowsServiceInstalled();
+    }
+}
+// ─── macOS LaunchAgent ───────────────────────────────────────────────────────
+function buildPlist(serverMode) {
+    const envVars = serverMode
+        ? `    <key>EnvironmentVariables</key>
+    <dict>
+      <key>PATH</key>
+      <string>${process.env["PATH"] ?? "/usr/local/bin:/usr/bin:/bin"}</string>
+      <key>HOST</key>
+      <string>0.0.0.0</string>
+      <key>CC_ROUTER_SERVICE</key>
+      <string>1</string>
+    </dict>`
+        : `    <key>EnvironmentVariables</key>
+    <dict>
+      <key>PATH</key>
+      <string>${process.env["PATH"] ?? "/usr/local/bin:/usr/bin:/bin"}</string>
+      <key>CC_ROUTER_SERVICE</key>
+      <string>1</string>
+    </dict>`;
+    return `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>Label</key>
+    <string>${LAUNCHD_LABEL}</string>
+    <key>ProgramArguments</key>
+    <array>
+        <string>${process.execPath}</string>
+        <string>${CLI_ENTRY}</string>
+        <string>start</string>
+        <string>--foreground</string>
+    </array>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>KeepAlive</key>
+    <dict>
+        <key>SuccessfulExit</key>
+        <false/>
+    </dict>
+    <key>StandardOutPath</key>
+    <string>${LOG_PATH}</string>
+    <key>StandardErrorPath</key>
+    <string>${LOG_PATH}</string>
+    <key>WorkingDirectory</key>
+    <string>${os.homedir()}</string>
+${envVars}
+</dict>
+</plist>
+`;
+}
+async function installMacOS(serverMode) {
+    // Ensure LaunchAgents dir exists
+    const launchAgentsDir = dirname(LAUNCHD_PLIST);
+    if (!existsSync(launchAgentsDir))
+        mkdirSync(launchAgentsDir, { recursive: true });
+    // Unload existing if present (ignore errors)
+    if (existsSync(LAUNCHD_PLIST)) {
+        await launchctlUnload();
+    }
+    writeFileSync(LAUNCHD_PLIST, buildPlist(serverMode), "utf-8");
+    // Load — try modern `bootstrap` first, fallback to legacy `load`
+    const uid = String(process.getuid?.() ?? 501);
+    try {
+        await execFileAsync("launchctl", ["bootstrap", `gui/${uid}`, LAUNCHD_PLIST]);
+    }
+    catch {
+        try {
+            await execFileAsync("launchctl", ["load", LAUNCHD_PLIST]);
+        }
+        catch (err) {
+            console.log(chalk.yellow(`⚠ Could not auto-load the LaunchAgent: ${err.message}`));
+            console.log(chalk.gray(`  Load manually: launchctl load ${LAUNCHD_PLIST}`));
+            return;
+        }
+    }
+    console.log(chalk.green("✓ Auto-start on boot configured (macOS LaunchAgent)"));
+}
+async function uninstallMacOS() {
+    if (!existsSync(LAUNCHD_PLIST))
+        return;
+    await launchctlUnload();
+    try {
+        unlinkSync(LAUNCHD_PLIST);
+    }
+    catch { /* ok */ }
+    if (isServiceInstalled()) {
+        console.log(chalk.yellow("  ⚠ Service may still be installed — check manually"));
+    }
+    else {
+        console.log(chalk.green("  ✓ Auto-start removed"));
+    }
+}
+async function launchctlUnload() {
+    const uid = String(process.getuid?.() ?? 501);
+    try {
+        await execFileAsync("launchctl", ["bootout", `gui/${uid}/${LAUNCHD_LABEL}`]);
+    }
+    catch {
+        try {
+            await execFileAsync("launchctl", ["unload", LAUNCHD_PLIST]);
+        }
+        catch { /* already unloaded */ }
+    }
+}
+// ─── Linux systemd user service ──────────────────────────────────────────────
+function buildSystemdUnit(serverMode) {
+    const envLine = serverMode
+        ? `Environment=PATH=${process.env["PATH"] ?? "/usr/local/bin:/usr/bin:/bin"}\nEnvironment=HOST=0.0.0.0\nEnvironment=CC_ROUTER_SERVICE=1`
+        : `Environment=PATH=${process.env["PATH"] ?? "/usr/local/bin:/usr/bin:/bin"}\nEnvironment=CC_ROUTER_SERVICE=1`;
+    return `[Unit]
+Description=CC-Router — round-robin proxy for Claude Max
+After=network-online.target
+Wants=network-online.target
+
+[Service]
+Type=simple
+ExecStart=${process.execPath} ${CLI_ENTRY} start --foreground
+Restart=on-failure
+RestartSec=5
+StartLimitIntervalSec=60
+StartLimitBurst=5
+${envLine}
+
+[Install]
+WantedBy=default.target
+`;
+}
+async function installLinux(serverMode) {
+    if (!existsSync(SYSTEMD_DIR))
+        mkdirSync(SYSTEMD_DIR, { recursive: true });
+    writeFileSync(SYSTEMD_SERVICE, buildSystemdUnit(serverMode), "utf-8");
+    try {
+        await execFileAsync("systemctl", ["--user", "daemon-reload"]);
+        await execFileAsync("systemctl", ["--user", "enable", "cc-router"]);
+        await execFileAsync("systemctl", ["--user", "start", "cc-router"]);
+        console.log(chalk.green("✓ Auto-start on boot configured (systemd user service)"));
+        console.log(chalk.gray("  Logs: journalctl --user-unit cc-router -f"));
+        console.log(chalk.gray("  Tip for headless servers: loginctl enable-linger $(whoami)"));
+    }
+    catch (err) {
+        console.log(chalk.yellow(`⚠ systemd setup issue: ${err.message}`));
+        console.log(chalk.gray(`  Enable manually: systemctl --user enable --now cc-router`));
+    }
+}
+async function uninstallLinux() {
+    if (!existsSync(SYSTEMD_SERVICE))
+        return;
+    try {
+        await execFileAsync("systemctl", ["--user", "stop", "cc-router"]);
+        await execFileAsync("systemctl", ["--user", "disable", "cc-router"]);
+    }
+    catch { /* may already be stopped */ }
+    try {
+        unlinkSync(SYSTEMD_SERVICE);
+    }
+    catch { /* ok */ }
+    try {
+        await execFileAsync("systemctl", ["--user", "daemon-reload"]);
+    }
+    catch { /* ok */ }
+    if (isServiceInstalled()) {
+        console.log(chalk.yellow("  ⚠ Service may still be installed — check manually"));
+    }
+    else {
+        console.log(chalk.green("  ✓ Auto-start removed"));
+    }
+}
+// ─── Windows Registry ────────────────────────────────────────────────────────
+function buildWindowsCommand() {
+    return `"${process.execPath}" "${CLI_ENTRY}" start --foreground`;
+}
+async function installWindows(serverMode) {
+    const cmd = serverMode
+        ? `cmd /c "set HOST=0.0.0.0 && set CC_ROUTER_SERVICE=1 && ${buildWindowsCommand()}"`
+        : `cmd /c "set CC_ROUTER_SERVICE=1 && ${buildWindowsCommand()}"`;
+    try {
+        await execFileAsync("reg", [
+            "add", WINDOWS_REG_KEY,
+            "/v", WINDOWS_REG_NAME,
+            "/t", "REG_SZ",
+            "/d", cmd,
+            "/f",
+        ]);
+        console.log(chalk.green("✓ Auto-start on login configured (Windows Registry)"));
+    }
+    catch (err) {
+        console.log(chalk.yellow(`⚠ Registry write failed: ${err.message}`));
+        console.log(chalk.gray(`  Add manually via Task Scheduler or registry editor.`));
+    }
+}
+async function uninstallWindows() {
+    try {
+        await execFileAsync("reg", [
+            "delete", WINDOWS_REG_KEY,
+            "/v", WINDOWS_REG_NAME,
+            "/f",
+        ]);
+    }
+    catch { /* not installed */ }
+    if (isServiceInstalled()) {
+        console.log(chalk.yellow("  ⚠ Service may still be installed — check manually"));
+    }
+    else {
+        console.log(chalk.green("  ✓ Auto-start removed"));
+    }
+}
+function isWindowsServiceInstalled() {
+    try {
+        execFileSync("reg", ["query", WINDOWS_REG_KEY, "/v", WINDOWS_REG_NAME]);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}