@yeshwanthyk/coding-agent 0.2.2

package/src/shell-runner.ts

@@ -0,0 +1,134 @@
+/**
+ * Shell command runner for TUI ! prefix commands.
+ * Executes commands and returns truncated output.
+ */
+
+import { spawn } from "node:child_process"
+import { createWriteStream, mkdirSync } from "node:fs"
+import { tmpdir } from "node:os"
+import { join } from "node:path"
+import {
+	getShellConfig,
+	killProcessTree,
+	truncateTail,
+	DEFAULT_MAX_LINES,
+	DEFAULT_MAX_BYTES,
+	type TruncationResult,
+} from "@yeshwanthyk/base-tools"
+
+export interface ShellResult {
+	output: string
+	exitCode: number | null
+	truncated: boolean
+	truncation?: TruncationResult
+	tempFilePath?: string
+	cancelled: boolean
+}
+
+function getTempFilePath(): string {
+	const dir = join(tmpdir(), "marvin-shell")
+	mkdirSync(dir, { recursive: true })
+	return join(dir, `output-${Date.now()}-${Math.random().toString(36).slice(2, 8)}.txt`)
+}
+
+/**
+ * Execute a shell command and return the result with truncation applied.
+ */
+export function runShellCommand(
+	command: string,
+	options?: { signal?: AbortSignal; timeout?: number }
+): Promise<ShellResult> {
+	return new Promise((resolve) => {
+		const { shell, args } = getShellConfig()
+		const child = spawn(shell, [...args, command], {
+			detached: true,
+			stdio: ["ignore", "pipe", "pipe"],
+		})
+
+		const chunks: Buffer[] = []
+		let totalBytes = 0
+		let tempFilePath: string | undefined
+		let tempFileStream: ReturnType<typeof createWriteStream> | undefined
+		let cancelled = false
+		let timeoutId: ReturnType<typeof setTimeout> | undefined
+
+		// Timeout handling
+		if (options?.timeout) {
+			timeoutId = setTimeout(() => {
+				cancelled = true
+				if (child.pid) killProcessTree(child.pid)
+			}, options.timeout)
+		}
+
+		// Abort signal handling
+		if (options?.signal) {
+			options.signal.addEventListener("abort", () => {
+				cancelled = true
+				if (child.pid) killProcessTree(child.pid)
+			})
+		}
+
+		const handleData = (data: Buffer) => {
+			totalBytes += data.length
+
+			// Start temp file if exceeding threshold
+			if (totalBytes > DEFAULT_MAX_BYTES && !tempFilePath) {
+				tempFilePath = getTempFilePath()
+				tempFileStream = createWriteStream(tempFilePath)
+				// Write existing chunks to temp file
+				for (const chunk of chunks) {
+					tempFileStream.write(chunk)
+				}
+			}
+
+			// Write to temp file if active
+			if (tempFileStream) {
+				tempFileStream.write(data)
+			}
+
+			// Keep rolling buffer (2x max for truncation headroom)
+			chunks.push(data)
+			let chunksBytes = chunks.reduce((sum, c) => sum + c.length, 0)
+			const maxChunksBytes = DEFAULT_MAX_BYTES * 2
+			while (chunksBytes > maxChunksBytes && chunks.length > 1) {
+				const removed = chunks.shift()!
+				chunksBytes -= removed.length
+			}
+		}
+
+		child.stdout?.on("data", handleData)
+		child.stderr?.on("data", handleData)
+
+		child.on("close", (code) => {
+			if (timeoutId) clearTimeout(timeoutId)
+			if (tempFileStream) tempFileStream.end()
+
+			const fullOutput = Buffer.concat(chunks).toString("utf-8")
+			const truncation = truncateTail(fullOutput, {
+				maxLines: DEFAULT_MAX_LINES,
+				maxBytes: DEFAULT_MAX_BYTES,
+			})
+
+			resolve({
+				output: truncation.content,
+				exitCode: code,
+				truncated: truncation.truncated,
+				truncation: truncation.truncated ? truncation : undefined,
+				tempFilePath: truncation.truncated ? tempFilePath : undefined,
+				cancelled,
+			})
+		})
+
+		child.on("error", (err) => {
+			if (timeoutId) clearTimeout(timeoutId)
+			if (tempFileStream) tempFileStream.end()
+
+			resolve({
+				output: `Error: ${err.message}`,
+				exitCode: null,
+				truncated: false,
+				cancelled,
+			})
+		})
+	})
+}

package/src/syntax-highlighting.ts

@@ -0,0 +1,114 @@
+import chalk from "chalk";
+import { highlight } from "cli-highlight";
+
+type CliHighlightTheme = Record<string, (s: string) => string>;
+
+const cliHighlightTheme: CliHighlightTheme = {
+	keyword: (s: string) => chalk.hex("#569CD6")(s),
+	built_in: (s: string) => chalk.hex("#4EC9B0")(s),
+	literal: (s: string) => chalk.hex("#B5CEA8")(s),
+	number: (s: string) => chalk.hex("#B5CEA8")(s),
+	string: (s: string) => chalk.hex("#CE9178")(s),
+	comment: (s: string) => chalk.hex("#6A9955")(s),
+	function: (s: string) => chalk.hex("#DCDCAA")(s),
+	title: (s: string) => chalk.hex("#DCDCAA")(s),
+	class: (s: string) => chalk.hex("#4EC9B0")(s),
+	type: (s: string) => chalk.hex("#4EC9B0")(s),
+	attr: (s: string) => chalk.hex("#9CDCFE")(s),
+	variable: (s: string) => chalk.hex("#9CDCFE")(s),
+	params: (s: string) => chalk.hex("#9CDCFE")(s),
+	operator: (s: string) => chalk.hex("#D4D4D4")(s),
+	punctuation: (s: string) => chalk.hex("#D4D4D4")(s),
+};
+
+export function replaceTabs(text: string): string {
+	return text.replace(/\t/g, "   ");
+}
+
+export function highlightCode(code: string, lang?: string): string[] {
+	const opts = {
+		language: lang,
+		ignoreIllegals: true,
+		theme: cliHighlightTheme,
+	};
+
+	return highlight(code, opts).split("\n");
+}
+
+export function getLanguageFromPath(filePath: string): string | undefined {
+	const trimmed = (filePath || "").trim();
+	if (!trimmed) return undefined;
+
+	const base = trimmed.split(/[\\/]/).pop()?.toLowerCase();
+	if (!base) return undefined;
+
+	if (base === "dockerfile") return "dockerfile";
+	if (base === "makefile") return "makefile";
+
+	const ext = base.includes(".") ? base.split(".").pop()?.toLowerCase() : undefined;
+	if (!ext) return undefined;
+
+	const extToLang: Record<string, string> = {
+		ts: "typescript",
+		tsx: "typescript",
+		js: "javascript",
+		jsx: "javascript",
+		mjs: "javascript",
+		cjs: "javascript",
+		py: "python",
+		rb: "ruby",
+		rs: "rust",
+		go: "go",
+		java: "java",
+		kt: "kotlin",
+		swift: "swift",
+		c: "c",
+		h: "c",
+		cpp: "cpp",
+		cc: "cpp",
+		cxx: "cpp",
+		hpp: "cpp",
+		cs: "csharp",
+		php: "php",
+		sh: "bash",
+		bash: "bash",
+		zsh: "bash",
+		fish: "fish",
+		ps1: "powershell",
+		sql: "sql",
+		html: "html",
+		htm: "html",
+		css: "css",
+		scss: "scss",
+		sass: "sass",
+		less: "less",
+		json: "json",
+		yaml: "yaml",
+		yml: "yaml",
+		toml: "toml",
+		xml: "xml",
+		md: "markdown",
+		markdown: "markdown",
+		dockerfile: "dockerfile",
+		makefile: "makefile",
+		cmake: "cmake",
+		lua: "lua",
+		pl: "perl",
+		perl: "perl",
+		r: "r",
+		scala: "scala",
+		clj: "clojure",
+		ex: "elixir",
+		exs: "elixir",
+		erl: "erlang",
+		hs: "haskell",
+		ml: "ocaml",
+		vim: "vim",
+		graphql: "graphql",
+		proto: "protobuf",
+		tf: "hcl",
+		hcl: "hcl",
+	};
+
+	return extToLang[ext];
+}

package/src/theme-names.ts

@@ -0,0 +1,37 @@
+/**
+ * Single source of truth for available theme names.
+ * "marvin" is the built-in default, other names match BUILTIN_THEMES keys in @yeshwanthyk/open-tui.
+ * Verified by test in autocomplete-commands.test.ts.
+ */
+export const THEME_NAMES = [
+	"marvin",
+	"aura",
+	"ayu",
+	"catppuccin",
+	"catppuccin-macchiato",
+	"cobalt2",
+	"dracula",
+	"everforest",
+	"flexoki",
+	"github",
+	"gruvbox",
+	"kanagawa",
+	"lucent-orng",
+	"material",
+	"matrix",
+	"mercury",
+	"monokai",
+	"nightowl",
+	"nord",
+	"one-dark",
+	"opencode",
+	"orng",
+	"palenight",
+	"rosepine",
+	"solarized",
+	"synthwave84",
+	"tokyonight",
+	"vercel",
+	"vesper",
+	"zenburn",
+]

package/src/tool-ui-contracts.ts

@@ -0,0 +1,77 @@
+export type DelegationMode = "single" | "parallel" | "chain"
+export type DelegationStatus = "pending" | "running" | "ok" | "error"
+
+export interface AgentDelegationUiItem {
+	id: string
+	agent: string
+	task: string
+	status: DelegationStatus
+	preview?: string
+}
+
+export interface AgentDelegationUi {
+	kind: "agent_delegation"
+	mode: DelegationMode
+	items: AgentDelegationUiItem[]
+	activeId?: string
+}
+
+export interface AgentDelegationArgs {
+	agent?: string
+	task?: string
+	tasks?: Array<{ agent: string; task: string }>
+	chain?: Array<{ agent: string; task: string }>
+}
+
+const isRecord = (v: unknown): v is Record<string, unknown> => typeof v === "object" && v !== null
+
+export function getAgentDelegationArgs(args: unknown): AgentDelegationArgs | null {
+	if (!isRecord(args)) return null
+	const agent = typeof args.agent === "string" ? args.agent : undefined
+	const task = typeof args.task === "string" ? args.task : undefined
+	const tasks = Array.isArray(args.tasks) ? args.tasks : undefined
+	const chain = Array.isArray(args.chain) ? args.chain : undefined
+
+	const isTaskItem = (v: unknown): v is { agent: string; task: string } =>
+		isRecord(v) && typeof v.agent === "string" && typeof v.task === "string"
+
+	const normTasks = tasks?.every(isTaskItem) ? tasks : undefined
+	const normChain = chain?.every(isTaskItem) ? chain : undefined
+
+	if (normChain || normTasks || (agent && task)) return { agent, task, tasks: normTasks, chain: normChain }
+	return null
+}
+
+export function getAgentDelegationUi(details: unknown): AgentDelegationUi | null {
+	if (!isRecord(details)) return null
+	if (!isRecord(details.ui)) return null
+	if (details.ui.kind !== "agent_delegation") return null
+
+	const ui = details.ui as Record<string, unknown>
+	const mode = ui.mode
+	if (mode !== "single" && mode !== "parallel" && mode !== "chain") return null
+	if (!Array.isArray(ui.items)) return null
+
+	const isStatus = (s: unknown): s is DelegationStatus =>
+		s === "pending" || s === "running" || s === "ok" || s === "error"
+	const items: AgentDelegationUiItem[] = []
+	for (const raw of ui.items) {
+		if (!isRecord(raw)) return null
+		if (typeof raw.id !== "string" || typeof raw.agent !== "string" || typeof raw.task !== "string") return null
+		if (!isStatus(raw.status)) return null
+		items.push({
+			id: raw.id,
+			agent: raw.agent,
+			task: raw.task,
+			status: raw.status,
+			preview: typeof raw.preview === "string" ? raw.preview : undefined,
+		})
+	}
+
+	return {
+		kind: "agent_delegation",
+		mode,
+		items,
+		activeId: typeof ui.activeId === "string" ? ui.activeId : undefined,
+	}
+}