git-project-manager/cmd/util_completion.go

package cmd

import (
	"strconv"
	"strings"

	"github.com/spf13/cobra"
	"golang.org/x/exp/slices"
)

func validProjectsFunc(cmd *cobra.Command, args []string, toComplete string) (
	[]string, cobra.ShellCompDirective) {
	initProjectCache(cmd, args)
	return projectCache.ProjectStrings(toComplete), cobra.ShellCompDirectiveNoFileComp
}

func validAliasesFunc(cmd *cobra.Command, args []string, toComplete string) (
	[]string, cobra.ShellCompDirective) {
	initProjectCache(cmd, args)
	return projectCache.AliasStrings(toComplete), cobra.ShellCompDirectiveNoFileComp
}

func validProjectsOrAliasesFunc(cmd *cobra.Command, args []string, toComplete string) (
	[]string, cobra.ShellCompDirective) {
	projectStrings, _ := validAliasesFunc(cmd, args, toComplete)
	aliasStrings, _ := validProjectsFunc(cmd, args, toComplete)
	return append(projectStrings, aliasStrings...), cobra.ShellCompDirectiveDefault
}

func validRemotesFunc(cmd *cobra.Command, args []string, toComplete string) (
	[]string, cobra.ShellCompDirective) {
	remotes := make([]string, 0, len(conf.Remotes))
	for _, remote := range conf.Remotes {
		if strings.HasPrefix(remote.Host, toComplete) {
			remotes = append(remotes, remote.Host)
		}
	}
	return remotes, cobra.ShellCompDirectiveNoFileComp
}

func validLogLevelsFunc(cmd *cobra.Command, args []string, toComplete string) (
	[]string, cobra.ShellCompDirective) {
	levels := []string{"info", "warn", "error", "debug"}
	matchingLevels := make([]string, 0, len(levels))
	for _, level := range levels {
		if strings.HasPrefix(level, toComplete) {
			matchingLevels = append(matchingLevels, level)
		}
	}
	return matchingLevels, cobra.ShellCompDirectiveNoFileComp
}

func validProjectIdFunc(cmd *cobra.Command, args []string, toComplete string) (
	[]string, cobra.ShellCompDirective) {
	initProjectCache(cmd, args)
	matchingIds := make([]string, 0, len(projectCache.Projects))
	for _, p := range projectCache.Projects {
		idString := strconv.FormatInt(int64(p.ID), 10)
		if strings.HasPrefix(idString, toComplete) {
			matchingIds = append(matchingIds, idString)
		}
	}
	return slices.Clip(matchingIds), cobra.ShellCompDirectiveNoFileComp
}