@tracemarketplace/cli 0.0.18 → 0.0.19

package/src/submit-worker-state.ts

@@ -0,0 +1,127 @@
+import { existsSync, mkdirSync, readdirSync, readFileSync, unlinkSync, writeFileSync } from "fs";
+import { join } from "path";
+import { getSubmitLogPath, getSubmitWorkerStateDir } from "./config.js";
+
+export interface SubmitWorkerState {
+  pid: number;
+  startedAt: string;
+  readyChunkCount: number;
+  readySessionCount: number;
+  logPath: string;
+}
+
+interface SubmitWorkerStateOptions {
+  dir?: string;
+}
+
+export function isProcessRunning(pid: number): boolean {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch (err) {
+    return (err as NodeJS.ErrnoException).code === "EPERM";
+  }
+}
+
+export function writeSubmitWorkerState(
+  profile: string,
+  state: SubmitWorkerState,
+  opts: SubmitWorkerStateOptions = {},
+): void {
+  const dir = resolveStateDir(profile, opts.dir);
+  mkdirSync(dir, { recursive: true });
+  writeFileSync(join(dir, `${state.pid}.json`), JSON.stringify(state, null, 2) + "\n", "utf-8");
+}
+
+export function removeSubmitWorkerState(
+  profile: string,
+  pid: number,
+  opts: SubmitWorkerStateOptions = {},
+): void {
+  const dir = resolveStateDir(profile, opts.dir);
+  const path = join(dir, `${pid}.json`);
+  if (!existsSync(path)) {
+    return;
+  }
+
+  try {
+    unlinkSync(path);
+  } catch {}
+}
+
+export function listActiveSubmitWorkers(
+  profile: string,
+  opts: SubmitWorkerStateOptions = {},
+): SubmitWorkerState[] {
+  const dir = resolveStateDir(profile, opts.dir);
+  if (!existsSync(dir)) {
+    return [];
+  }
+
+  const workers: SubmitWorkerState[] = [];
+  for (const entry of readdirSync(dir)) {
+    if (!entry.endsWith(".json")) {
+      continue;
+    }
+
+    const path = join(dir, entry);
+    const parsed = readSubmitWorkerStateFile(path);
+    if (!parsed || !isProcessRunning(parsed.pid)) {
+      try {
+        unlinkSync(path);
+      } catch {}
+      continue;
+    }
+
+    workers.push(parsed);
+  }
+
+  return workers.sort((a, b) => a.startedAt.localeCompare(b.startedAt));
+}
+
+export function createSubmitWorkerState(
+  profile: string,
+  pid: number,
+  startedAt: string,
+  readyChunkCount: number,
+  readySessionCount: number,
+): SubmitWorkerState {
+  return {
+    pid,
+    startedAt,
+    readyChunkCount,
+    readySessionCount,
+    logPath: getSubmitLogPath(profile),
+  };
+}
+
+function resolveStateDir(profile: string, dir?: string): string {
+  return dir ?? getSubmitWorkerStateDir(profile);
+}
+
+function readSubmitWorkerStateFile(path: string): SubmitWorkerState | null {
+  try {
+    const value = JSON.parse(readFileSync(path, "utf-8")) as Partial<SubmitWorkerState>;
+    if (
+      typeof value.pid !== "number"
+      || !Number.isInteger(value.pid)
+      || value.pid <= 0
+      || typeof value.startedAt !== "string"
+      || typeof value.readyChunkCount !== "number"
+      || typeof value.readySessionCount !== "number"
+      || typeof value.logPath !== "string"
+    ) {
+      return null;
+    }
+
+    return {
+      pid: value.pid,
+      startedAt: value.startedAt,
+      readyChunkCount: value.readyChunkCount,
+      readySessionCount: value.readySessionCount,
+      logPath: value.logPath,
+    };
+  } catch {
+    return null;
+  }
+}