nolo-cli 0.1.41 → 0.1.43

package/taskRunCommand.ts

@@ -1,265 +0,0 @@
-import { DEFAULT_NOLO_SERVER_URL } from "./defaultServer";
-import { existsSync, readFileSync } from "node:fs";
-import { homedir } from "node:os";
-import { join } from "node:path";
-
-type EnvLike = Record<string, string | undefined>;
-type OutputLike = { write(chunk: string): unknown };
-
-type TaskRunCommandDeps = {
-  env?: EnvLike;
-  output?: OutputLike;
-  fetchImpl?: typeof fetch;
-};
-
-function parseEnvFile(path: string): EnvLike {
-  if (!path || !existsSync(path)) return {};
-  const parsed: EnvLike = {};
-  for (const rawLine of readFileSync(path, "utf8").split(/\r?\n/)) {
-    const line = rawLine.trim();
-    if (!line || line.startsWith("#")) continue;
-    const index = line.indexOf("=");
-    if (index <= 0) continue;
-    const key = line.slice(0, index).trim();
-    let value = line.slice(index + 1).trim();
-    if (
-      (value.startsWith('"') && value.endsWith('"')) ||
-      (value.startsWith("'") && value.endsWith("'"))
-    ) {
-      value = value.slice(1, -1);
-    }
-    if (key) parsed[key] = value;
-  }
-  return parsed;
-}
-
-function loadTaskRunEnv(env: EnvLike, options: { includeDefaultEnvFiles?: boolean } = {}) {
-  const envFileCandidates = [
-    env.NOLO_TASK_RUN_ENV_FILE,
-    env.NOLO_MACHINE_ENV_FILE,
-    ...(options.includeDefaultEnvFiles
-      ? ["/etc/nolo/machine-runner.env", join(homedir(), ".nolo", "machine-runner.env")]
-      : []),
-  ].filter((path): path is string => Boolean(path));
-  const fileEnv = Object.assign({}, ...envFileCandidates.map(parseEnvFile));
-  return { ...fileEnv, ...env };
-}
-
-function readOption(args: string[], flag: string) {
-  for (let i = 0; i < args.length; i++) {
-    const value = args[i];
-    if (value === flag) return args[i + 1] ?? "";
-    if (value.startsWith(`${flag}=`)) return value.slice(flag.length + 1);
-  }
-  return "";
-}
-
-function csv(raw: string) {
-  return raw.split(",").map((item) => item.trim()).filter(Boolean);
-}
-
-function optionalJson(args: string[], flag: string) {
-  const value = optional(args, flag);
-  if (!value) return undefined;
-  try {
-    return JSON.parse(value);
-  } catch (error) {
-    throw new Error(`${flag} must be valid JSON: ${error instanceof Error ? error.message : String(error)}`);
-  }
-}
-
-function resolveServerUrl(args: string[], env: EnvLike) {
-  return (
-    readOption(args, "--server-url") ||
-    readOption(args, "--server") ||
-    readOption(args, "--base-url") ||
-    env.NOLO_SERVER_URL ||
-    env.NOLO_SERVER ||
-    env.BASE_URL ||
-    DEFAULT_NOLO_SERVER_URL
-  ).replace(/\/+$/, "");
-}
-
-function resolveAuthToken(args: string[], env: EnvLike) {
-  return (
-    readOption(args, "--machine-key") ||
-    readOption(args, "--token") ||
-    env.NOLO_TASK_RUN_TOKEN ||
-    env.NOLO_MACHINE_API_KEY ||
-    env.AUTH_TOKEN ||
-    env.AUTH ||
-    ""
-  );
-}
-
-function required(args: string[], flag: string) {
-  const value = readOption(args, flag).trim();
-  if (!value) throw new Error(`${flag} is required`);
-  return value;
-}
-
-function optional(args: string[], flag: string) {
-  const value = readOption(args, flag).trim();
-  return value || undefined;
-}
-
-function buildBody(command: string, args: string[]) {
-  const dbKey = required(args, "--row-dbkey");
-  if (command === "read-context" || command === "readTaskContext") {
-    return { action: "readTaskContext", dbKey };
-  }
-  if (command === "claim-work-item" || command === "claimWorkItem") {
-    return {
-      action: "claimWorkItem",
-      dbKey,
-      workItemId: required(args, "--work-item-id"),
-      agentKey: required(args, "--agent-key"),
-      role: optional(args, "--role"),
-    };
-  }
-  if (command === "record-agent-run" || command === "recordAgentRun") {
-    const toolTraceSummary = optionalJson(args, "--tool-trace-summary");
-    return {
-      action: "recordAgentRun",
-      dbKey,
-      ...(optional(args, "--run-id") ? { runId: optional(args, "--run-id") } : {}),
-      agentKey: required(args, "--agent-key"),
-      ...(optional(args, "--role") ? { role: optional(args, "--role") } : {}),
-      runStatus: required(args, "--run-status"),
-      ...(optional(args, "--work-item-id") ? { workItemId: optional(args, "--work-item-id") } : {}),
-      ...(optional(args, "--dialog-id") ? { dialogId: optional(args, "--dialog-id") } : {}),
-      ...(optional(args, "--result-summary") ? { resultSummary: optional(args, "--result-summary") } : {}),
-      ...(optional(args, "--error-code") ? { errorCode: optional(args, "--error-code") } : {}),
-      ...(optional(args, "--tools-used") ? { toolsUsed: csv(optional(args, "--tools-used")!) } : {}),
-      ...(toolTraceSummary ? { toolTraceSummary } : {}),
-    };
-  }
-  if (command === "set-blocker" || command === "setBlocker") {
-    return {
-      action: "setBlocker",
-      dbKey,
-      workItemId: optional(args, "--work-item-id"),
-      blocker: {
-        layer: required(args, "--layer"),
-        code: required(args, "--code"),
-        message: required(args, "--message"),
-      },
-    };
-  }
-  if (command === "clear-blocker" || command === "clearBlocker") {
-    return {
-      action: "clearBlocker",
-      dbKey,
-      blockerId: required(args, "--blocker-id"),
-      workItemId: optional(args, "--work-item-id"),
-    };
-  }
-  if (command === "submit-outcome" || command === "submitOutcome") {
-    return {
-      action: "submitOutcome",
-      dbKey,
-      workItemId: required(args, "--work-item-id"),
-      summary: required(args, "--summary"),
-      ...(optional(args, "--dialog-id") ? { dialogId: optional(args, "--dialog-id") } : {}),
-      ...(optional(args, "--artifact-ids") ? { artifactIds: csv(optional(args, "--artifact-ids")!) } : {}),
-    };
-  }
-  if (command === "request-review" || command === "requestReview") {
-    return {
-      action: "requestReview",
-      dbKey,
-      workItemId: optional(args, "--work-item-id"),
-      reviewerAgentKey: required(args, "--reviewer-agent-key"),
-      ...(optional(args, "--dialog-id") ? { dialogId: optional(args, "--dialog-id") } : {}),
-      ...(optional(args, "--message") ? { message: optional(args, "--message") } : {}),
-      ...(optional(args, "--artifact-ids") ? { artifactIds: csv(optional(args, "--artifact-ids")!) } : {}),
-    };
-  }
-  if (command === "record-review" || command === "recordReview") {
-    return {
-      action: "recordReview",
-      dbKey,
-      ...(optional(args, "--work-item-id") ? { workItemId: optional(args, "--work-item-id") } : {}),
-      reviewStatus: required(args, "--review-status"),
-      ...(optional(args, "--reviewer-agent-key")
-        ? { reviewerAgentKey: optional(args, "--reviewer-agent-key") }
-        : {}),
-      ...(optional(args, "--dialog-id") ? { dialogId: optional(args, "--dialog-id") } : {}),
-      ...(optional(args, "--findings") ? { findings: optional(args, "--findings") } : {}),
-      ...(optional(args, "--artifact-ids") ? { artifactIds: csv(optional(args, "--artifact-ids")!) } : {}),
-    };
-  }
-  if (command === "close-work-item" || command === "complete-work-item" || command === "closeWorkItem" || command === "completeWorkItem") {
-    return {
-      action: "closeWorkItem",
-      dbKey,
-      workItemId: required(args, "--work-item-id"),
-      summary: required(args, "--summary"),
-      ...(optional(args, "--reason") ? { reason: optional(args, "--reason") } : {}),
-      ...(optional(args, "--dialog-id") ? { dialogId: optional(args, "--dialog-id") } : {}),
-      ...(optional(args, "--artifact-ids") ? { artifactIds: csv(optional(args, "--artifact-ids")!) } : {}),
-    };
-  }
-  throw new Error(`Unknown task-run command: ${command}`);
-}
-
-function usage() {
-  return [
-    "Usage:",
-    "  nolo task-run read-context --row-dbkey <row>",
-    "  nolo task-run claim-work-item --row-dbkey <row> --work-item-id <id> --agent-key <agent> [--role <role>]",
-    "  nolo task-run record-agent-run --row-dbkey <row> --agent-key <agent> --run-status <status> [--tools-used <csv>]",
-    "  nolo task-run set-blocker --row-dbkey <row> [--work-item-id <id>] --layer <layer> --code <code> --message <message>",
-    "  nolo task-run clear-blocker --row-dbkey <row> --blocker-id <id> [--work-item-id <id>]",
-    "  nolo task-run submit-outcome --row-dbkey <row> --work-item-id <id> --summary <summary> [--dialog-id <id>]",
-    "  nolo task-run request-review --row-dbkey <row> --reviewer-agent-key <agent> [--work-item-id <id>]",
-    "  nolo task-run record-review --row-dbkey <row> [--work-item-id <id>] --review-status <passed|needs_changes|blocked> [--findings <text>]",
-    "  nolo task-run close-work-item --row-dbkey <row> --work-item-id <id> --summary <summary>",
-    "",
-  ].join("\n");
-}
-
-export async function runTaskRunCommand(args: string[], deps: TaskRunCommandDeps = {}) {
-  const env = loadTaskRunEnv(deps.env ?? process.env, { includeDefaultEnvFiles: deps.env === undefined });
-  const output = deps.output ?? process.stdout;
-  const fetchImpl = deps.fetchImpl ?? fetch;
-  const command = args[0];
-  if (!command || command === "--help" || command === "-h") {
-    output.write(usage());
-    return 0;
-  }
-
-  let body: Record<string, unknown>;
-  try {
-    body = buildBody(command, args.slice(1));
-  } catch (error) {
-    output.write(`[nolo] task-run failed: ${error instanceof Error ? error.message : String(error)}\n`);
-    return 1;
-  }
-
-  const authToken = resolveAuthToken(args, env);
-  if (!authToken) {
-    output.write("[nolo] task-run failed: AUTH_TOKEN or NOLO_MACHINE_API_KEY is required.\n");
-    return 1;
-  }
-
-  const serverUrl = resolveServerUrl(args, env);
-  let response: Response;
-  try {
-    response = await fetchImpl(`${serverUrl}/api/task-run`, {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${authToken}`,
-        "Content-Type": "application/json",
-      },
-      body: JSON.stringify(body),
-    });
-  } catch (error) {
-    output.write(`[nolo] task-run failed: ${error instanceof Error ? error.message : String(error)}\n`);
-    return 1;
-  }
-
-  const text = await response.text();
-  output.write(text.trim() ? `${text.trim()}\n` : "");
-  return response.ok ? 0 : 1;
-}