WIP

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
+}