@townco/cli 0.1.23 → 0.1.25
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/commands/create.js +5 -0
- package/dist/commands/mcp-add.d.ts +10 -5
- package/dist/index.js +24 -33
- package/package.json +5 -5
- package/dist/commands/tool-stub.d.ts +0 -6
- package/dist/commands/tool-stub.js +0 -376
package/dist/commands/create.js
CHANGED
|
@@ -43,6 +43,11 @@ const AVAILABLE_TOOLS = [
|
|
|
43
43
|
value: "web_search",
|
|
44
44
|
description: "Exa-powered web search (requires EXA_API_KEY)",
|
|
45
45
|
},
|
|
46
|
+
{
|
|
47
|
+
label: "Filesystem",
|
|
48
|
+
value: "filesystem",
|
|
49
|
+
description: "Read, write, and search files in the project directory",
|
|
50
|
+
},
|
|
46
51
|
];
|
|
47
52
|
function NameInput({ nameInput, setNameInput, onSubmit }) {
|
|
48
53
|
useInput((_input, key) => {
|
|
@@ -1,9 +1,14 @@
|
|
|
1
1
|
interface MCPAddProps {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
2
|
+
name?: string;
|
|
3
|
+
url?: string;
|
|
4
|
+
command?: string;
|
|
5
|
+
args?: readonly string[];
|
|
6
6
|
}
|
|
7
|
-
declare function MCPAddApp({
|
|
7
|
+
declare function MCPAddApp({
|
|
8
|
+
name: initialName,
|
|
9
|
+
url: initialUrl,
|
|
10
|
+
command: initialCommand,
|
|
11
|
+
args: initialArgs,
|
|
12
|
+
}: MCPAddProps): import("react/jsx-runtime").JSX.Element | null;
|
|
8
13
|
export default MCPAddApp;
|
|
9
14
|
export declare function runMCPAdd(props?: MCPAddProps): Promise<void>;
|
package/dist/index.js
CHANGED
|
@@ -4,6 +4,7 @@ import { argument, command, constant, flag, multiple, object, option, optional,
|
|
|
4
4
|
import { message } from "@optique/core/message";
|
|
5
5
|
import { integer, string } from "@optique/core/valueparser";
|
|
6
6
|
import { run } from "@optique/run";
|
|
7
|
+
import { initForClaudeCode } from "@townco/agent/scaffold";
|
|
7
8
|
import { isInsideTownProject } from "@townco/agent/storage";
|
|
8
9
|
import { createSecret, deleteSecret, genenv, listSecrets, } from "@townco/secret";
|
|
9
10
|
import inquirer from "inquirer";
|
|
@@ -14,9 +15,6 @@ import { createProjectCommand } from "./commands/create-project.js";
|
|
|
14
15
|
import { deleteCommand } from "./commands/delete.js";
|
|
15
16
|
import { editCommand } from "./commands/edit.js";
|
|
16
17
|
import { listCommand } from "./commands/list.js";
|
|
17
|
-
import { runMCPAdd } from "./commands/mcp-add.js";
|
|
18
|
-
import { runMCPList } from "./commands/mcp-list.js";
|
|
19
|
-
import { runMCPRemove } from "./commands/mcp-remove.js";
|
|
20
18
|
import { runCommand } from "./commands/run.js";
|
|
21
19
|
import { runToolAdd } from "./commands/tool-add.js";
|
|
22
20
|
import { runToolList } from "./commands/tool-list.js";
|
|
@@ -45,7 +43,10 @@ const parser = or(command("deploy", constant("deploy"), { brief: message `Deploy
|
|
|
45
43
|
tools: multiple(option("-t", "--tool", string())),
|
|
46
44
|
systemPrompt: optional(option("-p", "--prompt", string())),
|
|
47
45
|
init: optional(option("--init", string())),
|
|
48
|
-
|
|
46
|
+
claude: optional(flag("--claude")),
|
|
47
|
+
}), {
|
|
48
|
+
brief: message `Create a new agent or project (with --init <path>). Use --claude to add Claude Code integration.`,
|
|
49
|
+
}), command("list", constant("list"), { brief: message `List all agents.` }), command("run", object({
|
|
49
50
|
command: constant("run"),
|
|
50
51
|
name: argument(string({ metavar: "NAME" })),
|
|
51
52
|
http: optional(flag("--http")),
|
|
@@ -57,20 +58,7 @@ const parser = or(command("deploy", constant("deploy"), { brief: message `Deploy
|
|
|
57
58
|
}), { brief: message `Edit an agent.` }), command("delete", object({
|
|
58
59
|
command: constant("delete"),
|
|
59
60
|
name: argument(string({ metavar: "NAME" })),
|
|
60
|
-
}), { brief: message `Delete an agent.` }), command("
|
|
61
|
-
command: constant("mcp"),
|
|
62
|
-
subcommand: or(command("add", object({
|
|
63
|
-
action: constant("add"),
|
|
64
|
-
name: optional(option("-n", "--name", string())),
|
|
65
|
-
url: optional(option("-u", "--url", string())),
|
|
66
|
-
command: optional(option("-c", "--command", string())),
|
|
67
|
-
args: multiple(option("-a", "--arg", string())),
|
|
68
|
-
}), { brief: message `Add a new MCP server.` }), command("list", constant("list"), {
|
|
69
|
-
brief: message `List all configured MCP servers.`,
|
|
70
|
-
}), command("remove", constant("remove"), {
|
|
71
|
-
brief: message `Remove an MCP server.`,
|
|
72
|
-
})),
|
|
73
|
-
}), { brief: message `Manage MCP (Model Context Protocol) servers.` }), command("tool", object({
|
|
61
|
+
}), { brief: message `Delete an agent.` }), command("tool", object({
|
|
74
62
|
command: constant("tool"),
|
|
75
63
|
subcommand: or(command("add", object({
|
|
76
64
|
action: constant("add"),
|
|
@@ -113,7 +101,24 @@ async function main(parser, meta) {
|
|
|
113
101
|
.with("configure", async () => {
|
|
114
102
|
await configureCommand();
|
|
115
103
|
})
|
|
116
|
-
.with({ command: "create" }, async ({ name, model, tools, systemPrompt, init }) => {
|
|
104
|
+
.with({ command: "create" }, async ({ name, model, tools, systemPrompt, init, claude }) => {
|
|
105
|
+
// Handle --claude flag (initialize .claude in existing project)
|
|
106
|
+
if (claude === true) {
|
|
107
|
+
if (init !== null && init !== undefined) {
|
|
108
|
+
console.error("Error: --claude flag is redundant with --init (projects include .claude by default)");
|
|
109
|
+
process.exit(1);
|
|
110
|
+
}
|
|
111
|
+
// Check if we're in a Town project
|
|
112
|
+
const projectRoot = await isInsideTownProject();
|
|
113
|
+
if (projectRoot === null) {
|
|
114
|
+
console.error("Error: Not inside a Town project. Use 'town create --init <path>' to create a new project.");
|
|
115
|
+
process.exit(1);
|
|
116
|
+
}
|
|
117
|
+
// Initialize .claude directory only
|
|
118
|
+
await initForClaudeCode(projectRoot);
|
|
119
|
+
console.log("\n✓ Claude Code workspace initialized successfully!");
|
|
120
|
+
return;
|
|
121
|
+
}
|
|
117
122
|
// Check if --init flag is present for project scaffolding
|
|
118
123
|
if (init !== null && init !== undefined) {
|
|
119
124
|
// Project mode - scaffold a standalone project
|
|
@@ -186,20 +191,6 @@ async function main(parser, meta) {
|
|
|
186
191
|
})
|
|
187
192
|
.with({ command: "delete" }, async ({ name }) => {
|
|
188
193
|
await deleteCommand(name);
|
|
189
|
-
})
|
|
190
|
-
.with({ command: "mcp" }, async ({ subcommand }) => {
|
|
191
|
-
await match(subcommand)
|
|
192
|
-
.with({ action: "add" }, async ({ name, url, command, args }) => {
|
|
193
|
-
await runMCPAdd({
|
|
194
|
-
...(name !== undefined && { name }),
|
|
195
|
-
...(url !== undefined && { url }),
|
|
196
|
-
...(command !== undefined && { command }),
|
|
197
|
-
...(args.length > 0 && { args }),
|
|
198
|
-
});
|
|
199
|
-
})
|
|
200
|
-
.with("list", async () => await runMCPList())
|
|
201
|
-
.with("remove", async () => await runMCPRemove())
|
|
202
|
-
.exhaustive();
|
|
203
194
|
})
|
|
204
195
|
.with({ command: "tool" }, async ({ subcommand }) => {
|
|
205
196
|
await match(subcommand)
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@townco/cli",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.25",
|
|
4
4
|
"type": "module",
|
|
5
5
|
"bin": {
|
|
6
6
|
"town": "./dist/index.js"
|
|
@@ -18,16 +18,16 @@
|
|
|
18
18
|
"build": "tsc"
|
|
19
19
|
},
|
|
20
20
|
"devDependencies": {
|
|
21
|
-
"@townco/tsconfig": "0.1.
|
|
21
|
+
"@townco/tsconfig": "0.1.17",
|
|
22
22
|
"@types/bun": "^1.3.1",
|
|
23
23
|
"@types/react": "^19.2.2"
|
|
24
24
|
},
|
|
25
25
|
"dependencies": {
|
|
26
26
|
"@optique/core": "^0.6.2",
|
|
27
27
|
"@optique/run": "^0.6.2",
|
|
28
|
-
"@townco/agent": "0.1.
|
|
29
|
-
"@townco/secret": "0.1.
|
|
30
|
-
"@townco/ui": "0.1.
|
|
28
|
+
"@townco/agent": "0.1.25",
|
|
29
|
+
"@townco/secret": "0.1.20",
|
|
30
|
+
"@townco/ui": "0.1.20",
|
|
31
31
|
"@types/inquirer": "^9.0.9",
|
|
32
32
|
"ink": "^6.4.0",
|
|
33
33
|
"ink-text-input": "^6.0.0",
|
|
@@ -1,6 +0,0 @@
|
|
|
1
|
-
interface ToolStubProps {
|
|
2
|
-
name?: string;
|
|
3
|
-
}
|
|
4
|
-
declare function ToolStubApp({ name: initialName }: ToolStubProps): import("react/jsx-runtime").JSX.Element | null;
|
|
5
|
-
export default ToolStubApp;
|
|
6
|
-
export declare function runToolStub(props?: ToolStubProps): Promise<void>;
|
|
@@ -1,376 +0,0 @@
|
|
|
1
|
-
import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
|
|
2
|
-
import { spawn } from "node:child_process";
|
|
3
|
-
import { copyFileSync, existsSync, readFileSync, unlinkSync, writeFileSync, } from "node:fs";
|
|
4
|
-
import { tmpdir } from "node:os";
|
|
5
|
-
import { join } from "node:path";
|
|
6
|
-
import { getAgentPath, listAgents } from "@townco/agent/storage";
|
|
7
|
-
import { MultiSelect } from "@townco/ui/tui";
|
|
8
|
-
import { Box, render, Text, useApp, useInput } from "ink";
|
|
9
|
-
import TextInput from "ink-text-input";
|
|
10
|
-
import { useEffect, useState } from "react";
|
|
11
|
-
import { ensureToolsDir, getToolsDir, saveToolConfig, toolConfigExists, } from "../lib/tool-storage";
|
|
12
|
-
// ============================================================================
|
|
13
|
-
// Helper Functions
|
|
14
|
-
// ============================================================================
|
|
15
|
-
/**
|
|
16
|
-
* Normalize a tool name to kebab-case for use as a filename
|
|
17
|
-
*/
|
|
18
|
-
function normalizeToolName(name) {
|
|
19
|
-
return name
|
|
20
|
-
.trim()
|
|
21
|
-
.toLowerCase()
|
|
22
|
-
.replace(/[^a-z0-9]+/g, "-")
|
|
23
|
-
.replace(/^-+|-+$/g, "");
|
|
24
|
-
}
|
|
25
|
-
/**
|
|
26
|
-
* Convert kebab-case or snake_case to camelCase for function names
|
|
27
|
-
*/
|
|
28
|
-
function toCamelCase(str) {
|
|
29
|
-
return str.replace(/[-_](.)/g, (_, char) => char.toUpperCase());
|
|
30
|
-
}
|
|
31
|
-
/**
|
|
32
|
-
* Convert a name to snake_case for the tool name export
|
|
33
|
-
*/
|
|
34
|
-
function toSnakeCase(str) {
|
|
35
|
-
return str
|
|
36
|
-
.trim()
|
|
37
|
-
.toLowerCase()
|
|
38
|
-
.replace(/[^a-z0-9]+/g, "_")
|
|
39
|
-
.replace(/^_+|_+$/g, "");
|
|
40
|
-
}
|
|
41
|
-
/**
|
|
42
|
-
* Generate the stub template content
|
|
43
|
-
*/
|
|
44
|
-
function generateStubTemplate(toolName) {
|
|
45
|
-
const snakeCaseName = toolName ? toSnakeCase(toolName) : "tool_name";
|
|
46
|
-
const camelCaseName = toolName ? toCamelCase(toolName) : "toolName";
|
|
47
|
-
const description = toolName
|
|
48
|
-
? `Describe what ${toolName} does here`
|
|
49
|
-
: "Describe what your tool does here";
|
|
50
|
-
return `// biome-ignore lint/suspicious/noExplicitAny: .
|
|
51
|
-
export const schema = (z: any) =>
|
|
52
|
-
z.object({
|
|
53
|
-
// Define your input parameters here
|
|
54
|
-
// Example: message: z.string().describe("The message to process"),
|
|
55
|
-
input: z.string().optional(),
|
|
56
|
-
});
|
|
57
|
-
|
|
58
|
-
export const name = "${snakeCaseName}";
|
|
59
|
-
export const description = "${description}";
|
|
60
|
-
|
|
61
|
-
export default function ${camelCaseName}(_input: unknown) {
|
|
62
|
-
// TODO: Implement your tool logic here
|
|
63
|
-
|
|
64
|
-
return "result";
|
|
65
|
-
}
|
|
66
|
-
`;
|
|
67
|
-
}
|
|
68
|
-
/**
|
|
69
|
-
* Open a file in the user's editor and wait for it to close
|
|
70
|
-
*/
|
|
71
|
-
async function openInEditor(filePath) {
|
|
72
|
-
return new Promise((resolve, reject) => {
|
|
73
|
-
// Get editor from environment, fallback to vi
|
|
74
|
-
const editor = process.env.EDITOR || process.env.VISUAL || "vi";
|
|
75
|
-
// Parse editor command (might include args like "code --wait")
|
|
76
|
-
const editorParts = editor.split(" ");
|
|
77
|
-
const editorCommand = editorParts[0] || "vi";
|
|
78
|
-
const editorArgs = [...editorParts.slice(1), filePath];
|
|
79
|
-
const child = spawn(editorCommand, editorArgs, {
|
|
80
|
-
stdio: "inherit",
|
|
81
|
-
});
|
|
82
|
-
child.on("exit", (code) => {
|
|
83
|
-
if (code === 0 || code === null) {
|
|
84
|
-
resolve();
|
|
85
|
-
}
|
|
86
|
-
else {
|
|
87
|
-
reject(new Error(`Editor exited with code ${code}`));
|
|
88
|
-
}
|
|
89
|
-
});
|
|
90
|
-
child.on("error", (error) => {
|
|
91
|
-
reject(error);
|
|
92
|
-
});
|
|
93
|
-
});
|
|
94
|
-
}
|
|
95
|
-
function TextInputStage({ title, value, onChange, onSubmit, onCancel, placeholder, }) {
|
|
96
|
-
useInput((_input, key) => {
|
|
97
|
-
if (key.escape) {
|
|
98
|
-
onCancel();
|
|
99
|
-
}
|
|
100
|
-
});
|
|
101
|
-
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" }) })] }));
|
|
102
|
-
}
|
|
103
|
-
function NameInputStage({ value, onChange, onNext, onBack, }) {
|
|
104
|
-
const [localError, setLocalError] = useState(null);
|
|
105
|
-
const handleSubmit = () => {
|
|
106
|
-
const trimmed = value.trim();
|
|
107
|
-
if (!trimmed) {
|
|
108
|
-
return;
|
|
109
|
-
}
|
|
110
|
-
// Check for duplicate
|
|
111
|
-
if (toolConfigExists(trimmed)) {
|
|
112
|
-
setLocalError(`Tool "${trimmed}" already exists`);
|
|
113
|
-
return;
|
|
114
|
-
}
|
|
115
|
-
onNext(trimmed);
|
|
116
|
-
};
|
|
117
|
-
const handleChange = (newValue) => {
|
|
118
|
-
if (localError) {
|
|
119
|
-
setLocalError(null);
|
|
120
|
-
}
|
|
121
|
-
onChange(newValue);
|
|
122
|
-
};
|
|
123
|
-
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] }) }))] }));
|
|
124
|
-
}
|
|
125
|
-
function NoAgentsMessage({ onNext }) {
|
|
126
|
-
useEffect(() => {
|
|
127
|
-
const timer = setTimeout(() => {
|
|
128
|
-
onNext();
|
|
129
|
-
}, 1000);
|
|
130
|
-
return () => clearTimeout(timer);
|
|
131
|
-
}, [onNext]);
|
|
132
|
-
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." })] }));
|
|
133
|
-
}
|
|
134
|
-
function AgentSelectionStage({ selectedAgents, onSelectedAgentsChange, onNext, onBack, }) {
|
|
135
|
-
const [availableAgents, setAvailableAgents] = useState([]);
|
|
136
|
-
const [isLoading, setIsLoading] = useState(true);
|
|
137
|
-
useEffect(() => {
|
|
138
|
-
// Fetch available agents
|
|
139
|
-
const fetchAgents = async () => {
|
|
140
|
-
try {
|
|
141
|
-
const agents = await listAgents();
|
|
142
|
-
setAvailableAgents(agents);
|
|
143
|
-
}
|
|
144
|
-
catch (error) {
|
|
145
|
-
console.error("Error fetching agents:", error);
|
|
146
|
-
setAvailableAgents([]);
|
|
147
|
-
}
|
|
148
|
-
finally {
|
|
149
|
-
setIsLoading(false);
|
|
150
|
-
}
|
|
151
|
-
};
|
|
152
|
-
fetchAgents();
|
|
153
|
-
}, []);
|
|
154
|
-
// If still loading, show loading message
|
|
155
|
-
if (isLoading) {
|
|
156
|
-
return (_jsx(Box, { flexDirection: "column", children: _jsx(Text, { children: "Loading agents..." }) }));
|
|
157
|
-
}
|
|
158
|
-
// If no agents available, show info message and auto-proceed
|
|
159
|
-
if (availableAgents.length === 0) {
|
|
160
|
-
return _jsx(NoAgentsMessage, { onNext: onNext });
|
|
161
|
-
}
|
|
162
|
-
// Show agent selection UI
|
|
163
|
-
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) => ({
|
|
164
|
-
label: agent,
|
|
165
|
-
value: agent,
|
|
166
|
-
})), selected: selectedAgents, onChange: onSelectedAgentsChange, onSubmit: onNext, onCancel: onBack })] }));
|
|
167
|
-
}
|
|
168
|
-
function DoneStage({ config, status, error, attachedAgents }) {
|
|
169
|
-
const { exit } = useApp();
|
|
170
|
-
useEffect(() => {
|
|
171
|
-
if (status === "done") {
|
|
172
|
-
setImmediate(() => {
|
|
173
|
-
exit();
|
|
174
|
-
});
|
|
175
|
-
}
|
|
176
|
-
}, [status, exit]);
|
|
177
|
-
if (status === "saving") {
|
|
178
|
-
return (_jsx(Box, { flexDirection: "column", children: _jsx(Text, { children: "\u23F3 Saving tool configuration..." }) }));
|
|
179
|
-
}
|
|
180
|
-
if (status === "error") {
|
|
181
|
-
return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { color: "red", children: "\u274C Error saving tool" }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { children: error }) })] }));
|
|
182
|
-
}
|
|
183
|
-
if (status === "done") {
|
|
184
|
-
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(", ")] }) }))] }));
|
|
185
|
-
}
|
|
186
|
-
return null;
|
|
187
|
-
}
|
|
188
|
-
// ============================================================================
|
|
189
|
-
// Main Component
|
|
190
|
-
// ============================================================================
|
|
191
|
-
function ToolStubApp({ name: initialName }) {
|
|
192
|
-
const { exit } = useApp();
|
|
193
|
-
const [stage, setStage] = useState("name");
|
|
194
|
-
const [config, setConfig] = useState({});
|
|
195
|
-
const [nameInput, setNameInput] = useState(initialName || "");
|
|
196
|
-
const [saveStatus, setSaveStatus] = useState("pending");
|
|
197
|
-
const [saveError, setSaveError] = useState(null);
|
|
198
|
-
const [selectedAgents, setSelectedAgents] = useState([]);
|
|
199
|
-
const [tempFilePath, setTempFilePath] = useState(null);
|
|
200
|
-
const [editorOpened, setEditorOpened] = useState(false);
|
|
201
|
-
// On mount, create temp file and open editor
|
|
202
|
-
useEffect(() => {
|
|
203
|
-
if (editorOpened)
|
|
204
|
-
return;
|
|
205
|
-
const setupAndOpenEditor = async () => {
|
|
206
|
-
try {
|
|
207
|
-
// Generate temp file path
|
|
208
|
-
const tempFile = join(tmpdir(), `town-tool-${Date.now()}.ts`);
|
|
209
|
-
setTempFilePath(tempFile);
|
|
210
|
-
// Generate and write stub template
|
|
211
|
-
const stubContent = generateStubTemplate(initialName);
|
|
212
|
-
writeFileSync(tempFile, stubContent, "utf-8");
|
|
213
|
-
// Open in editor
|
|
214
|
-
await openInEditor(tempFile);
|
|
215
|
-
// After editor closes, move to name input stage
|
|
216
|
-
setEditorOpened(true);
|
|
217
|
-
}
|
|
218
|
-
catch (error) {
|
|
219
|
-
setSaveStatus("error");
|
|
220
|
-
setSaveError(`Failed to open editor: ${error instanceof Error ? error.message : String(error)}`);
|
|
221
|
-
setStage("done");
|
|
222
|
-
}
|
|
223
|
-
};
|
|
224
|
-
setupAndOpenEditor();
|
|
225
|
-
}, [initialName, editorOpened]);
|
|
226
|
-
const handleSave = (toolName, agentsToAttach) => {
|
|
227
|
-
setSaveStatus("saving");
|
|
228
|
-
try {
|
|
229
|
-
if (!tempFilePath) {
|
|
230
|
-
setSaveStatus("error");
|
|
231
|
-
setSaveError("Temp file path not found");
|
|
232
|
-
return;
|
|
233
|
-
}
|
|
234
|
-
// Check if temp file still exists
|
|
235
|
-
if (!existsSync(tempFilePath)) {
|
|
236
|
-
setSaveStatus("error");
|
|
237
|
-
setSaveError("Temp file was deleted");
|
|
238
|
-
return;
|
|
239
|
-
}
|
|
240
|
-
// Read the edited content
|
|
241
|
-
const editedContent = readFileSync(tempFilePath, "utf-8");
|
|
242
|
-
// Ensure tools directory exists
|
|
243
|
-
ensureToolsDir();
|
|
244
|
-
// Normalize the tool name for the filename
|
|
245
|
-
const normalizedName = normalizeToolName(toolName);
|
|
246
|
-
const toolsDir = getToolsDir();
|
|
247
|
-
const targetPath = join(toolsDir, `${normalizedName}.ts`);
|
|
248
|
-
// Write the edited content to the target path
|
|
249
|
-
try {
|
|
250
|
-
writeFileSync(targetPath, editedContent, "utf-8");
|
|
251
|
-
}
|
|
252
|
-
catch (error) {
|
|
253
|
-
setSaveStatus("error");
|
|
254
|
-
setSaveError(`Failed to save file: ${error instanceof Error ? error.message : String(error)}`);
|
|
255
|
-
return;
|
|
256
|
-
}
|
|
257
|
-
// Create tool config
|
|
258
|
-
const toolConfig = {
|
|
259
|
-
name: toolName,
|
|
260
|
-
path: targetPath,
|
|
261
|
-
};
|
|
262
|
-
// Save to global tool storage
|
|
263
|
-
saveToolConfig(toolConfig);
|
|
264
|
-
// Update selected agents
|
|
265
|
-
for (const agentName of agentsToAttach) {
|
|
266
|
-
try {
|
|
267
|
-
const agentPath = getAgentPath(agentName);
|
|
268
|
-
const agentJsonPath = join(agentPath, "agent.json");
|
|
269
|
-
// Read existing agent.json
|
|
270
|
-
const agentJsonContent = readFileSync(agentJsonPath, "utf-8");
|
|
271
|
-
const agentDef = JSON.parse(agentJsonContent);
|
|
272
|
-
// Add tool to tools array (create array if doesn't exist)
|
|
273
|
-
if (!agentDef.tools) {
|
|
274
|
-
agentDef.tools = [];
|
|
275
|
-
}
|
|
276
|
-
// Create custom tool object
|
|
277
|
-
const customTool = {
|
|
278
|
-
type: "custom",
|
|
279
|
-
modulePath: targetPath,
|
|
280
|
-
};
|
|
281
|
-
// Check if this tool is already in the agent's config
|
|
282
|
-
const existingIndex = agentDef.tools.findIndex((tool) => typeof tool === "object" &&
|
|
283
|
-
tool !== null &&
|
|
284
|
-
"type" in tool &&
|
|
285
|
-
tool.type === "custom" &&
|
|
286
|
-
"modulePath" in tool &&
|
|
287
|
-
tool.modulePath === targetPath);
|
|
288
|
-
if (existingIndex >= 0) {
|
|
289
|
-
// Update existing config
|
|
290
|
-
agentDef.tools[existingIndex] = customTool;
|
|
291
|
-
}
|
|
292
|
-
else {
|
|
293
|
-
// Add new config
|
|
294
|
-
agentDef.tools.push(customTool);
|
|
295
|
-
}
|
|
296
|
-
// Write back to agent.json
|
|
297
|
-
writeFileSync(agentJsonPath, JSON.stringify(agentDef, null, 2));
|
|
298
|
-
}
|
|
299
|
-
catch (error) {
|
|
300
|
-
console.error(`Error updating agent ${agentName}:`, error);
|
|
301
|
-
// Continue with other agents even if one fails
|
|
302
|
-
}
|
|
303
|
-
}
|
|
304
|
-
// Clean up temp file
|
|
305
|
-
try {
|
|
306
|
-
unlinkSync(tempFilePath);
|
|
307
|
-
}
|
|
308
|
-
catch (error) {
|
|
309
|
-
// Ignore cleanup errors
|
|
310
|
-
}
|
|
311
|
-
setConfig(toolConfig);
|
|
312
|
-
setSaveStatus("done");
|
|
313
|
-
}
|
|
314
|
-
catch (error) {
|
|
315
|
-
setSaveStatus("error");
|
|
316
|
-
setSaveError(error instanceof Error ? error.message : "Unknown error occurred");
|
|
317
|
-
}
|
|
318
|
-
};
|
|
319
|
-
// Wait for editor to open and close before showing UI
|
|
320
|
-
if (!editorOpened) {
|
|
321
|
-
return (_jsx(Box, { flexDirection: "column", children: _jsx(Text, { children: "Opening editor..." }) }));
|
|
322
|
-
}
|
|
323
|
-
// If there was an error during editor setup, show done stage with error
|
|
324
|
-
if (saveStatus === "error" && stage === "done") {
|
|
325
|
-
return (_jsx(DoneStage, { config: config, status: saveStatus, error: saveError, attachedAgents: selectedAgents }));
|
|
326
|
-
}
|
|
327
|
-
// Name input stage
|
|
328
|
-
if (stage === "name") {
|
|
329
|
-
return (_jsx(NameInputStage, { value: nameInput, onChange: setNameInput, onNext: (name) => {
|
|
330
|
-
setConfig({ ...config, name });
|
|
331
|
-
setStage("agentSelection");
|
|
332
|
-
}, onBack: () => {
|
|
333
|
-
// Clean up temp file
|
|
334
|
-
if (tempFilePath) {
|
|
335
|
-
try {
|
|
336
|
-
unlinkSync(tempFilePath);
|
|
337
|
-
}
|
|
338
|
-
catch (error) {
|
|
339
|
-
// Ignore cleanup errors
|
|
340
|
-
}
|
|
341
|
-
}
|
|
342
|
-
exit();
|
|
343
|
-
} }));
|
|
344
|
-
}
|
|
345
|
-
// Agent selection stage
|
|
346
|
-
if (stage === "agentSelection") {
|
|
347
|
-
return (_jsx(AgentSelectionStage, { selectedAgents: selectedAgents, onSelectedAgentsChange: setSelectedAgents, onNext: () => {
|
|
348
|
-
if (!config.name) {
|
|
349
|
-
setSaveStatus("error");
|
|
350
|
-
setSaveError("Tool name is required");
|
|
351
|
-
setStage("done");
|
|
352
|
-
return;
|
|
353
|
-
}
|
|
354
|
-
handleSave(config.name, selectedAgents);
|
|
355
|
-
setStage("done");
|
|
356
|
-
}, onBack: () => setStage("name") }));
|
|
357
|
-
}
|
|
358
|
-
// Done stage
|
|
359
|
-
if (stage === "done") {
|
|
360
|
-
return (_jsx(DoneStage, { config: config, status: saveStatus, error: saveError, attachedAgents: selectedAgents }));
|
|
361
|
-
}
|
|
362
|
-
return null;
|
|
363
|
-
}
|
|
364
|
-
// ============================================================================
|
|
365
|
-
// Export and Runner
|
|
366
|
-
// ============================================================================
|
|
367
|
-
export default ToolStubApp;
|
|
368
|
-
export async function runToolStub(props = {}) {
|
|
369
|
-
// Set stdin to raw mode to capture input
|
|
370
|
-
if (process.stdin.isTTY) {
|
|
371
|
-
process.stdin.setRawMode(true);
|
|
372
|
-
}
|
|
373
|
-
const { waitUntilExit } = render(_jsx(ToolStubApp, { ...props }));
|
|
374
|
-
// Wait for the app to exit before returning
|
|
375
|
-
await waitUntilExit();
|
|
376
|
-
}
|