Function testWshExec

cmd/test-conn/testutil.go:192–266

(connName string, cmd string, timeout time.Duration)

}

func testWshExec(connName string, cmd string, timeout time.Duration) error {
	opts, err := remote.ParseOpts(connName)
	if err != nil {
		return fmt.Errorf("failed to parse connection string: %w", err)
	}

	log.Printf("Connecting to %s with wsh enabled...", opts.String())

	conn := conncontroller.GetConn(opts)
	ctx, cancel := context.WithTimeout(context.Background(), timeout)
	defer cancel()

	wshEnabled := true
	err = conn.Connect(ctx, &wconfig.ConnKeywords{
		ConnWshEnabled: &wshEnabled,
	})
	if err != nil {
		return fmt.Errorf("connection failed: %w", err)
	}

	status := conn.DeriveConnStatus()
	log.Printf("✓ Connected! (wsh enabled: %v)", status.WshEnabled)
	if status.WshVersion != "" {
		log.Printf("  wsh version: %s", status.WshVersion)
	}
	if !status.WshEnabled {
		log.Printf("  WARNING: wsh not enabled - reason: %s", status.NoWshReason)
	}

	log.Printf("Starting wsh-enabled shell...")

	swapToken := &shellutil.TokenSwapEntry{
		Token: uuid.New().String(),
		Env:   make(map[string]string),
		Exp:   time.Now().Add(5 * time.Minute),
	}
	swapToken.Env["TERM_PROGRAM"] = "waveterm"
	swapToken.Env["WAVETERM"] = "1"
	swapToken.Env["WAVETERM_VERSION"] = wavebase.WaveVersion
	swapToken.Env["WAVETERM_CONN"] = connName

	cmdOpts := shellexec.CommandOptsType{
		SwapToken: swapToken,
	}

	termSize := waveobj.TermSize{Rows: 24, Cols: 80}
	shellProc, err := shellexec.StartRemoteShellProc(ctx, ctx, termSize, "", cmdOpts, conn)
	if err != nil {
		return fmt.Errorf("failed to start shell: %w", err)
	}
	defer shellProc.Close()

	log.Printf("✓ Shell started! Executing: %s", cmd)

	_, err = shellProc.Cmd.Write([]byte(cmd + "\n"))
	if err != nil {
		return fmt.Errorf("failed to write command: %w", err)
	}