init

internal/commands/status.go

@@ -0,0 +1,108 @@
+package commands
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/rathi/agentikube/internal/kube"
+	"github.com/spf13/cobra"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+)
+
+func NewStatusCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "status",
+		Short: "Show cluster and sandbox status",
+		Long:  "Displays warm pool status, sandbox counts, and compute node information.",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := context.Background()
+
+			cfg, err := loadConfig(cmd)
+			if err != nil {
+				return err
+			}
+
+			client, err := kube.NewClient()
+			if err != nil {
+				return fmt.Errorf("connecting to cluster: %w", err)
+			}
+
+			ns := cfg.Namespace
+
+			// Warm pool status
+			warmPoolGVR := schema.GroupVersionResource{
+				Group:    "agentsandbox.dev",
+				Version:  "v1",
+				Resource: "sandboxwarmpools",
+			}
+
+			wp, err := client.Dynamic().Resource(warmPoolGVR).Namespace(ns).Get(ctx, "sandbox-warm-pool", metav1.GetOptions{})
+			if err != nil {
+				fmt.Printf("warm pool: not found (%v)\n", err)
+			} else {
+				spec, _ := wp.Object["spec"].(map[string]interface{})
+				status, _ := wp.Object["status"].(map[string]interface{})
+
+				replicas := getInt64(spec, "replicas")
+				readyReplicas := getInt64(status, "readyReplicas")
+				pendingReplicas := getInt64(status, "pendingReplicas")
+
+				fmt.Println("warm pool:")
+				fmt.Printf("  desired:  %d\n", replicas)
+				fmt.Printf("  ready:    %d\n", readyReplicas)
+				fmt.Printf("  pending:  %d\n", pendingReplicas)
+			}
+
+			// Sandbox count
+			claimGVR := schema.GroupVersionResource{
+				Group:    "agentsandbox.dev",
+				Version:  "v1",
+				Resource: "sandboxclaims",
+			}
+
+			claims, err := client.Dynamic().Resource(claimGVR).Namespace(ns).List(ctx, metav1.ListOptions{})
+			if err != nil {
+				fmt.Printf("\nsandboxes: error listing (%v)\n", err)
+			} else {
+				fmt.Printf("\nsandboxes: %d\n", len(claims.Items))
+			}
+
+			// Karpenter nodes (if applicable)
+			if cfg.Compute.Type == "karpenter" {
+				nodes, err := client.Clientset().CoreV1().Nodes().List(ctx, metav1.ListOptions{
+					LabelSelector: "karpenter.sh/nodepool",
+				})
+				if err != nil {
+					fmt.Printf("\nkarpenter nodes: error listing (%v)\n", err)
+				} else {
+					fmt.Printf("\nkarpenter nodes: %d\n", len(nodes.Items))
+				}
+			}
+
+			return nil
+		},
+	}
+
+	return cmd
+}
+
+func getInt64(m map[string]interface{}, key string) int64 {
+	if m == nil {
+		return 0
+	}
+	v, ok := m[key]
+	if !ok {
+		return 0
+	}
+	switch n := v.(type) {
+	case int64:
+		return n
+	case float64:
+		return int64(n)
+	case int:
+		return int64(n)
+	default:
+		return 0
+	}
+}