@wanghuimvp/axon 0.2.0 → 0.4.0

package/README.md

@@ -1,8 +1,8 @@
 # Axon
 
-An agentic coding CLI. Axon streams from Anthropic and runs a multi-step tool loop over your codebase — it reads, searches, and reasons across your files to answer a prompt.
+A multi-provider agentic coding CLI. Axon runs a multi-step tool loop over your codebase — it reads, searches, edits, and runs commands to carry out a request — with an interactive chat or a one-shot non-interactive mode.
 
-> Foundation release (`0.0.x`). Non-interactive `axon -p` mode with read-only tools (always available) and write/edit/shell tools (enabled with `--yolo`). Interactive TUI is on the roadmap.
+> `0.x` — interactive TUI (`axon`) and non-interactive (`axon -p`), multi-provider (Anthropic / OpenAI + OpenAI-compatible / Gemini), read + write/edit/shell tools.
 
 ## Install
 
@@ -10,6 +10,29 @@ An agentic coding CLI. Axon streams from Anthropic and runs a multi-step tool lo
 npm install -g @wanghuimvp/axon
 ```
 
+## Interactive mode
+
+Run `axon` with no arguments to open the interactive chat:
+
+```bash
+axon
+```
+
+On first run without an API key, `axon` opens a short setup screen — paste a key for the active provider and it is saved to `~/.axon/config.json` (with restricted file permissions) so later runs start straight in the chat. You can also set the provider's environment variable instead (e.g. `ANTHROPIC_API_KEY`), which always takes precedence.
+
+Axon reads your project on startup — `package.json` and the first of `AGENTS.md` / `CLAUDE.md` / `README.md` — and gives the model that context.
+
+Type a request and press Enter. Axon streams its reasoning and tool calls. When it wants to write a file or run a command, you get an inline prompt:
+
+```
+🔒 write_file({"path":"NOTES.md", …})
+[a] allow once / [A] always this session / [d] deny
+```
+
+Press `a` to allow once, `A` to allow that tool for the rest of the session, or `d` to deny (the model is told and adapts). `Ctrl+C` quits. Pass `--yolo` to skip all prompts.
+
+For one-shot, non-interactive use, see [Usage](#usage) (`axon -p`).
+
 ## Providers
 
 Axon speaks to three backends. The OpenAI-compatible provider also drives any

package/dist/cli.js

@@ -54,7 +54,7 @@ function loadConfig() {
 }
 
 // src/config/configFile.ts
-import { readFileSync as readFileSync2, writeFileSync, mkdirSync } from "node:fs";
+import { readFileSync as readFileSync2, writeFileSync, mkdirSync, chmodSync } from "node:fs";
 import { homedir as homedir2 } from "node:os";
 import { join as join2, dirname } from "node:path";
 function configPath() {
@@ -86,6 +86,18 @@ function setConfigValue(key, value) {
   mkdirSync(dirname(path), { recursive: true });
   writeFileSync(path, JSON.stringify(cfg, null, 2) + "\n");
 }
+function setApiKey(provider, key) {
+  const cfg = readConfigFile();
+  const providers = cfg.providers ??= {};
+  (providers[provider] ??= {}).apiKey = key;
+  const path = configPath();
+  mkdirSync(dirname(path), { recursive: true });
+  writeFileSync(path, JSON.stringify(cfg, null, 2) + "\n", { mode: 384 });
+  try {
+    chmodSync(path, 384);
+  } catch {
+  }
+}
 
 // src/providers/registry.ts
 import Anthropic from "@anthropic-ai/sdk";
@@ -773,14 +785,395 @@ function truncate(s, max = 500) {
   return s.length > max ? `${s.slice(0, max)}\u2026` : s;
 }
 
+// src/ui/runTui.tsx
+import { useState as useState3, useRef } from "react";
+import { Box as Box6, Text as Text6, render } from "ink";
+
+// src/core/projectContext.ts
+import { readFileSync as readFileSync3 } from "node:fs";
+import { join as join4 } from "node:path";
+function truncate2(s, max) {
+  return s.length > max ? `${s.slice(0, max)}\u2026` : s;
+}
+function loadProjectContext(cwd) {
+  const parts = [];
+  try {
+    const pkg = JSON.parse(readFileSync3(join4(cwd, "package.json"), "utf8"));
+    const bits = [];
+    if (pkg.name) bits.push(`name: ${pkg.name}`);
+    if (pkg.description) bits.push(`description: ${pkg.description}`);
+    if (pkg.scripts && typeof pkg.scripts === "object") bits.push(`scripts: ${Object.keys(pkg.scripts).join(", ")}`);
+    if (bits.length) parts.push(`package.json \u2014 ${bits.join("; ")}`);
+  } catch {
+  }
+  for (const name of ["AGENTS.md", "CLAUDE.md", "README.md"]) {
+    try {
+      const text = readFileSync3(join4(cwd, name), "utf8").trim();
+      if (text) {
+        parts.push(`${name}:
+${truncate2(text, 2e3)}`);
+        break;
+      }
+    } catch {
+    }
+  }
+  return parts.join("\n\n");
+}
+
+// src/ui/permissionController.ts
+function createPermissionController() {
+  const sessionAllow = /* @__PURE__ */ new Set();
+  const subscribers = /* @__PURE__ */ new Set();
+  const queue = [];
+  let active = null;
+  let currentPending = null;
+  const notify = (p) => {
+    currentPending = p;
+    for (const fn of subscribers) fn(p);
+  };
+  const pump = () => {
+    if (active || queue.length === 0) return;
+    active = queue.shift();
+    notify({ req: active.req });
+  };
+  const gate = (req) => {
+    if (sessionAllow.has(req.name)) {
+      return Promise.resolve({ allow: true, reason: "allowed (remembered this session)" });
+    }
+    return new Promise((res) => {
+      queue.push({ req, settle: res });
+      pump();
+    });
+  };
+  const resolve3 = (decision) => {
+    if (!active) return;
+    const { req, settle } = active;
+    active = null;
+    if (decision === "always") sessionAllow.add(req.name);
+    settle({
+      allow: decision !== "deny",
+      reason: decision === "deny" ? `permission denied by user: ${req.name}` : `allowed (${decision})`
+    });
+    pump();
+    if (!active) notify(null);
+  };
+  const subscribe = (fn) => {
+    subscribers.add(fn);
+    return () => {
+      subscribers.delete(fn);
+    };
+  };
+  const getPending = () => currentPending;
+  return { gate, subscribe, getPending, resolve: resolve3 };
+}
+
+// src/config/credentials.ts
+function hasUsableKey(cfg) {
+  const k = cfg.providers[cfg.provider]?.apiKey;
+  return typeof k === "string" && k.trim().length > 0;
+}
+var INFO = {
+  anthropic: { envVar: "ANTHROPIC_API_KEY", url: "https://console.anthropic.com/settings/keys" },
+  openai: { envVar: "OPENAI_API_KEY", url: "https://platform.openai.com/api-keys" },
+  gemini: { envVar: "GEMINI_API_KEY", url: "https://aistudio.google.com/apikey" }
+};
+function keyProviderInfo(provider) {
+  return INFO[provider] ?? { envVar: `${provider.toUpperCase()}_API_KEY`, url: "" };
+}
+
+// src/ui/app.tsx
+import { useEffect, useState, useSyncExternalStore, useCallback } from "react";
+import { Box as Box4, Text as Text4, useInput as useInput2 } from "ink";
+import TextInput from "ink-text-input";
+
+// src/ui/components/MessageView.tsx
+import { Box, Text } from "ink";
+import { jsx, jsxs } from "react/jsx-runtime";
+function toolIcon(status) {
+  if (status === "running") return "\u23F3";
+  return status === "ok" ? "\u2705" : "\u274C";
+}
+function truncate3(s, max = 300) {
+  return s.length > max ? `${s.slice(0, max)}\u2026` : s;
+}
+function MessageView({ items }) {
+  return /* @__PURE__ */ jsx(Box, { flexDirection: "column", children: items.map((it, i) => {
+    if (it.kind === "user") {
+      return /* @__PURE__ */ jsxs(Text, { children: [
+        /* @__PURE__ */ jsxs(Text, { color: "cyan", bold: true, children: [
+          "You",
+          "  "
+        ] }),
+        it.text
+      ] }, i);
+    }
+    if (it.kind === "assistant") {
+      return /* @__PURE__ */ jsxs(Text, { children: [
+        /* @__PURE__ */ jsxs(Text, { color: "green", bold: true, children: [
+          "Axon",
+          " "
+        ] }),
+        it.text
+      ] }, i);
+    }
+    if (it.kind === "error") {
+      return /* @__PURE__ */ jsxs(Text, { color: "red", children: [
+        "\u{1F4A5} ",
+        it.text
+      ] }, i);
+    }
+    if (it.kind === "tool") {
+      return /* @__PURE__ */ jsxs(Text, { color: "gray", children: [
+        toolIcon(it.status),
+        " ",
+        it.name,
+        "(",
+        truncate3(JSON.stringify(it.args ?? {}), 60),
+        ")",
+        it.output ? ` \u2014 ${truncate3(it.output)}` : ""
+      ] }, i);
+    }
+    return null;
+  }) });
+}
+
+// src/ui/components/PermissionPrompt.tsx
+import { Box as Box2, Text as Text2, useInput } from "ink";
+import { jsx as jsx2, jsxs as jsxs2 } from "react/jsx-runtime";
+function summarize(args) {
+  const s = JSON.stringify(args ?? {});
+  return s.length > 80 ? `${s.slice(0, 80)}\u2026` : s;
+}
+function decisionForKey(input) {
+  if (input === "a") return "once";
+  if (input === "A") return "always";
+  if (input === "d") return "deny";
+  return null;
+}
+function PermissionPrompt({ req, onDecide }) {
+  useInput((input) => {
+    const d = decisionForKey(input);
+    if (d) onDecide(d);
+  });
+  return /* @__PURE__ */ jsxs2(Box2, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsxs2(Text2, { color: "yellow", children: [
+      "\u{1F512} ",
+      req.name,
+      "(",
+      summarize(req.args),
+      ")"
+    ] }),
+    /* @__PURE__ */ jsx2(Text2, { color: "yellow", children: "[a] allow once / [A] always this session / [d] deny" })
+  ] });
+}
+
+// src/ui/components/StatusBar.tsx
+import { Box as Box3, Text as Text3 } from "ink";
+import { jsx as jsx3, jsxs as jsxs3 } from "react/jsx-runtime";
+function StatusBar({ provider, model, running, yolo }) {
+  return /* @__PURE__ */ jsx3(Box3, { children: /* @__PURE__ */ jsxs3(Text3, { color: "gray", children: [
+    "[axon: ",
+    provider,
+    "/",
+    model,
+    yolo ? " \xB7 yolo" : "",
+    " \xB7 ",
+    running ? "working\u2026" : "ready",
+    " \xB7 ^C quit]"
+  ] }) });
+}
+
+// src/ui/app.tsx
+import { jsx as jsx4, jsxs as jsxs4 } from "react/jsx-runtime";
+function reduceEvent(items, e) {
+  switch (e.type) {
+    case "text_delta": {
+      const last = items[items.length - 1];
+      if (last && last.kind === "assistant") {
+        return [...items.slice(0, -1), { kind: "assistant", text: last.text + e.text }];
+      }
+      return [...items, { kind: "assistant", text: e.text }];
+    }
+    case "tool_start":
+      return [...items, { kind: "tool", id: e.id, name: e.name, args: e.args, status: "running" }];
+    case "tool_end":
+      return items.map(
+        (it) => it.kind === "tool" && it.id === e.id ? { ...it, status: e.ok ? "ok" : "fail", output: e.output } : it
+      );
+    default:
+      return items;
+  }
+}
+function usePendingPermission(controller) {
+  const subscribe = useCallback((onChange) => controller.subscribe(() => onChange()), [controller]);
+  const getSnapshot = useCallback(() => controller.getPending(), [controller]);
+  const pending = useSyncExternalStore(subscribe, getSnapshot, getSnapshot);
+  return pending ? pending.req : null;
+}
+function App({ engine, controller, provider, model, yolo }) {
+  const [items, setItems] = useState([]);
+  const [input, setInput] = useState("");
+  const [running, setRunning] = useState(false);
+  const pending = usePendingPermission(controller);
+  useEffect(() => {
+    engine.on((e) => setItems((prev) => reduceEvent(prev, e)));
+  }, [engine]);
+  useInput2(
+    (_input, key) => {
+      if (key.escape && !running) setInput("");
+    },
+    { isActive: !pending }
+  );
+  const handleSubmit = (text) => {
+    if (!text.trim() || running) return;
+    setItems((prev) => [...prev, { kind: "user", text }]);
+    setInput("");
+    setRunning(true);
+    engine.submit(text).catch((err) => {
+      const msg = err instanceof Error ? err.message : String(err);
+      setItems((prev) => [...prev, { kind: "error", text: msg }]);
+    }).finally(() => setRunning(false));
+  };
+  return /* @__PURE__ */ jsxs4(Box4, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsx4(MessageView, { items }),
+    pending ? /* @__PURE__ */ jsx4(PermissionPrompt, { req: pending, onDecide: (d) => controller.resolve(d) }) : running ? /* @__PURE__ */ jsx4(Text4, { color: "gray", children: "\u2026 working (Ctrl+C to quit)" }) : /* @__PURE__ */ jsxs4(Box4, { children: [
+      /* @__PURE__ */ jsx4(Text4, { color: "cyan", children: "\u203A " }),
+      /* @__PURE__ */ jsx4(TextInput, { value: input, onChange: setInput, onSubmit: handleSubmit })
+    ] }),
+    /* @__PURE__ */ jsx4(StatusBar, { provider, model, running, yolo })
+  ] });
+}
+
+// src/ui/Setup.tsx
+import { useState as useState2 } from "react";
+import { Box as Box5, Text as Text5 } from "ink";
+import TextInput2 from "ink-text-input";
+import { jsx as jsx5, jsxs as jsxs5 } from "react/jsx-runtime";
+function trimmedKey(v) {
+  const t = v.trim();
+  return t ? t : null;
+}
+function Setup({
+  provider,
+  info,
+  onSubmit
+}) {
+  const [value, setValue] = useState2("");
+  const submit = (v) => {
+    const k = trimmedKey(v);
+    if (k) onSubmit(k);
+  };
+  return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsxs5(Text5, { color: "yellow", children: [
+      'No API key found for "',
+      provider,
+      '".'
+    ] }),
+    /* @__PURE__ */ jsxs5(Text5, { children: [
+      "Set ",
+      /* @__PURE__ */ jsx5(Text5, { bold: true, children: info.envVar }),
+      " in your environment, or paste a key below to save it to ~/.axon/config.json."
+    ] }),
+    info.url ? /* @__PURE__ */ jsxs5(Text5, { color: "gray", children: [
+      "Get a key at: ",
+      info.url
+    ] }) : null,
+    /* @__PURE__ */ jsxs5(Box5, { children: [
+      /* @__PURE__ */ jsx5(Text5, { color: "cyan", children: "key \u203A " }),
+      /* @__PURE__ */ jsx5(TextInput2, { value, onChange: setValue, onSubmit: submit, mask: "*" })
+    ] })
+  ] });
+}
+
+// src/ui/runTui.tsx
+import { jsx as jsx6, jsxs as jsxs6 } from "react/jsx-runtime";
+var TUI_SYSTEM = `You are Axon, an interactive agentic coding assistant. Use the tools to inspect and modify the project: read_file, list_dir, glob, grep (read-only) and write_file, edit_file, shell (these change the workspace; the user is prompted to approve each). Prefer edit_file for surgical changes. Explain briefly what you are doing.`;
+function Root({ deps }) {
+  const { cfg, controller, yolo, welcome, buildEngine, persistKey } = deps;
+  const [ready, setReady] = useState3(hasUsableKey(cfg));
+  const engineRef = useRef(null);
+  if (ready && !engineRef.current) engineRef.current = buildEngine();
+  if (!ready || !engineRef.current) {
+    const info = keyProviderInfo(cfg.provider);
+    return /* @__PURE__ */ jsx6(
+      Setup,
+      {
+        provider: cfg.provider,
+        info,
+        onSubmit: (key) => {
+          persistKey(cfg.provider, key);
+          cfg.providers[cfg.provider] = { ...cfg.providers[cfg.provider] ?? {}, apiKey: key };
+          setReady(true);
+        }
+      }
+    );
+  }
+  return /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsx6(Text6, { color: "green", children: welcome }),
+    /* @__PURE__ */ jsx6(
+      App,
+      {
+        engine: engineRef.current,
+        controller,
+        provider: cfg.provider,
+        model: resolveModel(cfg),
+        yolo
+      }
+    )
+  ] });
+}
+function runTui(opts) {
+  const cfg = loadConfig();
+  if (opts.provider) cfg.provider = opts.provider;
+  if (opts.model) cfg.model = opts.model;
+  const controller = createPermissionController();
+  const deps = {
+    cfg,
+    controller,
+    yolo: Boolean(opts.yolo),
+    welcome: `Axon \xB7 ${cfg.provider}/${resolveModel(cfg)} \xB7 type a request, Ctrl+C to quit`,
+    persistKey: (provider, key) => {
+      try {
+        setApiKey(provider, key);
+      } catch {
+      }
+    },
+    buildEngine: () => {
+      const provider = createProvider(cfg);
+      const tools = buildAllTools();
+      const gate = opts.yolo ? allowAllGate : controller.gate;
+      const context = loadProjectContext(process.cwd());
+      const system = TUI_SYSTEM + (context ? `
+
+Project context:
+${context}` : "");
+      return new Engine({ provider, tools, system, cwd: process.cwd(), gate });
+    }
+  };
+  if (!process.stdin.isTTY) {
+    process.stderr.write('axon: the interactive chat needs a terminal. For non-interactive use, run: axon -p "your prompt"\n');
+    process.exit(1);
+  }
+  render(/* @__PURE__ */ jsx6(Root, { deps }));
+}
+
 // src/cli.ts
 var READONLY_SYSTEM = `You are Axon, an agentic coding assistant. Use the read-only tools \u2014 read_file, list_dir, glob, grep \u2014 to inspect the project and answer precisely. When done, stop calling tools.`;
 var YOLO_SYSTEM = `You are Axon, an agentic coding assistant. Use the provided tools to inspect AND modify the project: read_file, list_dir, glob, grep (read-only), and write_file, edit_file, shell (these change the workspace). Prefer edit_file for surgical changes. When done, stop calling tools.`;
+function redactKeys(cfg) {
+  const providers = cfg.providers;
+  if (!providers || typeof providers !== "object") return cfg;
+  const masked = {};
+  for (const [name, p] of Object.entries(providers)) {
+    masked[name] = p && typeof p === "object" && "apiKey" in p && p.apiKey ? { ...p, apiKey: "***redacted***" } : p;
+  }
+  return { ...cfg, providers: masked };
+}
 var program = new Command();
 program.name("axon").version(VERSION).option("-p, --print <prompt>", "run one prompt non-interactively and stream the result").option("--provider <name>", "override the provider for this run (anthropic | openai | gemini)").option("--model <name>", "override the model for this run").option("--yolo", "allow write/edit/shell tools without prompting (non-interactive)");
 program.command("config").argument("<action>", "get | set").argument("[key]", "config key (provider | model | <provider>.<baseUrl|model>)").argument("[value]", "value to set").action((action, key, value) => {
   if (action === "get") {
-    process.stdout.write(JSON.stringify(readConfigFile(), null, 2) + "\n");
+    process.stdout.write(JSON.stringify(redactKeys(readConfigFile()), null, 2) + "\n");
     return;
   }
   if (action === "set") {
@@ -803,7 +1196,7 @@ program.action(() => {
 program.parse();
 async function main(opts) {
   if (!opts.print) {
-    process.stdout.write('Interactive TUI not built yet \u2014 use: axon -p "your prompt"\n');
+    runTui({ provider: opts.provider, model: opts.model, yolo: opts.yolo });
     return;
   }
   const cfg = loadConfig();
@@ -812,7 +1205,12 @@ async function main(opts) {
   const provider = createProvider(cfg);
   const tools = opts.yolo ? buildAllTools() : buildReadOnlyTools();
   const gate = opts.yolo ? allowAllGate : denyGate;
-  const system = opts.yolo ? YOLO_SYSTEM : READONLY_SYSTEM;
+  const baseSystem = opts.yolo ? YOLO_SYSTEM : READONLY_SYSTEM;
+  const context = loadProjectContext(process.cwd());
+  const system = baseSystem + (context ? `
+
+Project context:
+${context}` : "");
   const engine = new Engine({ provider, tools, system, cwd: process.cwd(), gate });
   printRunner(engine, (s) => process.stdout.write(s));
   process.stderr.write(`[axon: ${cfg.provider} / ${resolveModel(cfg)}${opts.yolo ? " / yolo" : ""}]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wanghuimvp/axon",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "Axon — a multi-provider agentic coding CLI (Anthropic, OpenAI + OpenAI-compatible endpoints, Gemini). Runs a multi-step tool loop over your codebase.",
   "type": "module",
   "bin": {
@@ -38,7 +38,7 @@
   "scripts": {
     "dev": "tsx src/cli.ts",
     "test": "vitest run",
-    "build": "esbuild src/cli.ts --bundle --platform=node --format=esm --outfile=dist/cli.js --packages=external",
+    "build": "esbuild src/cli.ts --bundle --platform=node --format=esm --outfile=dist/cli.js --packages=external --jsx=automatic",
     "prepublishOnly": "npm test && npm run build"
   },
   "dependencies": {
@@ -46,11 +46,16 @@
     "@google/genai": "^1.52.0",
     "commander": "^12.0.0",
     "fast-glob": "^3.3.0",
-    "openai": "^4.104.0"
+    "ink": "^5.2.1",
+    "ink-text-input": "^6.0.0",
+    "openai": "^4.104.0",
+    "react": "^18.3.1"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
+    "@types/react": "^18.3.31",
     "esbuild": "^0.23.0",
+    "ink-testing-library": "^4.0.0",
     "tsx": "^4.0.0",
     "typescript": "^5.5.0",
     "vitest": "^2.0.0"