import { spawn, spawnSync, type ChildProcess } from "node:child_process";import { existsSync, statSync } from "node:fs";import { join, resolve } from "node:path"; export type CommandResult = { status: number | null; stdout: string; stderr: string;}; export type AsyncCommandResult = CommandResult & { timedOut: boolean;}; export type AsyncCommandOptions = { timeoutMs: number; onOutput?: (chunk: string) => void; onSpawn?: (child: ChildProcess) => void;}; const DEFAULT_TIMEOUT_MS = 3_000;const TIMEOUT_KILL_GRACE_MS = 5_000; export const runCommand = ( command: string, args: string[], timeoutMs = DEFAULT_TIMEOUT_MS,): CommandResult => { try { const result = spawnSync(command, args, { timeout: timeoutMs, env: process.env }); return { status: result.status, stdout: result.stdout ? result.stdout.toString("utf-8").trim() : "", stderr: result.stderr ? result.stderr.toString("utf-8").trim() : "", }; } catch (error) { return { status: null, stdout: "", stderr: error instanceof Error ? error.message : String(error), }; }}; export const runCommandAsync = ( command: string, args: string[], options: AsyncCommandOptions): Promise<AsyncCommandResult> => { return new Promise((resolveResult) => { const child = spawn(command, args, { env: process.env }); options.onSpawn?.(child); let stdout = ""; let stderr = ""; let timedOut = false; let forceKillTimer: ReturnType<typeof setTimeout> | null = null; const timeoutTimer = setTimeout(() => { timedOut = true; child.kill("SIGTERM"); forceKillTimer = setTimeout(() => child.kill("SIGKILL"), TIMEOUT_KILL_GRACE_MS); }, options.timeoutMs); const settle = (result: AsyncCommandResult): void => { clearTimeout(timeoutTimer); if (forceKillTimer) clearTimeout(forceKillTimer); resolveResult(result); }; child.stdout?.on("data", (data: Buffer) => { const chunk = data.toString("utf-8"); stdout += chunk; options.onOutput?.(chunk); }); child.stderr?.on("data", (data: Buffer) => { const chunk = data.toString("utf-8"); stderr += chunk; options.onOutput?.(chunk); }); child.on("error", (error) => { settle({ status: null, stdout: stdout.trim(), stderr: error.message, timedOut }); }); child.on("close", (code) => { settle({ status: code, stdout: stdout.trim(), stderr: stderr.trim(), timedOut }); }); });}; const isExecutableFile = (filePath: string): boolean => { try { const stats = statSync(filePath); return stats.isFile(); } catch { return false; }}; export const resolveBinary = (binaryName: string): string | null => { if (!binaryName) return null; if (binaryName.includes("/")) { const resolved = resolve(binaryName); return isExecutableFile(resolved) ? resolved : null; } const searchPaths: string[] = []; const runtimeOverride = process.env["VLLM_STUDIO_RUNTIME_BIN"]; const runtimeBin = runtimeOverride ?? (process.env["SNAP"] ? resolve(process.cwd(), "runtime", "bin") : null); if (runtimeBin && existsSync(runtimeBin)) { searchPaths.push(runtimeBin); } const pathValue = process.env["PATH"]; if (pathValue) { for (const entry of pathValue.split(":")) { if (entry) searchPaths.push(entry); } } const home = process.env["HOME"]; if (home) { searchPaths.push(join(home, ".local", "bin")); searchPaths.push(join(home, "bin")); } const user = process.env["USER"] ?? process.env["LOGNAME"]; if (user) { searchPaths.push(join("/home", user, ".local", "bin")); searchPaths.push(join("/home", user, "bin")); } for (const entry of searchPaths) { const candidate = join(entry, binaryName); if (isExecutableFile(candidate)) return candidate; } return null;};
import { spawn, spawnSync, type ChildProcess } from "node:child_process";
import { existsSync, statSync } from "node:fs";
import { join, resolve } from "node:path";
export type CommandResult = {
status: number | null;
stdout: string;
stderr: string;
};
export type AsyncCommandResult = CommandResult & {
timedOut: boolean;
};
export type AsyncCommandOptions = {
timeoutMs: number;
onOutput?: (chunk: string) => void;
onSpawn?: (child: ChildProcess) => void;
};
const DEFAULT_TIMEOUT_MS = 3_000;
const TIMEOUT_KILL_GRACE_MS = 5_000;
export const runCommand = (
command: string,
args: string[],
timeoutMs = DEFAULT_TIMEOUT_MS,
): CommandResult => {
try {
const result = spawnSync(command, args, { timeout: timeoutMs, env: process.env });
return {
status: result.status,
stdout: result.stdout ? result.stdout.toString("utf-8").trim() : "",
stderr: result.stderr ? result.stderr.toString("utf-8").trim() : "",
};
} catch (error) {
return {
status: null,
stdout: "",
stderr: error instanceof Error ? error.message : String(error),
};
}
};
export const runCommandAsync = (
command: string,
args: string[],
options: AsyncCommandOptions
): Promise<AsyncCommandResult> => {
return new Promise((resolveResult) => {
const child = spawn(command, args, { env: process.env });
options.onSpawn?.(child);
let stdout = "";
let stderr = "";
let timedOut = false;
let forceKillTimer: ReturnType<typeof setTimeout> | null = null;
const timeoutTimer = setTimeout(() => {
timedOut = true;
child.kill("SIGTERM");
forceKillTimer = setTimeout(() => child.kill("SIGKILL"), TIMEOUT_KILL_GRACE_MS);
}, options.timeoutMs);
const settle = (result: AsyncCommandResult): void => {
clearTimeout(timeoutTimer);
if (forceKillTimer) clearTimeout(forceKillTimer);
resolveResult(result);
};
child.stdout?.on("data", (data: Buffer) => {
const chunk = data.toString("utf-8");
stdout += chunk;
options.onOutput?.(chunk);
});
child.stderr?.on("data", (data: Buffer) => {
const chunk = data.toString("utf-8");
stderr += chunk;
options.onOutput?.(chunk);
});
child.on("error", (error) => {
settle({ status: null, stdout: stdout.trim(), stderr: error.message, timedOut });
});
child.on("close", (code) => {
settle({ status: code, stdout: stdout.trim(), stderr: stderr.trim(), timedOut });
});
});
};
const isExecutableFile = (filePath: string): boolean => {
try {
const stats = statSync(filePath);
return stats.isFile();
} catch {
return false;
}
};
export const resolveBinary = (binaryName: string): string | null => {
if (!binaryName) return null;
if (binaryName.includes("/")) {
const resolved = resolve(binaryName);
return isExecutableFile(resolved) ? resolved : null;
}
const searchPaths: string[] = [];
const runtimeOverride = process.env["VLLM_STUDIO_RUNTIME_BIN"];
const runtimeBin = runtimeOverride ?? (process.env["SNAP"] ? resolve(process.cwd(), "runtime", "bin") : null);
if (runtimeBin && existsSync(runtimeBin)) {
searchPaths.push(runtimeBin);
}
const pathValue = process.env["PATH"];
if (pathValue) {
for (const entry of pathValue.split(":")) {
if (entry) searchPaths.push(entry);
}
}
const home = process.env["HOME"];
if (home) {
searchPaths.push(join(home, ".local", "bin"));
searchPaths.push(join(home, "bin"));
}
const user = process.env["USER"] ?? process.env["LOGNAME"];
if (user) {
searchPaths.push(join("/home", user, ".local", "bin"));
searchPaths.push(join("/home", user, "bin"));
}
for (const entry of searchPaths) {
const candidate = join(entry, binaryName);
if (isExecutableFile(candidate)) return candidate;
}
return null;
};