github.com/tilt-dev/tilt@v0.33.15-0.20240515162809-0a22ed45d8a0/internal/cli/enable.go

package cli

import (
	"context"
	"fmt"
	"strconv"
	"time"

	"github.com/pkg/errors"
	"github.com/spf13/cobra"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"

	"github.com/tilt-dev/tilt/internal/analytics"
	engineanalytics "github.com/tilt-dev/tilt/internal/engine/analytics"
	"github.com/tilt-dev/tilt/pkg/apis/core/v1alpha1"
	"github.com/tilt-dev/tilt/pkg/model"
)

type enableCmd struct {
	all    bool
	only   bool
	labels []string
}

func newEnableCmd() *enableCmd {
	return &enableCmd{}
}

func (c *enableCmd) name() model.TiltSubcommand { return "enable" }

func (c *enableCmd) register() *cobra.Command {
	cmd := &cobra.Command{
		Use:                   "enable {-all | [--only] <resource>...}",
		DisableFlagsInUseLine: true,
		Short:                 "Enables resources",
		Long: `Enables the specified resources in Tilt.

# enables the resources named 'frontend' and 'backend'
tilt enable frontend backend

# enables frontend and backend and disables all others
tilt enable --only frontend backend

# enables all resources
tilt enable --all
`,
	}

	addConnectServerFlags(cmd)
	cmd.Flags().StringSliceVarP(&c.labels, "labels", "l", c.labels, "Enable all resources with the specified labels")
	cmd.Flags().BoolVar(&c.only, "only", false, "Enable the specified resources, disable all others")
	cmd.Flags().BoolVar(&c.all, "all", false, "Enable all resources")

	return cmd
}

func (c *enableCmd) run(ctx context.Context, args []string) error {
	ctrlclient, err := newClient(ctx)
	if err != nil {
		return err
	}

	if c.all {
		if c.only {
			return errors.New("cannot use --all with --only")
		} else if len(args) > 0 {
			return errors.New("cannot use --all with resource names")
		}
	} else if len(args) == 0 && len(c.labels) == 0 {
		return errors.New("must specify at least one resource")
	}

	a := analytics.Get(ctx)
	cmdTags := engineanalytics.CmdTags(map[string]string{})
	cmdTags["only"] = strconv.FormatBool(c.only)
	cmdTags["all"] = strconv.FormatBool(c.all)
	a.Incr("cmd.enable", cmdTags.AsMap())
	defer a.Flush(time.Second)

	names := make(map[string]bool)
	for _, name := range args {
		names[name] = true
	}

	err = changeEnabledResources(ctx, ctrlclient, args, enableOptions{enable: true, all: c.all, only: c.only, labels: c.labels})
	if err != nil {
		return err
	}

	return nil
}

type enableOptions struct {
	enable bool
	all    bool
	only   bool
	labels []string
}

// Changes which resources are enabled in Tilt.
// For resources in `selectedResources`, enable them if `opts.enable` is true, else disable them.
// If `opts.only` is true, enable/disable `selectedResources` as above, and do the opposite to all other resources.
// If `opts.all` is true, ignore `selectedResources` and act on all resources.
func changeEnabledResources(
	ctx context.Context,
	cli client.Client,
	selectedResources []string,
	opts enableOptions) error {
	var uirs v1alpha1.UIResourceList
	err := cli.List(ctx, &uirs)
	if err != nil {
		return err
	}

	// before making any changes, validate that all selected names actually exist
	uirByName := make(map[string]v1alpha1.UIResource)
	for _, uir := range uirs.Items {
		uirByName[uir.Name] = uir
	}
	selectedResourcesByName := make(map[string]bool)
	for _, name := range selectedResources {
		uir, ok := uirByName[name]
		if !ok {
			return fmt.Errorf("no such resource %q", name)
		}
		if len(uir.Status.DisableStatus.Sources) == 0 {
			return fmt.Errorf("%s cannot be enabled or disabled", name)
		}
		selectedResourcesByName[name] = true
	}

	for _, uir := range uirs.Items {
		// resources w/o disable sources are always enabled (e.g., (Tiltfile))
		if len(uir.Status.DisableStatus.Sources) == 0 {
			continue
		}

		var enable bool
		if selectedResourcesByName[uir.Name] {
			enable = opts.enable
		} else if len(opts.labels) > 0 {
			var hasLabel bool
			for _, label := range opts.labels {
				if _, hasLabel = uir.Labels[label]; hasLabel {
					enable = opts.enable
					break
				}
			}
			if !hasLabel {
				continue
			}
		} else if opts.all {
			enable = opts.enable
		} else if opts.only {
			enable = !opts.enable
		} else {
			continue
		}

		for _, source := range uir.Status.DisableStatus.Sources {
			if source.ConfigMap == nil {
				return fmt.Errorf("internal error: resource %s's DisableSource does not have a ConfigMap'", uir.Name)
			}
			cm := &v1alpha1.ConfigMap{ObjectMeta: metav1.ObjectMeta{Name: source.ConfigMap.Name}}
			_, err := controllerutil.CreateOrUpdate(ctx, cli, cm, func() error {
				if cm.Data == nil {
					cm.Data = make(map[string]string)
				}
				cm.Data[source.ConfigMap.Key] = strconv.FormatBool(!enable)
				return nil
			})
			if err != nil {
				return err
			}
		}
	}

	return nil
}