From f7bd0a495ac40c5165e699f2eee31a33c3c0850f Mon Sep 17 00:00:00 2001
From: Paul Holzinger <paul.holzinger@web.de>
Date: Sat, 12 Dec 2020 17:42:53 +0100
Subject: shell completion for the network flag

Complete all the options e.g. `container:`,`ns:`,`host`, etc...

Signed-off-by: Paul Holzinger <paul.holzinger@web.de>
---
 cmd/podman/common/completion.go | 58 ++++++++++++++++++++++++++++++++---------
 cmd/podman/common/netflags.go   |  2 +-
 cmd/podman/play/kube.go         |  2 +-
 3 files changed, 47 insertions(+), 15 deletions(-)

(limited to 'cmd/podman')

diff --git a/cmd/podman/common/completion.go b/cmd/podman/common/completion.go
index f792b2713..83fe0723c 100644
--- a/cmd/podman/common/completion.go
+++ b/cmd/podman/common/completion.go
@@ -313,6 +313,10 @@ func completeKeyValues(toComplete string, k keyValueCompletion) ([]string, cobra
 	return suggestions, directive
 }
 
+func getBoolCompletion(_ string) ([]string, cobra.ShellCompDirective) {
+	return []string{"true", "false"}, cobra.ShellCompDirectiveNoFileComp
+}
+
 /* Autocomplete Functions for cobra ValidArgsFunction */
 
 // AutocompleteContainers - Autocomplete all container names.
@@ -797,6 +801,39 @@ func AutocompleteVolumeFlag(cmd *cobra.Command, args []string, toComplete string
 	return volumes, directive
 }
 
+// AutocompleteNetworkFlag - Autocomplete network flag options.
+func AutocompleteNetworkFlag(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+	kv := keyValueCompletion{
+		"container:": func(s string) ([]string, cobra.ShellCompDirective) { return getContainers(cmd, s, completeDefault) },
+		"ns:": func(_ string) ([]string, cobra.ShellCompDirective) {
+			return nil, cobra.ShellCompDirectiveDefault
+		},
+		"bridge":  nil,
+		"none":    nil,
+		"host":    nil,
+		"private": nil,
+		"slirp4netns:": func(s string) ([]string, cobra.ShellCompDirective) {
+			skv := keyValueCompletion{
+				"allow_host_loopback=": getBoolCompletion,
+				"cidr=":                nil,
+				"enable_ipv6=":         getBoolCompletion,
+				"outbound_addr=":       nil,
+				"outbound_addr6=":      nil,
+				"port_handler=": func(_ string) ([]string, cobra.ShellCompDirective) {
+					return []string{"rootlesskit", "slirp4netns"}, cobra.ShellCompDirectiveNoFileComp
+				},
+			}
+			return completeKeyValues(s, skv)
+		},
+	}
+
+	networks, _ := getNetworks(cmd, toComplete)
+	suggestions, dir := completeKeyValues(toComplete, kv)
+	// add slirp4netns here it does not work correct if we add it to the kv map
+	suggestions = append(suggestions, "slirp4netns")
+	return append(networks, suggestions...), dir
+}
+
 // AutocompleteJSONFormat - Autocomplete format flag option.
 // -> "json"
 func AutocompleteJSONFormat(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
@@ -974,17 +1011,14 @@ func AutocompletePodPsFilters(cmd *cobra.Command, args []string, toComplete stri
 
 // AutocompleteImageFilters - Autocomplete image ls --filter options.
 func AutocompleteImageFilters(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
-	getBool := func(_ string) ([]string, cobra.ShellCompDirective) {
-		return []string{"true", "false"}, cobra.ShellCompDirectiveNoFileComp
-	}
 	getImg := func(s string) ([]string, cobra.ShellCompDirective) { return getImages(cmd, s) }
 	kv := keyValueCompletion{
 		"before=":    getImg,
 		"since=":     getImg,
 		"label=":     nil,
 		"reference=": nil,
-		"dangling=":  getBool,
-		"readonly=":  getBool,
+		"dangling=":  getBoolCompletion,
+		"readonly=":  getBoolCompletion,
 	}
 	return completeKeyValues(toComplete, kv)
 }
@@ -1004,14 +1038,12 @@ func AutocompleteVolumeFilters(cmd *cobra.Command, args []string, toComplete str
 		return []string{"local"}, cobra.ShellCompDirectiveNoFileComp
 	}
 	kv := keyValueCompletion{
-		"name=":   func(s string) ([]string, cobra.ShellCompDirective) { return getVolumes(cmd, s) },
-		"driver=": local,
-		"scope=":  local,
-		"label=":  nil,
-		"opt=":    nil,
-		"dangling=": func(_ string) ([]string, cobra.ShellCompDirective) {
-			return []string{"true", "false"}, cobra.ShellCompDirectiveNoFileComp
-		},
+		"name=":     func(s string) ([]string, cobra.ShellCompDirective) { return getVolumes(cmd, s) },
+		"driver=":   local,
+		"scope=":    local,
+		"label=":    nil,
+		"opt=":      nil,
+		"dangling=": getBoolCompletion,
 	}
 	return completeKeyValues(toComplete, kv)
 }
diff --git a/cmd/podman/common/netflags.go b/cmd/podman/common/netflags.go
index cae52ccaa..9cb4ed550 100644
--- a/cmd/podman/common/netflags.go
+++ b/cmd/podman/common/netflags.go
@@ -63,7 +63,7 @@ func DefineNetFlags(cmd *cobra.Command) {
 		networkFlagName, containerConfig.NetNS(),
 		"Connect a container to a network",
 	)
-	_ = cmd.RegisterFlagCompletionFunc(networkFlagName, AutocompleteNetworks)
+	_ = cmd.RegisterFlagCompletionFunc(networkFlagName, AutocompleteNetworkFlag)
 
 	networkAliasFlagName := "network-alias"
 	netFlags.StringSlice(
diff --git a/cmd/podman/play/kube.go b/cmd/podman/play/kube.go
index 5e227d05a..db7280b1d 100644
--- a/cmd/podman/play/kube.go
+++ b/cmd/podman/play/kube.go
@@ -61,7 +61,7 @@ func init() {
 
 	networkFlagName := "network"
 	flags.StringVar(&kubeOptions.Network, networkFlagName, "", "Connect pod to CNI network(s)")
-	_ = kubeCmd.RegisterFlagCompletionFunc(networkFlagName, common.AutocompleteNetworks)
+	_ = kubeCmd.RegisterFlagCompletionFunc(networkFlagName, common.AutocompleteNetworkFlag)
 
 	logDriverFlagName := "log-driver"
 	flags.StringVar(&kubeOptions.LogDriver, logDriverFlagName, "", "Logging driver for the container")
-- 
cgit v1.2.3-54-g00ecf