mcp/cmd/mcp/adopt.go

package main

import (
	"context"
	"fmt"

	mcpv1 "git.wntrmute.dev/kyle/mcp/gen/mcp/v1"
	"git.wntrmute.dev/kyle/mcp/internal/config"
	"github.com/spf13/cobra"
)

func adoptCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "adopt <service>",
		Short: "Adopt all <service>-* containers into a service",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			cfg, err := config.LoadCLIConfig(cfgPath)
			if err != nil {
				return fmt.Errorf("load config: %w", err)
			}

			nodeName, _ := cmd.Flags().GetString("node")

			var addr string
			if nodeName != "" {
				for _, n := range cfg.Nodes {
					if n.Name == nodeName {
						addr = n.Address
						break
					}
				}
				if addr == "" {
					return fmt.Errorf("node %q not found in config", nodeName)
				}
			} else {
				if len(cfg.Nodes) == 0 {
					return fmt.Errorf("no nodes configured")
				}
				nodeName = cfg.Nodes[0].Name
				addr = cfg.Nodes[0].Address
			}

			client, conn, err := dialAgent(addr, cfg)
			if err != nil {
				return fmt.Errorf("dial %s: %w", nodeName, err)
			}
			defer func() { _ = conn.Close() }()

			resp, err := client.AdoptContainers(context.Background(), &mcpv1.AdoptContainersRequest{
				Service: args[0],
			})
			if err != nil {
				return fmt.Errorf("adopt on %s: %w", nodeName, err)
			}

			results := resp.GetResults()
			if len(results) == 0 {
				fmt.Printf("no containers matching %q found on %s\n", args[0]+"-*", nodeName)
				return nil
			}

			for _, r := range results {
				if r.GetSuccess() {
					fmt.Printf("  adopted %s -> %s\n", r.GetContainer(), r.GetComponent())
				} else {
					fmt.Printf("  failed  %s -> %s: %s\n", r.GetContainer(), r.GetComponent(), r.GetError())
				}
			}

			return nil
		},
	}

	cmd.Flags().String("node", "", "target node (default: first node in config)")

	return cmd
}