stackai 0.1.2 → 0.1.4

package/dist/cli.js

@@ -173,6 +173,40 @@ function toLLMError(err) {
 // ../core/dist/file-agent.js
 import { promises as fs } from "fs";
 import path from "path";
+var IGNORE_DIRS = /* @__PURE__ */ new Set([
+  "node_modules",
+  ".git",
+  "dist",
+  ".next",
+  ".turbo",
+  "build",
+  "out",
+  "coverage",
+  ".cache"
+]);
+function globToRegExp(glob) {
+  let re = "";
+  for (let i = 0; i < glob.length; i++) {
+    const c = glob[i];
+    if (c === "*") {
+      if (glob[i + 1] === "*") {
+        re += ".*";
+        i++;
+        if (glob[i + 1] === "/")
+          i++;
+      } else {
+        re += "[^/]*";
+      }
+    } else if (c === "?") {
+      re += "[^/]";
+    } else if (".+^${}()|[]\\".includes(c)) {
+      re += `\\${c}`;
+    } else {
+      re += c;
+    }
+  }
+  return new RegExp(`^${re}$`);
+}
 var FileAgent = class {
   root;
   constructor(cwd) {
@@ -220,6 +254,70 @@ var FileAgent = class {
   async createDir(relPath) {
     await fs.mkdir(this.resolve(relPath), { recursive: true });
   }
+  /** Recursively yield every file path under `dir`, skipping ignored dirs. */
+  async *walkFiles(dir) {
+    let entries;
+    try {
+      entries = await fs.readdir(dir, { withFileTypes: true });
+    } catch {
+      return;
+    }
+    for (const e of entries) {
+      if (e.isDirectory()) {
+        if (IGNORE_DIRS.has(e.name))
+          continue;
+        yield* this.walkFiles(path.join(dir, e.name));
+      } else {
+        yield path.join(dir, e.name);
+      }
+    }
+  }
+  rel(abs) {
+    return path.relative(this.root, abs).split(path.sep).join("/");
+  }
+  /** Search file contents (regex). Returns `path:line: text` matches. */
+  async grep(pattern, maxResults = 80) {
+    let re;
+    try {
+      re = new RegExp(pattern);
+    } catch {
+      re = new RegExp(pattern.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"));
+    }
+    const out = [];
+    for await (const abs of this.walkFiles(this.root)) {
+      if (out.length >= maxResults)
+        break;
+      let content;
+      try {
+        content = await fs.readFile(abs, "utf8");
+      } catch {
+        continue;
+      }
+      const lines = content.split("\n");
+      for (let i = 0; i < lines.length; i++) {
+        if (re.test(lines[i])) {
+          out.push(`${this.rel(abs)}:${i + 1}: ${lines[i].trim().slice(0, 160)}`);
+          if (out.length >= maxResults)
+            break;
+        }
+      }
+    }
+    return out;
+  }
+  /** Find files whose relative path matches a glob pattern. */
+  async glob(pattern, maxResults = 200) {
+    const re = globToRegExp(pattern);
+    const out = [];
+    for await (const abs of this.walkFiles(this.root)) {
+      const rel = this.rel(abs);
+      if (re.test(rel)) {
+        out.push(rel);
+        if (out.length >= maxResults)
+          break;
+      }
+    }
+    return out.sort((a, b) => a.localeCompare(b));
+  }
 };
 
 // ../core/dist/system-prompt.js
@@ -227,10 +325,14 @@ var SYSTEM_PROMPT = `You are StackAI, an autonomous coding agent operating insid
 
 You can read, write, and edit files by calling the provided tools. Be DECISIVE and make the FEWEST tool calls possible \u2014 every call is a slow round-trip.
 
+Tools: read_file, write_file, edit_file, list_files, create_dir, search_files (regex content search), find_files (glob).
+
 How to work:
-1. To create a new file: call write_file once. Do NOT list_files or read_file first.
-2. To change an existing file: call read_file ONCE to see its contents, then make ONE edit_file (or one write_file) call to apply the change.
-3. When the task is done, stop calling tools and reply with ONE short sentence summarizing what you changed.
+1. To explore an unfamiliar project, use find_files (e.g. "src/**/*.ts") and search_files (regex) instead of reading everything.
+2. To create a new file: call write_file once. Do NOT list_files or read_file first.
+3. To change an existing file: call read_file ONCE to see its contents, then make ONE edit_file (or one write_file) call to apply the change.
+4. If project instructions were provided (from STACKAI.md / AGENTS.md), follow them.
+5. When the task is done, stop calling tools and reply with ONE short sentence summarizing what you changed.
 
 Hard rules:
 - Make each distinct tool call AT MOST ONCE. Never repeat the same read or edit.
@@ -307,6 +409,30 @@ var TOOLS = [
         required: ["path"]
       }
     }
+  },
+  {
+    type: "function",
+    function: {
+      name: "search_files",
+      description: "Search file contents across the project with a regular expression. Returns matching `path:line: text`.",
+      parameters: {
+        type: "object",
+        properties: { pattern: { type: "string" } },
+        required: ["pattern"]
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "find_files",
+      description: "Find files by glob pattern (e.g. `src/**/*.ts`, `*.json`). Returns matching paths.",
+      parameters: {
+        type: "object",
+        properties: { pattern: { type: "string" } },
+        required: ["pattern"]
+      }
+    }
   }
 ];
 var AgentRunner = class {
@@ -318,16 +444,40 @@ var AgentRunner = class {
   }
   async run({ prompt, cwd, onStep }) {
     const files = new FileAgent(cwd);
-    const messages = [
-      { role: "system", content: SYSTEM_PROMPT },
-      { role: "user", content: prompt }
-    ];
+    const messages = await this.systemMessages(files);
+    messages.push({ role: "user", content: prompt });
     return this.runLoop(messages, files, onStep);
   }
   /** Start a stateful chat session that retains history across prompts. */
   session(cwd) {
     return new AgentSession(this, cwd);
   }
+  /**
+   * Build the leading system messages: the base system prompt plus any
+   * project context found in STACKAI.md / AGENTS.md at the project root.
+   * @internal
+   */
+  async systemMessages(files) {
+    const messages = [
+      { role: "system", content: SYSTEM_PROMPT }
+    ];
+    for (const name of ["STACKAI.md", "AGENTS.md"]) {
+      try {
+        const ctx = (await files.readFile(name)).trim();
+        if (ctx) {
+          messages.push({
+            role: "system",
+            content: `Project instructions from ${name}:
+
+${ctx}`
+          });
+          break;
+        }
+      } catch {
+      }
+    }
+    return messages;
+  }
   /**
    * Run the tool-call loop over an existing message history (mutated in place).
    * Shared by one-shot run() and the interactive AgentSession.
@@ -409,6 +559,14 @@ var AgentRunner = class {
       case "create_dir":
         await files.createDir(str(args.path));
         return `Created ${str(args.path)}`;
+      case "search_files": {
+        const matches = await files.grep(str(args.pattern));
+        return matches.join("\n") || "(no matches)";
+      }
+      case "find_files": {
+        const found = await files.glob(str(args.pattern));
+        return found.join("\n") || "(no files matched)";
+      }
       default:
         return `Error: unknown tool ${name}`;
     }
@@ -422,16 +580,19 @@ function str(value) {
 }
 var AgentSession = class {
   runner;
-  messages = [
-    { role: "system", content: SYSTEM_PROMPT }
-  ];
+  messages = [];
   files;
+  initialized = false;
   constructor(runner, cwd) {
     this.runner = runner;
     this.files = new FileAgent(cwd);
   }
   /** Send a user message; the agent acts with full prior context. */
   async send(prompt, onStep) {
+    if (!this.initialized) {
+      this.messages = await this.runner.systemMessages(this.files);
+      this.initialized = true;
+    }
     this.messages.push({ role: "user", content: prompt });
     return this.runner.runLoop(this.messages, this.files, onStep);
   }
@@ -501,36 +662,99 @@ import { Box as Box2, Text as Text2, useApp } from "ink";
 import Spinner from "ink-spinner";
 
 // src/ui/format.ts
-function describe(step) {
-  const p = step.args.path ?? step.args.dir ?? "";
+var ACCENT = "#e8ff47";
+var MAX_PREVIEW = 6;
+function splitLines(s) {
+  return s.replace(/\n+$/, "").split("\n");
+}
+function clip(s, n = 76) {
+  return s.length > n ? `${s.slice(0, n)}\u2026` : s;
+}
+function preview(lines, prefix, color) {
+  const out = lines.slice(0, MAX_PREVIEW).map((l) => ({ kind: "sub", color, text: `${prefix} ${clip(l)}` }));
+  if (lines.length > MAX_PREVIEW) {
+    out.push({
+      kind: "sub",
+      color: "gray",
+      text: `\u2026 ${lines.length - MAX_PREVIEW} more lines`
+    });
+  }
+  return out;
+}
+function toolEntries(step) {
+  const args = step.args;
+  const path3 = String(args.path ?? args.dir ?? "");
   switch (step.name) {
+    case "write_file": {
+      const lines = splitLines(String(args.content ?? ""));
+      return [
+        { kind: "tool", color: ACCENT, label: `Write(${path3})` },
+        { kind: "sub", color: "green", text: `+${lines.length} lines` },
+        ...preview(lines, "+", "green")
+      ];
+    }
+    case "edit_file": {
+      const removed = splitLines(String(args.oldStr ?? ""));
+      const added = splitLines(String(args.newStr ?? ""));
+      return [
+        { kind: "tool", color: ACCENT, label: `Edit(${path3})` },
+        {
+          kind: "sub",
+          color: "gray",
+          text: `+${added.length} -${removed.length} lines`
+        },
+        ...preview(removed, "-", "red"),
+        ...preview(added, "+", "green")
+      ];
+    }
     case "read_file":
-      return `Reading ${p}`;
-    case "write_file":
-      return `Writing ${p}`;
-    case "edit_file":
-      return `Editing ${p}`;
+      return [{ kind: "tool", color: ACCENT, label: `Read(${path3})` }];
     case "list_files":
-      return `Listing ${p || "."}`;
+      return [{ kind: "tool", color: ACCENT, label: `List(${path3 || "."})` }];
     case "create_dir":
-      return `Creating ${p}/`;
+      return [{ kind: "tool", color: ACCENT, label: `Create(${path3}/)` }];
+    case "search_files":
+      return [
+        { kind: "tool", color: ACCENT, label: `Search(${String(args.pattern ?? "")})` }
+      ];
+    case "find_files":
+      return [
+        { kind: "tool", color: ACCENT, label: `Find(${String(args.pattern ?? "")})` }
+      ];
     default:
-      return step.name;
+      return [{ kind: "tool", color: ACCENT, label: step.name }];
   }
 }
 function applyStep(entries, step) {
   if (step.type === "token") {
     const last = entries[entries.length - 1];
     if (last && last.kind === "text") {
-      return [...entries.slice(0, -1), { kind: "text", text: last.text + step.text }];
+      return [
+        ...entries.slice(0, -1),
+        { kind: "text", text: last.text + step.text }
+      ];
     }
     return [...entries, { kind: "text", text: step.text }];
   }
   if (step.type === "tool_call") {
-    return [...entries, { kind: "tool", color: "gray", text: describe(step) }];
+    return [...entries, ...toolEntries(step)];
   }
-  if (step.type === "tool_result" && !step.ok) {
-    return [...entries, { kind: "tool", color: "red", text: step.detail }];
+  if (step.type === "tool_result") {
+    if (!step.ok) {
+      return [
+        ...entries,
+        { kind: "sub", color: "red", text: `\u2717 ${step.detail}` }
+      ];
+    }
+    if (step.name === "read_file") {
+      const n = step.detail ? splitLines(step.detail).length : 0;
+      return [...entries, { kind: "sub", color: "gray", text: `read ${n} lines` }];
+    }
+    if (step.name === "search_files" || step.name === "find_files") {
+      const empty = /^\(no /.test(step.detail);
+      const n = empty ? 0 : splitLines(step.detail).length;
+      return [...entries, { kind: "sub", color: "gray", text: `${n} results` }];
+    }
   }
   return entries;
 }
@@ -539,12 +763,18 @@ function applyStep(entries, step) {
 import { Box, Text } from "ink";
 import { Fragment, jsx, jsxs } from "react/jsx-runtime";
 function EntryLines({ entries }) {
-  return /* @__PURE__ */ jsx(Fragment, { children: entries.map(
-    (e, i) => e.kind === "tool" ? /* @__PURE__ */ jsx(Box, { marginLeft: 2, children: /* @__PURE__ */ jsxs(Text, { color: e.color, children: [
-      "\u2192 ",
-      e.text
-    ] }) }, i) : /* @__PURE__ */ jsx(Box, { marginTop: 1, marginLeft: 2, children: /* @__PURE__ */ jsx(Text, { children: e.text }) }, i)
-  ) });
+  return /* @__PURE__ */ jsx(Fragment, { children: entries.map((e, i) => {
+    if (e.kind === "tool") {
+      return /* @__PURE__ */ jsx(Box, { marginTop: 1, children: /* @__PURE__ */ jsxs(Text, { color: e.color, bold: true, children: [
+        "\u25CF ",
+        e.label
+      ] }) }, i);
+    }
+    if (e.kind === "sub") {
+      return /* @__PURE__ */ jsx(Box, { marginLeft: 2, children: /* @__PURE__ */ jsx(Text, { color: e.color, children: e.text }) }, i);
+    }
+    return /* @__PURE__ */ jsx(Box, { marginTop: 1, marginLeft: 2, children: /* @__PURE__ */ jsx(Text, { children: e.text }) }, i);
+  }) });
 }
 
 // src/ui/run-view.tsx
@@ -597,7 +827,7 @@ import { Box as Box3, Text as Text3, useApp as useApp2 } from "ink";
 import Spinner2 from "ink-spinner";
 import TextInput from "ink-text-input";
 import { Fragment as Fragment2, jsx as jsx3, jsxs as jsxs3 } from "react/jsx-runtime";
-var ACCENT = "#e8ff47";
+var ACCENT2 = "#e8ff47";
 var LOGO = [
   "\u2588\u2580\u2580 \u2580\u2588\u2580 \u2588\u2580\u2588 \u2588\u2580\u2580 \u2588\u2584\u2580 \u2588\u2580\u2588 \u2588",
   "\u2584\u2584\u2588  \u2588  \u2588\u2580\u2588 \u2588\u2584\u2584 \u2588\u2580\u2584 \u2588\u2580\u2588 \u2588"
@@ -647,11 +877,11 @@ function Interactive({ session, cwd, version }) {
       {
         flexDirection: "column",
         borderStyle: "round",
-        borderColor: ACCENT,
+        borderColor: ACCENT2,
         paddingX: 2,
         paddingY: 1,
         children: [
-          LOGO.map((line, i) => /* @__PURE__ */ jsx3(Text3, { color: ACCENT, bold: true, children: line }, i)),
+          LOGO.map((line, i) => /* @__PURE__ */ jsx3(Text3, { color: ACCENT2, bold: true, children: line }, i)),
           /* @__PURE__ */ jsx3(Box3, { marginTop: 1, children: /* @__PURE__ */ jsxs3(Text3, { color: "gray", children: [
             "v",
             version,
@@ -664,12 +894,12 @@ function Interactive({ session, cwd, version }) {
     /* @__PURE__ */ jsx3(Box3, { marginTop: 1, marginBottom: 1, children: /* @__PURE__ */ jsxs3(Text3, { color: "gray", children: [
       "Describe what you want to build \xB7 type",
       " ",
-      /* @__PURE__ */ jsx3(Text3, { color: ACCENT, children: "/exit" }),
+      /* @__PURE__ */ jsx3(Text3, { color: ACCENT2, children: "/exit" }),
       " to quit"
     ] }) }),
     history.map(
       (block, i) => block.kind === "user" ? /* @__PURE__ */ jsxs3(Box3, { marginTop: 1, children: [
-        /* @__PURE__ */ jsxs3(Text3, { color: ACCENT, bold: true, children: [
+        /* @__PURE__ */ jsxs3(Text3, { color: ACCENT2, bold: true, children: [
           "\u203A",
           " "
         ] }),
@@ -685,13 +915,13 @@ function Interactive({ session, cwd, version }) {
       {
         marginTop: 1,
         borderStyle: "round",
-        borderColor: busy ? "gray" : ACCENT,
+        borderColor: busy ? "gray" : ACCENT2,
         paddingX: 1,
         children: busy ? /* @__PURE__ */ jsxs3(Text3, { color: "gray", children: [
           /* @__PURE__ */ jsx3(Spinner2, { type: "dots" }),
           " working\u2026"
         ] }) : /* @__PURE__ */ jsxs3(Fragment2, { children: [
-          /* @__PURE__ */ jsx3(Text3, { color: ACCENT, children: "\u203A " }),
+          /* @__PURE__ */ jsx3(Text3, { color: ACCENT2, children: "\u203A " }),
           /* @__PURE__ */ jsx3(
             TextInput,
             {
@@ -776,7 +1006,7 @@ function LoginView() {
 
 // src/cli.tsx
 import { jsx as jsx5 } from "react/jsx-runtime";
-var VERSION = "0.1.2";
+var VERSION = "0.1.4";
 var HELP = `
   StackAI \u2014 AI coding agent in your terminal
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stackai",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "StackAI — AI coding agent in your terminal. Read, write, and edit code with AI.",
   "type": "module",
   "bin": {