@townco/cli 0.1.11 → 0.1.12

package/dist/commands/tool-stub.js

@@ -0,0 +1,376 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { spawn } from "node:child_process";
+import { copyFileSync, existsSync, readFileSync, unlinkSync, writeFileSync, } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { getAgentPath, listAgents } from "@townco/agent/storage";
+import { MultiSelect } from "@townco/ui/tui";
+import { Box, render, Text, useApp, useInput } from "ink";
+import TextInput from "ink-text-input";
+import { useEffect, useState } from "react";
+import { ensureToolsDir, getToolsDir, saveToolConfig, toolConfigExists, } from "../lib/tool-storage";
+// ============================================================================
+// Helper Functions
+// ============================================================================
+/**
+ * Normalize a tool name to kebab-case for use as a filename
+ */
+function normalizeToolName(name) {
+    return name
+        .trim()
+        .toLowerCase()
+        .replace(/[^a-z0-9]+/g, "-")
+        .replace(/^-+|-+$/g, "");
+}
+/**
+ * Convert kebab-case or snake_case to camelCase for function names
+ */
+function toCamelCase(str) {
+    return str.replace(/[-_](.)/g, (_, char) => char.toUpperCase());
+}
+/**
+ * Convert a name to snake_case for the tool name export
+ */
+function toSnakeCase(str) {
+    return str
+        .trim()
+        .toLowerCase()
+        .replace(/[^a-z0-9]+/g, "_")
+        .replace(/^_+|_+$/g, "");
+}
+/**
+ * Generate the stub template content
+ */
+function generateStubTemplate(toolName) {
+    const snakeCaseName = toolName ? toSnakeCase(toolName) : "tool_name";
+    const camelCaseName = toolName ? toCamelCase(toolName) : "toolName";
+    const description = toolName
+        ? `Describe what ${toolName} does here`
+        : "Describe what your tool does here";
+    return `// biome-ignore lint/suspicious/noExplicitAny: .
+export const schema = (z: any) =>
+  z.object({
+    // Define your input parameters here
+    // Example: message: z.string().describe("The message to process"),
+    input: z.string().optional(),
+  });
+
+export const name = "${snakeCaseName}";
+export const description = "${description}";
+
+export default function ${camelCaseName}(_input: unknown) {
+  // TODO: Implement your tool logic here
+
+  return "result";
+}
+`;
+}
+/**
+ * Open a file in the user's editor and wait for it to close
+ */
+async function openInEditor(filePath) {
+    return new Promise((resolve, reject) => {
+        // Get editor from environment, fallback to vi
+        const editor = process.env.EDITOR || process.env.VISUAL || "vi";
+        // Parse editor command (might include args like "code --wait")
+        const editorParts = editor.split(" ");
+        const editorCommand = editorParts[0] || "vi";
+        const editorArgs = [...editorParts.slice(1), filePath];
+        const child = spawn(editorCommand, editorArgs, {
+            stdio: "inherit",
+        });
+        child.on("exit", (code) => {
+            if (code === 0 || code === null) {
+                resolve();
+            }
+            else {
+                reject(new Error(`Editor exited with code ${code}`));
+            }
+        });
+        child.on("error", (error) => {
+            reject(error);
+        });
+    });
+}
+function TextInputStage({ title, value, onChange, onSubmit, onCancel, placeholder, }) {
+    useInput((_input, key) => {
+        if (key.escape) {
+            onCancel();
+        }
+    });
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Box, { marginBottom: 1, children: _jsx(Text, { bold: true, children: title }) }), _jsxs(Box, { children: [_jsxs(Text, { children: [">", " "] }), _jsx(TextInput, { value: value, onChange: onChange, onSubmit: onSubmit, ...(placeholder && { placeholder }) })] }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { dimColor: true, children: "Enter: Continue \u2022 Esc: Back" }) })] }));
+}
+function NameInputStage({ value, onChange, onNext, onBack, }) {
+    const [localError, setLocalError] = useState(null);
+    const handleSubmit = () => {
+        const trimmed = value.trim();
+        if (!trimmed) {
+            return;
+        }
+        // Check for duplicate
+        if (toolConfigExists(trimmed)) {
+            setLocalError(`Tool "${trimmed}" already exists`);
+            return;
+        }
+        onNext(trimmed);
+    };
+    const handleChange = (newValue) => {
+        if (localError) {
+            setLocalError(null);
+        }
+        onChange(newValue);
+    };
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsx(TextInputStage, { title: "Enter tool name:", value: value, onChange: handleChange, onSubmit: handleSubmit, onCancel: onBack, placeholder: "my-custom-tool" }), localError && (_jsx(Box, { marginTop: 1, children: _jsxs(Text, { color: "red", children: ["\u274C ", localError] }) }))] }));
+}
+function NoAgentsMessage({ onNext }) {
+    useEffect(() => {
+        const timer = setTimeout(() => {
+            onNext();
+        }, 1000);
+        return () => clearTimeout(timer);
+    }, [onNext]);
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { dimColor: true, children: "No agents found. Tool will be saved globally." }), _jsx(Text, { dimColor: true, children: "You can attach it to agents later." })] }));
+}
+function AgentSelectionStage({ selectedAgents, onSelectedAgentsChange, onNext, onBack, }) {
+    const [availableAgents, setAvailableAgents] = useState([]);
+    const [isLoading, setIsLoading] = useState(true);
+    useEffect(() => {
+        // Fetch available agents
+        const fetchAgents = async () => {
+            try {
+                const agents = await listAgents();
+                setAvailableAgents(agents);
+            }
+            catch (error) {
+                console.error("Error fetching agents:", error);
+                setAvailableAgents([]);
+            }
+            finally {
+                setIsLoading(false);
+            }
+        };
+        fetchAgents();
+    }, []);
+    // If still loading, show loading message
+    if (isLoading) {
+        return (_jsx(Box, { flexDirection: "column", children: _jsx(Text, { children: "Loading agents..." }) }));
+    }
+    // If no agents available, show info message and auto-proceed
+    if (availableAgents.length === 0) {
+        return _jsx(NoAgentsMessage, { onNext: onNext });
+    }
+    // Show agent selection UI
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Box, { marginBottom: 1, children: _jsx(Text, { bold: true, children: "Register tool with agents (optional):" }) }), _jsxs(Box, { marginBottom: 1, children: [_jsx(Text, { dimColor: true, children: "Select which agents should have access to this tool." }), _jsx(Text, { dimColor: true, children: "You can skip this step and register agents later." })] }), _jsx(MultiSelect, { options: availableAgents.map((agent) => ({
+                    label: agent,
+                    value: agent,
+                })), selected: selectedAgents, onChange: onSelectedAgentsChange, onSubmit: onNext, onCancel: onBack })] }));
+}
+function DoneStage({ config, status, error, attachedAgents }) {
+    const { exit } = useApp();
+    useEffect(() => {
+        if (status === "done") {
+            setImmediate(() => {
+                exit();
+            });
+        }
+    }, [status, exit]);
+    if (status === "saving") {
+        return (_jsx(Box, { flexDirection: "column", children: _jsx(Text, { children: "\u23F3 Saving tool configuration..." }) }));
+    }
+    if (status === "error") {
+        return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { color: "red", children: "\u274C Error saving tool" }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { children: error }) })] }));
+    }
+    if (status === "done") {
+        return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { color: "green", children: "\u2705 Tool saved successfully!" }), _jsxs(Box, { marginTop: 1, children: [_jsxs(Text, { dimColor: true, children: ["Name: ", config.name] }), _jsxs(Text, { dimColor: true, children: ["Path: ", config.path] })] }), attachedAgents.length > 0 && (_jsx(Box, { marginTop: 1, flexDirection: "column", children: _jsxs(Text, { dimColor: true, children: ["Registered with agents: ", attachedAgents.join(", ")] }) }))] }));
+    }
+    return null;
+}
+// ============================================================================
+// Main Component
+// ============================================================================
+function ToolStubApp({ name: initialName }) {
+    const { exit } = useApp();
+    const [stage, setStage] = useState("name");
+    const [config, setConfig] = useState({});
+    const [nameInput, setNameInput] = useState(initialName || "");
+    const [saveStatus, setSaveStatus] = useState("pending");
+    const [saveError, setSaveError] = useState(null);
+    const [selectedAgents, setSelectedAgents] = useState([]);
+    const [tempFilePath, setTempFilePath] = useState(null);
+    const [editorOpened, setEditorOpened] = useState(false);
+    // On mount, create temp file and open editor
+    useEffect(() => {
+        if (editorOpened)
+            return;
+        const setupAndOpenEditor = async () => {
+            try {
+                // Generate temp file path
+                const tempFile = join(tmpdir(), `town-tool-${Date.now()}.ts`);
+                setTempFilePath(tempFile);
+                // Generate and write stub template
+                const stubContent = generateStubTemplate(initialName);
+                writeFileSync(tempFile, stubContent, "utf-8");
+                // Open in editor
+                await openInEditor(tempFile);
+                // After editor closes, move to name input stage
+                setEditorOpened(true);
+            }
+            catch (error) {
+                setSaveStatus("error");
+                setSaveError(`Failed to open editor: ${error instanceof Error ? error.message : String(error)}`);
+                setStage("done");
+            }
+        };
+        setupAndOpenEditor();
+    }, [initialName, editorOpened]);
+    const handleSave = (toolName, agentsToAttach) => {
+        setSaveStatus("saving");
+        try {
+            if (!tempFilePath) {
+                setSaveStatus("error");
+                setSaveError("Temp file path not found");
+                return;
+            }
+            // Check if temp file still exists
+            if (!existsSync(tempFilePath)) {
+                setSaveStatus("error");
+                setSaveError("Temp file was deleted");
+                return;
+            }
+            // Read the edited content
+            const editedContent = readFileSync(tempFilePath, "utf-8");
+            // Ensure tools directory exists
+            ensureToolsDir();
+            // Normalize the tool name for the filename
+            const normalizedName = normalizeToolName(toolName);
+            const toolsDir = getToolsDir();
+            const targetPath = join(toolsDir, `${normalizedName}.ts`);
+            // Write the edited content to the target path
+            try {
+                writeFileSync(targetPath, editedContent, "utf-8");
+            }
+            catch (error) {
+                setSaveStatus("error");
+                setSaveError(`Failed to save file: ${error instanceof Error ? error.message : String(error)}`);
+                return;
+            }
+            // Create tool config
+            const toolConfig = {
+                name: toolName,
+                path: targetPath,
+            };
+            // Save to global tool storage
+            saveToolConfig(toolConfig);
+            // Update selected agents
+            for (const agentName of agentsToAttach) {
+                try {
+                    const agentPath = getAgentPath(agentName);
+                    const agentJsonPath = join(agentPath, "agent.json");
+                    // Read existing agent.json
+                    const agentJsonContent = readFileSync(agentJsonPath, "utf-8");
+                    const agentDef = JSON.parse(agentJsonContent);
+                    // Add tool to tools array (create array if doesn't exist)
+                    if (!agentDef.tools) {
+                        agentDef.tools = [];
+                    }
+                    // Create custom tool object
+                    const customTool = {
+                        type: "custom",
+                        modulePath: targetPath,
+                    };
+                    // Check if this tool is already in the agent's config
+                    const existingIndex = agentDef.tools.findIndex((tool) => typeof tool === "object" &&
+                        tool !== null &&
+                        "type" in tool &&
+                        tool.type === "custom" &&
+                        "modulePath" in tool &&
+                        tool.modulePath === targetPath);
+                    if (existingIndex >= 0) {
+                        // Update existing config
+                        agentDef.tools[existingIndex] = customTool;
+                    }
+                    else {
+                        // Add new config
+                        agentDef.tools.push(customTool);
+                    }
+                    // Write back to agent.json
+                    writeFileSync(agentJsonPath, JSON.stringify(agentDef, null, 2));
+                }
+                catch (error) {
+                    console.error(`Error updating agent ${agentName}:`, error);
+                    // Continue with other agents even if one fails
+                }
+            }
+            // Clean up temp file
+            try {
+                unlinkSync(tempFilePath);
+            }
+            catch (error) {
+                // Ignore cleanup errors
+            }
+            setConfig(toolConfig);
+            setSaveStatus("done");
+        }
+        catch (error) {
+            setSaveStatus("error");
+            setSaveError(error instanceof Error ? error.message : "Unknown error occurred");
+        }
+    };
+    // Wait for editor to open and close before showing UI
+    if (!editorOpened) {
+        return (_jsx(Box, { flexDirection: "column", children: _jsx(Text, { children: "Opening editor..." }) }));
+    }
+    // If there was an error during editor setup, show done stage with error
+    if (saveStatus === "error" && stage === "done") {
+        return (_jsx(DoneStage, { config: config, status: saveStatus, error: saveError, attachedAgents: selectedAgents }));
+    }
+    // Name input stage
+    if (stage === "name") {
+        return (_jsx(NameInputStage, { value: nameInput, onChange: setNameInput, onNext: (name) => {
+                setConfig({ ...config, name });
+                setStage("agentSelection");
+            }, onBack: () => {
+                // Clean up temp file
+                if (tempFilePath) {
+                    try {
+                        unlinkSync(tempFilePath);
+                    }
+                    catch (error) {
+                        // Ignore cleanup errors
+                    }
+                }
+                exit();
+            } }));
+    }
+    // Agent selection stage
+    if (stage === "agentSelection") {
+        return (_jsx(AgentSelectionStage, { selectedAgents: selectedAgents, onSelectedAgentsChange: setSelectedAgents, onNext: () => {
+                if (!config.name) {
+                    setSaveStatus("error");
+                    setSaveError("Tool name is required");
+                    setStage("done");
+                    return;
+                }
+                handleSave(config.name, selectedAgents);
+                setStage("done");
+            }, onBack: () => setStage("name") }));
+    }
+    // Done stage
+    if (stage === "done") {
+        return (_jsx(DoneStage, { config: config, status: saveStatus, error: saveError, attachedAgents: selectedAgents }));
+    }
+    return null;
+}
+// ============================================================================
+// Export and Runner
+// ============================================================================
+export default ToolStubApp;
+export async function runToolStub(props = {}) {
+    // Set stdin to raw mode to capture input
+    if (process.stdin.isTTY) {
+        process.stdin.setRawMode(true);
+    }
+    const { waitUntilExit } = render(_jsx(ToolStubApp, { ...props }));
+    // Wait for the app to exit before returning
+    await waitUntilExit();
+}

package/dist/index.js

@@ -3,7 +3,7 @@ import { argument, command, constant, flag, multiple, object, option, optional,
 import { message } from "@optique/core/message";
 import { integer, string } from "@optique/core/valueparser";
 import { run } from "@optique/run";
-import { createSecret, deleteSecret, genenv, listSecrets } from "@townco/secret";
+import { createSecret, deleteSecret, genenv, listSecrets, } from "@townco/secret";
 import inquirer from "inquirer";
 import { match } from "ts-pattern";
 import { configureCommand } from "./commands/configure.js";
@@ -15,6 +15,10 @@ import { runMCPAdd } from "./commands/mcp-add.js";
 import { runMCPList } from "./commands/mcp-list.js";
 import { runMCPRemove } from "./commands/mcp-remove.js";
 import { runCommand } from "./commands/run.js";
+import { runToolAdd } from "./commands/tool-add.js";
+import { runToolList } from "./commands/tool-list.js";
+import { runToolRegister } from "./commands/tool-register.js";
+import { runToolRemove } from "./commands/tool-remove.js";
 /**
  * Securely prompt for a secret value without echoing to the terminal
  */
@@ -62,7 +66,21 @@ const parser = or(command("deploy", constant("deploy"), { brief: message `Deploy
     }), command("remove", constant("remove"), {
         brief: message `Remove an MCP server.`,
     })),
-}), { brief: message `Manage MCP (Model Context Protocol) servers.` }), command("secret", object({
+}), { brief: message `Manage MCP (Model Context Protocol) servers.` }), command("tool", object({
+    command: constant("tool"),
+    subcommand: or(command("add", object({
+        action: constant("add"),
+        name: optional(argument(string({ metavar: "NAME" }))),
+    }), { brief: message `Create a new tool from a template.` }), command("list", constant("list"), {
+        brief: message `List all configured custom tools.`,
+    }), command("register", object({
+        action: constant("register"),
+        name: optional(option("-n", "--name", string())),
+        path: optional(option("-p", "--path", string())),
+    }), { brief: message `Register an existing custom tool.` }), command("remove", constant("remove"), {
+        brief: message `Remove a custom tool.`,
+    })),
+}), { brief: message `Manage custom tools.` }), command("secret", object({
     command: constant("secret"),
     subcommand: or(command("list", constant("list"), { brief: message `List secrets.` }), command("add", object({
         action: constant("add"),
@@ -134,6 +152,23 @@ async function main(parser, meta) {
             .with("list", async () => await runMCPList())
             .with("remove", async () => await runMCPRemove())
             .exhaustive();
+    })
+        .with({ command: "tool" }, async ({ subcommand }) => {
+        await match(subcommand)
+            .with({ action: "add" }, async ({ name }) => {
+            await runToolAdd({
+                ...(name !== undefined && { name }),
+            });
+        })
+            .with("list", async () => await runToolList())
+            .with({ action: "register" }, async ({ name, path }) => {
+            await runToolRegister({
+                ...(name !== undefined && { name }),
+                ...(path !== undefined && { path }),
+            });
+        })
+            .with("remove", async () => await runToolRemove())
+            .exhaustive();
     })
         .with({ command: "secret" }, async ({ subcommand }) => {
         await match(subcommand)

package/dist/lib/mcp-storage.d.ts

@@ -1,9 +1,9 @@
 export type MCPConfig = {
-	name: string;
-	transport: "stdio" | "http";
-	command?: string;
-	args?: string[];
-	url?: string;
+    name: string;
+    transport: "stdio" | "http";
+    command?: string;
+    args?: string[];
+    url?: string;
 };
 /**
  * Save an MCP config to the store

package/dist/lib/mcp-storage.js

@@ -1,7 +1,6 @@
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
-
 // ============================================================================
 // Constants
 // ============================================================================
@@ -14,40 +13,38 @@ const MCPS_FILE = join(TOWN_CONFIG_DIR, "mcps.json");
  * Ensure the config directory exists
  */
 function ensureConfigDir() {
-	if (!existsSync(TOWN_CONFIG_DIR)) {
-		mkdirSync(TOWN_CONFIG_DIR, { recursive: true });
-	}
+    if (!existsSync(TOWN_CONFIG_DIR)) {
+        mkdirSync(TOWN_CONFIG_DIR, { recursive: true });
+    }
 }
 /**
  * Load all MCP configs from the JSON file
  */
 function loadStore() {
-	ensureConfigDir();
-	if (!existsSync(MCPS_FILE)) {
-		return {};
-	}
-	try {
-		const content = readFileSync(MCPS_FILE, "utf-8");
-		return JSON.parse(content);
-	} catch (error) {
-		throw new Error(
-			`Failed to load MCP configs: ${error instanceof Error ? error.message : String(error)}`,
-		);
-	}
+    ensureConfigDir();
+    if (!existsSync(MCPS_FILE)) {
+        return {};
+    }
+    try {
+        const content = readFileSync(MCPS_FILE, "utf-8");
+        return JSON.parse(content);
+    }
+    catch (error) {
+        throw new Error(`Failed to load MCP configs: ${error instanceof Error ? error.message : String(error)}`);
+    }
 }
 /**
  * Save all MCP configs to the JSON file
  */
 function saveStore(store) {
-	ensureConfigDir();
-	try {
-		const content = JSON.stringify(store, null, 2);
-		writeFileSync(MCPS_FILE, content, "utf-8");
-	} catch (error) {
-		throw new Error(
-			`Failed to save MCP configs: ${error instanceof Error ? error.message : String(error)}`,
-		);
-	}
+    ensureConfigDir();
+    try {
+        const content = JSON.stringify(store, null, 2);
+        writeFileSync(MCPS_FILE, content, "utf-8");
+    }
+    catch (error) {
+        throw new Error(`Failed to save MCP configs: ${error instanceof Error ? error.message : String(error)}`);
+    }
 }
 // ============================================================================
 // Public API
@@ -56,54 +53,55 @@ function saveStore(store) {
  * Save an MCP config to the store
  */
 export function saveMCPConfig(config) {
-	const store = loadStore();
-	store[config.name] = config;
-	saveStore(store);
+    const store = loadStore();
+    store[config.name] = config;
+    saveStore(store);
 }
 /**
  * Load an MCP config by name
  */
 export function loadMCPConfig(name) {
-	const store = loadStore();
-	return store[name] || null;
+    const store = loadStore();
+    return store[name] || null;
 }
 /**
  * Delete an MCP config by name
  */
 export function deleteMCPConfig(name) {
-	const store = loadStore();
-	if (store[name]) {
-		delete store[name];
-		saveStore(store);
-		return true;
-	}
-	return false;
+    const store = loadStore();
+    if (store[name]) {
+        delete store[name];
+        saveStore(store);
+        return true;
+    }
+    return false;
 }
 /**
  * List all MCP configs
  */
 export function listMCPConfigs() {
-	const store = loadStore();
-	return Object.values(store).sort((a, b) => a.name.localeCompare(b.name));
+    const store = loadStore();
+    return Object.values(store).sort((a, b) => a.name.localeCompare(b.name));
 }
 /**
  * Check if an MCP config exists
  */
 export function mcpConfigExists(name) {
-	const store = loadStore();
-	return name in store;
+    const store = loadStore();
+    return name in store;
 }
 /**
  * Get a summary of an MCP config for display
  */
 export function getMCPSummary(config) {
-	if (config.transport === "http") {
-		return `HTTP: ${config.url}`;
-	} else {
-		const parts = [`Stdio: ${config.command}`];
-		if (config.args && config.args.length > 0) {
-			parts.push(`[${config.args.join(" ")}]`);
-		}
-		return parts.join(" ");
-	}
+    if (config.transport === "http") {
+        return `HTTP: ${config.url}`;
+    }
+    else {
+        const parts = [`Stdio: ${config.command}`];
+        if (config.args && config.args.length > 0) {
+            parts.push(`[${config.args.join(" ")}]`);
+        }
+        return parts.join(" ");
+    }
 }

package/dist/lib/tool-storage.d.ts

@@ -0,0 +1,32 @@
+export type ToolConfig = {
+    name: string;
+    path: string;
+};
+/**
+ * Ensure the tools directory exists
+ */
+export declare function ensureToolsDir(): void;
+/**
+ * Get the tools directory path
+ */
+export declare function getToolsDir(): string;
+/**
+ * Save a tool config to the store
+ */
+export declare function saveToolConfig(config: ToolConfig): void;
+/**
+ * Load a tool config by name
+ */
+export declare function loadToolConfig(name: string): ToolConfig | null;
+/**
+ * Delete a tool config by name
+ */
+export declare function deleteToolConfig(name: string): boolean;
+/**
+ * List all tool configs
+ */
+export declare function listToolConfigs(): ToolConfig[];
+/**
+ * Check if a tool config exists
+ */
+export declare function toolConfigExists(name: string): boolean;