feat(setup): add interactive/options mode with backup and dry-run

Author: fishxcode

src/commands/setup.ts

@@ -1,34 +1,158 @@
 import pc from "picocolors";
-import { getApiKey, maskKey } from "../lib/config.js";
+import { existsSync } from "node:fs";
+import { copyFile, mkdir } from "node:fs/promises";
+import { basename, dirname, join } from "node:path";
+import { ask } from "../lib/prompt.js";
+import { getApiKey, getPreferences, maskKey, savePreferences } from "../lib/config.js";
 import { API_BASE, API_BASE_OPENAI } from "../lib/constants.js";
 import { SUPPORTED_TOOLS, getImplementedAdapter } from "../lib/tools.js";
+import type { ToolAdapter } from "../tools/types.js";
 
-export async function setupCommand(): Promise<void> {
+type SetupOptions = {
+  interactive?: boolean;
+  tools?: string;
+  model?: string;
+  dryRun?: boolean;
+  backup?: boolean;
+};
+
+const DEFAULT_MODEL = "claude-sonnet-4-5";
+const MODEL_CHOICES = ["claude-sonnet-4-5", "claude-opus-4-5", "gpt-5.4", "gpt-5"];
+
+function parseToolIds(input: string, validIds: string[]): string[] {
+  const raw = input
+    .split(",")
+    .map((s) => s.trim())
+    .filter(Boolean);
+  if (raw.length === 0) return [];
+  if (raw.includes("all")) return validIds;
+  const unknown = raw.filter((id) => !validIds.includes(id));
+  if (unknown.length > 0) {
+    throw new Error(`无效工具 ID: ${unknown.join(", ")}`);
+  }
+  return [...new Set(raw)];
+}
+
+function getImplementedTools(): Array<{ id: string; name: string; adapter: ToolAdapter }> {
+  return SUPPORTED_TOOLS.map((tool) => {
+    const adapter = getImplementedAdapter(tool.id);
+    return adapter ? { id: tool.id, name: tool.name, adapter } : null;
+  }).filter((item): item is { id: string; name: string; adapter: ToolAdapter } => item !== null);
+}
+
+async function maybeBackupFiles(files: string[]): Promise<string[]> {
+  const backedUp: string[] = [];
+  const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
+  for (const file of files) {
+    if (!existsSync(file)) continue;
+    const backupDir = join(dirname(file), ".fishxcode-backups");
+    await mkdir(backupDir, { recursive: true });
+    const backupFile = join(backupDir, `${basename(file)}.${timestamp}.bak`);
+    await copyFile(file, backupFile);
+    backedUp.push(backupFile);
+  }
+  return backedUp;
+}
+
+async function resolveInteractiveOptions(
+  initial: SetupOptions,
+  validToolIds: string[],
+  preferenceModel: string,
+): Promise<Required<Pick<SetupOptions, "interactive" | "tools" | "model" | "dryRun" | "backup">>> {
+  const toolsHint = validToolIds.join(",");
+  const defaultTools = initial.tools ?? "all";
+  const defaultModel = initial.model ?? preferenceModel;
+  const toolAnswer = (
+    await ask(`选择工具 ID（逗号分隔，默认 all，可选 ${toolsHint}）: `)
+  ).trim();
+  const modelAnswer = (await ask(`模型（默认 ${defaultModel}）: `)).trim();
+  const dryRunAnswer = (await ask("仅预览不写入？(y/N): ")).trim().toLowerCase();
+  const backupAnswer = (await ask("写入前备份配置？(y/N): ")).trim().toLowerCase();
+
+  return {
+    interactive: true,
+    tools: toolAnswer || defaultTools,
+    model: modelAnswer || defaultModel,
+    dryRun: dryRunAnswer === "y" || dryRunAnswer === "yes",
+    backup: backupAnswer === "y" || backupAnswer === "yes",
+  };
+}
+
+export async function setupCommand(options: SetupOptions = {}): Promise<void> {
   const key = await getApiKey();
   if (!key) {
     throw new Error("未检测到 API Key，请先执行 fishx login");
   }
 
+  const preferences = await getPreferences();
+  const implementedTools = getImplementedTools();
+  const validToolIds = implementedTools.map((t) => t.id);
+  const preferenceModel = preferences.defaultModel ?? DEFAULT_MODEL;
+
+  let effectiveOptions: SetupOptions = {
+    interactive: options.interactive ?? preferences.interactive ?? false,
+    tools: options.tools ?? preferences.defaultTools?.join(",") ?? "all",
+    model: options.model ?? preferenceModel,
+    dryRun: options.dryRun ?? false,
+    backup: options.backup ?? preferences.backup ?? false,
+  };
+
+  if (effectiveOptions.interactive) {
+    effectiveOptions = await resolveInteractiveOptions(effectiveOptions, validToolIds, preferenceModel);
+  }
+
+  const selectedToolIds = parseToolIds(effectiveOptions.tools ?? "all", validToolIds);
+  const selectedModel = effectiveOptions.model?.trim() || DEFAULT_MODEL;
+  const dryRun = !!effectiveOptions.dryRun;
+  const backup = !!effectiveOptions.backup;
+
+  await savePreferences({
+    defaultTools: selectedToolIds,
+    defaultModel: selectedModel,
+    interactive: !!effectiveOptions.interactive,
+    backup,
+  });
+
   console.log(pc.bold("开始配置 FishXCode 工具"));
   console.log(`- API Key: ${maskKey(key)}`);
   console.log(`- Anthropic Base URL: ${API_BASE}`);
   console.log(`- OpenAI Base URL: ${API_BASE_OPENAI}`);
+  console.log(`- 模型: ${selectedModel}`);
+  console.log(`- 模式: ${dryRun ? "预览 (dry-run)" : "写入"}`);
+  if (backup) {
+    console.log("- 备份: 已启用");
+  }
 
   const applied: string[] = [];
   const skipped: string[] = [];
+  const backups: string[] = [];
 
-  for (const tool of SUPPORTED_TOOLS) {
+  for (const tool of SUPPORTED_TOOLS.filter((t) => selectedToolIds.includes(t.id))) {
     const adapter = getImplementedAdapter(tool.id);
     if (!adapter) {
       skipped.push(tool.name);
       continue;
     }
 
     try {
+      const targetFiles = adapter.getTargetFiles?.() ?? [];
+      if (dryRun) {
+        const previewFile = targetFiles[0] ?? "(未知路径)";
+        applied.push(`${tool.name} -> ${previewFile}`);
+        console.log(pc.cyan(`~ ${tool.name}`), pc.gray(previewFile), pc.gray("[dry-run]"));
+        continue;
+      }
+
+      if (backup && targetFiles.length > 0) {
+        const backupFiles = await maybeBackupFiles(targetFiles);
+        backups.push(...backupFiles);
+      }
+
       const result = await adapter.configure({
         apiKey: key,
         baseAnthropic: API_BASE,
         baseOpenAI: API_BASE_OPENAI,
+        model: selectedModel,
       });
       applied.push(`${tool.name} -> ${result.file}`);
       console.log(pc.green(`✓ ${tool.name}`), pc.gray(result.file));
@@ -45,4 +169,8 @@ export async function setupCommand(): Promise<void> {
 
   console.log(`- 未实装: ${skipped.length}`);
   for (const item of skipped) console.log(`  ${pc.yellow("•")} ${item}`);
+  if (backups.length > 0) {
+    console.log(`- 备份文件: ${backups.length}`);
+    for (const file of backups) console.log(`  ${pc.blue("•")} ${file}`);
+  }
 }

src/index.ts

@@ -28,7 +28,17 @@ program
 
 program.command("whoami").description("查看当前登录状态").action(whoamiCommand);
 program.command("logout").description("清除本地 API Key").action(logoutCommand);
-program.command("setup").description("执行工具配置流程（迁移版）").action(setupCommand);
+program
+  .command("setup")
+  .description("执行工具配置流程（迁移版）")
+  .option("-i, --interactive", "交互式选择配置项")
+  .option("--tools <ids>", "仅配置指定工具（逗号分隔），如 codex,aider")
+  .option("--model <model>", "覆盖默认模型，如 claude-opus-4-5")
+  .option("--dry-run", "仅预览将要修改的配置，不写文件")
+  .option("--backup", "写入前备份目标配置文件")
+  .action(async (opts) => {
+    await setupCommand(opts);
+  });
 program.command("doctor").description("检查环境与配置状态").action(doctorCommand);
 program.command("tools").description("列出支持工具").action(toolsCommand);
 program.command("reset").description("重置本地配置").action(resetCommand);

src/lib/config.ts

@@ -7,6 +7,14 @@ import { CONFIG_DIRNAME } from "./constants.js";
 export type AppConfig = {
   apiKey?: string;
   savedAt?: string;
+  preferences?: AppPreferences;
+};
+
+export type AppPreferences = {
+  defaultTools?: string[];
+  defaultModel?: string;
+  interactive?: boolean;
+  backup?: boolean;
 };
 
 export const configDir = join(homedir(), CONFIG_DIRNAME);
@@ -41,6 +49,17 @@ export async function getApiKey(): Promise<string> {
   return cfg.apiKey ?? process.env.FISHXCODE_API_KEY ?? "";
 }
 
+export async function getPreferences(): Promise<AppPreferences> {
+  const cfg = await loadConfig();
+  return cfg.preferences ?? {};
+}
+
+export async function savePreferences(patch: AppPreferences): Promise<AppConfig> {
+  const current = await loadConfig();
+  const nextPreferences = { ...(current.preferences ?? {}), ...patch };
+  return saveConfig({ preferences: nextPreferences });
+}
+
 export function maskKey(key: string): string {
   if (!key || key.length < 10) return "****";
   return `${key.slice(0, 8)}...${key.slice(-4)}`;

src/tools/aider.ts

@@ -30,6 +30,12 @@ function removeManagedBlock(content: string): string {
   return kept.join("\n").trim();
 }
 
+function resolveAiderModel(model?: string): string {
+  if (!model) return "openai/claude-sonnet-4-5";
+  if (model.includes("/")) return model;
+  return `openai/${model}`;
+}
+
 export const aiderAdapter: ToolAdapter = {
   id: "aider",
   name: "Aider",
@@ -38,6 +44,9 @@ export const aiderAdapter: ToolAdapter = {
   checkInstalled() {
     return commandExists("aider");
   },
+  getTargetFiles() {
+    return [aiderFile];
+  },
   isConfigured() {
     if (!existsSync(aiderFile)) return false;
     const content = readFileSync(aiderFile, "utf8");
@@ -50,7 +59,7 @@ export const aiderAdapter: ToolAdapter = {
       `${marker} — https://fishxcode.com`,
       `openai-api-key: ${ctx.apiKey}`,
       `openai-api-base: ${ctx.baseOpenAI}`,
-      "model: openai/claude-sonnet-4-5",
+      `model: ${resolveAiderModel(ctx.model)}`,
       "",
     ].join("\n");
     const next = [clean, block].filter(Boolean).join("\n\n").trim();

src/tools/claude-code.ts

@@ -29,6 +29,9 @@ export const claudeCodeAdapter: ToolAdapter = {
   checkInstalled() {
     return commandExists("claude");
   },
+  getTargetFiles() {
+    return [settingsFile];
+  },
   isConfigured() {
     const s = readSettings();
     return !!s.env?.ANTHROPIC_AUTH_TOKEN;

src/tools/codex.ts

@@ -31,17 +31,21 @@ export const codexAdapter: ToolAdapter = {
   checkInstalled() {
     return commandExists("codex");
   },
+  getTargetFiles() {
+    return [tomlFile, jsonFile];
+  },
   isConfigured() {
     if (!existsSync(tomlFile)) return false;
     const content = readFileSync(tomlFile, "utf8");
     return content.includes('model_provider = "fishxcode"') && content.includes("fishxcode.com");
   },
   async configure(ctx) {
+    const model = ctx.model ?? "gpt-5.4";
     await mkdir(codexDir, { recursive: true });
     const current = await readToml();
     const cleaned = removeFishxcodeBlocks(current).trim();
     const next = [
-      'model = "gpt-5.4"',
+      `model = "${model}"`,
       'model_provider = "fishxcode"',
       "",
       cleaned,
@@ -61,7 +65,7 @@ export const codexAdapter: ToolAdapter = {
       try {
         const raw = await readFile(jsonFile, "utf8");
         const json = JSON.parse(raw) as Record<string, unknown>;
-        json.model = "gpt-5.4";
+        json.model = model;
         json.provider = "fishxcode";
         const providers = (json.providers ?? {}) as Record<string, unknown>;
         providers.fishxcode = {

src/tools/continue.ts

@@ -9,11 +9,18 @@ const continueDir = join(toolHome, ".continue");
 const yamlFile = join(continueDir, "config.yaml");
 const jsonFile = join(continueDir, "config.json");
 
-const FISHXCODE_MODELS = [
+const DEFAULT_MODELS = [
   { name: "FishXCode — Claude Sonnet", model: "claude-sonnet-4-5" },
   { name: "FishXCode — Claude Opus", model: "claude-opus-4-5" },
 ];
 
+function getFishxcodeModels(model?: string): Array<{ name: string; model: string }> {
+  if (!model) return DEFAULT_MODELS;
+  const custom = { name: `FishXCode — ${model}`, model };
+  const rest = DEFAULT_MODELS.filter((item) => item.model !== model);
+  return [custom, ...rest];
+}
+
 function getConfigFormat(): "yaml" | "json" {
   if (existsSync(yamlFile)) return "yaml";
   if (existsSync(jsonFile)) return "json";
@@ -29,15 +36,19 @@ function removeFishxcodeYamlBlocks(content: string): string {
   return content.replace(/ {2}- name: FishXCode[^\n]*\n( {4}[^\n]*\n)*/g, "");
 }
 
-async function writeYamlConfig(apiKey: string, baseUrl: string): Promise<void> {
+async function writeYamlConfig(
+  apiKey: string,
+  baseUrl: string,
+  models: Array<{ name: string; model: string }>,
+): Promise<void> {
   await mkdir(continueDir, { recursive: true });
 
   let content = removeFishxcodeYamlBlocks(readYaml());
   if (!content.includes("models:")) {
     content = `models:\n${content}`;
   }
 
-  const modelBlocks = FISHXCODE_MODELS.map((m) => {
+  const modelBlocks = models.map((m) => {
     return [
       `  - name: ${m.name}`,
       "    provider: openai",
@@ -82,6 +93,9 @@ export const continueAdapter: ToolAdapter = {
   checkInstalled() {
     return existsSync(continueDir);
   },
+  getTargetFiles() {
+    return [yamlFile, jsonFile];
+  },
   isConfigured() {
     if (existsSync(yamlFile)) {
       return readYaml().includes("fishxcode.com");
@@ -90,9 +104,10 @@ export const continueAdapter: ToolAdapter = {
     return (cfg.models ?? []).some((m) => String(m.apiBase ?? "").includes("fishxcode.com"));
   },
   async configure(ctx) {
+    const models = getFishxcodeModels(ctx.model);
     const format = getConfigFormat();
     if (format === "yaml") {
-      await writeYamlConfig(ctx.apiKey, ctx.baseOpenAI);
+      await writeYamlConfig(ctx.apiKey, ctx.baseOpenAI, models);
       return { file: yamlFile, hot: true };
     }
 
@@ -101,7 +116,7 @@ export const continueAdapter: ToolAdapter = {
       (m) => !String(m.apiBase ?? "").includes("fishxcode.com"),
     );
     cfg.models = [
-      ...FISHXCODE_MODELS.map((m) => ({
+      ...models.map((m) => ({
         title: m.name,
         provider: "openai",
         model: m.model,