Improve rollout monitoring by discovering deployments dynamically

Author: AryanSharma9917

pkg/deploy/rollout.go

@@ -3,47 +3,94 @@ package deploy
 import (
 	"fmt"
 	"os/exec"
+	"strings"
 	"time"
 
 	"github.com/aryansharma9917/codewise-cli/pkg/env"
 )
 
-func MonitorRollout(environment *env.Env) error {
+func getDeployments(namespace string, context string) ([]string, error) {
 
-	release := environment.Helm.Release
-	namespace := environment.K8s.Namespace
-	context := environment.K8s.Context
+	args := []string{
+		"get",
+		"deployments",
+		"-n",
+		namespace,
+		"-o",
+		"name",
+	}
 
-	fmt.Println("Waiting for deployment rollout...")
+	if context != "" {
+		args = append(args, "--context", context)
+	}
+
+	cmd := exec.Command("kubectl", args...)
+	out, err := cmd.Output()
+	if err != nil {
+		return nil, fmt.Errorf("failed to list deployments")
+	}
+
+	lines := strings.Split(strings.TrimSpace(string(out)), "\n")
 
-	// We assume Helm creates a Deployment with the release name.
-	// This is standard Helm behavior unless overridden.
+	var deployments []string
+	for _, line := range lines {
+		if strings.TrimSpace(line) != "" {
+			deployments = append(deployments, line)
+		}
+	}
+
+	return deployments, nil
+}
+
+func waitForDeployment(deployment string, namespace string, context string) error {
 
 	args := []string{
 		"rollout",
 		"status",
-		"deployment/" + release,
+		deployment,
 		"-n",
 		namespace,
+		"--timeout=120s",
 	}
 
 	if context != "" {
 		args = append(args, "--context", context)
 	}
 
 	cmd := exec.Command("kubectl", args...)
-
-	// Stream output live
 	cmd.Stdout = nil
 	cmd.Stderr = nil
 
-	err := cmd.Run()
+	return cmd.Run()
+}
+
+func MonitorRollout(environment *env.Env) error {
+
+	ns := environment.K8s.Namespace
+	ctx := environment.K8s.Context
+
+	fmt.Println("Waiting for deployment rollout...")
+
+	deployments, err := getDeployments(ns, ctx)
 	if err != nil {
-		return fmt.Errorf("deployment failed to roll out")
+		return err
+	}
+
+	if len(deployments) == 0 {
+		fmt.Println("No deployments found. Skipping rollout monitoring.")
+		return nil
 	}
 
-	time.Sleep(1 * time.Second)
+	for _, deploy := range deployments {
+		fmt.Printf("Monitoring %s...\n", deploy)
+
+		if err := waitForDeployment(deploy, ns, ctx); err != nil {
+			return fmt.Errorf("deployment %s failed to roll out", deploy)
+		}
+
+		time.Sleep(500 * time.Millisecond)
+	}
 
-	fmt.Println("Deployment successfully rolled out.")
+	fmt.Println("All deployments successfully rolled out.")
 	return nil
 }