37 lines
717 B
Go
37 lines
717 B
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/cobra"
|
|
"os"
|
|
"path/filepath"
|
|
"playbookctl/internal/types"
|
|
)
|
|
|
|
func ArgRoleCompletion(_ *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)
|
|
}
|
|
|
|
playbook, err := types.ReadPlaybook(workDir)
|
|
if err != nil {
|
|
exitApp(err)
|
|
}
|
|
|
|
for _, role := range playbook.Roles {
|
|
if toComplete == "" || len(role) >= len(toComplete) && role[:len(toComplete)] == toComplete {
|
|
suggestions = append(suggestions, role)
|
|
}
|
|
}
|
|
|
|
return suggestions, cobra.ShellCompDirectiveDefault
|
|
}
|