gograpple/cmd/interactive.go

package cmd

import (
	"path"

	"github.com/foomo/gograpple/internal/config"
	"github.com/foomo/gograpple/internal/grapple"
	"github.com/foomo/gograpple/internal/kubectl"
	"github.com/spf13/cobra"
)

func init() {
	interactiveCmd.Flags().BoolVar(&flagAttach, "attach", false, "debug with attach (default will patch)")
	interactiveCmd.Flags().StringVar(&flagSaveDir, "save", ".", "directory to save interactive configuration")
	rootCmd.AddCommand(interactiveCmd)
}

var (
	flagAttach     bool
	flagSaveDir    string
	interactiveCmd = &cobra.Command{
		Use:   "interactive",
		Short: "setup and run patch or attach interactively",
		Args:  cobra.MinimumNArgs(0),
		RunE: func(cmd *cobra.Command, args []string) error {
			if flagAttach {
				return attachDebug(flagSaveDir)
			}
			return patchDebug(flagSaveDir)
		},
	}
)

func attachDebug(baseDir string) error {
	fp := ""
	if baseDir != "" {
		fp = path.Join(baseDir, "gograpple-attach.yaml")
	}
	var c config.AttachConfig
	err := config.Interact(fp, &c)
	if err != nil {
		return err
	}
	g, err := grapple.NewGrapple(newLogEntry(flagDebug), c.Namespace, c.Deployment)
	if err != nil {
		return err
	}
	host, port, err := c.Addr()
	if err != nil {
		return err
	}
	if err := kubectl.SetContext(c.Cluster); err != nil {
		return err
	}
	return g.Attach(c.Namespace, c.Deployment, c.Container, c.AttachTo, c.Arch, host, port, flagDebug)
}

func patchDebug(baseDir string) error {
	fp := ""
	if baseDir != "" {
		fp = path.Join(baseDir, "gograpple-patch.yaml")
	}
	var c config.PatchConfig
	err := config.Interact(fp, &c)
	if err != nil {
		return err
	}
	if &c == nil {
		return nil
	}
	g, err := grapple.NewGrapple(newLogEntry(flagDebug), c.Namespace, c.Deployment)
	if err != nil {
		return err
	}
	host, port, err := c.Addr()
	if err != nil {
		return err
	}
	if err := kubectl.SetContext(c.Cluster); err != nil {
		return err
	}
	if err := g.Patch(c.Image, c.Container, nil); err != nil {
		return err
	}
	defer g.Rollback()
	return g.Delve("", c.Container, c.SourcePath, nil, host, port, c.LaunchVscode, c.DelveContinue)
}