libretto 0.1.5 → 0.2.0

package/dist/cli/core/ai-config.js

@@ -0,0 +1,149 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { homedir } from "node:os";
+import { z } from "zod";
+import { LIBRETTO_CONFIG_PATH } from "./context.js";
+const CURRENT_CONFIG_VERSION = 1;
+const AiPresetSchema = z.enum(["codex", "claude", "gemini"]);
+const AiConfigSchema = z.object({
+  preset: AiPresetSchema,
+  commandPrefix: z.array(z.string()).min(1),
+  updatedAt: z.string()
+}).strict();
+const LibrettoConfigSchema = z.object({
+  version: z.literal(CURRENT_CONFIG_VERSION),
+  ai: AiConfigSchema.optional()
+}).passthrough();
+const AI_CONFIG_PRESETS = {
+  codex: ["codex", "exec", "--skip-git-repo-check", "--sandbox", "read-only"],
+  claude: [join(homedir(), ".claude", "local", "claude"), "-p"],
+  gemini: ["gemini", "--output-format", "json"]
+};
+function invalidConfigError(configPath) {
+  return new Error(
+    `AI config is invalid at ${configPath}. Fix the file to match the expected schema or delete it.`
+  );
+}
+function parseConfig(raw, configPath) {
+  try {
+    return LibrettoConfigSchema.parse(JSON.parse(raw));
+  } catch {
+    throw invalidConfigError(configPath);
+  }
+}
+function readLibrettoConfig(configPath = LIBRETTO_CONFIG_PATH) {
+  if (!existsSync(configPath)) {
+    return { version: CURRENT_CONFIG_VERSION };
+  }
+  return parseConfig(readFileSync(configPath, "utf-8"), configPath);
+}
+function writeLibrettoConfig(config, configPath = LIBRETTO_CONFIG_PATH) {
+  const parsed = LibrettoConfigSchema.parse(config);
+  mkdirSync(dirname(configPath), { recursive: true });
+  writeFileSync(configPath, JSON.stringify(parsed, null, 2), "utf-8");
+  return parsed;
+}
+function readAiConfig(configPath = LIBRETTO_CONFIG_PATH) {
+  return readLibrettoConfig(configPath).ai ?? null;
+}
+function quoteShellArg(value) {
+  if (/^[a-zA-Z0-9_./:@=-]+$/.test(value)) return value;
+  return JSON.stringify(value);
+}
+function formatCommandPrefix(prefix) {
+  return prefix.map((arg) => quoteShellArg(arg)).join(" ");
+}
+function writeAiConfig(preset, commandPrefix, configPath = LIBRETTO_CONFIG_PATH) {
+  const librettoConfig = readLibrettoConfig(configPath);
+  const ai = AiConfigSchema.parse({
+    preset,
+    commandPrefix,
+    updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+  });
+  writeLibrettoConfig(
+    {
+      ...librettoConfig,
+      version: CURRENT_CONFIG_VERSION,
+      ai
+    },
+    configPath
+  );
+  return ai;
+}
+function clearAiConfig(configPath = LIBRETTO_CONFIG_PATH) {
+  const librettoConfig = readLibrettoConfig(configPath);
+  if (!librettoConfig.ai) return false;
+  writeLibrettoConfig(
+    {
+      version: librettoConfig.version
+    },
+    configPath
+  );
+  return true;
+}
+function printAiConfig(config, configPath) {
+  console.log(`AI preset: ${config.preset}`);
+  console.log(`Command prefix: ${formatCommandPrefix(config.commandPrefix)}`);
+  console.log(`Config file: ${configPath}`);
+  console.log(`Updated at: ${config.updatedAt}`);
+}
+function printConfigureUsage(commandName) {
+  console.log(
+    `Usage: ${commandName} <codex|claude|gemini> [-- <command prefix...>]
+       ${commandName}
+       ${commandName} --clear`
+  );
+}
+function runAiConfigure(input, options = {}) {
+  const configureCommandName = options.configureCommandName ?? "libretto-cli ai configure";
+  const configPath = options.configPath ?? LIBRETTO_CONFIG_PATH;
+  const presetArg = input.preset?.trim();
+  const customPrefix = (input.customPrefix ?? []).filter(Boolean);
+  if (!presetArg && customPrefix.length === 0 && !input.clear) {
+    const config2 = readAiConfig(configPath);
+    if (!config2) {
+      console.log(`No AI config set. Run '${configureCommandName} codex' to set one.`);
+      return;
+    }
+    printAiConfig(config2, configPath);
+    return;
+  }
+  if (input.clear) {
+    const removed = clearAiConfig(configPath);
+    if (removed) {
+      console.log(`Cleared AI config: ${configPath}`);
+    } else {
+      console.log("No AI config was set.");
+    }
+    return;
+  }
+  const parsedPreset = AiPresetSchema.safeParse(presetArg);
+  if (!parsedPreset.success) {
+    printConfigureUsage(configureCommandName);
+    throw new Error(
+      "Missing or invalid preset. Use one of: codex, claude, gemini."
+    );
+  }
+  if (input.customPrefix && input.customPrefix.length > 0 && customPrefix.length === 0) {
+    throw new Error("Custom command prefix cannot be empty.");
+  }
+  const preset = parsedPreset.data;
+  const commandPrefix = customPrefix.length > 0 ? customPrefix : AI_CONFIG_PRESETS[preset];
+  const config = writeAiConfig(preset, commandPrefix, configPath);
+  console.log("AI config saved.");
+  printAiConfig(config, configPath);
+}
+export {
+  AI_CONFIG_PRESETS,
+  AiConfigSchema,
+  AiPresetSchema,
+  CURRENT_CONFIG_VERSION,
+  LibrettoConfigSchema,
+  clearAiConfig,
+  formatCommandPrefix,
+  readAiConfig,
+  readLibrettoConfig,
+  runAiConfigure,
+  writeAiConfig,
+  writeLibrettoConfig
+};