@zhanngning/hecode 0.7.0 → 0.8.0

package/dist/index.js

@@ -1737,14 +1737,14 @@ Expecting one of '${allowedValues.join("', '")}'`);
    * @return {Command} `this` command for chaining
    * @private
    */
-  _optionEx(config, flags, description, fn, defaultValue) {
+  _optionEx(config2, flags, description, fn, defaultValue) {
     if (typeof flags === "object" && flags instanceof Option) {
       throw new Error(
         "To add an Option object use addOption() instead of option() or requiredOption()"
       );
     }
     const option = this.createOption(flags, description);
-    option.makeOptionMandatory(!!config.mandatory);
+    option.makeOptionMandatory(!!config2.mandatory);
     if (typeof fn === "function") {
       option.default(defaultValue).argParser(fn);
     } else if (fn instanceof RegExp) {
@@ -2730,9 +2730,9 @@ Expecting one of '${allowedValues.join("', '")}'`);
       this._outputConfiguration.writeErr("\n");
       this.outputHelp({ error: true });
     }
-    const config = errorOptions || {};
-    const exitCode = config.exitCode || 1;
-    const code = config.code || "commander.error";
+    const config2 = errorOptions || {};
+    const exitCode = config2.exitCode || 1;
+    const code = config2.code || "commander.error";
     this._exit(exitCode, code, message);
   }
   /**
@@ -3884,9 +3884,9 @@ async function loadConfig() {
     return { defaultModel: "openai:gpt-4o", models: {} };
   }
 }
-async function saveConfig(config) {
+async function saveConfig(config2) {
   await mkdir(CONFIG_DIR, { recursive: true });
-  await writeFile(CONFIG_FILE, JSON.stringify(config, null, 2), "utf-8");
+  await writeFile(CONFIG_FILE, JSON.stringify(config2, null, 2), "utf-8");
 }
 
 // node_modules/openai/internal/tslib.mjs
@@ -5434,11 +5434,11 @@ var SUBJECT_TOKEN_TYPES = {
 };
 var TOKEN_EXCHANGE_GRANT_TYPE = "urn:ietf:params:oauth:grant-type:token-exchange";
 var WorkloadIdentityAuth = class {
-  constructor(config, fetch2) {
+  constructor(config2, fetch2) {
     this.cachedToken = null;
     this.refreshPromise = null;
     this.tokenExchangeUrl = "https://auth.openai.com/oauth/token";
-    this.config = config;
+    this.config = config2;
     this.fetch = fetch2 ?? getDefaultFetch();
   }
   async getToken() {
@@ -14028,12 +14028,12 @@ var OpenAIProvider = class {
   name = "openai";
   client;
   model;
-  constructor(config) {
+  constructor(config2) {
     this.client = new OpenAI({
-      apiKey: config.apiKey ?? process.env.OPENAI_API_KEY,
-      baseURL: config.baseUrl
+      apiKey: config2.apiKey ?? process.env.OPENAI_API_KEY,
+      baseURL: config2.baseUrl
     });
-    this.model = config.model ?? "gpt-4o";
+    this.model = config2.model ?? "gpt-4o";
   }
   async chat(messages, tools) {
     const params = {
@@ -14106,9 +14106,9 @@ var AnthropicProvider = class {
   name = "anthropic";
   client;
   model;
-  constructor(config) {
-    this.client = new Anthropic({ apiKey: config.apiKey ?? process.env.ANTHROPIC_API_KEY });
-    this.model = config.model ?? "claude-sonnet-4-20250514";
+  constructor(config2) {
+    this.client = new Anthropic({ apiKey: config2.apiKey ?? process.env.ANTHROPIC_API_KEY });
+    this.model = config2.model ?? "claude-sonnet-4-20250514";
   }
   async chat(messages, tools) {
     const system = messages.find((m) => m.role === "system")?.content ?? "";
@@ -14170,9 +14170,9 @@ var OllamaProvider = class {
   name = "ollama";
   baseUrl;
   model;
-  constructor(config) {
-    this.baseUrl = config.baseUrl ?? "http://localhost:11434";
-    this.model = config.model ?? "llama3.1";
+  constructor(config2) {
+    this.baseUrl = config2.baseUrl ?? "http://localhost:11434";
+    this.model = config2.model ?? "llama3.1";
   }
   async chat(messages, tools) {
     const res = await fetch(`${this.baseUrl}/api/chat`, {
@@ -14247,13 +14247,13 @@ registry.set("openai", (c) => new OpenAIProvider(c));
 registry.set("claude", (c) => new AnthropicProvider(c));
 registry.set("anthropic", (c) => new AnthropicProvider(c));
 registry.set("ollama", (c) => new OllamaProvider(c));
-function createProvider(name, config) {
+function createProvider(name, config2) {
   const factory = registry.get(name);
   if (!factory)
     throw new Error(
       `Unknown model provider: ${name}. Available: ${[...registry.keys()].join(", ")}`
     );
-  return factory(config);
+  return factory(config2);
 }
 function parseModelString(modelStr) {
   const [provider, ...rest] = modelStr.split(":");
@@ -14385,26 +14385,50 @@ function matchSkill(skills, userMessage) {
 }
 
 // src/cli/repl.ts
-var VERSION2 = "0.6.3";
-function drawBox(lines, color = source_default) {
-  const maxLen = Math.max(...lines.map((l) => l.replace(/\x1b\[[0-9;]*m/g, "").length));
-  const border = color("\u2500".repeat(maxLen + 2));
-  console.log(color(`\u250C${border}\u2510`));
-  for (const line of lines) {
-    const plainLen = line.replace(/\x1b\[[0-9;]*m/g, "").length;
-    const padding = " ".repeat(maxLen - plainLen);
-    console.log(color("\u2502") + " " + line + padding + " " + color("\u2502"));
-  }
-  console.log(color(`\u2514${border}\u2518`));
+var VERSION2 = "0.8.0";
+var theme = {
+  primary: source_default.hex("#00d4ff"),
+  secondary: source_default.hex("#8b5cf6"),
+  accent: source_default.hex("#ec4899"),
+  success: source_default.hex("#10b981"),
+  warning: source_default.hex("#f59e0b"),
+  error: source_default.hex("#ef4444"),
+  muted: source_default.hex("#6b7280"),
+  dim: source_default.hex("#374151"),
+  bg: source_default.hex("#1a1a2e"),
+  border: source_default.hex("#2d2d44")
+};
+function drawLine(char = "\u2500", width = 56) {
+  console.log(theme.dim(char.repeat(width)));
 }
-function drawSeparator() {
-  console.log(source_default.gray("\u2500".repeat(50)));
+function getSkillIcon(name) {
+  const icons = {
+    coding: "\u03BB",
+    testing: "\u2713",
+    "code-review": "\u25CE",
+    docs: "\xB6",
+    git: "\u2387",
+    explain: "?",
+    refactor: "\u21C4",
+    security: "\u2295",
+    perf: "\u26A1",
+    content: "\u270E",
+    "data-analysis": "\u25A3",
+    translate: "\u2194",
+    ppt: "\u25EB",
+    planning: "\u229E",
+    finance: "$"
+  };
+  return icons[name] || "\u2022";
+}
+function getStatusIcon(status) {
+  return status ? theme.success("\u25CF") : theme.dim("\u25CB");
 }
 async function startREPL(model, opts) {
   const session = new SessionMemory();
   const skillsDir = join4(process.cwd(), "skills");
   const skills = await loadSkills(skillsDir);
-  const config = await loadConfig();
+  const config2 = await loadConfig();
   let skillInstructions;
   if (opts?.skillName) {
     const skill = skills.find((s) => s.name === opts.skillName);
@@ -14414,71 +14438,23 @@ async function startREPL(model, opts) {
     model,
     skillInstructions,
     onToolStart: (name) => {
-      console.log(source_default.yellow(`  \u2699 ${name}...`));
+      console.log(theme.warning(`  \u25CC ${name}...`));
     },
     onToolEnd: (_name, result) => {
       if (result.error) {
-        console.log(source_default.red(`  \u2717 ${_name}: ${result.error}`));
+        console.log(theme.error(`  \u2717 ${_name}: ${result.error}`));
       } else {
-        console.log(source_default.green(`  \u2713 ${_name}`));
+        console.log(theme.success(`  \u2713 ${_name}`));
       }
     }
   });
   const rl = readline.createInterface({
     input: process.stdin,
     output: process.stdout,
-    prompt: source_default.cyan("hecode> ")
+    prompt: theme.primary("\u276F ") + theme.dim("")
   });
   console.clear();
-  const logo = [
-    source_default.cyan("  \u2588\u2588\u2557  \u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557"),
-    source_default.cyan("  \u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u2588\u2588\u2554\u2550\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D"),
-    source_default.cyan("  \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2551\u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2551     \u2588\u2588\u2551   \u2588\u2588\u2551\u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2588\u2588\u2588\u2557  "),
-    source_default.cyan("  \u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u255D  \u2588\u2588\u2551     \u2588\u2588\u2551   \u2588\u2588\u2551\u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u255D  "),
-    source_default.cyan("  \u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u255A\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u255A\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557"),
-    source_default.cyan("  \u255A\u2550\u255D  \u255A\u2550\u255D\u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D")
-  ];
-  logo.forEach((line) => console.log(line));
-  console.log();
-  drawBox([
-    source_default.bold.white("Your AI Coding Assistant") + source_default.gray(` v${VERSION2}`),
-    source_default.gray("\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500"),
-    source_default.white("Model:   ") + source_default.cyan(config.defaultModel),
-    source_default.white("Session: ") + source_default.gray(session.sessionId.substring(0, 12) + "..."),
-    source_default.white("Skills:  ") + source_default.gray(`${skills.length} loaded`)
-  ]);
-  console.log();
-  if (skills.length > 0) {
-    console.log(source_default.bold.white("  Available Skills:"));
-    drawSeparator();
-    const skillChunks = [];
-    for (let i = 0; i < skills.length; i += 3) {
-      skillChunks.push(skills.slice(i, i + 3));
-    }
-    for (const chunk of skillChunks) {
-      const line = chunk.map((s) => {
-        const icon = getSkillIcon(s.name);
-        return source_default.cyan(`${icon} ${s.name.padEnd(12)}`);
-      }).join("");
-      console.log("  " + line);
-    }
-    console.log();
-  }
-  console.log(source_default.bold.white("  Quick Commands:"));
-  drawSeparator();
-  console.log(source_default.gray("  /help      Show help          /skills    List skills"));
-  console.log(source_default.gray("  /config    Show config         /clear     Clear chat"));
-  console.log(source_default.gray("  /model     Switch model        /save      Save session"));
-  console.log();
-  console.log(source_default.bold.white("  Usage:"));
-  drawSeparator();
-  console.log(source_default.gray("  \u2022 Type naturally to chat with AI"));
-  console.log(source_default.gray('  \u2022 Use --skill flag: hecode --skill coding "write a sort"'));
-  console.log(source_default.gray("  \u2022 Skills auto-activate based on keywords"));
-  console.log();
-  drawSeparator();
-  console.log(source_default.gray("  Press Ctrl+C to exit"));
-  console.log();
+  renderHeader(config2, skills, session);
   rl.prompt();
   rl.on("line", async (line) => {
     const input = line.trim();
@@ -14495,144 +14471,156 @@ async function startREPL(model, opts) {
       const matched = matchSkill(skills, input);
       if (matched) {
         const icon = getSkillIcon(matched.name);
-        console.log(source_default.gray(`
-  ${icon} Skill: ${matched.name}`));
+        console.log(theme.muted(`
+  ${icon} ${matched.name}
+`));
       }
     }
     try {
-      console.log();
       let hasOutput = false;
       for await (const token of orchestrator.runStream(input)) {
         process.stdout.write(token);
         hasOutput = true;
       }
-      if (hasOutput) {
-        console.log("\n");
-      }
+      if (hasOutput) console.log("\n");
       await session.save(orchestrator.getMessages());
     } catch (e) {
-      console.log(source_default.red(`
-  \u2717 Error: ${e.message}
+      console.log(theme.error(`
+  \u2717 ${e.message}
 `));
     }
     rl.prompt();
   });
   rl.on("close", () => {
-    console.log(source_default.gray("\n  Goodbye! \u{1F44B}\n"));
+    console.log(theme.muted("\n  bye\n"));
     process.exit(0);
   });
 }
-function getSkillIcon(name) {
-  const icons = {
-    coding: "\u{1F4BB}",
-    testing: "\u{1F9EA}",
-    "code-review": "\u{1F50D}",
-    docs: "\u{1F4DD}",
-    git: "\u{1F500}",
-    explain: "\u{1F4A1}",
-    refactor: "\u267B\uFE0F",
-    security: "\u{1F512}",
-    perf: "\u26A1",
-    content: "\u270D\uFE0F",
-    "data-analysis": "\u{1F4CA}",
-    translate: "\u{1F310}",
-    ppt: "\u{1F3A8}",
-    planning: "\u{1F4CB}",
-    finance: "\u{1F4B0}"
-  };
-  return icons[name] || "\u{1F4CC}";
+function renderHeader(config2, skills, session) {
+  const logo = [
+    theme.primary("  \u2566 \u2566\u2554\u2550\u2557\u2554\u2550\u2557\u2554\u2550\u2557\u2554\u2550\u2557\u2554\u2550\u2557"),
+    theme.primary("  \u2560\u2550\u2563\u2551\u2563 \u2560\u2550\u2563\u2551  \u2551  \u2551\u2563 "),
+    theme.primary("  \u2569 \u2569\u255A\u2550\u255D\u2569 \u2569\u255A\u2550\u255D\u255A\u2550\u255D\u255A\u2550\u255D") + theme.dim("  v" + VERSION2)
+  ];
+  logo.forEach((l) => console.log(l));
+  console.log();
+  drawLine("\u2500");
+  console.log(
+    theme.muted("  model ") + theme.primary(config2.defaultModel) + theme.muted("  \u2502  ") + theme.muted("skills ") + theme.success(`${skills.length}`) + theme.muted("  \u2502  ") + theme.muted("session ") + theme.dim(session.sessionId.substring(0, 8))
+  );
+  drawLine("\u2500");
+  console.log();
+  const cols = 3;
+  const colWidth = 20;
+  for (let i = 0; i < skills.length; i += cols) {
+    const row = skills.slice(i, i + cols);
+    const cells = row.map((s) => {
+      const icon = getSkillIcon(s.name);
+      const name = s.name.padEnd(14);
+      return `  ${theme.primary(icon)} ${theme.dim(name)}`;
+    });
+    console.log(cells.join(""));
+  }
+  console.log();
+  drawLine("\u2500");
+  console.log(theme.muted("  /help /skills /config /model /clear /save /exit"));
+  drawLine("\u2500");
+  console.log();
 }
 function handleCommand(input, orchestrator, session, skills) {
   const [cmd, ...args] = input.split(" ");
   switch (cmd) {
     case "/help":
       console.log();
-      console.log(source_default.bold.white("  Commands:"));
-      drawSeparator();
-      console.log(source_default.cyan("  /help      ") + source_default.gray("Show this help"));
-      console.log(source_default.cyan("  /skills    ") + source_default.gray("List all skills with triggers"));
-      console.log(source_default.cyan("  /config    ") + source_default.gray("Show current configuration"));
-      console.log(source_default.cyan("  /model     ") + source_default.gray("Show/switch model"));
-      console.log(source_default.cyan("  /clear     ") + source_default.gray("Clear conversation"));
-      console.log(source_default.cyan("  /history   ") + source_default.gray("Show message count"));
-      console.log(source_default.cyan("  /save      ") + source_default.gray("Save session"));
-      console.log(source_default.cyan("  /session   ") + source_default.gray("Show session ID"));
-      console.log(source_default.cyan("  /exit      ") + source_default.gray("Exit hecode"));
+      console.log(theme.primary("  Commands"));
+      drawLine();
+      const cmds = [
+        ["/help", "Show this help"],
+        ["/skills", "List all skills"],
+        ["/config", "Show configuration"],
+        ["/model [name]", "Show or switch model"],
+        ["/clear", "Clear conversation"],
+        ["/save", "Save session"],
+        ["/session", "Show session ID"],
+        ["/exit", "Exit hecode"]
+      ];
+      for (const [name, desc] of cmds) {
+        console.log(theme.primary(`  ${name.padEnd(18)}`) + theme.muted(desc));
+      }
       console.log();
       break;
     case "/skills":
       console.log();
-      console.log(source_default.bold.white("  Skills:"));
-      drawSeparator();
+      console.log(theme.primary("  Skills"));
+      drawLine();
       for (const skill of skills) {
         const icon = getSkillIcon(skill.name);
-        const name = source_default.cyan(skill.name.padEnd(15));
-        const triggers = skill.trigger ? skill.trigger.split(",").slice(0, 3).map((t) => t.trim()).join(", ") : "none";
-        console.log(`  ${icon} ${name} ${source_default.gray(triggers)}`);
+        const name = skill.name.padEnd(15);
+        const triggers = skill.trigger ? skill.trigger.split(",").slice(0, 3).map((t) => t.trim()).join(", ") : "";
+        console.log(`  ${theme.primary(icon)} ${theme.dim(name)} ${theme.muted(triggers)}`);
       }
       console.log();
       break;
     case "/config":
-      loadConfig().then((config) => {
+      loadConfig().then((cfg) => {
         console.log();
-        console.log(source_default.bold.white("  Configuration:"));
-        drawSeparator();
-        console.log(source_default.white("  Default Model: ") + source_default.cyan(config.defaultModel));
-        console.log(source_default.white("  Models:"));
-        for (const [name, cfg] of Object.entries(config.models || {})) {
-          const hasKey = cfg.apiKey ? source_default.green("\u2713") : source_default.red("\u2717");
-          console.log(`    ${hasKey} ${source_default.cyan(name)}`);
+        console.log(theme.primary("  Configuration"));
+        drawLine();
+        console.log(theme.muted("  Model:   ") + theme.primary(cfg.defaultModel));
+        console.log(theme.muted("  Providers:"));
+        for (const [name, c] of Object.entries(cfg.models || {})) {
+          const hasKey = c.apiKey ? getStatusIcon(true) : getStatusIcon(false);
+          console.log(`    ${hasKey} ${theme.dim(name)}`);
         }
         console.log();
       });
       break;
     case "/model":
       if (args[0]) {
-        console.log(source_default.green(`
-  \u2713 Switched to: ${args[0]}
+        console.log(theme.success(`
+  \u2713 ${args[0]}
 `));
       } else {
-        loadConfig().then((config) => {
-          console.log(source_default.cyan(`
-  Current: ${config.defaultModel}
+        loadConfig().then((cfg) => {
+          console.log(theme.primary(`
+  ${cfg.defaultModel}
 `));
-          console.log(source_default.gray("  Usage: /model provider:model"));
-          console.log(source_default.gray("  Example: /model claude:sonnet\n"));
+          console.log(theme.muted("  /model provider:model\n"));
         });
       }
       break;
     case "/clear":
       orchestrator.clearMessages();
       console.clear();
-      console.log(source_default.green("  \u2713 Conversation cleared\n"));
+      renderHeader(config, skills, session);
       break;
     case "/history":
-      console.log(source_default.gray(`
-  Messages: ${orchestrator.getMessages().length}
+      console.log(theme.muted(`
+  ${orchestrator.getMessages().length} messages
 `));
       break;
     case "/save":
       session.save(orchestrator.getMessages()).then(() => {
-        console.log(source_default.green("\n  \u2713 Session saved\n"));
+        console.log(theme.success("\n  \u2713 saved\n"));
       });
       break;
     case "/session":
-      console.log(source_default.gray(`
-  Session ID: ${session.sessionId}
+      console.log(theme.muted(`
+  ${session.sessionId}
 `));
       break;
     case "/exit":
-      console.log(source_default.gray("\n  Goodbye! \u{1F44B}\n"));
+      console.log(theme.muted("\n  bye\n"));
       process.exit(0);
     default:
-      console.log(source_default.red(`
-  Unknown command: ${cmd}`));
-      console.log(source_default.gray("  Type /help for available commands\n"));
+      console.log(theme.error(`
+  ? ${cmd}`));
+      console.log(theme.muted("  /help\n"));
   }
 }
 
 // src/cli/commands.ts
-var VERSION3 = "0.7.0";
+var VERSION3 = "0.8.0";
 function checkApiKey(provider, modelConfig) {
   const envKeys = {
     openai: "OPENAI_API_KEY",
@@ -14661,10 +14649,10 @@ function createProgram() {
   program3.option("-s, --skill <skill>", "Skill to activate");
   program3.argument("[prompt...]", "Single-shot prompt (non-interactive)");
   program3.action(async (prompt, opts) => {
-    const config = await loadConfig();
-    const modelStr = opts.model ?? config.defaultModel;
+    const config2 = await loadConfig();
+    const modelStr = opts.model ?? config2.defaultModel;
     const { provider, model } = parseModelString(modelStr);
-    const modelConfig = config.models[provider] ?? {};
+    const modelConfig = config2.models[provider] ?? {};
     if (model) modelConfig.model = model;
     if (!checkApiKey(provider, modelConfig)) {
       return;
@@ -14684,14 +14672,14 @@ function createProgram() {
   configCmd.argument("[key]", "Config key to get/set");
   configCmd.argument("[value]", "Value to set");
   configCmd.action(async (key, value) => {
-    const config = await loadConfig();
+    const config2 = await loadConfig();
     if (!key) {
-      console.log(JSON.stringify(config, null, 2));
+      console.log(JSON.stringify(config2, null, 2));
       return;
     }
     if (!value) {
       const keys2 = key.split(".");
-      let result = config;
+      let result = config2;
       for (const k of keys2) {
         result = result?.[k];
       }
@@ -14699,13 +14687,13 @@ function createProgram() {
       return;
     }
     const keys = key.split(".");
-    let target = config;
+    let target = config2;
     for (let i = 0; i < keys.length - 1; i++) {
       if (!target[keys[i]]) target[keys[i]] = {};
       target = target[keys[i]];
     }
     target[keys[keys.length - 1]] = value;
-    await saveConfig(config);
+    await saveConfig(config2);
     console.log(source_default.green(`\u2713 Set ${key} = ${value}`));
   });
   const initCmd = program3.command("init");