feat(S-302): Test-drive and implement job lifecycle subcommands
This commit is contained in:
@@ -1,7 +1,14 @@
|
||||
import minimist from "minimist";
|
||||
import { detectClients as realDetectClients } from "./detect.js";
|
||||
import { executePrompt as realExecutePrompt } from "./execute.js";
|
||||
import { startJob as realStartJob } from "./jobs.js";
|
||||
import {
|
||||
startJob as realStartJob,
|
||||
getJob as realGetJob,
|
||||
getJobResult as realGetJobResult,
|
||||
cancelJob as realCancelJob,
|
||||
listJobs as realListJobs,
|
||||
cleanupJobs as realCleanupJobs,
|
||||
} from "./jobs.js";
|
||||
import { resolveClient as realResolveClient } from "./dispatch.js";
|
||||
import { resolveConfig as realResolveConfig } from "./config.js";
|
||||
import { CLIENT_NAMES } from "./constants.js";
|
||||
@@ -11,6 +18,7 @@ import {
|
||||
type ExecResult,
|
||||
type DebugInfo,
|
||||
type Job,
|
||||
type JobStatus,
|
||||
ClientNotFoundError,
|
||||
} from "./types.js";
|
||||
|
||||
@@ -33,6 +41,29 @@ export interface CliDeps {
|
||||
resolveConfig?: () => { paths: Partial<Record<ClientName, string>>; defaultClient?: ClientName; timeout?: number };
|
||||
stdoutWrite?: (chunk: string) => void;
|
||||
stderrWrite?: (chunk: string) => void;
|
||||
getJob?: (jobId: string) => Job;
|
||||
getJobResult?: (jobId: string) => ExecResult;
|
||||
cancelJob?: (jobId: string) => void;
|
||||
listJobs?: (options?: { filter?: JobStatus }) => Job[];
|
||||
cleanupJobs?: (options?: { maxAgeMs?: number }) => void;
|
||||
}
|
||||
|
||||
function parseMaxAge(value: string): number | null {
|
||||
const match = value.match(/^(\d+(?:\.\d+)?)\s*([hmsd]?)$/i);
|
||||
if (!match) return null;
|
||||
const num = parseFloat(match[1]);
|
||||
if (!Number.isFinite(num) || num < 0) return null;
|
||||
const unit = match[2].toLowerCase();
|
||||
const multipliers: Record<string, number> = {
|
||||
s: 1000,
|
||||
m: 60 * 1000,
|
||||
h: 60 * 60 * 1000,
|
||||
d: 24 * 60 * 60 * 1000,
|
||||
"": 60 * 60 * 1000,
|
||||
};
|
||||
const multiplier = multipliers[unit];
|
||||
if (multiplier === undefined) return null;
|
||||
return num * multiplier;
|
||||
}
|
||||
|
||||
function printHelp(): void {
|
||||
@@ -58,13 +89,18 @@ export async function main(
|
||||
const resolveConfig = deps.resolveConfig ?? realResolveConfig;
|
||||
const stdoutWrite = deps.stdoutWrite ?? ((c: string) => process.stdout.write(c));
|
||||
const stderrWrite = deps.stderrWrite ?? ((c: string) => process.stderr.write(c));
|
||||
const getJob = deps.getJob ?? realGetJob;
|
||||
const getJobResult = deps.getJobResult ?? realGetJobResult;
|
||||
const cancelJob = deps.cancelJob ?? realCancelJob;
|
||||
const listJobs = deps.listJobs ?? realListJobs;
|
||||
const cleanupJobs = deps.cleanupJobs ?? realCleanupJobs;
|
||||
|
||||
const rawArgs = argv.slice(2);
|
||||
const parseArgs =
|
||||
rawArgs[0]?.includes("cli.ts") ? rawArgs.slice(1) : rawArgs;
|
||||
|
||||
const args = minimist(parseArgs, {
|
||||
string: ["client", "prompt", "timeout"],
|
||||
string: ["client", "prompt", "timeout", "max-age"],
|
||||
boolean: ["json", "text", "help", "debug", "sync"],
|
||||
alias: { h: "help" },
|
||||
});
|
||||
@@ -265,6 +301,227 @@ export async function main(
|
||||
}
|
||||
}
|
||||
|
||||
if (command === "start") {
|
||||
const client = args.client as ClientName | undefined;
|
||||
const prompt = args.prompt as string | undefined;
|
||||
|
||||
if (!client || !CLIENT_NAMES.includes(client)) {
|
||||
const message = !client
|
||||
? "--client is required"
|
||||
: `Unknown client: ${client}`;
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!prompt) {
|
||||
const message = "--prompt is required";
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
const config = resolveConfig();
|
||||
const parsedTimeout =
|
||||
typeof args.timeout === "string" ? Number(args.timeout) : undefined;
|
||||
const timeoutMs =
|
||||
Number.isFinite(parsedTimeout) ? parsedTimeout : config.timeout;
|
||||
|
||||
try {
|
||||
const job = await startJob(client, prompt, {
|
||||
timeoutMs,
|
||||
debug,
|
||||
onDebug: debug ? (info) => stderrWrite(JSON.stringify(info) + "\n") : undefined,
|
||||
});
|
||||
if (jsonMode) {
|
||||
console.log(JSON.stringify({ jobId: job.id, client: job.client, status: job.status }, null, 2));
|
||||
} else {
|
||||
console.log(`Job ${job.id} started (${job.client}): ${job.status}`);
|
||||
}
|
||||
return 0;
|
||||
} catch (err) {
|
||||
const message = err instanceof Error ? err.message : String(err);
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(message);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (command === "status") {
|
||||
const jobId = args._[1];
|
||||
if (!jobId) {
|
||||
const message = "job-id is required";
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
try {
|
||||
const job = getJob(jobId);
|
||||
if (jsonMode) {
|
||||
console.log(JSON.stringify(job, null, 2));
|
||||
} else {
|
||||
console.log(`Job ${jobId}: ${job.status}`);
|
||||
}
|
||||
return 0;
|
||||
} catch (err) {
|
||||
const message = err instanceof Error ? err.message : String(err);
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(message);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (command === "results") {
|
||||
const jobId = args._[1];
|
||||
if (!jobId) {
|
||||
const message = "job-id is required";
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
try {
|
||||
const result = getJobResult(jobId);
|
||||
if (jsonMode) {
|
||||
console.log(JSON.stringify(result, null, 2));
|
||||
} else {
|
||||
if (result.stdout) stdoutWrite(result.stdout);
|
||||
if (result.stderr) stderrWrite(result.stderr);
|
||||
}
|
||||
return 0;
|
||||
} catch (err) {
|
||||
const message = err instanceof Error ? err.message : String(err);
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(message);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (command === "cancel") {
|
||||
const jobId = args._[1];
|
||||
if (!jobId) {
|
||||
const message = "job-id is required";
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
try {
|
||||
const job = getJob(jobId);
|
||||
if (job.status !== "running") {
|
||||
const message = `Job is not running (status: ${job.status})`;
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
cancelJob(jobId);
|
||||
if (jsonMode) {
|
||||
console.log(JSON.stringify({ jobId, cancelled: true }, null, 2));
|
||||
} else {
|
||||
console.log(`Job ${jobId} cancelled`);
|
||||
}
|
||||
return 0;
|
||||
} catch (err) {
|
||||
const message = err instanceof Error ? err.message : String(err);
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(message);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (command === "list-jobs") {
|
||||
try {
|
||||
const filter = args.status as JobStatus | undefined;
|
||||
const jobs = listJobs({ filter });
|
||||
if (jsonMode) {
|
||||
console.log(JSON.stringify(jobs, null, 2));
|
||||
} else {
|
||||
for (const job of jobs) {
|
||||
console.log(`${job.id} (${job.client}): ${job.status}`);
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
} catch (err) {
|
||||
const message = err instanceof Error ? err.message : String(err);
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(message);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (command === "cleanup-jobs") {
|
||||
const maxAgeRaw = args["max-age"] as string | undefined;
|
||||
let maxAgeMs: number | undefined;
|
||||
if (maxAgeRaw !== undefined) {
|
||||
const parsed = parseMaxAge(maxAgeRaw);
|
||||
if (parsed === null) {
|
||||
const message = "Invalid --max-age format. Use: <number>[h|m|s|d], e.g. 24h";
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(`Error: ${message}`);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
maxAgeMs = parsed;
|
||||
}
|
||||
|
||||
try {
|
||||
const jobsBefore = listJobs();
|
||||
cleanupJobs({ maxAgeMs });
|
||||
const jobsAfter = listJobs();
|
||||
const count = jobsBefore.length - jobsAfter.length;
|
||||
if (jsonMode) {
|
||||
console.log(JSON.stringify({ count }, null, 2));
|
||||
} else {
|
||||
console.log(`Cleaned ${count} jobs`);
|
||||
}
|
||||
return 0;
|
||||
} catch (err) {
|
||||
const message = err instanceof Error ? err.message : String(err);
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
} else {
|
||||
console.error(message);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
const message = `Unknown command: ${command}`;
|
||||
if (jsonMode) {
|
||||
console.error(JSON.stringify({ error: message }, null, 2));
|
||||
|
||||
Reference in New Issue
Block a user