@treeport/treeport 0.3.0 → 0.5.0

package/dist/node/server/core/launcher.js

@@ -1,7 +1,8 @@
 #!/usr/bin/env node
-import { t as integrateShellLaunch } from "../../../shell-integration-7aBNr-p0.js";
+import { t as integrateShellLaunch } from "../../../shell-integration-Be_c91lw.js";
 import fs from "node:fs/promises";
 import path from "node:path";
+import { z } from "zod";
 import { spawn } from "node:child_process";
 import { fileURLToPath } from "node:url";
 //#region src/server/core/launcher.ts
@@ -10,6 +11,22 @@ const FORWARDED_SIGNALS = [
 	"SIGINT",
 	"SIGHUP"
 ];
+const launchSpecSchema = z.object({
+	argv: z.array(z.string()),
+	fallbackArgv: z.array(z.string()).optional(),
+	cwd: z.string(),
+	env: z.record(z.string(), z.string()),
+	shellIntegrationDir: z.string().optional(),
+	tmuxExecutable: z.string().optional(),
+	setupTasks: z.array(z.object({
+		label: z.string(),
+		argv: z.array(z.string()),
+		cwd: z.string(),
+		env: z.record(z.string(), z.string()),
+		timeoutMs: z.number()
+	}).strict()).optional(),
+	setupError: z.string().optional()
+}).strict();
 function safeDiagnostic(value) {
 	return [...value].map((character) => {
 		const codePoint = character.codePointAt(0) ?? 0;
@@ -95,6 +112,7 @@ async function runLaunchSpec(spec, dependencies = {}) {
 	const stdout = dependencies.stdout ?? process.stdout;
 	const stderr = dependencies.stderr ?? process.stderr;
 	const signalSource = dependencies.signalSource ?? process;
+	const tmuxPane = dependencies.tmuxPane === void 0 ? process.env.TMUX_PANE : dependencies.tmuxPane ?? void 0;
 	if (spec.setupError) {
 		stderr.write(`[Treeport setup] ${safeDiagnostic(spec.setupError) || "setup preparation failed"}\n`);
 		return 1;
@@ -150,6 +168,21 @@ async function runLaunchSpec(spec, dependencies = {}) {
 	if (result.spawnError) stderr.write(`Treeport launcher: ${safeDiagnostic(result.spawnError.message) || "spawn error"}\n`);
 	if (result.forwardedSignal && (!spec.fallbackArgv || result.forwardedSignal !== "SIGINT")) return 1;
 	if (spec.fallbackArgv) {
+		if (spec.tmuxExecutable && tmuxPane && spec.fallbackArgv[0]) await runChild([
+			spec.tmuxExecutable,
+			"set-option",
+			"-p",
+			"-t",
+			tmuxPane,
+			"--",
+			"@treeport-fallback-shell",
+			Buffer.from(JSON.stringify(spec.fallbackArgv[0]), "utf8").toString("base64url")
+		], {
+			cwd: spec.cwd,
+			env: process.env,
+			spawnProcess,
+			signalSource
+		});
 		const fallback = integrateShellLaunch(spec.fallbackArgv, commandEnvironment, spec.shellIntegrationDir, spec.tmuxExecutable);
 		const fallbackResult = await runChild(fallback.argv, {
 			cwd: spec.cwd,
@@ -176,7 +209,7 @@ async function main() {
 	}
 	let spec;
 	try {
-		spec = JSON.parse(await fs.readFile(specPath, "utf8"));
+		spec = launchSpecSchema.parse(JSON.parse(await fs.readFile(specPath, "utf8")));
 	} catch (error) {
 		process.stderr.write(`Treeport launcher: cannot read launch spec: ${error instanceof Error ? error.message : String(error)}\n`);
 		process.exit(127);