WIP

cmd/hosts.go

@@ -163,6 +163,7 @@ func newCommandHostsDefault() *cobra.Command {
 		Use:               "set-default",
 		Short:             "установка хоста по-умолчанию",
 		Args:              cobra.ExactArgs(1),
+		ValidArgsFunction: ArgHostCompletion,
 		RunE:              hostsDefaultRunE,
 	}
 

cmd/utils.go

@@ -34,3 +34,30 @@ func ArgRoleCompletion(_ *cobra.Command, _ []string, toComplete string) ([]strin
 
 	return suggestions, cobra.ShellCompDirectiveDefault
 }
+
+func ArgHostCompletion(_ *cobra.Command, _ []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+	exitApp := func(err error) {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+
+	var suggestions []string
+
+	workDir, err := filepath.Abs(flagWorkdir)
+	if err != nil {
+		exitApp(err)
+	}
+
+	hosts, err := types.ReadHosts(workDir)
+	if err != nil {
+		exitApp(err)
+	}
+
+	for host := range *hosts {
+		if toComplete == "" || len(host) >= len(toComplete) && host[:len(toComplete)] == toComplete {
+			suggestions = append(suggestions, host)
+		}
+	}
+
+	return suggestions, cobra.ShellCompDirectiveDefault
+}