@iislee/opencodex 2.11.0

package/src/cli/pi.ts

@@ -0,0 +1,188 @@
+/**
+ * `ocx pi [args...]` — launch Pi wired to the local proxy, and
+ * `ocx pi <status|apply|remove|settings|packages>` management subcommands.
+ *
+ * Launch path: ensure proxy is up, apply models.json providers.opencodex when
+ * possible, then exec `pi` with OPENCODEX_API_KEY in the environment (if an
+ * admission key is configured). Management subcommands talk to the live proxy API.
+ */
+import { spawn } from "node:child_process";
+import { loadConfig } from "../config";
+import { PI_API_KEY_ENV } from "../clients/config-export";
+import { resolvePiBinary } from "../pi/packages";
+import { commandInvocation } from "../lib/win-exec";
+import { loadServiceTokenFromFile, serviceApiTokenFilePath } from "../lib/service-secrets";
+import { findLiveProxy } from "../server/proxy-liveness";
+import {
+  CliUsageError,
+  printData,
+  rejectArgs,
+  runCliAction,
+  runtimeRequest,
+  summaryLines,
+  takeFlag,
+  takeOption,
+  type RuntimeApiDeps,
+} from "./runtime-api";
+
+const PI_USAGE = `Usage:
+  ocx pi [pi args...]                 Launch Pi (applies models.json first)
+  ocx pi status [--json]              Show Pi integration status
+  ocx pi apply [--json]               Write providers.opencodex into models.json
+  ocx pi remove [--json]              Remove providers.opencodex only
+  ocx pi settings [show] [--json]     Show curated settings
+  ocx pi settings set --key value...  Patch curated settings
+  ocx pi packages [list] [--json]     List installed packages
+  ocx pi packages install <source>    Install a package (runs \`pi install\`)
+  ocx pi packages remove <source>     Remove a package (runs \`pi remove\`)`;
+
+const MANAGEMENT = new Set(["status", "apply", "remove", "settings", "packages"]);
+
+export async function handlePiCommand(argv: string[], deps: RuntimeApiDeps = {}): Promise<number> {
+  const head = (argv[0] ?? "").toLowerCase();
+  if (MANAGEMENT.has(head) || head === "show") {
+    return handlePiManagement(argv, deps);
+  }
+  return cmdPi(argv);
+}
+
+async function handlePiManagement(argv: string[], deps: RuntimeApiDeps): Promise<number> {
+  return runCliAction(async () => {
+    const args = [...argv];
+    const action = (args.shift() ?? "status").toLowerCase();
+    const wantsJson = takeFlag(args, "--json");
+
+    if (action === "status" || action === "show") {
+      rejectArgs(args, PI_USAGE);
+      const result = await runtimeRequest("/api/pi", {}, deps);
+      printData(result, wantsJson, summaryLines(result));
+      return;
+    }
+    if (action === "apply") {
+      rejectArgs(args, PI_USAGE);
+      const result = await runtimeRequest("/api/pi/apply", { method: "POST" }, deps);
+      printData(result, wantsJson, [String((result as Record<string, unknown>).message ?? "Pi models applied.")]);
+      return;
+    }
+    if (action === "remove") {
+      rejectArgs(args, PI_USAGE);
+      const result = await runtimeRequest("/api/pi/remove", { method: "POST" }, deps);
+      printData(result, wantsJson, [String((result as Record<string, unknown>).message ?? "Pi models removed.")]);
+      return;
+    }
+    if (action === "settings") {
+      const sub = (args.shift() ?? "show").toLowerCase();
+      if (sub === "show" || sub === "status") {
+        rejectArgs(args, PI_USAGE);
+        const result = await runtimeRequest("/api/pi/settings", {}, deps);
+        printData(result, wantsJson, summaryLines(result));
+        return;
+      }
+      if (sub !== "set") throw new CliUsageError(`unknown Pi settings command ${sub}`, PI_USAGE);
+      const body: Record<string, unknown> = {};
+      const defaultProvider = takeOption(args, "--default-provider");
+      const defaultModel = takeOption(args, "--default-model");
+      const thinking = takeOption(args, "--thinking");
+      const theme = takeOption(args, "--theme");
+      const trust = takeOption(args, "--project-trust");
+      const quiet = takeFlag(args, "--quiet-startup") ? true : takeFlag(args, "--no-quiet-startup") ? false : undefined;
+      const hideThinking = takeFlag(args, "--hide-thinking") ? true : takeFlag(args, "--show-thinking") ? false : undefined;
+      rejectArgs(args, PI_USAGE);
+      if (defaultProvider !== undefined) body.defaultProvider = defaultProvider === "-" ? null : defaultProvider;
+      if (defaultModel !== undefined) body.defaultModel = defaultModel === "-" ? null : defaultModel;
+      if (thinking !== undefined) body.defaultThinkingLevel = thinking === "-" ? null : thinking;
+      if (theme !== undefined) body.theme = theme === "-" ? null : theme;
+      if (trust !== undefined) body.defaultProjectTrust = trust === "-" ? null : trust;
+      if (quiet !== undefined) body.quietStartup = quiet;
+      if (hideThinking !== undefined) body.hideThinkingBlock = hideThinking;
+      if (Object.keys(body).length === 0) {
+        throw new CliUsageError("at least one settings flag is required", PI_USAGE);
+      }
+      const result = await runtimeRequest("/api/pi/settings", { method: "PUT", body: JSON.stringify(body) }, deps);
+      printData(result, wantsJson, [String((result as Record<string, unknown>).message ?? "Pi settings updated.")]);
+      return;
+    }
+    if (action === "packages") {
+      const sub = (args.shift() ?? "list").toLowerCase();
+      if (sub === "list" || sub === "status") {
+        rejectArgs(args, PI_USAGE);
+        const result = await runtimeRequest("/api/pi/packages", {}, deps);
+        printData(result, wantsJson, summaryLines(result));
+        return;
+      }
+      if (sub === "install" || sub === "remove") {
+        const source = args.shift();
+        if (!source) throw new CliUsageError("package source is required", PI_USAGE);
+        rejectArgs(args, PI_USAGE);
+        const result = await runtimeRequest(`/api/pi/packages/${sub}`, {
+          method: "POST",
+          body: JSON.stringify({ source }),
+        }, deps);
+        printData(result, wantsJson, [String((result as Record<string, unknown>).message ?? `pi ${sub} done`)]);
+        return;
+      }
+      throw new CliUsageError(`unknown Pi packages command ${sub}`, PI_USAGE);
+    }
+    throw new CliUsageError(`unknown Pi command ${action}`, PI_USAGE);
+  });
+}
+
+/**
+ * Launch path. When the first arg is a management verb it is handled above; everything
+ * else is forwarded to the real `pi` binary after a best-effort apply.
+ */
+export async function cmdPi(argv: string[]): Promise<number> {
+  const binary = resolvePiBinary();
+  if (!binary) {
+    console.error("pi not found on PATH. Install: npm i -g @earendil-works/pi-coding-agent");
+    return 1;
+  }
+
+  // Best-effort: ensure proxy is discoverable and models are wired.
+  try {
+    const live = await findLiveProxy();
+    if (live) {
+      const config = loadConfig();
+      const { syncPiConfig } = await import("../pi/sync");
+      const result = await syncPiConfig(live.port, config, { hostname: live.hostname });
+      if (result.changed) console.error(result.message);
+      else if (!result.ok) console.error(result.message);
+    } else {
+      console.error("No running opencodex proxy found. Start with `ocx start` for routed models.");
+    }
+  } catch (error) {
+    console.error(`Pi models apply skipped: ${error instanceof Error ? error.message : String(error)}`);
+  }
+
+  const config = loadConfig();
+  const env: NodeJS.ProcessEnv = { ...process.env };
+  // Admission key for the proxy env reference in models.json (`$OPENCODEX_API_KEY`).
+  if (!env[PI_API_KEY_ENV]?.trim()) {
+    const authToken = env.OPENCODEX_API_AUTH_TOKEN?.trim();
+    if (authToken) env[PI_API_KEY_ENV] = authToken;
+    else {
+      const tokenEnv = env.OCX_API_TOKEN_FILE?.trim()
+        ? env
+        : { ...env, OCX_API_TOKEN_FILE: serviceApiTokenFilePath() };
+      const serviceToken = loadServiceTokenFromFile(tokenEnv);
+      env[PI_API_KEY_ENV] = serviceToken || config.apiKeys?.[0]?.key || "ocx";
+    }
+  }
+
+  const inv = commandInvocation(binary, argv);
+  return await new Promise<number>(resolve => {
+    const child = spawn(inv.file, inv.args, {
+      stdio: "inherit",
+      env,
+      windowsHide: true,
+      ...inv.options,
+    });
+    child.on("error", error => {
+      console.error(error.message);
+      resolve(1);
+    });
+    child.on("close", code => resolve(code ?? 1));
+  });
+}
+
+export const PI_CLI_USAGE = PI_USAGE;

package/src/cli/provider-runtime.ts

@@ -0,0 +1,162 @@
+import {
+  CliUsageError,
+  csv,
+  printData,
+  rejectArgs,
+  runCliAction,
+  runtimeRequest,
+  summaryLines,
+  takeBooleanOption,
+  takeFlag,
+  takeOption,
+  type RuntimeApiDeps,
+} from "./runtime-api";
+
+const USAGE = `Usage:
+  ocx provider edit <name> [--adapter <id>] [--base-url <url>] [--default-model <id|->]
+      [--auth-mode <key|forward|oauth|local|->] [--note <text|->]
+      [--api-key-transport <x-api-key|bearer|->]
+      [--enabled <on|off>] [--live-models <on|off>] [--allow-private-network <on|off>] [--json]
+  ocx provider test <name> [--json]
+  ocx provider quota [--refresh] [--json]
+  ocx provider presets [--json]
+  ocx provider account-mode <pool|direct> [--json]
+  ocx provider selected <name> [--set <model,model...>] [--clear] [--json]`;
+
+function cleared(value: string | undefined): string | undefined {
+  return value === "-" ? "" : value;
+}
+
+async function edit(argv: string[], deps: RuntimeApiDeps): Promise<void> {
+  const args = [...argv];
+  const name = args.shift()?.trim();
+  if (!name) throw new CliUsageError("provider name is required", USAGE);
+  const wantsJson = takeFlag(args, "--json");
+  const patch: Record<string, unknown> = {};
+  const adapter = takeOption(args, "--adapter");
+  const baseUrl = takeOption(args, "--base-url");
+  const defaultModel = cleared(takeOption(args, "--default-model"));
+  const authMode = cleared(takeOption(args, "--auth-mode"));
+  const note = cleared(takeOption(args, "--note"));
+  const apiKeyTransport = cleared(takeOption(args, "--api-key-transport"));
+  const enabled = takeBooleanOption(args, "--enabled");
+  const liveModels = takeBooleanOption(args, "--live-models");
+  const allowPrivateNetwork = takeBooleanOption(args, "--allow-private-network");
+  rejectArgs(args, USAGE);
+  if (adapter !== undefined) patch.adapter = adapter;
+  if (baseUrl !== undefined) patch.baseUrl = baseUrl;
+  if (defaultModel !== undefined) patch.defaultModel = defaultModel;
+  if (authMode !== undefined) patch.authMode = authMode;
+  if (note !== undefined) patch.note = note;
+  if (apiKeyTransport !== undefined) patch.apiKeyTransport = apiKeyTransport;
+  if (enabled !== undefined) patch.disabled = !enabled;
+  if (liveModels !== undefined) patch.liveModels = liveModels;
+  if (allowPrivateNetwork !== undefined) patch.allowPrivateNetwork = allowPrivateNetwork;
+  if (Object.keys(patch).length === 0) throw new CliUsageError("at least one edit option is required", USAGE);
+  const result = await runtimeRequest(`/api/providers?name=${encodeURIComponent(name)}`, {
+    method: "PATCH",
+    body: JSON.stringify(patch),
+  }, deps);
+  printData(result, wantsJson, [`Updated provider ${name}.`]);
+}
+
+async function testProvider(argv: string[], deps: RuntimeApiDeps): Promise<void> {
+  const args = [...argv];
+  const name = args.shift()?.trim();
+  const wantsJson = takeFlag(args, "--json");
+  if (!name) throw new CliUsageError("provider name is required", USAGE);
+  rejectArgs(args, USAGE);
+  const result = await runtimeRequest<Record<string, unknown>>(`/api/providers/test?name=${encodeURIComponent(name)}`, {
+    method: "POST",
+  }, deps);
+  if (result.applicable === false) {
+    printData(result, wantsJson, [
+      `${name}: not applicable`,
+      "Static catalog; no live model-discovery endpoint to test.",
+    ]);
+    return;
+  }
+  const ok = result.ok === true;
+  printData(result, wantsJson, [
+    `${name}: ${ok ? "connected" : "failed"}`,
+    String(result.message ?? result.error ?? "No detail"),
+    `Latency: ${String(result.latencyMs ?? "?")} ms`,
+  ]);
+  if (!ok) process.exitCode = 1;
+}
+
+async function quota(argv: string[], deps: RuntimeApiDeps): Promise<void> {
+  const args = [...argv];
+  const wantsJson = takeFlag(args, "--json");
+  const refresh = takeFlag(args, "--refresh");
+  rejectArgs(args, USAGE);
+  const result = await runtimeRequest(`/api/provider-quotas${refresh ? "?refresh=1" : ""}`, {}, deps);
+  printData(result, wantsJson, summaryLines(result));
+}
+
+async function presets(argv: string[], deps: RuntimeApiDeps): Promise<void> {
+  const args = [...argv];
+  const wantsJson = takeFlag(args, "--json");
+  rejectArgs(args, USAGE);
+  const result = await runtimeRequest<{ providers?: unknown[] } | unknown[]>("/api/provider-presets", {}, deps);
+  const rows = Array.isArray(result) ? result : result.providers ?? [];
+  printData(result, wantsJson, rows.map(row => {
+    const record = row as Record<string, unknown>;
+    return `${String(record.id ?? record.name ?? "?")}  ${String(record.label ?? record.adapter ?? "")}`.trimEnd();
+  }));
+}
+
+async function accountMode(argv: string[], deps: RuntimeApiDeps): Promise<void> {
+  const args = [...argv];
+  const mode = args.shift();
+  const wantsJson = takeFlag(args, "--json");
+  if (mode !== "pool" && mode !== "direct") throw new CliUsageError("mode must be pool or direct", USAGE);
+  rejectArgs(args, USAGE);
+  const result = await runtimeRequest("/api/providers?name=openai", {
+    method: "PATCH",
+    body: JSON.stringify({ codexAccountMode: mode }),
+  }, deps);
+  printData(result, wantsJson, [`OpenAI Codex account mode: ${mode}`]);
+}
+
+async function selected(argv: string[], deps: RuntimeApiDeps): Promise<void> {
+  const args = [...argv];
+  const name = args.shift()?.trim();
+  const wantsJson = takeFlag(args, "--json");
+  const set = csv(takeOption(args, "--set"));
+  const clear = takeFlag(args, "--clear");
+  if (!name) throw new CliUsageError("provider name is required", USAGE);
+  if (set !== undefined && clear) throw new CliUsageError("--set and --clear cannot be combined", USAGE);
+  rejectArgs(args, USAGE);
+  if (set === undefined && !clear) {
+    const result = await runtimeRequest<Record<string, unknown>>("/api/selected-models", {}, deps);
+    const selectedMap = (result.selected ?? {}) as Record<string, string[]>;
+    const availableMap = (result.available ?? {}) as Record<string, string[]>;
+    printData({ provider: name, selected: selectedMap[name] ?? [], available: availableMap[name] ?? [] }, wantsJson,
+      [`Selected models for ${name}: ${(selectedMap[name] ?? []).join(", ") || "all"}`]);
+    return;
+  }
+  const models = clear ? [] : (set ?? []);
+  const result = await runtimeRequest("/api/selected-models", {
+    method: "PUT",
+    body: JSON.stringify({ provider: name, models }),
+  }, deps);
+  printData(result, wantsJson, [`${name}: ${models.length ? models.join(", ") : "all models"}`]);
+}
+
+export async function handleProviderRuntimeCommand(sub: string, argv: string[], deps: RuntimeApiDeps = {}): Promise<number | null> {
+  const handlers: Record<string, (args: string[], deps: RuntimeApiDeps) => Promise<void>> = {
+    edit,
+    update: edit,
+    test: testProvider,
+    quota,
+    presets,
+    "account-mode": accountMode,
+    selected,
+  };
+  const handler = handlers[sub];
+  if (!handler) return null;
+  return runCliAction(() => handler(argv, deps));
+}
+
+export const PROVIDER_RUNTIME_USAGE = USAGE;