nolo-cli 0.1.23 → 0.1.29

package/agentRunParsing.ts

@@ -0,0 +1,279 @@
+import { existsSync, readFileSync } from "node:fs";
+import { extname, resolve } from "node:path";
+
+import { parseSkillDocProtocol, type WorkflowReferenceConfig } from "./ai/skills/skillDocProtocol";
+
+import { resolveCliAgentKeyInput } from "./agentAliases";
+import type { AgentRuntimeRequestedMode } from "./agentRuntimeLocal";
+import type { TaskRunPromptContext } from "./client/taskRunPrompt";
+
+type ParseAgentRunArgsOptions = {
+  readTextFile?: (path: string) => string;
+};
+
+export type ParsedAgentRunArgs = {
+  agentKey: string;
+  message: string;
+  imageUrls: string[];
+  allowShell: boolean;
+  runtimeMode?: AgentRuntimeRequestedMode;
+  continueDialogId?: string;
+  spaceId?: string;
+  category?: string;
+  inheritedFromDialogKey?: string;
+  parentDialogId?: string;
+  background: boolean;
+  noStream: boolean;
+  noDefaultTestRoot: boolean;
+  cwd?: string;
+  maxToolRounds?: number;
+  timeoutMs?: number;
+  traceTools: boolean;
+  useTaskWorktree: boolean;
+  taskRunContext?: TaskRunPromptContext;
+  workflowRef?: string;
+};
+
+export type ResolvedWorkflowReference = {
+  ref: string;
+  content: string;
+  config?: Partial<WorkflowReferenceConfig>;
+};
+
+const DEFAULT_TEST_ROOT_DIALOG_ID = "01KN6V3MVEQA4Q1PNKVM2W7YY8";
+const DEFAULT_TEST_ROOT_DIALOG_KEY = `dialog-b2e06f801f-${DEFAULT_TEST_ROOT_DIALOG_ID}`;
+const DEFAULT_TEST_DIALOG_CATEGORY_ID = "test-dialogs";
+
+function readFlagValue(args: string[], flag: string) {
+  const index = args.indexOf(flag);
+  if (index < 0) return undefined;
+  return args[index + 1];
+}
+
+function readRepeatedFlagValues(args: string[], flag: string) {
+  const values: string[] = [];
+  for (let index = 0; index < args.length; index += 1) {
+    if (args[index] === flag && args[index + 1]) {
+      values.push(args[index + 1]);
+      index += 1;
+    }
+  }
+  return values;
+}
+
+function runtimeModeFromArgs(args: string[]): AgentRuntimeRequestedMode | undefined {
+  if (args.includes("--local")) return "local";
+  if (args.includes("--server")) return "server";
+  if (args.includes("--auto")) return "auto";
+  return undefined;
+}
+
+function parseNonNegativeInteger(value: string | undefined) {
+  if (!value) return undefined;
+  const parsed = Number(value);
+  if (!Number.isFinite(parsed) || parsed < 0) return undefined;
+  return Math.floor(parsed);
+}
+
+function parsePositiveInteger(value: string | undefined) {
+  if (!value) return undefined;
+  const parsed = Number(value);
+  if (!Number.isFinite(parsed) || parsed <= 0) return undefined;
+  return Math.floor(parsed);
+}
+
+function positionalArgs(args: string[]) {
+  const values: string[] = [];
+  const valuelessFlags = new Set([
+    "--local",
+    "--server",
+    "--auto",
+    "--dangerously-allow-shell",
+    "--worktree-task",
+    "--trace-tools",
+    "--bg",
+    "--no-stream",
+    "--debug",
+    "--no-default-test-root",
+  ]);
+  for (let index = 0; index < args.length; index += 1) {
+    const arg = args[index];
+    if (valuelessFlags.has(arg)) continue;
+    if (arg.startsWith("--")) {
+      index += 1;
+      continue;
+    }
+    values.push(arg);
+  }
+  return values;
+}
+
+function parseDialogReference(rawInput: string) {
+  const normalized = rawInput.trim();
+  if (normalized.startsWith("dialog-")) {
+    const parts = normalized.split("-");
+    return {
+      dialogKey: normalized,
+      dialogId: parts.at(-1) ?? normalized,
+    };
+  }
+  const dialogMatch = normalized.match(/dialog-([a-zA-Z0-9]+)-([a-zA-Z0-9]+)/);
+  if (dialogMatch) {
+    return {
+      dialogKey: `dialog-${dialogMatch[1]}-${dialogMatch[2]}`,
+      dialogId: dialogMatch[2],
+    };
+  }
+  return {
+    dialogId: normalized,
+  };
+}
+
+function workflowRefToCandidatePath(cwd: string, ref: string) {
+  const normalized = ref.trim();
+  if (!normalized) return "";
+  if (normalized.endsWith(".md") || normalized.includes("/") || normalized.includes("\\")) {
+    const directPath = resolve(cwd, normalized);
+    if (existsSync(directPath)) return directPath;
+  }
+  const fileName = normalized.replace(/[^a-zA-Z0-9\u4e00-\u9fa5]+/g, "-").replace(/^-+|-+$/g, "");
+  return resolve(cwd, "docs", "workflows", `${fileName}.md`);
+}
+
+function mimeTypeForPath(path: string) {
+  switch (extname(path).toLowerCase()) {
+    case ".jpg":
+    case ".jpeg":
+      return "image/jpeg";
+    case ".webp":
+      return "image/webp";
+    case ".gif":
+      return "image/gif";
+    default:
+      return "image/png";
+  }
+}
+
+export function parseAgentRunArgs(
+  args: string[],
+  options: ParseAgentRunArgsOptions = {}
+): ParsedAgentRunArgs | null {
+  const agentKey = readFlagValue(args, "--agent") ?? positionalArgs(args)[0];
+  const explicitMsg = readFlagValue(args, "--msg");
+  const msgFile = readFlagValue(args, "--msg-file");
+  const fileMessage = msgFile
+    ? (options.readTextFile ?? ((path: string) => readFileSync(path, "utf8")))(msgFile)
+    : undefined;
+  const rawMessage = explicitMsg ?? fileMessage ?? positionalArgs(args).slice(1).join(" ");
+  if (!agentKey || !rawMessage.trim()) return null;
+  const message = rawMessage.trim();
+  const runtimeMode = runtimeModeFromArgs(args);
+  const continueDialogId = readFlagValue(args, "--continue") ?? readFlagValue(args, "--dialog");
+  const spaceId = readFlagValue(args, "--space");
+  const category = readFlagValue(args, "--category");
+  const inheritedFromDialog = readFlagValue(args, "--inherit-from-dialog");
+  const noDefaultTestRoot = args.includes("--no-default-test-root");
+  const cwd = readFlagValue(args, "--cwd");
+  const maxToolRounds = parseNonNegativeInteger(
+    readFlagValue(args, "--max-tool-rounds") ?? readFlagValue(args, "--rounds")
+  );
+  const timeoutMs = parsePositiveInteger(readFlagValue(args, "--timeout-ms"));
+  const inheritedRef = inheritedFromDialog
+    ? parseDialogReference(inheritedFromDialog)
+    : !continueDialogId && !noDefaultTestRoot
+      ? {
+          dialogKey: DEFAULT_TEST_ROOT_DIALOG_KEY,
+          dialogId: DEFAULT_TEST_ROOT_DIALOG_ID,
+        }
+      : undefined;
+  const imageUrls = [
+    ...readRepeatedFlagValues(args, "--image"),
+    ...readRepeatedFlagValues(args, "--image-url"),
+  ];
+  const useTaskWorktree = args.includes("--worktree-task");
+  const workflowRef = readFlagValue(args, "--workflow");
+  const taskRowDbKey = readFlagValue(args, "--task-row-dbkey");
+  const artifactIds = readFlagValue(args, "--artifact-ids")
+    ?.split(",")
+    .map((item) => item.trim())
+    .filter(Boolean);
+  return {
+    agentKey: resolveCliAgentKeyInput(agentKey),
+    message,
+    imageUrls,
+    allowShell: args.includes("--dangerously-allow-shell"),
+    traceTools: args.includes("--trace-tools"),
+    useTaskWorktree,
+    ...(workflowRef ? { workflowRef } : {}),
+    ...(taskRowDbKey
+      ? {
+          taskRunContext: {
+            rowDbKey: taskRowDbKey,
+            ...(readFlagValue(args, "--task-run-id") ? { taskRunId: readFlagValue(args, "--task-run-id") } : {}),
+            ...(readFlagValue(args, "--work-item-id") ? { workItemId: readFlagValue(args, "--work-item-id") } : {}),
+            ...(artifactIds?.length ? { artifactIds } : {}),
+          },
+        }
+      : {}),
+    ...(runtimeMode ? { runtimeMode } : useTaskWorktree ? { runtimeMode: "local" as const } : {}),
+    background: args.includes("--bg"),
+    noStream: args.includes("--no-stream"),
+    noDefaultTestRoot,
+    ...(continueDialogId ? { continueDialogId } : {}),
+    ...(!continueDialogId && spaceId ? { spaceId } : {}),
+    ...(category ? { category } : !continueDialogId && !noDefaultTestRoot ? { category: DEFAULT_TEST_DIALOG_CATEGORY_ID } : {}),
+    ...(inheritedRef?.dialogKey ? { inheritedFromDialogKey: inheritedRef.dialogKey } : {}),
+    ...(inheritedRef?.dialogId ? { parentDialogId: inheritedRef.dialogId } : {}),
+    ...(cwd ? { cwd } : {}),
+    ...(typeof maxToolRounds === "number" ? { maxToolRounds } : {}),
+    ...(typeof timeoutMs === "number" ? { timeoutMs } : {}),
+  };
+}
+
+export async function resolveWorkflowReference(ref: string, cwd = process.cwd()): Promise<ResolvedWorkflowReference> {
+  const path = workflowRefToCandidatePath(cwd, ref);
+  if (!path || !existsSync(path)) {
+    throw new Error(`Workflow reference not found: ${ref}`);
+  }
+  const markdown = readFileSync(path, "utf8");
+  const parsed = parseSkillDocProtocol(markdown);
+  return {
+    ref,
+    content: parsed.content,
+    ...(parsed.meta?.workflowConfig ? { config: parsed.meta.workflowConfig } : {}),
+  };
+}
+
+export function prependWorkflowReferencePrompt(message: string, workflow?: ResolvedWorkflowReference): string {
+  if (!workflow) return message;
+  const config = workflow.config;
+  return [
+    "AI-native workflow reference:",
+    "- This reference is guidance for the agent, not a central workflow engine.",
+    `- ref: ${workflow.ref}`,
+    ...(config?.id ? [`- id: ${config.id}`] : []),
+    ...(config?.name ? [`- name: ${config.name}`] : []),
+    ...(config?.defaultAgent ? [`- suggested defaultAgent: ${config.defaultAgent}`] : []),
+    ...(config?.inputs?.length ? [`- inputs: ${config.inputs.join(", ")}`] : []),
+    ...(config?.recommendedTools?.length ? [`- recommendedTools: ${config.recommendedTools.join(", ")}`] : []),
+    ...(config?.requiredTools?.length ? [`- requiredTools: ${config.requiredTools.join(", ")}`] : []),
+    ...(config?.requiredOutputs?.length ? [`- requiredOutputs: ${config.requiredOutputs.join(", ")}`] : []),
+    ...(config?.gates?.length ? [`- gates: ${config.gates.join(", ")}`] : []),
+    ...(config?.contextStrategy ? [`- contextStrategy: ${config.contextStrategy}`] : []),
+    ...(config?.failureProtocol ? [`- failureProtocol: ${config.failureProtocol}`] : []),
+    "",
+    "Reference body:",
+    workflow.content,
+    "",
+    "User task:",
+    message,
+  ].join("\n");
+}
+
+export function normalizeCliImageInput(input: string) {
+  if (/^(https?:|data:|file:)/i.test(input)) return input;
+  const absolutePath = resolve(input);
+  if (!existsSync(absolutePath)) return input;
+  const base64 = readFileSync(absolutePath).toString("base64");
+  return `data:${mimeTypeForPath(absolutePath)};base64,${base64}`;
+}