@agent-team-foundation/first-tree-hub 0.2.0 → 0.3.1

package/dist/cli/index.mjs

@@ -1,9 +1,12 @@
 #!/usr/bin/env node
-import { A as loadRuntimeConfig, B as readConfigFile, C as stopPostgres, D as FirstTreeHubSDK, E as AgentSlot, F as agentConfigSchema, H as resetConfigMeta, I as clientConfigSchema, L as getConfigValue, M as createAdminUser, O as SdkError, P as DEFAULT_CONFIG_DIR, R as initConfig, T as AgentRuntime, U as serverConfigSchema, V as resetConfig, W as setConfigValue, _ as checkWebSocket, a as runMigrations, c as checkClientConfig, d as checkDocker, f as checkGitHubToken, g as checkServerReachable, h as checkServerHealth, i as promptMissingFields, j as registerBuiltinHandlers, k as getHandlerFactory, l as checkContextTreeRepo, m as checkServerConfig, o as checkAgentConfigs, p as checkNodeVersion, r as promptAddAgent, s as checkAgentTokens, t as startServer, u as checkDatabase, v as printResults, w as ClientRuntime, z as loadAgents } from "../core-CD3xEbyB.mjs";
+import { _ as resetConfig, b as serverConfigSchema, c as DEFAULT_CONFIG_DIR, d as clientConfigSchema, g as readConfigFile, h as loadAgents, l as DEFAULT_DATA_DIR, m as initConfig, o as resolveAgentToken, p as getConfigValue, s as resolveServerUrl, t as bootstrapToken, u as agentConfigSchema, v as resetConfigMeta, x as setConfigValue } from "../bootstrap-B9JsJR3Z.mjs";
+import { A as ClientRuntime, B as registerBuiltinHandlers, C as checkWebSocket, F as SdkError, I as SessionRegistry, L as cleanWorkspaces, M as AgentSlot, N as DEFAULT_WORKSPACE_TTL_MS, P as FirstTreeHubSDK, R as getHandlerFactory, S as checkServerReachable, V as createAdminUser, _ as checkDocker, a as formatCheckReport, b as checkServerConfig, c as onboardContinue, d as runMigrations, f as checkAgentConfigs, g as checkDatabase, h as checkContextTreeRepo, i as promptMissingFields, j as AgentRuntime, k as stopPostgres, l as onboardCreate, m as checkClientConfig, n as isInteractive, o as loadOnboardState, p as checkAgentTokens, r as promptAddAgent, s as onboardCheck, t as startServer, u as saveOnboardState, v as checkGitHubToken, w as printResults, x as checkServerHealth, y as checkNodeVersion, z as loadRuntimeConfig } from "../core-D-c9r7D6.mjs";
+import { n as bindFeishuUser, t as bindFeishuBot } from "../feishu-Y4m2zFc3.mjs";
 import { createRequire } from "node:module";
 import { Command } from "commander";
-import { existsSync, mkdirSync, rmSync } from "node:fs";
+import { existsSync, mkdirSync, readdirSync, rmSync } from "node:fs";
 import { join } from "node:path";
+import { confirm, input, select } from "@inquirer/prompts";
 //#region src/commands/admin.ts
 function registerAdminCommands(program) {
 	program.command("admin").description("Admin user management").command("create").description("Create an admin user").option("-u, --username <name>", "Admin username", "admin").option("-p, --password <pass>", "Admin password (auto-generated if omitted)").action(async (options) => {
@@ -84,6 +87,37 @@ function registerAgentCommands(program) {
 	});
 }
 //#endregion
+//#region src/commands/bind.ts
+function registerBindCommands(program) {
+	program.command("bind-bot").description("Bind a Feishu bot to this agent (self-service)").requiredOption("--platform <platform>", "Platform: feishu").requiredOption("--app-id <id>", "Feishu bot App ID").requiredOption("--app-secret <secret>", "Feishu bot App Secret").option("--server <url>", "Hub server URL").action(async (options) => {
+		try {
+			if (options.platform !== "feishu") fail("UNSUPPORTED_PLATFORM", `Platform "${options.platform}" is not supported. Use "feishu".`);
+			await bindFeishuBot(resolveServerUrl(options.server), resolveAgentToken(), options.appId, options.appSecret);
+			process.stderr.write("Feishu bot bound successfully.\n");
+			success({
+				platform: "feishu",
+				bound: true
+			});
+		} catch (error) {
+			fail("BIND_BOT_ERROR", error instanceof Error ? error.message : String(error));
+		}
+	});
+	program.command("bind-user <humanAgentId>").description("Bind a Feishu user to a human agent (via delegate_mention)").requiredOption("--platform <platform>", "Platform: feishu").requiredOption("--feishu-id <id>", "Feishu user ID (ou_xxx)").option("--server <url>", "Hub server URL").action(async (humanAgentId, options) => {
+		try {
+			if (options.platform !== "feishu") fail("UNSUPPORTED_PLATFORM", `Platform "${options.platform}" is not supported. Use "feishu".`);
+			await bindFeishuUser(resolveServerUrl(options.server), resolveAgentToken(), humanAgentId, options.feishuId);
+			process.stderr.write(`Feishu user ${options.feishuId} bound to ${humanAgentId}.\n`);
+			success({
+				platform: "feishu",
+				humanAgentId,
+				feishuUserId: options.feishuId
+			});
+		} catch (error) {
+			fail("BIND_USER_ERROR", error instanceof Error ? error.message : String(error));
+		}
+	});
+}
+//#endregion
 //#region src/cli/util.ts
 function resolveConfig() {
 	const token = process.env.FIRST_TREE_HUB_TOKEN;
@@ -237,7 +271,7 @@ function registerClientCommands(program) {
 			process.exit(1);
 		}
 	});
-	client.command("remove <name>").description("Remove an agent instance").action((name) => {
+	client.command("remove <name>").description("Remove an agent instance and its runtime data").action((name) => {
 		const agentDir = join(DEFAULT_CONFIG_DIR, "agents", name);
 		if (!existsSync(agentDir)) {
 			process.stderr.write(`  Agent "${name}" not found.\n`);
@@ -247,6 +281,11 @@ function registerClientCommands(program) {
 			recursive: true,
 			force: true
 		});
+		rmSync(join(DEFAULT_DATA_DIR, "workspaces", name), {
+			recursive: true,
+			force: true
+		});
+		rmSync(join(DEFAULT_DATA_DIR, "sessions", `${name}.json`), { force: true });
 		process.stderr.write(`  Agent "${name}" removed.\n`);
 	});
 	client.command("list").description("List configured agents").action(() => {
@@ -268,6 +307,28 @@ function registerClientCommands(program) {
 			process.stderr.write("  No agents configured.\n");
 		}
 	});
+	client.command("workspace").description("Manage agent workspaces").command("clean [agent-name]").description("Remove stale workspace directories (older than TTL with no active session)").option("--ttl <days>", "TTL in days", String(DEFAULT_WORKSPACE_TTL_MS / (1440 * 60 * 1e3))).action((agentName, options) => {
+		const defaultDays = DEFAULT_WORKSPACE_TTL_MS / (1440 * 60 * 1e3);
+		const ttlMs = Number.parseInt(options?.ttl ?? String(defaultDays), 10) * 24 * 60 * 60 * 1e3;
+		const workspacesDir = join(DEFAULT_DATA_DIR, "workspaces");
+		if (!existsSync(workspacesDir)) {
+			process.stderr.write("  No workspaces found.\n");
+			return;
+		}
+		const agentNames = agentName ? [agentName] : readdirSync(workspacesDir);
+		let totalRemoved = 0;
+		for (const name of agentNames) {
+			const agentWorkspaceRoot = join(workspacesDir, name);
+			if (!existsSync(agentWorkspaceRoot)) continue;
+			const persisted = new SessionRegistry(join(DEFAULT_DATA_DIR, "sessions", `${name}.json`)).load();
+			const activeChatIds = /* @__PURE__ */ new Set();
+			for (const [chatId, data] of persisted) if (data.status !== "evicted") activeChatIds.add(chatId);
+			const removed = cleanWorkspaces(agentWorkspaceRoot, activeChatIds, ttlMs);
+			totalRemoved += removed.length;
+			for (const chatId of removed) process.stderr.write(`  Removed: ${name}/${chatId}\n`);
+		}
+		process.stderr.write(`  ${totalRemoved} workspace(s) cleaned.\n`);
+	});
 }
 //#endregion
 //#region src/commands/config.ts
@@ -395,6 +456,142 @@ function registerHistoryCommand(program) {
 	});
 }
 //#endregion
+//#region src/commands/onboard.ts
+async function promptMissing(args) {
+	let ghUsername = null;
+	try {
+		const { getGitHubUsername } = await import("../bootstrap-B9JsJR3Z.mjs").then((n) => n.n);
+		ghUsername = getGitHubUsername();
+	} catch {}
+	if (!args.id) {
+		args.id = await input({
+			message: "Member ID (directory name):",
+			default: ghUsername ?? void 0
+		});
+		saveOnboardState(args);
+	}
+	if (!args.type) {
+		args.type = await select({
+			message: "Agent type:",
+			choices: [
+				{
+					name: "human",
+					value: "human"
+				},
+				{
+					name: "personal_assistant",
+					value: "personal_assistant"
+				},
+				{
+					name: "autonomous_agent",
+					value: "autonomous_agent"
+				}
+			]
+		});
+		saveOnboardState(args);
+	}
+	if (!args.role) {
+		args.role = await input({ message: "Role:" });
+		saveOnboardState(args);
+	}
+	if (!args.domains) {
+		args.domains = await input({ message: "Domains (comma-separated):" });
+		saveOnboardState(args);
+	}
+	if (!args.displayName) {
+		const name = await input({ message: `Display name (Enter to use "${args.id}"):` });
+		if (name) {
+			args.displayName = name;
+			saveOnboardState(args);
+		}
+	}
+	if (!args.assistant) {
+		if (await confirm({
+			message: "Create a personal assistant?",
+			default: false
+		})) {
+			args.assistant = await input({
+				message: "Assistant ID:",
+				default: `${args.id}-assistant`
+			});
+			saveOnboardState(args);
+		}
+	}
+	if (!args.server) try {
+		const { resolveServerUrl } = await import("../bootstrap-B9JsJR3Z.mjs").then((n) => n.n);
+		resolveServerUrl();
+	} catch {
+		args.server = await input({ message: "Hub server URL:" });
+		saveOnboardState(args);
+	}
+	if (!args.feishuBotAppId) {
+		if (await confirm({
+			message: "Bind Feishu bot?",
+			default: false
+		})) {
+			args.feishuBotAppId = await input({ message: "Feishu App ID:" });
+			args.feishuBotAppSecret = await input({ message: "Feishu App Secret:" });
+			saveOnboardState(args);
+		}
+	}
+}
+function registerOnboardCommand(program) {
+	program.command("onboard").description("Onboard a new member to First Tree Hub (end-to-end)").option("--id <id>", "Member ID (directory name)").option("--type <type>", "Agent type: human | personal_assistant | autonomous_agent").option("--display-name <name>", "Display name (defaults to id)").option("--role <role>", "Role description").option("--domains <domains>", "Comma-separated domains").option("--assistant <id>", "Also create a personal_assistant with this ID").option("--delegate-mention <id>", "Set delegate_mention field").option("--server <url>", "Hub server URL").option("--feishu-bot-app-id <id>", "Feishu bot App ID").option("--feishu-bot-app-secret <secret>", "Feishu bot App Secret").option("--check", "Dry-run: show readiness checklist without executing").option("--continue", "Resume after PR merge (Phase 2)").action(async (options) => {
+		try {
+			const args = {
+				...loadOnboardState() ?? {},
+				...options.id && { id: options.id },
+				...options.type && { type: options.type },
+				...options.displayName && { displayName: options.displayName },
+				...options.role && { role: options.role },
+				...options.domains && { domains: options.domains },
+				...options.assistant && { assistant: options.assistant },
+				...options.delegateMention && { delegateMention: options.delegateMention },
+				...options.server && { server: options.server },
+				...options.feishuBotAppId && { feishuBotAppId: options.feishuBotAppId },
+				...options.feishuBotAppSecret && { feishuBotAppSecret: options.feishuBotAppSecret },
+				check: options.check,
+				continue: options.continue
+			};
+			if (!args.feishuBotAppId && process.env.FEISHU_APP_ID) args.feishuBotAppId = process.env.FEISHU_APP_ID;
+			if (!args.feishuBotAppSecret && process.env.FEISHU_APP_SECRET) args.feishuBotAppSecret = process.env.FEISHU_APP_SECRET;
+			if (args.continue) {
+				await onboardContinue(args);
+				return;
+			}
+			if (args.check) {
+				const items = await onboardCheck(args);
+				const report = formatCheckReport(items);
+				process.stderr.write(`\nOnboard Check: ${args.id ?? "(no id)"}\n\n${report}\n\n`);
+				if (items.some((i) => i.status === "missing_required" || i.status === "error")) process.exit(1);
+				return;
+			}
+			if (isInteractive()) await promptMissing(args);
+			const items = await onboardCheck(args);
+			if (items.some((i) => i.status === "missing_required" || i.status === "error")) {
+				const report = formatCheckReport(items);
+				process.stderr.write(`\nOnboard Check: ${args.id ?? "(no id)"}\n\n${report}\n\n`);
+				fail("MISSING_PARAMS", "Required parameters are missing. See checklist above.");
+			}
+			const result = await onboardCreate(args);
+			process.stderr.write(`\nPR created: ${result.prUrl}\n`);
+			process.stderr.write("Review and merge the PR, then run:\n");
+			process.stderr.write("  first-tree-hub onboard --continue\n\n");
+			success({
+				phase: "create",
+				prUrl: result.prUrl
+			});
+		} catch (error) {
+			const msg = error instanceof Error ? error.message : String(error);
+			if (isInteractive()) {
+				process.stderr.write(`\n❌ ${msg}\n\n`);
+				process.exit(1);
+			}
+			fail("ONBOARD_ERROR", msg);
+		}
+	});
+}
+//#endregion
 //#region src/commands/send.ts
 const MAX_STDIN_BYTES = 10 * 1024 * 1024;
 /** Read all of stdin as a string. Returns null if stdin is a TTY. */
@@ -555,6 +752,23 @@ function formatUptime(seconds) {
 	return `${mins}m`;
 }
 //#endregion
+//#region src/commands/token.ts
+function registerTokenCommands(program) {
+	program.command("token").description("Agent token management").command("bootstrap <agentId>").description("Bootstrap a token using GitHub identity (requires gh CLI)").option("--save-to <target>", "Save token to: \"agent\" (default) or a file path", "agent").option("--server <url>", "Hub server URL").action(async (agentId, options) => {
+		try {
+			const result = await bootstrapToken(resolveServerUrl(options.server), agentId, { saveTo: options.saveTo });
+			if (options.saveTo === "agent") process.stderr.write(`Token saved to ~/.first-tree-hub/agents/${agentId}/agent.yaml\n`);
+			else process.stderr.write(`Token saved to ${options.saveTo}\n`);
+			success({
+				agentId: result.agentId,
+				tokenSaved: true
+			});
+		} catch (error) {
+			fail("BOOTSTRAP_ERROR", error instanceof Error ? error.message : String(error));
+		}
+	});
+}
+//#endregion
 //#region src/cli/connect.ts
 function registerConnectCommand(program) {
 	program.command("connect").description("Connect a single agent to server and process messages").option("-t, --type <type>", "Handler type", "claude-code").option("--concurrency <n>", "Max parallel message processing", "5").option("--server <url>", "Override FIRST_TREE_HUB_SERVER").action(async (options) => {
@@ -628,6 +842,9 @@ registerStatusCommand(program);
 registerConnectCommand(program);
 registerStartCommand(program);
 registerAgentCommands(program);
+registerOnboardCommand(program);
+registerTokenCommands(program);
+registerBindCommands(program);
 registerSendCommand(program);
 registerChatsCommand(program);
 registerHistoryCommand(program);