@tomsun28/pizza 0.0.1 → 0.0.5

package/README.md

@@ -0,0 +1,154 @@
+# 🍕 Pizza
+
+> **CLI is all you need**
+
+Pizza is a command-line AI assistant that replaces traditional JSON tool calls with a unique `[CLI]` tag format. Built on top of [pi-coding-agent](https://github.com/badlogic/pi-coding-agent), it provides developers with an efficient terminal interaction experience.
+
+---
+
+## Features
+
+- **[CLI] Tag Format** — Invoke tools using natural text tags, no complex JSON structures needed
+- **Multi-Tool Support** — Built-in commands: `ls`, `read`, `write`, `edit`, `grep`, `bash`
+- **Interactive TUI** — Beautiful terminal interface based on pi-tui with keyboard shortcuts
+- **Multi-Model Support** — Compatible with OpenAI, Anthropic, and other LLM providers
+- **Session Management** — Automatic saving and management of conversation history
+
+---
+
+## Installation
+
+```bash
+npm install -g @tomsun28/pizza
+```
+
+---
+
+## Usage
+
+### Start
+
+```bash
+# Interactive mode
+pizza
+
+# Execute a single command
+pizza "analyze the current project structure"
+```
+
+### Command Line Options
+
+```
+pizza [options] [message]
+
+OPTIONS:
+  -p, --provider <name>   LLM provider (reads from config by default)
+  -m, --model <id>        Model ID
+  -h, --help              Show help information
+```
+
+### Keyboard Shortcuts
+
+| Shortcut | Function |
+|----------|----------|
+| `Ctrl+P` | Switch model |
+| `Ctrl+T` | Think mode |
+| `Ctrl+C` | Stop current operation |
+
+### In-Session Commands
+
+- `/cli-mode` — Toggle CLI tag mode (enabled by default)
+- `/help` — Show help
+
+---
+
+## [CLI] Tag Format
+
+Pizza's core innovation replaces traditional JSON tool calls with `[CLI]` tags:
+
+### Format
+
+```
+[CLI id=1] ls src[/CLI]
+[CLI id=2] read package.json[/CLI]
+```
+
+### Result Return
+
+```
+[RESULT id=1]
+file    1.2KB   index.ts
+file    2.5KB   utils.ts
+dir     -       components
+[/RESULT]
+```
+
+### Supported Commands
+
+| Command | Usage | Description |
+|---------|-------|-------------|
+| `ls` | `[CLI]ls <path>[/CLI]` | List directory contents |
+| `read` | `[CLI]read <path> [--offset N] [--limit N][/CLI]` | Read file |
+| `write` | `[CLI]write --path <path> --content <content>[/CLI]` | Write file |
+| `edit` | `[CLI]edit --path <path> --old <text> --new <text>[/CLI]` | Edit file |
+| `grep` | `[CLI]grep <pattern> --path <dir> --include <glob>[/CLI]` | Search files |
+| `bash` | `[CLI]bash <command>[/CLI]` | Execute shell command |
+
+---
+
+## Development
+
+### Project Structure
+
+```
+.
+├── src/
+│   ├── main.ts           # Entry file
+│   └── extension/
+│       ├── index.ts      # Extension main logic
+│       └── core.ts       # Core tool implementations
+├── themes/
+│   └── pizza.json        # Theme configuration
+├── package.json
+└── tsconfig.json
+```
+
+### Local Development
+
+```bash
+# Clone the repository
+git clone <repo-url>
+cd pizza
+
+# Install dependencies
+pnpm install
+
+# Development mode
+pnpm dev
+
+# Build
+pnpm build
+
+# Run
+pnpm start
+```
+
+---
+
+## Dependencies
+
+- [pi-coding-agent](https://github.com/badlogic/pi-coding-agent) — Agent session management and extension system
+- [pi-ai](https://github.com/badlogic/pi-coding-agent) — AI model interface
+- [pi-tui](https://github.com/badlogic/pi-coding-agent) — Terminal user interface
+
+---
+
+## Configuration
+
+Pizza uses the `.pizza` directory to store configuration and session data. It is created automatically on first run.
+
+---
+
+## License
+
+MIT

package/dist/extension/core.d.ts

@@ -0,0 +1,53 @@
+/**
+ * Core logic for pizza-mode: [CLI] tag parsing, tool execution helpers,
+ * and conversation history rewriting.
+ */
+export declare const MAX_CALLS_PER_TURN = 5;
+export declare const MAX_OUTPUT_BYTES: number;
+export declare const MAX_OUTPUT_LINES = 1000;
+export declare const MAX_FILE_BYTES: number;
+export declare const MAX_FILE_LINES = 2000;
+export declare const MAX_GREP_RESULTS = 200;
+export declare const MAX_GREP_FILE_SIZE: number;
+export interface ParsedCLICall {
+    id: number;
+    toolName: string;
+    rawArgs: string;
+}
+export interface CLIToolResult {
+    output: string;
+    isError: boolean;
+}
+/**
+ * Parse [CLI] toolName args[/CLI] blocks from model text output.
+ * Returns all calls found and the text before the first call.
+ */
+export declare function parseCLICalls(text: string): {
+    calls: ParsedCLICall[];
+    textBeforeCalls: string;
+};
+/**
+ * Parse a shell-style args string into a Record.
+ * Handles: --key "value", --key value, --flag, positional args.
+ */
+export declare function parseCliArgs(raw: string): Record<string, any>;
+/**
+ * Map parsed args to tool-specific argument schema.
+ */
+export declare function mapArgsToTool(toolName: string, rawArgs: string): Record<string, any>;
+/**
+ * Convert tool call arguments back to CLI flag string for history reconstruction.
+ */
+export declare function argsToCliString(toolName: string, arguments_: Record<string, any>): string;
+export declare function execLs(cwd: string, args: Record<string, any>): Promise<CLIToolResult>;
+export declare function execRead(cwd: string, args: Record<string, any>): Promise<CLIToolResult>;
+export declare function execWrite(cwd: string, args: Record<string, any>): Promise<CLIToolResult>;
+export declare function execEdit(cwd: string, args: Record<string, any>): Promise<CLIToolResult>;
+export declare function execGrep(cwd: string, args: Record<string, any>): Promise<CLIToolResult>;
+export declare function execBash(cwd: string, args: Record<string, any>, signal?: AbortSignal): Promise<CLIToolResult>;
+export declare function dispatchTool(toolName: string, rawArgs: string, cwd: string, signal?: AbortSignal): Promise<CLIToolResult>;
+export declare function buildResultBlock(output: string, isError: boolean, id?: number, total?: number, cliCommand?: string): string;
+/**
+ * Reconstruct the [CLI] text that the model originally emitted for a tool call.
+ */
+export declare function reconstructCliText(toolName: string, arguments_: Record<string, any>): string;

package/dist/extension/core.js

@@ -0,0 +1,466 @@
+/**
+ * Core logic for pizza-mode: [CLI] tag parsing, tool execution helpers,
+ * and conversation history rewriting.
+ */
+import { readFile, writeFile, mkdir, readdir, stat } from "node:fs/promises";
+import { resolve, isAbsolute, dirname, join, relative } from "node:path";
+import { spawn } from "node:child_process";
+// ─── Constants ────────────────────────────────────────────────────────────────
+export const MAX_CALLS_PER_TURN = 5;
+export const MAX_OUTPUT_BYTES = 128 * 1024;
+export const MAX_OUTPUT_LINES = 1000;
+export const MAX_FILE_BYTES = 256 * 1024;
+export const MAX_FILE_LINES = 2000;
+export const MAX_GREP_RESULTS = 200;
+export const MAX_GREP_FILE_SIZE = 1024 * 1024;
+// ─── [CLI] tag parsing ────────────────────────────────────────────────────────
+/**
+ * Parse [CLI] toolName args[/CLI] blocks from model text output.
+ * Returns all calls found and the text before the first call.
+ */
+export function parseCLICalls(text) {
+    const regex = /\[CLI(?:\s+id=(\d+))?\]([\s\S]*?)\[\/CLI\]/g;
+    const calls = [];
+    let firstMatchStart = -1;
+    let autoId = 1;
+    let match;
+    while ((match = regex.exec(text)) !== null) {
+        if (firstMatchStart === -1)
+            firstMatchStart = match.index;
+        const explicitId = match[1] ? parseInt(match[1], 10) : undefined;
+        const inner = match[2].trim();
+        const spaceIdx = inner.search(/\s/);
+        const toolName = spaceIdx === -1 ? inner : inner.slice(0, spaceIdx);
+        const rawArgs = spaceIdx === -1 ? "" : inner.slice(spaceIdx + 1).trim();
+        if (toolName) {
+            calls.push({ id: explicitId ?? autoId, toolName, rawArgs });
+            autoId++;
+        }
+    }
+    const textBeforeCalls = firstMatchStart >= 0 ? text.substring(0, firstMatchStart).trimEnd() : text;
+    return { calls, textBeforeCalls };
+}
+// ─── Shell-style arg parser ───────────────────────────────────────────────────
+/**
+ * Parse a shell-style args string into a Record.
+ * Handles: --key "value", --key value, --flag, positional args.
+ */
+export function parseCliArgs(raw) {
+    const args = {};
+    const positional = [];
+    const tokens = [];
+    let i = 0;
+    while (i < raw.length) {
+        while (i < raw.length && /\s/.test(raw[i]))
+            i++;
+        if (i >= raw.length)
+            break;
+        if (raw[i] === '"' || raw[i] === "'") {
+            const quote = raw[i++];
+            let val = "";
+            while (i < raw.length && raw[i] !== quote) {
+                if (raw[i] === "\\" && i + 1 < raw.length) {
+                    i++;
+                    val += raw[i++];
+                }
+                else {
+                    val += raw[i++];
+                }
+            }
+            i++;
+            tokens.push(val);
+        }
+        else {
+            let val = "";
+            while (i < raw.length && !/\s/.test(raw[i])) {
+                val += raw[i++];
+            }
+            tokens.push(val);
+        }
+    }
+    let ti = 0;
+    while (ti < tokens.length) {
+        const tok = tokens[ti];
+        if (tok.startsWith("--")) {
+            const key = tok.slice(2);
+            if (ti + 1 < tokens.length && !tokens[ti + 1].startsWith("--")) {
+                args[key] = tokens[ti + 1];
+                ti += 2;
+            }
+            else {
+                args[key] = true;
+                ti++;
+            }
+        }
+        else {
+            positional.push(tok);
+            ti++;
+        }
+    }
+    if (positional.length > 0) {
+        if (!("path" in args) && !("command" in args) && !("pattern" in args)) {
+            args._positional = positional.join(" ");
+        }
+    }
+    return args;
+}
+/**
+ * Map parsed args to tool-specific argument schema.
+ */
+export function mapArgsToTool(toolName, rawArgs) {
+    const args = parseCliArgs(rawArgs);
+    const pos = args._positional;
+    delete args._positional;
+    switch (toolName) {
+        case "ls":
+            if (pos && !args.path)
+                args.path = pos;
+            break;
+        case "read":
+            if (pos && !args.path)
+                args.path = pos;
+            break;
+        case "write":
+            // --path and --content required
+            break;
+        case "edit":
+            if (args.old) {
+                args.oldText = args.old;
+                delete args.old;
+            }
+            if (args.new) {
+                args.newText = args.new;
+                delete args.new;
+            }
+            if (pos && !args.path)
+                args.path = pos;
+            break;
+        case "grep":
+            if (pos && !args.pattern)
+                args.pattern = pos;
+            break;
+        case "bash":
+            if (rawArgs && !args.command)
+                args.command = rawArgs;
+            break;
+        default:
+            if (pos)
+                args.input = pos;
+    }
+    return args;
+}
+/**
+ * Convert tool call arguments back to CLI flag string for history reconstruction.
+ */
+export function argsToCliString(toolName, arguments_) {
+    if (toolName === "bash") {
+        return arguments_.command ?? "";
+    }
+    if (toolName === "ls" || toolName === "read") {
+        const path = arguments_.path;
+        const rest = Object.entries(arguments_)
+            .filter(([k]) => k !== "path")
+            .map(([k, v]) => `--${k} "${String(v).replace(/"/g, '\\"')}"`)
+            .join(" ");
+        return path ? (rest ? `${path} ${rest}` : path) : rest;
+    }
+    if (toolName === "edit") {
+        const mapped = { ...arguments_ };
+        if (mapped.oldText !== undefined) {
+            mapped.old = mapped.oldText;
+            delete mapped.oldText;
+        }
+        if (mapped.newText !== undefined) {
+            mapped.new = mapped.newText;
+            delete mapped.newText;
+        }
+        return Object.entries(mapped)
+            .map(([k, v]) => {
+            if (v === true)
+                return `--${k}`;
+            const s = String(v);
+            return s.includes(" ") || s.includes('"') || s.includes("\n")
+                ? `--${k} "${s.replace(/"/g, '\\"').replace(/\n/g, "\\n")}"`
+                : `--${k} ${s}`;
+        })
+            .join(" ");
+    }
+    return Object.entries(arguments_)
+        .map(([k, v]) => {
+        if (v === true)
+            return `--${k}`;
+        const s = String(v);
+        return s.includes(" ") || s.includes('"') || s.includes("\n")
+            ? `--${k} "${s.replace(/"/g, '\\"').replace(/\n/g, "\\n")}"`
+            : `--${k} ${s}`;
+    })
+        .join(" ");
+}
+// ─── Tool implementations ─────────────────────────────────────────────────────
+export async function execLs(cwd, args) {
+    const dirPath = args.path
+        ? isAbsolute(args.path) ? args.path : resolve(cwd, args.path)
+        : cwd;
+    const entries = await readdir(dirPath);
+    const lines = [];
+    for (const entry of entries.sort()) {
+        try {
+            const fullPath = join(dirPath, entry);
+            const s = await stat(fullPath);
+            const type = s.isDirectory() ? "dir" : "file";
+            const size = s.isDirectory() ? "-" : formatSize(s.size);
+            lines.push(`${type}\t${size}\t${entry}`);
+        }
+        catch {
+            lines.push(`?\t?\t${entry}`);
+        }
+    }
+    return {
+        output: lines.length === 0 ? "(empty directory)" : lines.join("\n"),
+        isError: false,
+    };
+}
+export async function execRead(cwd, args) {
+    const absolutePath = args.path
+        ? (isAbsolute(args.path) ? args.path : resolve(cwd, args.path))
+        : resolve(cwd, ".");
+    const buffer = await readFile(absolutePath);
+    const text = buffer.toString("utf-8");
+    const allLines = text.split("\n");
+    const totalLines = allLines.length;
+    const offset = args.offset ? Number(args.offset) : undefined;
+    const limit = args.limit ? Number(args.limit) : undefined;
+    const startLine = offset ? Math.max(0, offset - 1) : 0;
+    if (startLine >= totalLines) {
+        throw new Error(`Offset ${offset} is beyond end of file (${totalLines} lines)`);
+    }
+    let selected = limit !== undefined
+        ? allLines.slice(startLine, startLine + limit)
+        : allLines.slice(startLine);
+    if (selected.length > MAX_FILE_LINES) {
+        selected = selected.slice(0, MAX_FILE_LINES);
+    }
+    let output = selected.join("\n");
+    if (Buffer.byteLength(output, "utf-8") > MAX_FILE_BYTES) {
+        const buf = Buffer.from(output, "utf-8");
+        output = buf.subarray(0, MAX_FILE_BYTES).toString("utf-8");
+    }
+    const shownLines = output.split("\n").length;
+    const remaining = totalLines - startLine - shownLines;
+    if (remaining > 0) {
+        const nextOffset = startLine + shownLines + 1;
+        output += `\n\n[Showing ${shownLines} of ${totalLines} lines. Use --offset ${nextOffset} to continue.]`;
+    }
+    return { output, isError: false };
+}
+export async function execWrite(cwd, args) {
+    if (!args.path)
+        throw new Error("write requires --path");
+    if (args.content === undefined)
+        throw new Error("write requires --content");
+    const absolutePath = isAbsolute(args.path) ? args.path : resolve(cwd, args.path);
+    await mkdir(dirname(absolutePath), { recursive: true });
+    await writeFile(absolutePath, args.content, "utf-8");
+    return {
+        output: `Wrote ${String(args.content).length} bytes to ${args.path}`,
+        isError: false,
+    };
+}
+export async function execEdit(cwd, args) {
+    if (!args.path)
+        throw new Error("edit requires --path");
+    if (args.oldText === undefined)
+        throw new Error("edit requires --old");
+    if (args.newText === undefined)
+        throw new Error("edit requires --new");
+    const absolutePath = isAbsolute(args.path) ? args.path : resolve(cwd, args.path);
+    const content = await readFile(absolutePath, "utf-8");
+    const index = content.indexOf(args.oldText);
+    if (index === -1) {
+        throw new Error(`Could not find the exact text in ${args.path}. The --old text must match exactly including whitespace.`);
+    }
+    const second = content.indexOf(args.oldText, index + 1);
+    if (second !== -1) {
+        throw new Error(`Found multiple occurrences of the text in ${args.path}. Provide more context to make it unique.`);
+    }
+    const newContent = content.substring(0, index) + args.newText + content.substring(index + args.oldText.length);
+    await writeFile(absolutePath, newContent, "utf-8");
+    return { output: `Edited ${args.path}`, isError: false };
+}
+export async function execGrep(cwd, args) {
+    if (!args.pattern)
+        throw new Error("grep requires a pattern");
+    const searchDir = args.path
+        ? (isAbsolute(args.path) ? args.path : resolve(cwd, args.path))
+        : cwd;
+    let regex;
+    try {
+        regex = new RegExp(args.pattern, "g");
+    }
+    catch (e) {
+        throw new Error(`Invalid regex pattern: ${e.message}`);
+    }
+    const extFilter = args.include ? parseExtFilter(args.include) : null;
+    const results = [];
+    await grepDirectory(searchDir, regex, extFilter, results, cwd);
+    if (results.length === 0) {
+        return { output: "No matches found.", isError: false };
+    }
+    let output = results.join("\n");
+    if (results.length >= MAX_GREP_RESULTS) {
+        output += `\n\n[Truncated at ${MAX_GREP_RESULTS} results]`;
+    }
+    return { output, isError: false };
+}
+export async function execBash(cwd, args, signal) {
+    const command = args.command;
+    if (!command)
+        throw new Error("bash requires a command");
+    return new Promise((resolve_, reject) => {
+        const shell = process.env.SHELL || "/bin/bash";
+        const child = spawn(shell, ["-c", command], {
+            cwd,
+            stdio: ["ignore", "pipe", "pipe"],
+            env: { ...process.env },
+        });
+        const chunks = [];
+        const onAbort = () => {
+            child.kill("SIGTERM");
+            setTimeout(() => child.kill("SIGKILL"), 2000);
+        };
+        if (signal) {
+            if (signal.aborted) {
+                onAbort();
+            }
+            else {
+                signal.addEventListener("abort", onAbort, { once: true });
+            }
+        }
+        child.stdout?.on("data", (d) => chunks.push(d));
+        child.stderr?.on("data", (d) => chunks.push(d));
+        child.on("close", (code) => {
+            if (signal)
+                signal.removeEventListener("abort", onAbort);
+            const raw = Buffer.concat(chunks).toString("utf-8");
+            const output = truncateOutput(raw);
+            const isError = code !== 0 && code !== null;
+            resolve_({ output: output || "(no output)", isError });
+        });
+        child.on("error", (err) => {
+            if (signal)
+                signal.removeEventListener("abort", onAbort);
+            reject(err);
+        });
+    });
+}
+// ─── Dispatch ─────────────────────────────────────────────────────────────────
+export async function dispatchTool(toolName, rawArgs, cwd, signal) {
+    const args = mapArgsToTool(toolName, rawArgs);
+    switch (toolName) {
+        case "ls": return execLs(cwd, args);
+        case "read": return execRead(cwd, args);
+        case "write": return execWrite(cwd, args);
+        case "edit": return execEdit(cwd, args);
+        case "grep": return execGrep(cwd, args);
+        case "bash": return execBash(cwd, args, signal);
+        default: {
+            // Fallback: treat unknown commands as bash
+            const bashCmd = `${toolName}${rawArgs ? " " + rawArgs : ""}`;
+            return execBash(cwd, { command: bashCmd }, signal);
+        }
+    }
+}
+// ─── Result formatting ────────────────────────────────────────────────────────
+export function buildResultBlock(output, isError, id, total, cliCommand) {
+    const idAttr = id !== undefined ? ` id=${id}` : "";
+    const cliAttr = cliCommand ? ` cli="${cliCommand.replace(/"/g, '\\"')}"` : "";
+    const errAttr = isError ? " error=true" : "";
+    return `[RESULT${idAttr}${cliAttr}${errAttr}]\n${output}\n[/RESULT]`;
+}
+// ─── History rewriting ────────────────────────────────────────────────────────
+/**
+ * Reconstruct the [CLI] text that the model originally emitted for a tool call.
+ */
+export function reconstructCliText(toolName, arguments_) {
+    const cliArgs = argsToCliString(toolName, arguments_);
+    const inner = cliArgs ? `${toolName} ${cliArgs}` : toolName;
+    return `[CLI] ${inner}[/CLI]`;
+}
+// ─── Helpers ──────────────────────────────────────────────────────────────────
+function formatSize(bytes) {
+    if (bytes < 1024)
+        return `${bytes}B`;
+    if (bytes < 1024 * 1024)
+        return `${(bytes / 1024).toFixed(1)}KB`;
+    return `${(bytes / (1024 * 1024)).toFixed(1)}MB`;
+}
+function parseExtFilter(include) {
+    if (include.startsWith("*."))
+        return include.substring(1);
+    return include;
+}
+const SKIP_DIRS = new Set([
+    "node_modules", ".git", "dist", "build", ".next", "__pycache__",
+    ".venv", "venv", ".cache", "coverage",
+]);
+async function grepDirectory(dir, regex, extFilter, results, cwd) {
+    if (results.length >= MAX_GREP_RESULTS)
+        return;
+    let entries;
+    try {
+        entries = await readdir(dir);
+    }
+    catch {
+        return;
+    }
+    for (const entry of entries) {
+        if (results.length >= MAX_GREP_RESULTS)
+            return;
+        const fullPath = join(dir, entry);
+        let s;
+        try {
+            s = await stat(fullPath);
+        }
+        catch {
+            continue;
+        }
+        if (s.isDirectory()) {
+            if (!SKIP_DIRS.has(entry)) {
+                await grepDirectory(fullPath, regex, extFilter, results, cwd);
+            }
+        }
+        else if (s.isFile()) {
+            if (s.size > MAX_GREP_FILE_SIZE)
+                continue;
+            if (extFilter && !entry.endsWith(extFilter))
+                continue;
+            try {
+                const content = await readFile(fullPath, "utf-8");
+                const lines = content.split("\n");
+                const relPath = relative(cwd, fullPath);
+                for (let i = 0; i < lines.length; i++) {
+                    regex.lastIndex = 0;
+                    if (regex.test(lines[i])) {
+                        results.push(`${relPath}:${i + 1}: ${lines[i].trimEnd()}`);
+                        if (results.length >= MAX_GREP_RESULTS)
+                            return;
+                    }
+                }
+            }
+            catch { /* skip binary/unreadable */ }
+        }
+    }
+}
+function truncateOutput(text) {
+    const lines = text.split("\n");
+    if (lines.length > MAX_OUTPUT_LINES) {
+        const kept = lines.slice(lines.length - MAX_OUTPUT_LINES);
+        return `[${lines.length - MAX_OUTPUT_LINES} lines truncated]\n${kept.join("\n")}`;
+    }
+    if (Buffer.byteLength(text, "utf-8") > MAX_OUTPUT_BYTES) {
+        const buf = Buffer.from(text, "utf-8");
+        return buf.subarray(buf.length - MAX_OUTPUT_BYTES).toString("utf-8");
+    }
+    return text;
+}
+//# sourceMappingURL=core.js.map