@aigne/cli 1.60.0-beta → 1.74.0-beta

package/dist/commands/hub.cjs

@@ -0,0 +1,297 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const require_constants = require('../utils/aigne-hub/constants.cjs');
+const require_index = require('../utils/aigne-hub/store/index.cjs');
+const require_credential = require('../utils/aigne-hub/credential.cjs');
+const require_utils_get_url_origin = require('../utils/get-url-origin.cjs');
+const require_utils_aigne_hub_user = require('../utils/aigne-hub-user.cjs');
+let chalk = require("chalk");
+chalk = require_rolldown_runtime.__toESM(chalk);
+let _aigne_aigne_hub = require("@aigne/aigne-hub");
+let inquirer = require("inquirer");
+inquirer = require_rolldown_runtime.__toESM(inquirer);
+let cli_table3 = require("cli-table3");
+cli_table3 = require_rolldown_runtime.__toESM(cli_table3);
+
+//#region src/commands/hub.ts
+const formatNumber = (balance) => {
+	const balanceNum = String(balance).replace(/[^\d.-]/g, "").split(".")[0];
+	if (!balanceNum) return "0";
+	return (balanceNum || "").trim().replace(/\B(?=(\d{3})+(?!\d))/g, ",");
+};
+const formatNumberWithColor = (balance) => {
+	return chalk.default.yellow(formatNumber(balance));
+};
+function formatHubInfoName(name) {
+	return chalk.default.bold(`${name}:`.padEnd(10));
+}
+function printHubStatus(data) {
+	const divider = "─".repeat(46);
+	console.log(chalk.default.bold("AIGNE Hub Connection"));
+	console.log(chalk.default.gray(divider));
+	console.log(`${chalk.default.bold("Hub:".padEnd(10))} ${data.hub}`);
+	console.log(`${chalk.default.bold("Status:".padEnd(10))} ${data.status === "Connected" ? chalk.default.green(`${data.status} ✅`) : `${data.status}`}`);
+	console.log("");
+	console.log(chalk.default.bold("User:"));
+	console.log(`  ${formatHubInfoName("Name")} ${data.user.name}`);
+	console.log(`  ${formatHubInfoName("DID")} ${data.user.did}`);
+	console.log(`  ${formatHubInfoName("Email")} ${data.user.email}`);
+	console.log("");
+	if (data.enableCredit) {
+		console.log(chalk.default.bold("Credits:"));
+		console.log(`  ${formatHubInfoName("Total")} ${formatNumberWithColor(data.credits.total)}`);
+		console.log(`  ${formatHubInfoName("Used")} ${formatNumberWithColor(data.credits.used)}`);
+		console.log(`  ${formatHubInfoName("Available")} ${formatNumberWithColor(data.credits.available)}`);
+		console.log("");
+		console.log(chalk.default.bold("Links:"));
+		if (data.links.payment) console.log(`  ${formatHubInfoName("Payment")} ${data.links.payment}`);
+		if (data.links.profile) console.log(`  ${formatHubInfoName("Credits")} ${data.links.profile}`);
+	}
+}
+async function getHubs() {
+	try {
+		const hosts = await (await require_index.default()).listHosts();
+		const statusList = [];
+		for (const host of hosts) statusList.push({
+			host: new URL(host.AIGNE_HUB_API_URL).host,
+			apiUrl: host.AIGNE_HUB_API_URL,
+			apiKey: host.AIGNE_HUB_API_KEY || ""
+		});
+		return statusList;
+	} catch {
+		return [];
+	}
+}
+const getDefaultHub = async () => {
+	try {
+		return (await (await require_index.default()).getDefault())?.AIGNE_HUB_API_URL || _aigne_aigne_hub.AIGNE_HUB_URL;
+	} catch {
+		return _aigne_aigne_hub.AIGNE_HUB_URL;
+	}
+};
+async function formatHubsList(statusList) {
+	if (statusList?.length === 0) {
+		console.log(chalk.default.yellow("No AIGNE Hub connected."));
+		console.log("Use 'aigne hub connect' to connect to a hub.");
+		return;
+	}
+	const defaultHub = await getDefaultHub();
+	const table = new cli_table3.default({
+		head: ["URL", "ACTIVE"],
+		colWidths: [70, 10],
+		style: {
+			head: ["cyan"],
+			border: ["grey"]
+		}
+	});
+	console.log(chalk.default.blue("Connected AIGNE Hubs:\n"));
+	for (const status of statusList) {
+		const isConnected = require_utils_get_url_origin.getUrlOrigin(status.apiUrl) === require_utils_get_url_origin.getUrlOrigin(defaultHub);
+		table.push([require_utils_get_url_origin.getUrlOrigin(status.apiUrl), isConnected ? "YES" : "NO"]);
+	}
+	console.log(table.toString());
+	console.log(chalk.default.blue("Use 'aigne hub use' to switch to a different hub."));
+}
+function createHubCommand() {
+	return {
+		command: "hub <command>",
+		describe: "Manage AIGNE Hub connections",
+		builder: (yargs) => yargs.command(["list", "ls"], "List all connected AIGNE Hubs", listHubs).command({
+			command: "connect [url]",
+			describe: "Connect to an AIGNE Hub",
+			builder: (yargs$1) => yargs$1.positional("url", {
+				type: "string",
+				describe: "The URL of the AIGNE Hub to connect to",
+				default: null
+			}),
+			handler: (args) => {
+				if (args.url) saveAndConnect(args.url);
+				else connectHub();
+			}
+		}).command("use", "Switch to a different AIGNE Hub", useHub).command(["status", "st"], "Show details of a connected hub", showInfo).command(["remove", "rm"], "Remove a connected hub", removeHub).demandCommand(1, "Please provide a valid hub command"),
+		handler: () => {}
+	};
+}
+const listHubs = async () => {
+	await formatHubsList(await getHubs());
+};
+async function connectHub() {
+	const defaultUrl = "https://hub.aigne.io";
+	const { isOfficial } = await inquirer.default.prompt({
+		type: "select",
+		name: "isOfficial",
+		message: `Choose a hub to connect:`,
+		choices: [{
+			name: `Official Hub (${defaultUrl})`,
+			value: true
+		}, {
+			name: `Custom Hub URL`,
+			value: false
+		}],
+		default: true
+	});
+	let currentUrl = defaultUrl;
+	if (!isOfficial) {
+		const { customUrl } = await inquirer.default.prompt({
+			type: "input",
+			name: "customUrl",
+			message: "Enter the URL of your AIGNE Hub:",
+			validate: validateUrl
+		});
+		currentUrl = customUrl;
+	}
+	await saveAndConnect(currentUrl);
+}
+async function useHub() {
+	const hubs = await getHubs();
+	if (!hubs.length) {
+		console.log(chalk.default.yellow("No AIGNE Hub connected."));
+		return;
+	}
+	const { hubApiKey } = await inquirer.default.prompt({
+		type: "select",
+		name: "hubApiKey",
+		message: `Choose a hub to switch to:`,
+		choices: hubs.map((h) => ({
+			name: require_utils_get_url_origin.getUrlOrigin(h.apiUrl),
+			value: h.apiUrl
+		}))
+	});
+	await setDefaultHub(hubApiKey);
+}
+async function removeHub() {
+	const hubs = await getHubs();
+	if (!hubs.length) {
+		console.log(chalk.default.yellow("No AIGNE Hub connected."));
+		return;
+	}
+	const { hubApiKey } = await inquirer.default.prompt({
+		type: "select",
+		name: "hubApiKey",
+		message: `Choose a hub to remove:`,
+		choices: hubs.map((h) => ({
+			name: require_utils_get_url_origin.getUrlOrigin(h.apiUrl),
+			value: h.apiUrl
+		}))
+	});
+	await deleteHub(hubApiKey);
+}
+async function showInfo() {
+	const hubs = await getHubs();
+	if (!hubs.length) {
+		console.log(chalk.default.yellow("No AIGNE Hub connected."));
+		return;
+	}
+	const defaultHub = await getDefaultHub();
+	const defaultHubIndex = hubs.findIndex((h) => require_utils_get_url_origin.getUrlOrigin(h.apiUrl) === require_utils_get_url_origin.getUrlOrigin(defaultHub));
+	const { hubApiKey } = await inquirer.default.prompt({
+		type: "select",
+		name: "hubApiKey",
+		message: `Choose a hub to view info:`,
+		choices: hubs.map((h, index) => ({
+			name: `${require_utils_get_url_origin.getUrlOrigin(h.apiUrl)} ${defaultHubIndex === index ? "(connected)" : ""}`,
+			value: h.apiUrl
+		}))
+	});
+	try {
+		await printHubDetails(hubApiKey);
+	} catch (error) {
+		console.error(chalk.default.red("✗ Failed to print hub details:"), error.message);
+	}
+}
+function validateUrl(input) {
+	try {
+		return new URL(input).protocol.startsWith("http") ? true : "Must be http or https";
+	} catch {
+		return "Invalid URL";
+	}
+}
+async function saveAndConnect(url) {
+	const secretStore = await require_index.default();
+	const currentKey = await secretStore.getKey(url);
+	if (currentKey?.AIGNE_HUB_API_URL) {
+		await setDefaultHub(currentKey.AIGNE_HUB_API_URL);
+		console.log(chalk.default.green(`✓ Hub ${require_utils_get_url_origin.getUrlOrigin(currentKey.AIGNE_HUB_API_URL)} connected successfully.`));
+		return;
+	}
+	try {
+		if (require_constants.isTest) {
+			await secretStore.setKey("https://hub.aigne.io/ai-kit", "test-key");
+			await secretStore.setDefault("https://hub.aigne.io/ai-kit");
+			console.log(chalk.default.green(`✓ Hub https://hub.aigne.io/ai-kit connected successfully.`));
+			return;
+		}
+		await require_credential.connectToAIGNEHub(url);
+		console.log(chalk.default.green(`✓ Hub ${require_utils_get_url_origin.getUrlOrigin(url)} connected successfully.`));
+	} catch (error) {
+		console.error(chalk.default.red("✗ Failed to connect:"), error.message);
+	}
+}
+async function setDefaultHub(url) {
+	try {
+		const secretStore = await require_index.default();
+		const key = await secretStore.getKey(url);
+		if (!key) {
+			console.error(chalk.default.red("✗ Hub not found"));
+			return;
+		}
+		await secretStore.setDefault(key.AIGNE_HUB_API_URL);
+		console.log(chalk.default.green(`✓ Switched active hub to ${require_utils_get_url_origin.getUrlOrigin(url)}`));
+	} catch {
+		console.error(chalk.default.red("✗ Failed to set default hub"));
+	}
+}
+async function deleteHub(url) {
+	try {
+		const secretStore = await require_index.default();
+		if (!await secretStore.getKey(url)) {
+			console.error(chalk.default.red("✗ Hub not found"));
+			return;
+		}
+		const defaultHub = await getDefaultHub();
+		await secretStore.deleteKey(url);
+		if (!defaultHub) return;
+		if (require_utils_get_url_origin.getUrlOrigin(url) === require_utils_get_url_origin.getUrlOrigin(defaultHub)) {
+			await secretStore.deleteDefault();
+			const nextHub = (await getHubs())[0];
+			if (nextHub) {
+				await secretStore.setDefault(nextHub?.apiUrl);
+				console.log(chalk.default.green(`✓ Hub ${require_utils_get_url_origin.getUrlOrigin(url)} removed, switched to ${require_utils_get_url_origin.getUrlOrigin(nextHub?.apiUrl)}`));
+				return;
+			}
+		}
+		console.log(chalk.default.green(`✓ Hub ${require_utils_get_url_origin.getUrlOrigin(url)} removed`));
+	} catch {
+		console.error(chalk.default.red("✗ Failed to delete hub"));
+	}
+}
+async function printHubDetails(url) {
+	const key = await (await require_index.default()).getKey(url);
+	const defaultHub = await getDefaultHub();
+	const isDefault = require_utils_get_url_origin.getUrlOrigin(url) === require_utils_get_url_origin.getUrlOrigin(defaultHub);
+	const userInfo = await require_utils_aigne_hub_user.getUserInfo({
+		baseUrl: key?.AIGNE_HUB_API_URL || "",
+		apiKey: key?.AIGNE_HUB_API_KEY || ""
+	}).catch(() => null);
+	printHubStatus({
+		hub: require_utils_get_url_origin.getUrlOrigin(url),
+		status: isDefault ? "Connected" : "Not Used",
+		user: {
+			name: userInfo?.user.fullName || "",
+			did: userInfo?.user.did || "",
+			email: userInfo?.user.email || ""
+		},
+		credits: {
+			available: userInfo?.creditBalance?.balance || "0",
+			total: userInfo?.creditBalance?.total || "0",
+			used: String(parseFloat(userInfo?.creditBalance?.total || "0") - parseFloat(userInfo?.creditBalance?.balance || "0"))
+		},
+		links: {
+			payment: userInfo?.paymentLink || "",
+			profile: userInfo?.profileLink || ""
+		},
+		enableCredit: userInfo?.enableCredit || false
+	});
+}
+
+//#endregion
+exports.createHubCommand = createHubCommand;

package/dist/commands/hub.mjs

@@ -0,0 +1,294 @@
+import { isTest } from "../utils/aigne-hub/constants.mjs";
+import store_default from "../utils/aigne-hub/store/index.mjs";
+import { connectToAIGNEHub } from "../utils/aigne-hub/credential.mjs";
+import { getUrlOrigin } from "../utils/get-url-origin.mjs";
+import { getUserInfo } from "../utils/aigne-hub-user.mjs";
+import chalk from "chalk";
+import { AIGNE_HUB_URL } from "@aigne/aigne-hub";
+import inquirer from "inquirer";
+import Table from "cli-table3";
+
+//#region src/commands/hub.ts
+const formatNumber = (balance) => {
+	const balanceNum = String(balance).replace(/[^\d.-]/g, "").split(".")[0];
+	if (!balanceNum) return "0";
+	return (balanceNum || "").trim().replace(/\B(?=(\d{3})+(?!\d))/g, ",");
+};
+const formatNumberWithColor = (balance) => {
+	return chalk.yellow(formatNumber(balance));
+};
+function formatHubInfoName(name) {
+	return chalk.bold(`${name}:`.padEnd(10));
+}
+function printHubStatus(data) {
+	const divider = "─".repeat(46);
+	console.log(chalk.bold("AIGNE Hub Connection"));
+	console.log(chalk.gray(divider));
+	console.log(`${chalk.bold("Hub:".padEnd(10))} ${data.hub}`);
+	console.log(`${chalk.bold("Status:".padEnd(10))} ${data.status === "Connected" ? chalk.green(`${data.status} ✅`) : `${data.status}`}`);
+	console.log("");
+	console.log(chalk.bold("User:"));
+	console.log(`  ${formatHubInfoName("Name")} ${data.user.name}`);
+	console.log(`  ${formatHubInfoName("DID")} ${data.user.did}`);
+	console.log(`  ${formatHubInfoName("Email")} ${data.user.email}`);
+	console.log("");
+	if (data.enableCredit) {
+		console.log(chalk.bold("Credits:"));
+		console.log(`  ${formatHubInfoName("Total")} ${formatNumberWithColor(data.credits.total)}`);
+		console.log(`  ${formatHubInfoName("Used")} ${formatNumberWithColor(data.credits.used)}`);
+		console.log(`  ${formatHubInfoName("Available")} ${formatNumberWithColor(data.credits.available)}`);
+		console.log("");
+		console.log(chalk.bold("Links:"));
+		if (data.links.payment) console.log(`  ${formatHubInfoName("Payment")} ${data.links.payment}`);
+		if (data.links.profile) console.log(`  ${formatHubInfoName("Credits")} ${data.links.profile}`);
+	}
+}
+async function getHubs() {
+	try {
+		const hosts = await (await store_default()).listHosts();
+		const statusList = [];
+		for (const host of hosts) statusList.push({
+			host: new URL(host.AIGNE_HUB_API_URL).host,
+			apiUrl: host.AIGNE_HUB_API_URL,
+			apiKey: host.AIGNE_HUB_API_KEY || ""
+		});
+		return statusList;
+	} catch {
+		return [];
+	}
+}
+const getDefaultHub = async () => {
+	try {
+		return (await (await store_default()).getDefault())?.AIGNE_HUB_API_URL || AIGNE_HUB_URL;
+	} catch {
+		return AIGNE_HUB_URL;
+	}
+};
+async function formatHubsList(statusList) {
+	if (statusList?.length === 0) {
+		console.log(chalk.yellow("No AIGNE Hub connected."));
+		console.log("Use 'aigne hub connect' to connect to a hub.");
+		return;
+	}
+	const defaultHub = await getDefaultHub();
+	const table = new Table({
+		head: ["URL", "ACTIVE"],
+		colWidths: [70, 10],
+		style: {
+			head: ["cyan"],
+			border: ["grey"]
+		}
+	});
+	console.log(chalk.blue("Connected AIGNE Hubs:\n"));
+	for (const status of statusList) {
+		const isConnected = getUrlOrigin(status.apiUrl) === getUrlOrigin(defaultHub);
+		table.push([getUrlOrigin(status.apiUrl), isConnected ? "YES" : "NO"]);
+	}
+	console.log(table.toString());
+	console.log(chalk.blue("Use 'aigne hub use' to switch to a different hub."));
+}
+function createHubCommand() {
+	return {
+		command: "hub <command>",
+		describe: "Manage AIGNE Hub connections",
+		builder: (yargs) => yargs.command(["list", "ls"], "List all connected AIGNE Hubs", listHubs).command({
+			command: "connect [url]",
+			describe: "Connect to an AIGNE Hub",
+			builder: (yargs$1) => yargs$1.positional("url", {
+				type: "string",
+				describe: "The URL of the AIGNE Hub to connect to",
+				default: null
+			}),
+			handler: (args) => {
+				if (args.url) saveAndConnect(args.url);
+				else connectHub();
+			}
+		}).command("use", "Switch to a different AIGNE Hub", useHub).command(["status", "st"], "Show details of a connected hub", showInfo).command(["remove", "rm"], "Remove a connected hub", removeHub).demandCommand(1, "Please provide a valid hub command"),
+		handler: () => {}
+	};
+}
+const listHubs = async () => {
+	await formatHubsList(await getHubs());
+};
+async function connectHub() {
+	const defaultUrl = "https://hub.aigne.io";
+	const { isOfficial } = await inquirer.prompt({
+		type: "select",
+		name: "isOfficial",
+		message: `Choose a hub to connect:`,
+		choices: [{
+			name: `Official Hub (${defaultUrl})`,
+			value: true
+		}, {
+			name: `Custom Hub URL`,
+			value: false
+		}],
+		default: true
+	});
+	let currentUrl = defaultUrl;
+	if (!isOfficial) {
+		const { customUrl } = await inquirer.prompt({
+			type: "input",
+			name: "customUrl",
+			message: "Enter the URL of your AIGNE Hub:",
+			validate: validateUrl
+		});
+		currentUrl = customUrl;
+	}
+	await saveAndConnect(currentUrl);
+}
+async function useHub() {
+	const hubs = await getHubs();
+	if (!hubs.length) {
+		console.log(chalk.yellow("No AIGNE Hub connected."));
+		return;
+	}
+	const { hubApiKey } = await inquirer.prompt({
+		type: "select",
+		name: "hubApiKey",
+		message: `Choose a hub to switch to:`,
+		choices: hubs.map((h) => ({
+			name: getUrlOrigin(h.apiUrl),
+			value: h.apiUrl
+		}))
+	});
+	await setDefaultHub(hubApiKey);
+}
+async function removeHub() {
+	const hubs = await getHubs();
+	if (!hubs.length) {
+		console.log(chalk.yellow("No AIGNE Hub connected."));
+		return;
+	}
+	const { hubApiKey } = await inquirer.prompt({
+		type: "select",
+		name: "hubApiKey",
+		message: `Choose a hub to remove:`,
+		choices: hubs.map((h) => ({
+			name: getUrlOrigin(h.apiUrl),
+			value: h.apiUrl
+		}))
+	});
+	await deleteHub(hubApiKey);
+}
+async function showInfo() {
+	const hubs = await getHubs();
+	if (!hubs.length) {
+		console.log(chalk.yellow("No AIGNE Hub connected."));
+		return;
+	}
+	const defaultHub = await getDefaultHub();
+	const defaultHubIndex = hubs.findIndex((h) => getUrlOrigin(h.apiUrl) === getUrlOrigin(defaultHub));
+	const { hubApiKey } = await inquirer.prompt({
+		type: "select",
+		name: "hubApiKey",
+		message: `Choose a hub to view info:`,
+		choices: hubs.map((h, index) => ({
+			name: `${getUrlOrigin(h.apiUrl)} ${defaultHubIndex === index ? "(connected)" : ""}`,
+			value: h.apiUrl
+		}))
+	});
+	try {
+		await printHubDetails(hubApiKey);
+	} catch (error) {
+		console.error(chalk.red("✗ Failed to print hub details:"), error.message);
+	}
+}
+function validateUrl(input) {
+	try {
+		return new URL(input).protocol.startsWith("http") ? true : "Must be http or https";
+	} catch {
+		return "Invalid URL";
+	}
+}
+async function saveAndConnect(url) {
+	const secretStore = await store_default();
+	const currentKey = await secretStore.getKey(url);
+	if (currentKey?.AIGNE_HUB_API_URL) {
+		await setDefaultHub(currentKey.AIGNE_HUB_API_URL);
+		console.log(chalk.green(`✓ Hub ${getUrlOrigin(currentKey.AIGNE_HUB_API_URL)} connected successfully.`));
+		return;
+	}
+	try {
+		if (isTest) {
+			await secretStore.setKey("https://hub.aigne.io/ai-kit", "test-key");
+			await secretStore.setDefault("https://hub.aigne.io/ai-kit");
+			console.log(chalk.green(`✓ Hub https://hub.aigne.io/ai-kit connected successfully.`));
+			return;
+		}
+		await connectToAIGNEHub(url);
+		console.log(chalk.green(`✓ Hub ${getUrlOrigin(url)} connected successfully.`));
+	} catch (error) {
+		console.error(chalk.red("✗ Failed to connect:"), error.message);
+	}
+}
+async function setDefaultHub(url) {
+	try {
+		const secretStore = await store_default();
+		const key = await secretStore.getKey(url);
+		if (!key) {
+			console.error(chalk.red("✗ Hub not found"));
+			return;
+		}
+		await secretStore.setDefault(key.AIGNE_HUB_API_URL);
+		console.log(chalk.green(`✓ Switched active hub to ${getUrlOrigin(url)}`));
+	} catch {
+		console.error(chalk.red("✗ Failed to set default hub"));
+	}
+}
+async function deleteHub(url) {
+	try {
+		const secretStore = await store_default();
+		if (!await secretStore.getKey(url)) {
+			console.error(chalk.red("✗ Hub not found"));
+			return;
+		}
+		const defaultHub = await getDefaultHub();
+		await secretStore.deleteKey(url);
+		if (!defaultHub) return;
+		if (getUrlOrigin(url) === getUrlOrigin(defaultHub)) {
+			await secretStore.deleteDefault();
+			const nextHub = (await getHubs())[0];
+			if (nextHub) {
+				await secretStore.setDefault(nextHub?.apiUrl);
+				console.log(chalk.green(`✓ Hub ${getUrlOrigin(url)} removed, switched to ${getUrlOrigin(nextHub?.apiUrl)}`));
+				return;
+			}
+		}
+		console.log(chalk.green(`✓ Hub ${getUrlOrigin(url)} removed`));
+	} catch {
+		console.error(chalk.red("✗ Failed to delete hub"));
+	}
+}
+async function printHubDetails(url) {
+	const key = await (await store_default()).getKey(url);
+	const defaultHub = await getDefaultHub();
+	const isDefault = getUrlOrigin(url) === getUrlOrigin(defaultHub);
+	const userInfo = await getUserInfo({
+		baseUrl: key?.AIGNE_HUB_API_URL || "",
+		apiKey: key?.AIGNE_HUB_API_KEY || ""
+	}).catch(() => null);
+	printHubStatus({
+		hub: getUrlOrigin(url),
+		status: isDefault ? "Connected" : "Not Used",
+		user: {
+			name: userInfo?.user.fullName || "",
+			did: userInfo?.user.did || "",
+			email: userInfo?.user.email || ""
+		},
+		credits: {
+			available: userInfo?.creditBalance?.balance || "0",
+			total: userInfo?.creditBalance?.total || "0",
+			used: String(parseFloat(userInfo?.creditBalance?.total || "0") - parseFloat(userInfo?.creditBalance?.balance || "0"))
+		},
+		links: {
+			payment: userInfo?.paymentLink || "",
+			profile: userInfo?.profileLink || ""
+		},
+		enableCredit: userInfo?.enableCredit || false
+	});
+}
+
+//#endregion
+export { createHubCommand };
+//# sourceMappingURL=hub.mjs.map

package/dist/commands/hub.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"hub.mjs","names":["getSecretStore","yargs"],"sources":["../../src/commands/hub.ts"],"sourcesContent":["import { AIGNE_HUB_URL } from \"@aigne/aigne-hub\";\nimport chalk from \"chalk\";\nimport Table from \"cli-table3\";\nimport inquirer from \"inquirer\";\nimport type { CommandModule } from \"yargs\";\nimport { isTest } from \"../utils/aigne-hub/constants.js\";\nimport { connectToAIGNEHub } from \"../utils/aigne-hub/credential.js\";\nimport getSecretStore from \"../utils/aigne-hub/store/index.js\";\nimport { getUserInfo } from \"../utils/aigne-hub-user.js\";\nimport { getUrlOrigin } from \"../utils/get-url-origin.js\";\n\ninterface StatusInfo {\n  host: string;\n  apiUrl: string;\n  apiKey: string;\n}\n\nexport const formatNumber = (balance: string) => {\n  const cleanNumber = String(balance).replace(/[^\\d.-]/g, \"\");\n  const balanceNum = cleanNumber.split(\".\")[0];\n\n  if (!balanceNum) {\n    return \"0\";\n  }\n\n  return (balanceNum || \"\").trim().replace(/\\B(?=(\\d{3})+(?!\\d))/g, \",\");\n};\n\nconst formatNumberWithColor = (balance: string) => {\n  return chalk.yellow(formatNumber(balance));\n};\n\nfunction formatHubInfoName(name: string) {\n  return chalk.bold(`${name}:`.padEnd(10));\n}\n\nfunction printHubStatus(data: {\n  hub: string;\n  status: string;\n  user: {\n    name: string;\n    did: string;\n    email: string;\n  };\n  credits: {\n    available: string;\n    used: string;\n    total: string;\n  };\n  links: {\n    payment: string;\n    profile: string;\n  };\n  enableCredit: boolean;\n}) {\n  const divider = \"─\".repeat(46);\n\n  console.log(chalk.bold(\"AIGNE Hub Connection\"));\n  console.log(chalk.gray(divider));\n\n  console.log(`${chalk.bold(\"Hub:\".padEnd(10))} ${data.hub}`);\n  console.log(\n    `${chalk.bold(\"Status:\".padEnd(10))} ${\n      data.status === \"Connected\" ? chalk.green(`${data.status} ✅`) : `${data.status}`\n    }`,\n  );\n  console.log(\"\");\n\n  console.log(chalk.bold(\"User:\"));\n  console.log(`  ${formatHubInfoName(\"Name\")} ${data.user.name}`);\n  console.log(`  ${formatHubInfoName(\"DID\")} ${data.user.did}`);\n  console.log(`  ${formatHubInfoName(\"Email\")} ${data.user.email}`);\n  console.log(\"\");\n\n  if (data.enableCredit) {\n    console.log(chalk.bold(\"Credits:\"));\n    console.log(`  ${formatHubInfoName(\"Total\")} ${formatNumberWithColor(data.credits.total)}`);\n    console.log(`  ${formatHubInfoName(\"Used\")} ${formatNumberWithColor(data.credits.used)}`);\n    console.log(\n      `  ${formatHubInfoName(\"Available\")} ${formatNumberWithColor(data.credits.available)}`,\n    );\n    console.log(\"\");\n\n    console.log(chalk.bold(\"Links:\"));\n    if (data.links.payment) {\n      console.log(`  ${formatHubInfoName(\"Payment\")} ${data.links.payment}`);\n    }\n    if (data.links.profile) {\n      console.log(`  ${formatHubInfoName(\"Credits\")} ${data.links.profile}`);\n    }\n  }\n}\n\nasync function getHubs(): Promise<StatusInfo[]> {\n  try {\n    const secretStore = await getSecretStore();\n    const hosts = await secretStore.listHosts();\n\n    const statusList: StatusInfo[] = [];\n    for (const host of hosts) {\n      statusList.push({\n        host: new URL(host.AIGNE_HUB_API_URL).host,\n        apiUrl: host.AIGNE_HUB_API_URL,\n        apiKey: host.AIGNE_HUB_API_KEY || \"\",\n      });\n    }\n\n    return statusList;\n  } catch {\n    return [];\n  }\n}\n\nconst getDefaultHub = async () => {\n  try {\n    const secretStore = await getSecretStore();\n    const defaultHost = await secretStore.getDefault();\n    return defaultHost?.AIGNE_HUB_API_URL || AIGNE_HUB_URL;\n  } catch {\n    return AIGNE_HUB_URL;\n  }\n};\n\nasync function formatHubsList(statusList: StatusInfo[]) {\n  if (statusList?.length === 0) {\n    console.log(chalk.yellow(\"No AIGNE Hub connected.\"));\n    console.log(\"Use 'aigne hub connect' to connect to a hub.\");\n    return;\n  }\n\n  const defaultHub = await getDefaultHub();\n\n  const table = new Table({\n    head: [\"URL\", \"ACTIVE\"],\n    colWidths: [70, 10],\n    style: {\n      head: [\"cyan\"],\n      border: [\"grey\"],\n    },\n  });\n\n  console.log(chalk.blue(\"Connected AIGNE Hubs:\\n\"));\n\n  for (const status of statusList) {\n    const isConnected = getUrlOrigin(status.apiUrl) === getUrlOrigin(defaultHub);\n    table.push([getUrlOrigin(status.apiUrl), isConnected ? \"YES\" : \"NO\"]);\n  }\n\n  console.log(table.toString());\n  console.log(chalk.blue(\"Use 'aigne hub use' to switch to a different hub.\"));\n}\n\nexport function createHubCommand(): CommandModule {\n  return {\n    command: \"hub <command>\",\n    describe: \"Manage AIGNE Hub connections\",\n    builder: (yargs) =>\n      yargs\n        .command([\"list\", \"ls\"], \"List all connected AIGNE Hubs\", listHubs)\n        .command({\n          command: \"connect [url]\",\n          describe: \"Connect to an AIGNE Hub\",\n          builder: (yargs) =>\n            yargs.positional(\"url\", {\n              type: \"string\",\n              describe: \"The URL of the AIGNE Hub to connect to\",\n              default: null,\n            }),\n          handler: (args) => {\n            if (args.url) {\n              saveAndConnect(args.url);\n            } else {\n              connectHub();\n            }\n          },\n        })\n        .command(\"use\", \"Switch to a different AIGNE Hub\", useHub)\n        .command([\"status\", \"st\"], \"Show details of a connected hub\", showInfo)\n        .command([\"remove\", \"rm\"], \"Remove a connected hub\", removeHub)\n        .demandCommand(1, \"Please provide a valid hub command\"),\n    handler: () => {},\n  };\n}\n\nconst listHubs = async () => {\n  const list = await getHubs();\n  await formatHubsList(list);\n};\n\nasync function connectHub() {\n  const defaultUrl = \"https://hub.aigne.io\";\n  const { isOfficial } = await inquirer.prompt({\n    type: \"select\",\n    name: \"isOfficial\",\n    message: `Choose a hub to connect:`,\n    choices: [\n      { name: `Official Hub (${defaultUrl})`, value: true },\n      { name: `Custom Hub URL`, value: false },\n    ],\n    default: true,\n  });\n\n  let currentUrl = defaultUrl;\n  if (!isOfficial) {\n    const { customUrl } = await inquirer.prompt({\n      type: \"input\",\n      name: \"customUrl\",\n      message: \"Enter the URL of your AIGNE Hub:\",\n      validate: validateUrl,\n    });\n    currentUrl = customUrl;\n  }\n\n  await saveAndConnect(currentUrl);\n}\n\nasync function useHub() {\n  const hubs = await getHubs();\n\n  if (!hubs.length) {\n    console.log(chalk.yellow(\"No AIGNE Hub connected.\"));\n    return;\n  }\n\n  const { hubApiKey } = await inquirer.prompt({\n    type: \"select\",\n    name: \"hubApiKey\",\n    message: `Choose a hub to switch to:`,\n    choices: hubs.map((h) => ({\n      name: getUrlOrigin(h.apiUrl),\n      value: h.apiUrl,\n    })),\n  });\n\n  await setDefaultHub(hubApiKey);\n}\n\nasync function removeHub() {\n  const hubs = await getHubs();\n  if (!hubs.length) {\n    console.log(chalk.yellow(\"No AIGNE Hub connected.\"));\n    return;\n  }\n\n  const { hubApiKey } = await inquirer.prompt({\n    type: \"select\",\n    name: \"hubApiKey\",\n    message: `Choose a hub to remove:`,\n    choices: hubs.map((h) => ({\n      name: getUrlOrigin(h.apiUrl),\n      value: h.apiUrl,\n    })),\n  });\n\n  await deleteHub(hubApiKey);\n}\n\nasync function showInfo() {\n  const hubs = await getHubs();\n  if (!hubs.length) {\n    console.log(chalk.yellow(\"No AIGNE Hub connected.\"));\n    return;\n  }\n\n  const defaultHub = await getDefaultHub();\n  const defaultHubIndex = hubs.findIndex(\n    (h) => getUrlOrigin(h.apiUrl) === getUrlOrigin(defaultHub),\n  );\n\n  const { hubApiKey } = await inquirer.prompt({\n    type: \"select\",\n    name: \"hubApiKey\",\n    message: `Choose a hub to view info:`,\n    choices: hubs.map((h, index) => ({\n      name: `${getUrlOrigin(h.apiUrl)} ${defaultHubIndex === index ? \"(connected)\" : \"\"}`,\n      value: h.apiUrl,\n    })),\n  });\n\n  try {\n    await printHubDetails(hubApiKey);\n  } catch (error: any) {\n    console.error(chalk.red(\"✗ Failed to print hub details:\"), error.message);\n  }\n}\n\nfunction validateUrl(input: string) {\n  try {\n    const url = new URL(input);\n    return url.protocol.startsWith(\"http\") ? true : \"Must be http or https\";\n  } catch {\n    return \"Invalid URL\";\n  }\n}\n\nasync function saveAndConnect(url: string) {\n  const secretStore = await getSecretStore();\n  const currentKey = await secretStore.getKey(url);\n\n  if (currentKey?.AIGNE_HUB_API_URL) {\n    await setDefaultHub(currentKey.AIGNE_HUB_API_URL);\n    console.log(\n      chalk.green(`✓ Hub ${getUrlOrigin(currentKey.AIGNE_HUB_API_URL)} connected successfully.`),\n    );\n    return;\n  }\n\n  try {\n    if (isTest) {\n      await secretStore.setKey(\"https://hub.aigne.io/ai-kit\", \"test-key\");\n      await secretStore.setDefault(\"https://hub.aigne.io/ai-kit\");\n      console.log(chalk.green(`✓ Hub https://hub.aigne.io/ai-kit connected successfully.`));\n      return;\n    }\n\n    await connectToAIGNEHub(url);\n    console.log(chalk.green(`✓ Hub ${getUrlOrigin(url)} connected successfully.`));\n  } catch (error: any) {\n    console.error(chalk.red(\"✗ Failed to connect:\"), error.message);\n  }\n}\n\nasync function setDefaultHub(url: string) {\n  try {\n    const secretStore = await getSecretStore();\n    const key = await secretStore.getKey(url);\n\n    if (!key) {\n      console.error(chalk.red(\"✗ Hub not found\"));\n      return;\n    }\n\n    await secretStore.setDefault(key.AIGNE_HUB_API_URL);\n    console.log(chalk.green(`✓ Switched active hub to ${getUrlOrigin(url)}`));\n  } catch {\n    console.error(chalk.red(\"✗ Failed to set default hub\"));\n  }\n}\n\nasync function deleteHub(url: string) {\n  try {\n    const secretStore = await getSecretStore();\n    const key = await secretStore.getKey(url);\n\n    if (!key) {\n      console.error(chalk.red(\"✗ Hub not found\"));\n      return;\n    }\n\n    const defaultHub = await getDefaultHub();\n\n    await secretStore.deleteKey(url);\n\n    if (!defaultHub) {\n      return;\n    }\n\n    const isDefaultHub = getUrlOrigin(url) === getUrlOrigin(defaultHub);\n    if (isDefaultHub) {\n      await secretStore.deleteDefault();\n      const remainingHubs = await getHubs();\n\n      const nextHub = remainingHubs[0];\n      if (nextHub) {\n        await secretStore.setDefault(nextHub?.apiUrl);\n\n        console.log(\n          chalk.green(\n            `✓ Hub ${getUrlOrigin(url)} removed, switched to ${getUrlOrigin(nextHub?.apiUrl)}`,\n          ),\n        );\n        return;\n      }\n    }\n\n    console.log(chalk.green(`✓ Hub ${getUrlOrigin(url)} removed`));\n  } catch {\n    console.error(chalk.red(\"✗ Failed to delete hub\"));\n  }\n}\n\nasync function printHubDetails(url: string) {\n  const secretStore = await getSecretStore();\n  const key = await secretStore.getKey(url);\n  const defaultHub = await getDefaultHub();\n  const isDefault = getUrlOrigin(url) === getUrlOrigin(defaultHub);\n\n  const userInfo = await getUserInfo({\n    baseUrl: key?.AIGNE_HUB_API_URL || \"\",\n    apiKey: key?.AIGNE_HUB_API_KEY || \"\",\n  }).catch(() => null);\n\n  printHubStatus({\n    hub: getUrlOrigin(url),\n    status: isDefault ? \"Connected\" : \"Not Used\",\n    user: {\n      name: userInfo?.user.fullName || \"\",\n      did: userInfo?.user.did || \"\",\n      email: userInfo?.user.email || \"\",\n    },\n    credits: {\n      available: userInfo?.creditBalance?.balance || \"0\",\n      total: userInfo?.creditBalance?.total || \"0\",\n      used: String(\n        parseFloat(userInfo?.creditBalance?.total || \"0\") -\n          parseFloat(userInfo?.creditBalance?.balance || \"0\"),\n      ),\n    },\n    links: {\n      payment: userInfo?.paymentLink || \"\",\n      profile: userInfo?.profileLink || \"\",\n    },\n    enableCredit: userInfo?.enableCredit || false,\n  });\n}\n"],"mappings":";;;;;;;;;;;AAiBA,MAAa,gBAAgB,YAAoB;CAE/C,MAAM,aADc,OAAO,QAAQ,CAAC,QAAQ,YAAY,GAAG,CAC5B,MAAM,IAAI,CAAC;AAE1C,KAAI,CAAC,WACH,QAAO;AAGT,SAAQ,cAAc,IAAI,MAAM,CAAC,QAAQ,yBAAyB,IAAI;;AAGxE,MAAM,yBAAyB,YAAoB;AACjD,QAAO,MAAM,OAAO,aAAa,QAAQ,CAAC;;AAG5C,SAAS,kBAAkB,MAAc;AACvC,QAAO,MAAM,KAAK,GAAG,KAAK,GAAG,OAAO,GAAG,CAAC;;AAG1C,SAAS,eAAe,MAkBrB;CACD,MAAM,UAAU,IAAI,OAAO,GAAG;AAE9B,SAAQ,IAAI,MAAM,KAAK,uBAAuB,CAAC;AAC/C,SAAQ,IAAI,MAAM,KAAK,QAAQ,CAAC;AAEhC,SAAQ,IAAI,GAAG,MAAM,KAAK,OAAO,OAAO,GAAG,CAAC,CAAC,GAAG,KAAK,MAAM;AAC3D,SAAQ,IACN,GAAG,MAAM,KAAK,UAAU,OAAO,GAAG,CAAC,CAAC,GAClC,KAAK,WAAW,cAAc,MAAM,MAAM,GAAG,KAAK,OAAO,IAAI,GAAG,GAAG,KAAK,WAE3E;AACD,SAAQ,IAAI,GAAG;AAEf,SAAQ,IAAI,MAAM,KAAK,QAAQ,CAAC;AAChC,SAAQ,IAAI,KAAK,kBAAkB,OAAO,CAAC,GAAG,KAAK,KAAK,OAAO;AAC/D,SAAQ,IAAI,KAAK,kBAAkB,MAAM,CAAC,GAAG,KAAK,KAAK,MAAM;AAC7D,SAAQ,IAAI,KAAK,kBAAkB,QAAQ,CAAC,GAAG,KAAK,KAAK,QAAQ;AACjE,SAAQ,IAAI,GAAG;AAEf,KAAI,KAAK,cAAc;AACrB,UAAQ,IAAI,MAAM,KAAK,WAAW,CAAC;AACnC,UAAQ,IAAI,KAAK,kBAAkB,QAAQ,CAAC,GAAG,sBAAsB,KAAK,QAAQ,MAAM,GAAG;AAC3F,UAAQ,IAAI,KAAK,kBAAkB,OAAO,CAAC,GAAG,sBAAsB,KAAK,QAAQ,KAAK,GAAG;AACzF,UAAQ,IACN,KAAK,kBAAkB,YAAY,CAAC,GAAG,sBAAsB,KAAK,QAAQ,UAAU,GACrF;AACD,UAAQ,IAAI,GAAG;AAEf,UAAQ,IAAI,MAAM,KAAK,SAAS,CAAC;AACjC,MAAI,KAAK,MAAM,QACb,SAAQ,IAAI,KAAK,kBAAkB,UAAU,CAAC,GAAG,KAAK,MAAM,UAAU;AAExE,MAAI,KAAK,MAAM,QACb,SAAQ,IAAI,KAAK,kBAAkB,UAAU,CAAC,GAAG,KAAK,MAAM,UAAU;;;AAK5E,eAAe,UAAiC;AAC9C,KAAI;EAEF,MAAM,QAAQ,OADM,MAAMA,eAAgB,EACV,WAAW;EAE3C,MAAM,aAA2B,EAAE;AACnC,OAAK,MAAM,QAAQ,MACjB,YAAW,KAAK;GACd,MAAM,IAAI,IAAI,KAAK,kBAAkB,CAAC;GACtC,QAAQ,KAAK;GACb,QAAQ,KAAK,qBAAqB;GACnC,CAAC;AAGJ,SAAO;SACD;AACN,SAAO,EAAE;;;AAIb,MAAM,gBAAgB,YAAY;AAChC,KAAI;AAGF,UADoB,OADA,MAAMA,eAAgB,EACJ,YAAY,GAC9B,qBAAqB;SACnC;AACN,SAAO;;;AAIX,eAAe,eAAe,YAA0B;AACtD,KAAI,YAAY,WAAW,GAAG;AAC5B,UAAQ,IAAI,MAAM,OAAO,0BAA0B,CAAC;AACpD,UAAQ,IAAI,+CAA+C;AAC3D;;CAGF,MAAM,aAAa,MAAM,eAAe;CAExC,MAAM,QAAQ,IAAI,MAAM;EACtB,MAAM,CAAC,OAAO,SAAS;EACvB,WAAW,CAAC,IAAI,GAAG;EACnB,OAAO;GACL,MAAM,CAAC,OAAO;GACd,QAAQ,CAAC,OAAO;GACjB;EACF,CAAC;AAEF,SAAQ,IAAI,MAAM,KAAK,0BAA0B,CAAC;AAElD,MAAK,MAAM,UAAU,YAAY;EAC/B,MAAM,cAAc,aAAa,OAAO,OAAO,KAAK,aAAa,WAAW;AAC5E,QAAM,KAAK,CAAC,aAAa,OAAO,OAAO,EAAE,cAAc,QAAQ,KAAK,CAAC;;AAGvE,SAAQ,IAAI,MAAM,UAAU,CAAC;AAC7B,SAAQ,IAAI,MAAM,KAAK,oDAAoD,CAAC;;AAG9E,SAAgB,mBAAkC;AAChD,QAAO;EACL,SAAS;EACT,UAAU;EACV,UAAU,UACR,MACG,QAAQ,CAAC,QAAQ,KAAK,EAAE,iCAAiC,SAAS,CAClE,QAAQ;GACP,SAAS;GACT,UAAU;GACV,UAAU,YACRC,QAAM,WAAW,OAAO;IACtB,MAAM;IACN,UAAU;IACV,SAAS;IACV,CAAC;GACJ,UAAU,SAAS;AACjB,QAAI,KAAK,IACP,gBAAe,KAAK,IAAI;QAExB,aAAY;;GAGjB,CAAC,CACD,QAAQ,OAAO,mCAAmC,OAAO,CACzD,QAAQ,CAAC,UAAU,KAAK,EAAE,mCAAmC,SAAS,CACtE,QAAQ,CAAC,UAAU,KAAK,EAAE,0BAA0B,UAAU,CAC9D,cAAc,GAAG,qCAAqC;EAC3D,eAAe;EAChB;;AAGH,MAAM,WAAW,YAAY;AAE3B,OAAM,eADO,MAAM,SAAS,CACF;;AAG5B,eAAe,aAAa;CAC1B,MAAM,aAAa;CACnB,MAAM,EAAE,eAAe,MAAM,SAAS,OAAO;EAC3C,MAAM;EACN,MAAM;EACN,SAAS;EACT,SAAS,CACP;GAAE,MAAM,iBAAiB,WAAW;GAAI,OAAO;GAAM,EACrD;GAAE,MAAM;GAAkB,OAAO;GAAO,CACzC;EACD,SAAS;EACV,CAAC;CAEF,IAAI,aAAa;AACjB,KAAI,CAAC,YAAY;EACf,MAAM,EAAE,cAAc,MAAM,SAAS,OAAO;GAC1C,MAAM;GACN,MAAM;GACN,SAAS;GACT,UAAU;GACX,CAAC;AACF,eAAa;;AAGf,OAAM,eAAe,WAAW;;AAGlC,eAAe,SAAS;CACtB,MAAM,OAAO,MAAM,SAAS;AAE5B,KAAI,CAAC,KAAK,QAAQ;AAChB,UAAQ,IAAI,MAAM,OAAO,0BAA0B,CAAC;AACpD;;CAGF,MAAM,EAAE,cAAc,MAAM,SAAS,OAAO;EAC1C,MAAM;EACN,MAAM;EACN,SAAS;EACT,SAAS,KAAK,KAAK,OAAO;GACxB,MAAM,aAAa,EAAE,OAAO;GAC5B,OAAO,EAAE;GACV,EAAE;EACJ,CAAC;AAEF,OAAM,cAAc,UAAU;;AAGhC,eAAe,YAAY;CACzB,MAAM,OAAO,MAAM,SAAS;AAC5B,KAAI,CAAC,KAAK,QAAQ;AAChB,UAAQ,IAAI,MAAM,OAAO,0BAA0B,CAAC;AACpD;;CAGF,MAAM,EAAE,cAAc,MAAM,SAAS,OAAO;EAC1C,MAAM;EACN,MAAM;EACN,SAAS;EACT,SAAS,KAAK,KAAK,OAAO;GACxB,MAAM,aAAa,EAAE,OAAO;GAC5B,OAAO,EAAE;GACV,EAAE;EACJ,CAAC;AAEF,OAAM,UAAU,UAAU;;AAG5B,eAAe,WAAW;CACxB,MAAM,OAAO,MAAM,SAAS;AAC5B,KAAI,CAAC,KAAK,QAAQ;AAChB,UAAQ,IAAI,MAAM,OAAO,0BAA0B,CAAC;AACpD;;CAGF,MAAM,aAAa,MAAM,eAAe;CACxC,MAAM,kBAAkB,KAAK,WAC1B,MAAM,aAAa,EAAE,OAAO,KAAK,aAAa,WAAW,CAC3D;CAED,MAAM,EAAE,cAAc,MAAM,SAAS,OAAO;EAC1C,MAAM;EACN,MAAM;EACN,SAAS;EACT,SAAS,KAAK,KAAK,GAAG,WAAW;GAC/B,MAAM,GAAG,aAAa,EAAE,OAAO,CAAC,GAAG,oBAAoB,QAAQ,gBAAgB;GAC/E,OAAO,EAAE;GACV,EAAE;EACJ,CAAC;AAEF,KAAI;AACF,QAAM,gBAAgB,UAAU;UACzB,OAAY;AACnB,UAAQ,MAAM,MAAM,IAAI,iCAAiC,EAAE,MAAM,QAAQ;;;AAI7E,SAAS,YAAY,OAAe;AAClC,KAAI;AAEF,SADY,IAAI,IAAI,MAAM,CACf,SAAS,WAAW,OAAO,GAAG,OAAO;SAC1C;AACN,SAAO;;;AAIX,eAAe,eAAe,KAAa;CACzC,MAAM,cAAc,MAAMD,eAAgB;CAC1C,MAAM,aAAa,MAAM,YAAY,OAAO,IAAI;AAEhD,KAAI,YAAY,mBAAmB;AACjC,QAAM,cAAc,WAAW,kBAAkB;AACjD,UAAQ,IACN,MAAM,MAAM,SAAS,aAAa,WAAW,kBAAkB,CAAC,0BAA0B,CAC3F;AACD;;AAGF,KAAI;AACF,MAAI,QAAQ;AACV,SAAM,YAAY,OAAO,+BAA+B,WAAW;AACnE,SAAM,YAAY,WAAW,8BAA8B;AAC3D,WAAQ,IAAI,MAAM,MAAM,4DAA4D,CAAC;AACrF;;AAGF,QAAM,kBAAkB,IAAI;AAC5B,UAAQ,IAAI,MAAM,MAAM,SAAS,aAAa,IAAI,CAAC,0BAA0B,CAAC;UACvE,OAAY;AACnB,UAAQ,MAAM,MAAM,IAAI,uBAAuB,EAAE,MAAM,QAAQ;;;AAInE,eAAe,cAAc,KAAa;AACxC,KAAI;EACF,MAAM,cAAc,MAAMA,eAAgB;EAC1C,MAAM,MAAM,MAAM,YAAY,OAAO,IAAI;AAEzC,MAAI,CAAC,KAAK;AACR,WAAQ,MAAM,MAAM,IAAI,kBAAkB,CAAC;AAC3C;;AAGF,QAAM,YAAY,WAAW,IAAI,kBAAkB;AACnD,UAAQ,IAAI,MAAM,MAAM,4BAA4B,aAAa,IAAI,GAAG,CAAC;SACnE;AACN,UAAQ,MAAM,MAAM,IAAI,8BAA8B,CAAC;;;AAI3D,eAAe,UAAU,KAAa;AACpC,KAAI;EACF,MAAM,cAAc,MAAMA,eAAgB;AAG1C,MAAI,CAFQ,MAAM,YAAY,OAAO,IAAI,EAE/B;AACR,WAAQ,MAAM,MAAM,IAAI,kBAAkB,CAAC;AAC3C;;EAGF,MAAM,aAAa,MAAM,eAAe;AAExC,QAAM,YAAY,UAAU,IAAI;AAEhC,MAAI,CAAC,WACH;AAIF,MADqB,aAAa,IAAI,KAAK,aAAa,WAAW,EACjD;AAChB,SAAM,YAAY,eAAe;GAGjC,MAAM,WAFgB,MAAM,SAAS,EAEP;AAC9B,OAAI,SAAS;AACX,UAAM,YAAY,WAAW,SAAS,OAAO;AAE7C,YAAQ,IACN,MAAM,MACJ,SAAS,aAAa,IAAI,CAAC,wBAAwB,aAAa,SAAS,OAAO,GACjF,CACF;AACD;;;AAIJ,UAAQ,IAAI,MAAM,MAAM,SAAS,aAAa,IAAI,CAAC,UAAU,CAAC;SACxD;AACN,UAAQ,MAAM,MAAM,IAAI,yBAAyB,CAAC;;;AAItD,eAAe,gBAAgB,KAAa;CAE1C,MAAM,MAAM,OADQ,MAAMA,eAAgB,EACZ,OAAO,IAAI;CACzC,MAAM,aAAa,MAAM,eAAe;CACxC,MAAM,YAAY,aAAa,IAAI,KAAK,aAAa,WAAW;CAEhE,MAAM,WAAW,MAAM,YAAY;EACjC,SAAS,KAAK,qBAAqB;EACnC,QAAQ,KAAK,qBAAqB;EACnC,CAAC,CAAC,YAAY,KAAK;AAEpB,gBAAe;EACb,KAAK,aAAa,IAAI;EACtB,QAAQ,YAAY,cAAc;EAClC,MAAM;GACJ,MAAM,UAAU,KAAK,YAAY;GACjC,KAAK,UAAU,KAAK,OAAO;GAC3B,OAAO,UAAU,KAAK,SAAS;GAChC;EACD,SAAS;GACP,WAAW,UAAU,eAAe,WAAW;GAC/C,OAAO,UAAU,eAAe,SAAS;GACzC,MAAM,OACJ,WAAW,UAAU,eAAe,SAAS,IAAI,GAC/C,WAAW,UAAU,eAAe,WAAW,IAAI,CACtD;GACF;EACD,OAAO;GACL,SAAS,UAAU,eAAe;GAClC,SAAS,UAAU,eAAe;GACnC;EACD,cAAc,UAAU,gBAAgB;EACzC,CAAC"}

package/dist/commands/observe.cjs

@@ -0,0 +1,49 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+let chalk = require("chalk");
+chalk = require_rolldown_runtime.__toESM(chalk);
+let _aigne_core_utils_type_utils = require("@aigne/core/utils/type-utils");
+let _aigne_observability_api_cli = require("@aigne/observability-api/cli");
+let _aigne_observability_api_db_path = require("@aigne/observability-api/db-path");
+_aigne_observability_api_db_path = require_rolldown_runtime.__toESM(_aigne_observability_api_db_path);
+let detect_port = require("detect-port");
+detect_port = require_rolldown_runtime.__toESM(detect_port);
+
+//#region src/commands/observe.ts
+const DEFAULT_PORT = () => (0, _aigne_core_utils_type_utils.tryOrThrow)(() => {
+	const { PORT } = process.env;
+	if (!PORT) return 7890;
+	const port = Number.parseInt(PORT, 10);
+	if (!port || !Number.isInteger(port)) throw new Error(`Invalid PORT: ${PORT}`);
+	return port;
+}, (error) => /* @__PURE__ */ new Error(`parse PORT error ${error.message}`));
+function createObservabilityCommand() {
+	return {
+		command: "observe",
+		describe: "Start the observability server",
+		builder: (yargs) => {
+			return yargs.option("host", {
+				type: "string",
+				describe: "Host to run the observability server on, use 0.0.0.0 to publicly expose the server",
+				default: "localhost"
+			}).option("port", {
+				type: "number",
+				describe: "Port to run the observability server on"
+			});
+		},
+		handler: async (options) => {
+			const port = await (0, detect_port.default)(options.port || DEFAULT_PORT());
+			const dbUrl = (0, _aigne_observability_api_db_path.default)();
+			process.once("SIGINT", async () => {
+				process.exit(0);
+			});
+			console.log("Observability database path:", chalk.default.greenBright(dbUrl));
+			await (0, _aigne_observability_api_cli.startObservabilityCLIServer)({
+				port,
+				dbUrl
+			});
+		}
+	};
+}
+
+//#endregion
+exports.createObservabilityCommand = createObservabilityCommand;