vmsan 0.1.0-alpha.13 → 0.1.0-alpha.15

package/dist/_chunks/connect.mjs

@@ -1,13 +1,57 @@
-import { l as agentTimeoutError } from "./errors.mjs";
-async function waitForAgent(guestIp, port, timeoutMs = 6e4) {
-	const start = Date.now();
-	const url = `http://${guestIp}:${port}/health`;
-	while (Date.now() - start < timeoutMs) {
+import { n as vmsanPaths } from "./paths.mjs";
+import { t as handleCommandError } from "./errors.mjs";
+import { t as createCommandLogger } from "./logger.mjs";
+import "./vm-state.mjs";
+import { n as waitForAgent, t as resolveVmState } from "./vm-context.mjs";
+import { t as ShellSession } from "./shell.mjs";
+import { consola } from "consola";
+import { defineCommand } from "citty";
+const connectCommand = defineCommand({
+	meta: {
+		name: "connect",
+		description: "Connect to a running VM"
+	},
+	args: {
+		vmId: {
+			type: "positional",
+			description: "VM ID to connect to",
+			required: true
+		},
+		session: {
+			type: "string",
+			alias: "s",
+			description: "Attach to an existing shell session ID",
+			required: false
+		}
+	},
+	async run({ args }) {
+		const cmdLog = createCommandLogger("connect");
+		const paths = vmsanPaths();
 		try {
-			if ((await fetch(url, { signal: AbortSignal.timeout(2e3) })).ok) return;
-		} catch {}
-		await new Promise((r) => setTimeout(r, 500));
+			const { state, guestIp, port } = resolveVmState(args.vmId, paths);
+			const log = consola.withTag(args.vmId);
+			consola.debug(`Agent endpoint: ${guestIp}:${port}`);
+			log.start("Waiting for agent to become ready...");
+			await waitForAgent(guestIp, port);
+			log.success("Agent is ready. Connecting via PTY shell...");
+			const shell = new ShellSession({
+				host: guestIp,
+				port,
+				token: state.agentToken,
+				sessionId: args.session
+			});
+			const closeInfo = await shell.connect();
+			cmdLog.set({
+				vmId: args.vmId,
+				method: "pty"
+			});
+			cmdLog.emit();
+			if (!closeInfo.sessionDestroyed && shell.sessionId) process.stderr.write(`\nResume this session with:\n  vmsan connect ${args.vmId} --session ${shell.sessionId}\n`);
+			process.exit(0);
+		} catch (error) {
+			handleCommandError(error, cmdLog);
+			process.exit(1);
+		}
 	}
-	throw agentTimeoutError(guestIp, timeoutMs);
-}
-export { waitForAgent as t };
+});
+export { connectCommand as default };