features.go

package cmd

import (
	"fmt"
	"strings"

	"github.com/cego/gitte/config"
	"github.com/cego/gitte/features"
	"github.com/cego/gitte/output"
	"github.com/cego/gitte/state"

	"github.com/spf13/cobra"
)

func newFeaturesCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "features",
		Short: "Manage feature gates",
		RunE: func(cmd *cobra.Command, args []string) error {
			if outputMode() == output.ModePlain {
				return newFeaturesListCmd().RunE(cmd, args)
			}
			err := features.Run(globalCfg, globalCwd, globalSt)
			if err == features.ErrNoFeatureGates {
				fmt.Println(err)
				return nil
			}
			return err
		},
	}

	cmd.AddCommand(
		newFeaturesListCmd(),
		newFeaturesEnableCmd(),
		newFeaturesDisableCmd(),
	)

	return cmd
}

func newFeaturesListCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "list",
		Short: "List all feature gates and their state",
		RunE: func(cmd *cobra.Command, args []string) error {
			fmt.Printf("%-30s %-8s SCOPE\n", "GATE", "ENABLED")
			fmt.Printf("%-30s %-8s -----\n", "----", "-------")

			for gateName, gate := range globalCfg.FeatureGates {
				enabled := "no"
				scopeDesc := buildScopeDescription(gate.Scope.Projects, gate.Scope.GitlabGroups, gate.Scope.GithubOrgs)
				if fs, ok := globalSt.Features[gateName]; ok && fs.Enabled {
					enabled = "yes"
					if fs.OverrideScope != nil {
						scopeDesc = buildOverrideScopeDescription(fs.OverrideScope)
					}
				}

				fmt.Printf("%-30s %-8s %s\n", gateName, enabled, scopeDesc)
			}
			return nil
		},
	}
}

func newFeaturesEnableCmd() *cobra.Command {
	var (
		projects     []string
		gitlabGroups []string
		githubOrgs   []string
		excludes     []string
	)

	cmd := &cobra.Command{
		Use:   "enable <gate>",
		Short: "Enable a feature gate",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			gateName := args[0]
			if _, ok := globalCfg.FeatureGates[gateName]; !ok {
				return fmt.Errorf("unknown feature gate: %q", gateName)
			}

			if len(excludes) > 0 && len(gitlabGroups) == 0 && len(githubOrgs) == 0 {
				return fmt.Errorf("--exclude requires --gitlab-group or --github-org")
			}

			fs := state.FeatureState{Enabled: true}

			if len(projects) > 0 || len(gitlabGroups) > 0 || len(githubOrgs) > 0 {
				override, err := buildOverrideFromFlags(projects, gitlabGroups, githubOrgs, excludes)
				if err != nil {
					return err
				}
				fs.OverrideScope = override
			}

			globalSt.Features[gateName] = fs
			if err := state.Save(globalCwd, globalSt); err != nil {
				return fmt.Errorf("failed to save state: %w", err)
			}

			fmt.Printf("Feature gate %q enabled\n", gateName)
			return nil
		},
	}

	cmd.Flags().StringArrayVar(&projects, "project", nil, "limit to specific project(s)")
	cmd.Flags().StringArrayVar(&gitlabGroups, "gitlab-group", nil, "limit to gitlab group (host/group)")
	cmd.Flags().StringArrayVar(&githubOrgs, "github-org", nil, "limit to github org (host/org)")
	cmd.Flags().StringArrayVar(&excludes, "exclude", nil, "exclude project from all groups/orgs")
	return cmd
}

func newFeaturesDisableCmd() *cobra.Command {
	var (
		projects     []string
		gitlabGroups []string
		githubOrgs   []string
		excludes     []string
	)

	cmd := &cobra.Command{
		Use:   "disable <gate>",
		Short: "Disable a feature gate, or (with scope flags) disable it only for specific projects",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			gateName := args[0]

			if len(excludes) > 0 && len(gitlabGroups) == 0 && len(githubOrgs) == 0 {
				return fmt.Errorf("--exclude requires --gitlab-group or --github-org")
			}

			fs, ok := globalSt.Features[gateName]
			scoped := len(projects) > 0 || len(gitlabGroups) > 0 || len(githubOrgs) > 0

			// Unscoped disable only consults state, so it can also clean up a stale
			// entry for a gate that has since been removed from the config.
			if !scoped {
				if !ok {
					fmt.Printf("Feature gate %q was not enabled\n", gateName)
					return nil
				}
				delete(globalSt.Features, gateName)
				if err := state.Save(globalCwd, globalSt); err != nil {
					return fmt.Errorf("failed to save state: %w", err)
				}
				fmt.Printf("Feature gate %q disabled\n", gateName)
				return nil
			}

			// Scoped disable needs the gate's configured scope to reconstruct the override.
			gate, cfgOK := globalCfg.FeatureGates[gateName]
			if !cfgOK {
				return fmt.Errorf("unknown feature gate: %q", gateName)
			}
			if !ok || !fs.Enabled {
				fmt.Printf("Feature gate %q was not enabled\n", gateName)
				return nil
			}

			// Remove the matching projects from the gate's current scope, leaving it
			// enabled for the rest.
			removal, err := buildOverrideFromFlags(projects, gitlabGroups, githubOrgs, excludes)
			if err != nil {
				return err
			}

			scopeProjects := features.ProjectsInGateScope(globalCfg, gate)
			checked := features.OverrideToCheckedState(fs.OverrideScope, scopeProjects)

			removed := 0
			for name, sp := range scopeProjects {
				if checked[name] && features.ProjectMatchesOverrideScope(name, sp.Host, sp.Path, removal) {
					checked[name] = false
					removed++
				}
			}

			if removed == 0 {
				fmt.Printf("Feature gate %q was not enabled for the given project(s)\n", gateName)
				return nil
			}

			anyLeft := false
			for _, v := range checked {
				if v {
					anyLeft = true
					break
				}
			}

			if anyLeft {
				fs.OverrideScope = features.CheckedStateToOverride(checked, scopeProjects)
				globalSt.Features[gateName] = fs
			} else {
				delete(globalSt.Features, gateName)
			}

			if err := state.Save(globalCwd, globalSt); err != nil {
				return fmt.Errorf("failed to save state: %w", err)
			}

			if anyLeft {
				fmt.Printf("Feature gate %q disabled for the given project(s)\n", gateName)
			} else {
				fmt.Printf("Feature gate %q disabled\n", gateName)
			}
			return nil
		},
	}

	cmd.Flags().StringArrayVar(&projects, "project", nil, "disable only for specific project(s)")
	cmd.Flags().StringArrayVar(&gitlabGroups, "gitlab-group", nil, "disable only for gitlab group (host/group)")
	cmd.Flags().StringArrayVar(&githubOrgs, "github-org", nil, "disable only for github org (host/org)")
	cmd.Flags().StringArrayVar(&excludes, "exclude", nil, "exclude project from all groups/orgs")
	return cmd
}

// buildOverrideFromFlags turns the CLI scope flags into a ScopeOverride.
func buildOverrideFromFlags(projects, gitlabGroups, githubOrgs, excludes []string) (*state.ScopeOverride, error) {
	override := &state.ScopeOverride{Projects: projects}
	for _, g := range gitlabGroups {
		host, group, ok := strings.Cut(g, "/")
		if !ok {
			return nil, fmt.Errorf("invalid --gitlab-group format %q, expected host/group", g)
		}
		override.GitlabGroups = append(override.GitlabGroups, state.ScopeOverrideGroup{Host: host, Group: group, ExcludeProjects: excludes})
	}
	for _, o := range githubOrgs {
		host, org, ok := strings.Cut(o, "/")
		if !ok {
			return nil, fmt.Errorf("invalid --github-org format %q, expected host/org", o)
		}
		override.GithubOrgs = append(override.GithubOrgs, state.ScopeOverrideOrg{Host: host, Org: org, ExcludeProjects: excludes})
	}
	return override, nil
}

func buildOverrideScopeDescription(o *state.ScopeOverride) string {
	var parts []string
	if len(o.Projects) > 0 {
		parts = append(parts, "projects: "+strings.Join(o.Projects, ", "))
	}
	for _, g := range o.GitlabGroups {
		s := fmt.Sprintf("gitlab:%s/%s", g.Host, g.Group)
		if len(g.ExcludeProjects) > 0 {
			s += " (excl: " + strings.Join(g.ExcludeProjects, ", ") + ")"
		}
		parts = append(parts, s)
	}
	for _, g := range o.GithubOrgs {
		s := fmt.Sprintf("github:%s/%s", g.Host, g.Org)
		if len(g.ExcludeProjects) > 0 {
			s += " (excl: " + strings.Join(g.ExcludeProjects, ", ") + ")"
		}
		parts = append(parts, s)
	}
	if len(parts) == 0 {
		return "none (disabled)"
	}
	return strings.Join(parts, "; ")
}

func buildScopeDescription(projects []string, gitlabGroups []config.GitlabScope, githubOrgs []config.GithubScope) string {
	var parts []string
	if len(projects) > 0 {
		parts = append(parts, "projects: "+strings.Join(projects, ", "))
	}
	for _, g := range gitlabGroups {
		parts = append(parts, fmt.Sprintf("gitlab:%s/%s", g.Host, g.Group))
	}
	for _, g := range githubOrgs {
		parts = append(parts, fmt.Sprintf("github:%s/%s", g.Host, g.Org))
	}
	if len(parts) == 0 {
		return "all projects"
	}
	return strings.Join(parts, "; ")
}