@prismatic-io/lux 0.0.1

package/src/drivers/codex/config.ts

@@ -0,0 +1,128 @@
+import { z } from "zod";
+import { DEFAULT_CODEX_MODEL, ReasoningEffortSchema } from "../../core/index.js";
+
+export type CodexConfigValue =
+  | string
+  | number
+  | boolean
+  | CodexConfigValue[]
+  | { [key: string]: CodexConfigValue };
+
+const CodexConfigValueSchema: z.ZodType<CodexConfigValue> = z.lazy(() =>
+  z.union([
+    z.string(),
+    z.number().finite(),
+    z.boolean(),
+    z.array(CodexConfigValueSchema),
+    z.record(z.string(), CodexConfigValueSchema),
+  ]),
+);
+
+const IsolationSchema = z
+  .object({
+    allowRead: z.array(z.string()).default([]),
+  })
+  .strict();
+
+export const CodexDriverConfigSchema = z
+  .object({
+    command: z.string().default("codex"),
+    model: z.string().min(1).default(DEFAULT_CODEX_MODEL),
+    reasoningEffort: ReasoningEffortSchema.default("low"),
+    sandbox: z
+      .enum(["read-only", "workspace-write", "danger-full-access"])
+      .default("workspace-write"),
+    cwdMode: z.enum(["artifacts", "runDir"]).default("artifacts"),
+    cwd: z.string().optional(),
+    env: z.record(z.string(), z.string()).optional(),
+    addDirs: z.array(z.string()).optional(),
+    profile: z.string().optional(),
+    config: z.record(z.string(), CodexConfigValueSchema).optional(),
+    extraArgs: z.array(z.string()).optional(),
+    ephemeral: z.boolean().default(true),
+    skipGitRepoCheck: z.boolean().default(true),
+    ignoreUserConfig: z.boolean().default(false),
+    ignoreRules: z.boolean().default(false),
+    strictConfig: z.boolean().default(false),
+    interactionMode: z.enum(["app-server", "exec"]).default("app-server"),
+    approvalPolicy: z.enum(["untrusted", "on-request", "never"]).default("on-request"),
+    isolation: IsolationSchema.optional(),
+    maxInterrupts: z.number().positive().default(20),
+    maxLineBytes: z.number().positive().default(10_000_000),
+    idleTimeoutMs: z.number().positive().default(600_000),
+  })
+  .strict()
+  .superRefine((config, ctx) => {
+    if (config.interactionMode === "app-server" && config.profile) {
+      ctx.addIssue({
+        code: "custom",
+        path: ["profile"],
+        message: "app-server does not accept a Codex profile; use inline config",
+      });
+    }
+    if (config.interactionMode === "app-server" && config.ignoreRules) {
+      ctx.addIssue({
+        code: "custom",
+        path: ["ignoreRules"],
+        message: "ignoreRules is a codex exec option",
+      });
+    }
+    if (!config.isolation) return;
+    if (config.cwd || config.cwdMode !== "artifacts") {
+      ctx.addIssue({
+        code: "custom",
+        path: ["isolation"],
+        message: "isolation requires cwdMode 'artifacts' and does not allow an explicit cwd",
+      });
+    }
+    if (config.interactionMode === "exec") {
+      ctx.addIssue({
+        code: "custom",
+        path: ["interactionMode"],
+        message: "isolation requires app-server interaction mode",
+      });
+    }
+  });
+
+export type CodexDriverConfig = z.infer<typeof CodexDriverConfigSchema>;
+
+const tomlKey = (key: string): string => (/^[A-Za-z0-9_-]+$/.test(key) ? key : JSON.stringify(key));
+
+const tomlValue = (value: CodexConfigValue): string => {
+  if (typeof value === "string") return JSON.stringify(value);
+  if (typeof value === "number" || typeof value === "boolean") return String(value);
+  if (Array.isArray(value)) return `[${value.map(tomlValue).join(",")}]`;
+  return `{${Object.entries(value)
+    .map(([key, entry]) => `${tomlKey(key)}=${tomlValue(entry)}`)
+    .join(",")}}`;
+};
+
+export const buildExecArgs = (config: CodexDriverConfig): string[] => {
+  const args = ["exec", "--json", "--color", "never", "--sandbox", config.sandbox];
+  args.push("--model", config.model);
+  if (config.profile) args.push("--profile", config.profile);
+  if (config.ephemeral) args.push("--ephemeral");
+  if (config.skipGitRepoCheck) args.push("--skip-git-repo-check");
+  if (config.ignoreUserConfig) args.push("--ignore-user-config");
+  if (config.ignoreRules) args.push("--ignore-rules");
+  if (config.strictConfig) args.push("--strict-config");
+  for (const directory of config.addDirs ?? []) args.push("--add-dir", directory);
+  for (const [key, value] of Object.entries(config.config ?? {})) {
+    args.push("--config", `${key}=${tomlValue(value)}`);
+  }
+  args.push("--config", `model_reasoning_effort=${tomlValue(config.reasoningEffort)}`);
+  if (config.extraArgs) args.push(...config.extraArgs);
+  args.push("-");
+  return args;
+};
+
+export const buildAppServerArgs = (config: CodexDriverConfig): string[] => {
+  const args = ["app-server", "--stdio"];
+  if (config.strictConfig) args.push("--strict-config");
+  for (const [key, value] of Object.entries(config.config ?? {})) {
+    args.push("--config", `${key}=${tomlValue(value)}`);
+  }
+  args.push("--config", `model_reasoning_effort=${tomlValue(config.reasoningEffort)}`);
+  if (config.extraArgs) args.push(...config.extraArgs);
+  return args;
+};