| 336 | } |
| 337 | } |
| 338 | func (h *Harness) Exec(nodeName string, cmd []string) (string, string, error) { |
| 339 | h.mu.Lock() |
| 340 | container, ok := h.Nodes[nodeName] |
| 341 | h.mu.Unlock() |
| 342 | |
| 343 | if !ok { |
| 344 | return "", "", fmt.Errorf("node %s not found", nodeName) |
| 345 | } |
| 346 | |
| 347 | code, r, err := container.Exec(h.ctx, cmd) |
| 348 | if err != nil { |
| 349 | return "", "", err |
| 350 | } |
| 351 | |
| 352 | stdoutBuf := new(bytes.Buffer) |
| 353 | stderrBuf := new(bytes.Buffer) |
| 354 | |
| 355 | // Demultiplex the stream using stdcopy |
| 356 | _, err = stdcopy.StdCopy(stdoutBuf, stderrBuf, r) |
| 357 | if err != nil { |
| 358 | return "", "", fmt.Errorf("failed to copy output: %w", err) |
| 359 | } |
| 360 | |
| 361 | stdout := StripAnsi(stdoutBuf.String()) |
| 362 | stderr := StripAnsi(stderrBuf.String()) |
| 363 | |
| 364 | if code != 0 { |
| 365 | return stdout, stderr, fmt.Errorf("command exited with code %d: %s\nStderr: %s", code, stdout, stderr) |
| 366 | } |
| 367 | |
| 368 | return stdout, stderr, nil |
| 369 | } |
| 370 | |
| 371 | type BackgroundExec struct { |
| 372 | Stdout string |