Function completeTemplateNames

internal/cli/storage.go:849–888

completeTemplateNames is a ValidArgsFunction for in storage download template. It requires node (args[0]) and storage (args[1]) to already be present.

(cmd *cobra.Command, args []string, toComplete string)

// completeTemplateNames is a ValidArgsFunction for <template> in storage download template.
// It requires node (args[0]) and storage (args[1]) to already be present.
func completeTemplateNames(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
	if len(args) < 1 {
		return completeNodeNames(cmd, args, toComplete)
	}

	if len(args) < 2 {
		return completeStorageNames(cmd, args, toComplete)
	}

	nodeName := args[0]

	session, err := initCLISession(cmd)
	if err != nil || session == nil {
		return nil, cobra.ShellCompDirectiveError
	}

	client, err := session.clientForNode(context.Background(), nodeName)
	if err != nil {
		return nil, cobra.ShellCompDirectiveError
	}

	templates, err := client.GetAvailableTemplates(nodeName)
	if err != nil {
		return nil, cobra.ShellCompDirectiveError
	}

	seen := make(map[string]bool)
	var names []string

	for _, t := range templates {
		if strings.HasPrefix(t.Filename, toComplete) && !seen[t.Filename] {
			seen[t.Filename] = true
			names = append(names, t.Filename)
		}
	}

	sort.Strings(names)

	return names, cobra.ShellCompDirectiveNoFileComp
}