feat: add context to ssh execution

d-strobel · Oct 15, 2023 · 645caab · 645caab
1 parent c40c785
commit 645caab
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/connection/ssh.go b/connection/ssh.go
@@ -80,8 +80,14 @@ func (c *Connection) runSSH(ctx context.Context, cmd string) (string, string, er
 		return "", "", err
 	}
 
-	// Wait for the command to complete
-	err = s.Wait()
+	// Wait for the command to complete with context support
+	select {
+	case <-ctx.Done():
+		_ = s.Signal(ssh.SIGINT)
+		return "", "", ctx.Err()
+	default:
+		err = s.Wait()
+	}
 
 	// Return the error if stderr has no value
 	if err != nil && stderrBytes == nil {