1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
/**
* Daemon lifecycle helpers — spawn detection and lazy launch.
*
* The daemon is a standalone long-lived process that owns the single WebSocket
* to Chrome. pi instances connect to it via a Unix domain socket. This module
* provides the plumbing to detect a running daemon and spawn one if needed.
*
* Reference:
* - child_process.spawn detached: https://nodejs.org/docs/latest-v22.x/api/child_process.html#optionsdetached
*/
import { spawn, type ChildProcess } from "node:child_process";
import { access } from "node:fs/promises";
import { existsSync } from "node:fs";
import { fileURLToPath } from "node:url";
import { dirname, join } from "node:path";
import { DAEMON_SOCKET_PATH } from "./protocol";
const moduleDir = dirname(fileURLToPath(import.meta.url));
/**
* Check whether the daemon is running by testing if its Unix socket exists.
* The socket file is created by the daemon on startup and removed on shutdown.
*/
export const isDaemonRunning = async (): Promise<boolean> => {
try {
await access(DAEMON_SOCKET_PATH);
return true;
} catch {
return false;
}
};
/**
* Spawn the daemon as a detached child process. The daemon will continue
* running after the parent pi process exits.
*
* The daemon entry point is resolved relative to this module. When running
* from source (tsx), the entry is src/daemon/index.ts. When running compiled,
* it's dist/daemon/index.js.
*
* Returns the child process handle (caller should call child.unref()).
*/
export const spawnDaemon = (): ChildProcess | null => {
// Resolve the daemon entry point and tsx binary.
const daemonScript = join(moduleDir, "index.ts");
const tsxBin = join(moduleDir, "..", "..", "node_modules", ".bin", "tsx");
// Use tsx to run the daemon TypeScript source directly.
// The spawed process inherits the parent's env so module resolution works.
// Fall back to `npx tsx` if the local binary isn't available.
const cmd = existsSync(tsxBin) ? tsxBin : "npx";
const args = existsSync(tsxBin) ? [daemonScript] : ["tsx", daemonScript];
const child = spawn(cmd, args, {
detached: true,
stdio: "ignore",
env: { ...process.env },
});
child.unref();
return child;
};
/**
* Ensure the daemon is running. If not, spawn it and wait (up to timeoutMs)
* for the socket file to appear.
*
* Returns true if the daemon is running (or was successfully spawned).
*/
export const ensureDaemon = async (timeoutMs = 10_000): Promise<boolean> => {
if (await isDaemonRunning()) return true;
const child = spawnDaemon();
if (!child) return false;
// Poll for the socket file to appear
const deadline = Date.now() + timeoutMs;
while (Date.now() < deadline) {
if (await isDaemonRunning()) return true;
await new Promise((r) => setTimeout(r, 200));
}
return false;
};
|
