sharkcode 0.3.0 → 0.3.2

package/dist/cli.mjs

@@ -3,6 +3,7 @@
 // src/cli.ts
 import chalk3 from "chalk";
 import * as readline2 from "readline";
+import { select, input } from "@inquirer/prompts";
 
 // src/config.ts
 import { parse } from "smol-toml";
@@ -438,177 +439,179 @@ function renderWord(word, padLeft = 2) {
   return rows;
 }
 var BANNER = ["", ...renderWord("shark", 2), "", ...renderWord("code", 8), ""].join("\n");
-function printSlashHelp() {
-  const pad = (s, n) => s + " ".repeat(Math.max(0, n - s.length));
-  console.log(PURPLE2("\n  \u25C6 Slash Commands\n"));
-  const cmds = [
-    ["/provider", "show current provider & list options"],
-    ["/provider <name>", "switch provider  (deepseek | ark)"],
-    ["/key <api-key>", "set API key for current provider"],
-    ["/model <model-id>", "set model for current provider"],
-    ["/clear", "clear conversation history"],
-    ["/help", "show this menu"],
-    ["/exit", "quit"]
-  ];
-  for (const [cmd, desc] of cmds) {
-    console.log("  " + PURPLE2(pad(cmd, 26)) + GRAY(desc));
-  }
+async function showCommandMenu(multiConfig) {
   console.log();
-}
-function handleSlash(input, multiConfig) {
-  const parts = input.trim().split(/\s+/);
-  const cmd = parts[0].toLowerCase();
-  const arg = parts.slice(1).join(" ");
-  if (cmd === "/" || cmd === "/help") {
-    printSlashHelp();
-    return { multiConfig, config: resolveConfig(multiConfig) };
-  }
-  if (cmd === "/exit" || cmd === "/quit") {
-    console.log(GRAY("Bye! \u{1F988}"));
-    return { multiConfig, config: resolveConfig(multiConfig), exit: true };
-  }
-  if (cmd === "/clear") {
-    console.log(GRAY("  \u2713 Conversation cleared."));
-    return { multiConfig, config: resolveConfig(multiConfig), clearHistory: true };
-  }
-  if (cmd === "/provider") {
-    if (!arg) {
-      const current = multiConfig.activeProvider;
-      console.log(PURPLE2("\n  \u25C6 Providers\n"));
-      for (const [name2, meta] of Object.entries(PROVIDERS)) {
-        const entry = multiConfig.providers[name2];
-        const active = name2 === current;
-        const hasKey = !!entry?.key;
-        const marker = active ? PURPLE2("\u25B6") : " ";
-        const keyStatus = hasKey ? GREEN("\u2713 key set") : YELLOW("\u2717 no key");
-        console.log(
-          `  ${marker} ${active ? PURPLE2(name2) : GRAY(name2)}  ${GRAY(meta.label)}  ${keyStatus}` + (active ? `  ${GRAY("model: " + (entry?.model ?? meta.defaultModel))}` : "")
-        );
-      }
-      console.log(
-        `
-  ${GRAY("Usage:")} ${PURPLE2("/provider deepseek")} ${GRAY("or")} ${PURPLE2("/provider ark")}
-`
-      );
-      return { multiConfig, config: resolveConfig(multiConfig) };
+  try {
+    const action = await select({
+      message: PURPLE2("\u25C6 \u9009\u62E9\u64CD\u4F5C"),
+      choices: [
+        { name: "\u{1F50C}  \u5207\u6362 / \u914D\u7F6E Provider", value: "provider" },
+        { name: "\u{1F5D1}\uFE0F  \u6E05\u7A7A\u5BF9\u8BDD\u5386\u53F2", value: "clear" },
+        { name: "\u{1F6AA}  \u9000\u51FA", value: "exit" }
+      ]
+    });
+    switch (action) {
+      case "provider":
+        return showSetupFlow(multiConfig);
+      case "clear":
+        console.log(GRAY("\n  \u2713 \u5BF9\u8BDD\u5DF2\u6E05\u7A7A\n"));
+        return { multiConfig, config: resolveConfig(multiConfig), clearHistory: true };
+      case "exit":
+        console.log(GRAY("\nBye! \u{1F988}"));
+        return { multiConfig, config: resolveConfig(multiConfig), exit: true };
     }
-    const name = arg.toLowerCase();
-    if (!PROVIDERS[name]) {
-      console.log(RED(`  \u2717 Unknown provider: "${name}". Available: ${Object.keys(PROVIDERS).join(", ")}`));
-      return { multiConfig, config: resolveConfig(multiConfig) };
+  } catch {
+    console.log(GRAY("\n  \u53D6\u6D88\n"));
+  }
+  return { multiConfig, config: resolveConfig(multiConfig) };
+}
+async function showSetupFlow(multiConfig) {
+  console.log();
+  try {
+    const providerChoices = Object.entries(PROVIDERS).map(([id, meta]) => {
+      const hasKey = !!multiConfig.providers[id]?.key;
+      const badge = hasKey ? GREEN("\u2713 \u5DF2\u914D\u7F6E") : YELLOW("\u2717 \u672A\u914D\u7F6E");
+      return { name: `${meta.label}   ${badge}`, value: id };
+    });
+    const selectedProvider = await select({
+      message: PURPLE2("\u25C6 \u9009\u62E9 Provider"),
+      choices: providerChoices,
+      default: multiConfig.activeProvider
+    });
+    const currentKey = multiConfig.providers[selectedProvider]?.key ?? "";
+    const hint = currentKey ? GRAY("(\u56DE\u8F66\u4FDD\u7559  " + currentKey.slice(0, 6) + "\u2022\u2022\u2022)") : GRAY("(\u5FC5\u586B)");
+    const rawKey = await input({
+      message: PURPLE2("\u25C6 API Key ") + hint,
+      default: currentKey || void 0
+    });
+    const newKey = rawKey.trim() || currentKey;
+    let updated = { ...multiConfig, activeProvider: selectedProvider };
+    if (newKey) {
+      updated = {
+        ...updated,
+        providers: {
+          ...updated.providers,
+          [selectedProvider]: {
+            model: PROVIDERS[selectedProvider]?.defaultModel ?? "",
+            ...updated.providers[selectedProvider] ?? {},
+            key: newKey
+          }
+        }
+      };
     }
-    const updated = { ...multiConfig, activeProvider: name };
     saveMultiConfig(updated);
     const newConfig = resolveConfig(updated);
-    console.log(
-      GREEN(`
-  \u2713 Switched to ${PROVIDERS[name].label}`) + GRAY(` (${name}) \u2014 model: ${newConfig.model}`)
-    );
+    const keyMsg = newKey && newKey !== currentKey ? "\uFF0CAPI Key \u5DF2\u4FDD\u5B58" : "";
+    console.log(GREEN(`
+  \u2713 \u5DF2\u5207\u6362\u5230 ${PROVIDERS[selectedProvider].label}${keyMsg}`) + "\n");
     if (!newConfig.apiKey) {
-      console.log(YELLOW(`  \u26A0 No API key set. Use /key <your-key> to configure it.
-`));
-    } else {
-      console.log();
+      console.log(YELLOW("  \u26A0 \u8FD8\u672A\u586B\u5199 API Key\uFF0C\u65E0\u6CD5\u53D1\u9001\u6D88\u606F\n"));
     }
     return { multiConfig: updated, config: newConfig };
+  } catch {
+    console.log(GRAY("\n  \u53D6\u6D88\n"));
+    return { multiConfig, config: resolveConfig(multiConfig) };
   }
-  if (cmd === "/key") {
-    if (!arg) {
-      console.log(YELLOW(`  Usage: /key <your-api-key>`));
-      return { multiConfig, config: resolveConfig(multiConfig) };
-    }
-    const name = multiConfig.activeProvider;
-    const updated = {
-      ...multiConfig,
-      providers: {
-        ...multiConfig.providers,
-        [name]: {
-          ...multiConfig.providers[name] ?? { model: PROVIDERS[name]?.defaultModel ?? "" },
-          key: arg
-        }
+}
+async function readLineRaw(promptStr) {
+  process.stdout.write(promptStr);
+  return new Promise((resolve4) => {
+    let buffer = "";
+    let finished = false;
+    const finish = (value) => {
+      if (finished) return;
+      finished = true;
+      process.stdin.removeListener("data", onData);
+      try {
+        process.stdin.setRawMode(false);
+      } catch {
       }
+      process.stdin.pause();
+      resolve4(value);
     };
-    saveMultiConfig(updated);
-    const masked = arg.slice(0, 6) + "\u2022".repeat(Math.max(0, arg.length - 10)) + arg.slice(-4);
-    console.log(GREEN(`  \u2713 API key saved for ${name}: ${GRAY(masked)}
-`));
-    return { multiConfig: updated, config: resolveConfig(updated) };
-  }
-  if (cmd === "/model") {
-    if (!arg) {
-      const name2 = multiConfig.activeProvider;
-      const current = multiConfig.providers[name2]?.model ?? PROVIDERS[name2]?.defaultModel;
-      console.log(GRAY(`  Current model: `) + PURPLE2(current ?? "unknown"));
-      console.log(GRAY(`  Usage: /model <model-id>
-`));
-      return { multiConfig, config: resolveConfig(multiConfig) };
-    }
-    const name = multiConfig.activeProvider;
-    const updated = {
-      ...multiConfig,
-      providers: {
-        ...multiConfig.providers,
-        [name]: {
-          ...multiConfig.providers[name] ?? { key: "" },
-          model: arg
+    const onData = (chunk) => {
+      const str = chunk.toString("utf8");
+      if (str.startsWith("\x1B")) return;
+      for (const ch of str) {
+        const code = ch.codePointAt(0);
+        if (code === 3 || code === 4) {
+          process.stdout.write("\n");
+          finish(null);
+          return;
+        }
+        if (code === 13 || code === 10) {
+          process.stdout.write("\n");
+          finish(buffer);
+          return;
+        }
+        if (code === 127 || code === 8) {
+          if (buffer.length > 0) {
+            const chars = [...buffer];
+            chars.pop();
+            buffer = chars.join("");
+            process.stdout.write("\b \b");
+          }
+          continue;
         }
+        if (code < 32) continue;
+        if (ch === "/" && buffer === "") {
+          process.stdout.write("\n");
+          finish("/");
+          return;
+        }
+        buffer += ch;
+        process.stdout.write(ch);
       }
     };
-    saveMultiConfig(updated);
-    console.log(GREEN(`  \u2713 Model set to ${PURPLE2(arg)} for ${name}
-`));
-    return { multiConfig: updated, config: resolveConfig(updated) };
-  }
-  console.log(RED(`  \u2717 Unknown command: "${cmd}"`));
-  console.log(GRAY(`  Type /help to see available commands.
-`));
-  return { multiConfig, config: resolveConfig(multiConfig) };
-}
-async function readLine(prompt) {
-  return new Promise((resolve4) => {
-    const rl = readline2.createInterface({ input: process.stdin, output: process.stdout });
-    let answered = false;
-    rl.question(prompt, (answer) => {
-      answered = true;
-      rl.close();
-      resolve4(answer);
-    });
-    rl.on("close", () => {
-      if (!answered) resolve4(null);
-    });
+    try {
+      process.stdin.setRawMode(true);
+      process.stdin.resume();
+      process.stdin.on("data", onData);
+    } catch {
+      finished = true;
+      const rl = readline2.createInterface({ input: process.stdin, output: process.stdout });
+      let answered = false;
+      rl.question(promptStr, (answer) => {
+        answered = true;
+        rl.close();
+        resolve4(answer);
+      });
+      rl.on("close", () => {
+        if (!answered) resolve4(null);
+      });
+    }
   });
 }
 function statusLine(config) {
-  const providerLabel = PROVIDERS[config.providerName]?.label ?? config.providerName;
-  return PURPLE2("  \u25C6") + GRAY(` ${providerLabel}`) + CYAN(` [${config.model}]`) + GRAY("   type /help for commands\n");
+  const label = PROVIDERS[config.providerName]?.label ?? config.providerName;
+  return PURPLE2("  \u25C6") + GRAY(` ${label}`) + CYAN(`  [${config.model}]`) + GRAY("   \u8F93\u5165 / \u8C03\u51FA\u6307\u4EE4\u83DC\u5355\n");
 }
 async function main() {
   const args = process.argv.slice(2);
   if (args[0] === "--help" || args[0] === "-h") {
     console.log(BANNER);
     console.log(PURPLE2("  Usage:"));
-    console.log("    " + PURPLE2("sharkcode") + GRAY("                   \u2014 interactive mode"));
-    console.log("    " + PURPLE2("sharkcode") + YELLOW(' "prompt"') + GRAY("         \u2014 single-shot mode"));
-    console.log("\n  " + PURPLE2("Slash commands (in interactive mode):"));
-    printSlashHelp();
+    console.log("    " + PURPLE2("sharkcode") + GRAY("          \u2014 \u4EA4\u4E92\u6A21\u5F0F\uFF08\u76F4\u63A5\u542F\u52A8\uFF09"));
+    console.log("    " + PURPLE2("sharkcode") + YELLOW(' "prompt"') + GRAY("  \u2014 \u5355\u6B21\u6267\u884C"));
+    console.log(GRAY("\n  \u4EA4\u4E92\u6A21\u5F0F\u5185\u8F93\u5165 / \u8C03\u51FA\u6307\u4EE4\u83DC\u5355\n"));
     return;
   }
   if (args[0] === "--version" || args[0] === "-v") {
-    console.log("sharkcode v0.3.0");
+    console.log("sharkcode v0.3.2");
     return;
   }
   if (args.length > 0) {
     const mc = readMultiConfig();
     const config2 = resolveConfig(mc);
     if (!config2.apiKey) {
-      console.error(RED("\u274C No API key found for provider: " + config2.providerName));
-      console.error(GRAY("   Run sharkcode in interactive mode and use /key <your-key>"));
+      console.error(RED("\u274C \u672A\u914D\u7F6E API Key\u3002\u8BF7\u5148\u8FD0\u884C sharkcode \u5E76\u8F93\u5165 / \u914D\u7F6E Provider\u3002"));
       process.exit(1);
     }
-    console.log(PURPLE2("\n\u{1F988} SharkCode") + GRAY(` | ${PROVIDERS[config2.providerName]?.label ?? config2.providerName} | model: ${config2.model}
-`));
-    const messages2 = [{ role: "user", content: args.join(" ") }];
-    await runAgent(messages2, config2);
+    console.log(
+      PURPLE2("\n\u{1F988} SharkCode") + GRAY(` | ${PROVIDERS[config2.providerName]?.label ?? config2.providerName} | ${config2.model}
+`)
+    );
+    await runAgent([{ role: "user", content: args.join(" ") }], config2);
     return;
   }
   console.log(BANNER);
@@ -617,54 +620,59 @@ async function main() {
   console.log(statusLine(config));
   if (!config.apiKey) {
     console.log(
-      YELLOW("  \u26A0 No API key configured for ") + PURPLE2(PROVIDERS[config.providerName]?.label ?? config.providerName) + YELLOW(".")
+      YELLOW("  \u26A0 \u5C1A\u672A\u914D\u7F6E API Key\u3002") + GRAY("\u8F93\u5165 / \u7136\u540E\u9009\u62E9\u300C\u5207\u6362 / \u914D\u7F6E Provider\u300D\n")
     );
-    console.log(GRAY("    Use /key <your-api-key> to set it, or /provider to switch.\n"));
   }
   let messages = [];
   while (true) {
-    const input = await readLine(PURPLE2("\n\u25C6 "));
-    if (input === null) {
+    const raw = await readLineRaw(PURPLE2("\n\u25C6 "));
+    if (raw === null) {
       console.log(GRAY("\nBye! \u{1F988}"));
       break;
     }
-    const trimmed = input.trim();
+    const trimmed = raw.trim();
     if (!trimmed) continue;
     if (trimmed === "exit" || trimmed === "quit") {
       console.log(GRAY("Bye! \u{1F988}"));
       break;
     }
+    if (trimmed === "/") {
+      const r = await showCommandMenu(multiConfig);
+      multiConfig = r.multiConfig;
+      config = r.config;
+      if (r.clearHistory) messages = [];
+      if (r.exit) break;
+      console.log(statusLine(config));
+      continue;
+    }
+    if (trimmed === "/provider" || trimmed.startsWith("/provider ") || trimmed === "/model" || trimmed.startsWith("/model ") || trimmed === "/key" || trimmed.startsWith("/key ")) {
+      const r = await showSetupFlow(multiConfig);
+      multiConfig = r.multiConfig;
+      config = r.config;
+      if (r.exit) break;
+      console.log(statusLine(config));
+      continue;
+    }
     if (trimmed.startsWith("/")) {
-      const result = handleSlash(trimmed, multiConfig);
-      if (!result) continue;
-      multiConfig = result.multiConfig;
-      config = result.config;
-      if (result.clearHistory) messages = [];
-      if (result.exit) break;
+      console.log(GRAY("  \u672A\u77E5\u547D\u4EE4\u3002\u8F93\u5165 / \u8C03\u51FA\u6307\u4EE4\u83DC\u5355\n"));
       continue;
     }
     if (!config.apiKey) {
-      console.log(
-        YELLOW("\n  \u26A0 No API key for ") + PURPLE2(PROVIDERS[config.providerName]?.label ?? config.providerName) + YELLOW(". Set it with /key <your-api-key>\n")
-      );
+      console.log(YELLOW("  \u26A0 \u8FD8\u672A\u586B\u5199 API Key\u3002\u8F93\u5165 / \u2192 \u5207\u6362 / \u914D\u7F6E Provider\n"));
       continue;
     }
     messages.push({ role: "user", content: trimmed });
     try {
       messages = await runAgent(messages, config);
     } catch (err) {
-      const error = err;
       console.error(RED(`
-\u274C Error: ${error.message}`));
-      if (error.message?.includes("401") || error.message?.includes("Unauthorized")) {
-        console.error(YELLOW("   Check your API key with /key <your-api-key>"));
-      }
-      messages = messages.slice(0, -1);
+\u274C ${String(err)}
+`));
+      messages.pop();
     }
   }
 }
 main().catch((err) => {
-  console.error(RED(`
-\u274C Fatal: ${err.message}`));
+  console.error(chalk3.red(`Fatal: ${String(err)}`));
   process.exit(1);
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sharkcode",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "Local First, open-source AI Coding Agent",
   "type": "module",
   "bin": {
@@ -26,6 +26,7 @@
   },
   "dependencies": {
     "@ai-sdk/openai": "^3.0.48",
+    "@inquirer/prompts": "^8.3.2",
     "ai": "^6.0.141",
     "chalk": "^5.6.2",
     "smol-toml": "^1.6.1",