Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
115 changes: 115 additions & 0 deletions src/commands/tdd-daemon-process.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,115 @@
import { spawn } from 'node:child_process';
import { existsSync, readFileSync } from 'node:fs';
import { join } from 'node:path';

let defaultTimers = { setTimeout, clearTimeout };

function wait(ms, timers = defaultTimers) {
return new Promise(resolve => {
timers.setTimeout(resolve, ms);
});
}

function isProcessRunning(pid) {
try {
process.kill(pid, 0);
return true;
} catch {
return false;
}
}

function parsePositiveInteger(value) {
let text = String(value).trim();
if (!/^\d+$/.test(text)) {
return null;
}

let number = Number(text);
return Number.isSafeInteger(number) && number > 0 ? number : null;
}

export function readDaemonPidFile(pidFile, deps = {}) {
let fileExists = deps.existsSync || existsSync;
let readFile = deps.readFileSync || readFileSync;

if (!fileExists(pidFile)) {
return null;
}

try {
return parsePositiveInteger(readFile(pidFile, 'utf8'));
} catch {
return null;
}
}

export async function findDaemonPidByPort(port, { spawnProcess = spawn } = {}) {
try {
let lsofProcess = spawnProcess('lsof', ['-ti', `:${port}`], {
stdio: 'pipe',
});

let lsofOutput = '';
lsofProcess.stdout.on('data', data => {
lsofOutput += data.toString();
});

return await new Promise(resolve => {
lsofProcess.on('close', code => {
if (code === 0 && lsofOutput.trim()) {
let foundPid = parsePositiveInteger(lsofOutput.trim().split('\n')[0]);
resolve(foundPid);
return;
}

resolve(null);
});

lsofProcess.on('error', () => {
resolve(null);
});
});
} catch {
return null;
}
}

export async function resolveDaemonPid({
port,
pidFile = join(process.cwd(), '.vizzly', 'server.pid'),
readPid = readDaemonPidFile,
findByPort = findDaemonPidByPort,
fileDeps = {},
} = {}) {
let pid = readPid(pidFile, fileDeps);
if (pid) {
return pid;
}

return await findByPort(port);
}

export async function waitForProcessExit(
pid,
{
timeoutMs = 2000,
intervalMs = 100,
processRunning = isProcessRunning,
timers = defaultTimers,
} = {}
) {
let elapsedMs = 0;

while (elapsedMs < timeoutMs) {
if (!processRunning(pid)) {
return true;
}

let nextDelay = Math.min(intervalMs, timeoutMs - elapsedMs);
await wait(nextDelay, timers);
elapsedMs += nextDelay;
}

return !processRunning(pid);
}
112 changes: 8 additions & 104 deletions src/commands/tdd-daemon.js
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,14 @@ import { getServerRegistry } from '../tdd/server-registry.js';
import { withTimeout } from '../utils/async-utils.js';
import * as output from '../utils/output.js';
import { tddCommand } from './tdd.js';
import { resolveDaemonPid, waitForProcessExit } from './tdd-daemon-process.js';

export {
findDaemonPidByPort,
readDaemonPidFile,
resolveDaemonPid,
waitForProcessExit,
} from './tdd-daemon-process.js';

let defaultTimers = { setTimeout, clearTimeout };

Expand Down Expand Up @@ -143,86 +151,6 @@ function wait(ms, timers = defaultTimers) {
});
}

function isProcessRunning(pid) {
try {
process.kill(pid, 0);
return true;
} catch {
return false;
}
}

function parsePositiveInteger(value) {
let text = String(value).trim();
if (!/^\d+$/.test(text)) {
return null;
}

let number = Number(text);
return Number.isSafeInteger(number) && number > 0 ? number : null;
}

export function readDaemonPidFile(pidFile, deps = {}) {
let fileExists = deps.existsSync || existsSync;
let readFile = deps.readFileSync || readFileSync;

if (!fileExists(pidFile)) {
return null;
}

try {
return parsePositiveInteger(readFile(pidFile, 'utf8'));
} catch {
return null;
}
}

export async function findDaemonPidByPort(port, { spawnProcess = spawn } = {}) {
try {
let lsofProcess = spawnProcess('lsof', ['-ti', `:${port}`], {
stdio: 'pipe',
});

let lsofOutput = '';
lsofProcess.stdout.on('data', data => {
lsofOutput += data.toString();
});

return await new Promise(resolve => {
lsofProcess.on('close', code => {
if (code === 0 && lsofOutput.trim()) {
let foundPid = parsePositiveInteger(lsofOutput.trim().split('\n')[0]);
resolve(foundPid);
return;
}

resolve(null);
});

lsofProcess.on('error', () => {
resolve(null);
});
});
} catch {
return null;
}
}

export async function resolveDaemonPid({
port,
pidFile = getLocalDaemonFiles().pidFile,
readPid = readDaemonPidFile,
findByPort = findDaemonPidByPort,
fileDeps = {},
} = {}) {
let pid = readPid(pidFile, fileDeps);
if (pid) {
return pid;
}

return await findByPort(port);
}

export function buildDaemonChildArgs({
entrypoint = process.argv[1],
port,
Expand Down Expand Up @@ -355,30 +283,6 @@ export async function waitForServerRunning(
return false;
}

export async function waitForProcessExit(
pid,
{
timeoutMs = 2000,
intervalMs = 100,
processRunning = isProcessRunning,
timers = defaultTimers,
} = {}
) {
let elapsedMs = 0;

while (elapsedMs < timeoutMs) {
if (!processRunning(pid)) {
return true;
}

let nextDelay = Math.min(intervalMs, timeoutMs - elapsedMs);
await wait(nextDelay, timers);
elapsedMs += nextDelay;
}

return !processRunning(pid);
}

/**
* Start TDD server in daemon mode
* @param {Object} options - Command options
Expand Down
Loading