feat: implement attach flag for podman too (#72)

* feat: implement attach flag for podman too

* fix: podman tests

Author: dottorblaster

cmd/schedctl/run.go

@@ -89,12 +89,14 @@ func run(cmd *cobra.Command, args []string, attach bool) error {
 	}
 
 	if driver == constants.PODMAN {
-		err := podman.Run(result.ImageURI, schedulerID, containerArgs)
+		err := podman.Run(result.ImageURI, schedulerID, attach, containerArgs)
 		if err != nil {
 			panic(err)
 		}
 
-		_, _ = output.Out("Container %s started successfully\n", result.ImageURI)
+		if !attach {
+			_, _ = output.Out("Container %s started successfully\n", result.ImageURI)
+		}
 	}
 
 	return nil

internal/podman/podman.go

@@ -25,7 +25,7 @@ func generateRandomSuffix() (string, error) {
 	return hex.EncodeToString(bytes), nil
 }
 
-func Run(image, id string, args []string) error {
+func Run(image, id string, attach bool, args []string) error {
 	ctx := context.Background()
 	privileged := true
 
@@ -85,6 +85,35 @@ func Run(image, id string, args []string) error {
 		return fmt.Errorf("failed to start container: %w", err)
 	}
 
+	if attach {
+		errChan := make(chan error, 1)
+		go func() {
+			err := podman_containers.Attach(
+				client,
+				createResponse.ID,
+				os.Stdin,
+				os.Stdout,
+				os.Stderr,
+				nil,
+				new(podman_containers.AttachOptions).WithStream(true),
+			)
+			errChan <- err
+		}()
+
+		exitCode, err := podman_containers.Wait(client, createResponse.ID, nil)
+		if err != nil {
+			return fmt.Errorf("failed to wait for container: %w", err)
+		}
+
+		if attachErr := <-errChan; attachErr != nil {
+			return fmt.Errorf("attach failed: %w", attachErr)
+		}
+
+		if exitCode != 0 {
+			return fmt.Errorf("container exited with status: %d", exitCode)
+		}
+	}
+
 	return nil
 }
 

internal/podman/podman_test.go

@@ -24,7 +24,7 @@ func TestPodmanSpawnStopProcesses(t *testing.T) {
 
 	assert.Equal(t, 0, len(containers))
 
-	err = podman.Run("ghcr.io/schedkit/scx_rusty:latest", "test-scheduler", []string{})
+	err = podman.Run("ghcr.io/schedkit/scx_rusty:latest", "test-scheduler", false, []string{})
 	assert.Nil(t, err)
 
 	containers, err = podman.List()