@@ -0,0 +1,45 @@
+package plugin
+
+import (
+	"github.com/docker/cli/cli/command/completion"
+	"github.com/moby/moby/api/types/filters"
+	"github.com/spf13/cobra"
+)
+
+type pluginState string
+
+const (
+	stateAny      pluginState = ""
+	stateEnabled  pluginState = "enabled"
+	stateDisabled pluginState = "disabled"
+)
+
+// completeNames offers completion for plugin names in the given state.
+// The state argument can be one of:
+//
+// - "all": all plugins
+// - "enabled": all enabled plugins
+// - "disabled": all disabled plugins
+func completeNames(dockerCLI completion.APIClientProvider, state pluginState) cobra.CompletionFunc {
+	return func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		f := filters.NewArgs()
+		switch state {
+		case stateEnabled:
+			f.Add("enabled", "true")
+		case stateDisabled:
+			f.Add("enabled", "false")
+		case stateAny:
+			// no filter
+		}
+
+		list, err := dockerCLI.Client().PluginList(cmd.Context(), f)
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+		var names []string
+		for _, v := range list {
+			names = append(names, v.Name)
+		}
+		return names, cobra.ShellCompDirectiveNoFileComp
+	}
+}
@@ -76,7 +76,7 @@ func newCreateCommand(dockerCLI command.Cli) *cobra.Command {
 			options.context = args[1]
 			return runCreate(cmd.Context(), dockerCLI, options)
 		},
-		ValidArgsFunction:     cobra.NoFileCompletions,
+		ValidArgsFunction:     cobra.NoFileCompletions, // TODO(thaJeztah): should provide "directory" completion for the second arg
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -24,6 +24,7 @@ func newDisableCommand(dockerCLI command.Cli) *cobra.Command {
 			_, _ = fmt.Fprintln(dockerCLI.Out(), name)
 			return nil
 		},
+		ValidArgsFunction:     completeNames(dockerCLI, stateEnabled),
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -25,6 +25,7 @@ func newEnableCommand(dockerCLI command.Cli) *cobra.Command {
 			_, _ = fmt.Fprintln(dockerCLI.Out(), name)
 			return nil
 		},
+		ValidArgsFunction:     completeNames(dockerCLI, stateDisabled),
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -29,6 +29,7 @@ func newInspectCommand(dockerCLI command.Cli) *cobra.Command {
 			opts.pluginNames = args
 			return runInspect(cmd.Context(), dockerCLI, opts)
 		},
+		ValidArgsFunction:     completeNames(dockerCLI, stateAny),
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -36,6 +36,7 @@ func newInstallCommand(dockerCLI command.Cli) *cobra.Command {
 			}
 			return runInstall(cmd.Context(), dockerCLI, options)
 		},
+		ValidArgsFunction:     cobra.NoFileCompletions,
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -20,6 +20,7 @@ func newPushCommand(dockerCLI command.Cli) *cobra.Command {
 			name := args[0]
 			return runPush(cmd.Context(), dockerCLI, name)
 		},
+		ValidArgsFunction:     completeNames(dockerCLI, stateAny),
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -29,6 +29,7 @@ func newRemoveCommand(dockerCLI command.Cli) *cobra.Command {
 			opts.plugins = args
 			return runRemove(cmd.Context(), dockerCLI, &opts)
 		},
+		ValidArgsFunction:     completeNames(dockerCLI, stateAny),
 		DisableFlagsInUseLine: true,
 	}
 
 
@@ -14,6 +14,7 @@ func newSetCommand(dockerCLI command.Cli) *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			return dockerCLI.Client().PluginSet(cmd.Context(), args[0], args[1:])
 		},
+		ValidArgsFunction:     completeNames(dockerCLI, stateAny), // TODO(thaJeztah): should only complete for the first arg
 		DisableFlagsInUseLine: true,
 	}
 }
@@ -27,6 +27,7 @@ func newUpgradeCommand(dockerCLI command.Cli) *cobra.Command {
 			return runUpgrade(cmd.Context(), dockerCLI, options)
 		},
 		Annotations:           map[string]string{"version": "1.26"},
+		ValidArgsFunction:     completeNames(dockerCLI, stateAny), // TODO(thaJeztah): should only complete for the first arg
 		DisableFlagsInUseLine: true,
 	}
Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,7 @@ func newCreateCommand(dockerCLI command.Cli) *cobra.Command {`
`76`	`76`	`options.context = args[1]`
`77`	`77`	`return runCreate(cmd.Context(), dockerCLI, options)`
`78`	`78`	`},`
`79`		`- ValidArgsFunction: cobra.NoFileCompletions,`
	`79`	`+ ValidArgsFunction: cobra.NoFileCompletions, // TODO(thaJeztah): should provide "directory" completion for the second arg`
`80`	`80`	`DisableFlagsInUseLine: true,`
`81`	`81`	`}`
`82`	`82`
Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,7 @@ func newDisableCommand(dockerCLI command.Cli) *cobra.Command {`
`24`	`24`	`_, _ = fmt.Fprintln(dockerCLI.Out(), name)`
`25`	`25`	`return nil`
`26`	`26`	`},`
	`27`	`+ ValidArgsFunction: completeNames(dockerCLI, stateEnabled),`
`27`	`28`	`DisableFlagsInUseLine: true,`
`28`	`29`	`}`
`29`	`30`
Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,7 @@ func newEnableCommand(dockerCLI command.Cli) *cobra.Command {`
`25`	`25`	`_, _ = fmt.Fprintln(dockerCLI.Out(), name)`
`26`	`26`	`return nil`
`27`	`27`	`},`
	`28`	`+ ValidArgsFunction: completeNames(dockerCLI, stateDisabled),`
`28`	`29`	`DisableFlagsInUseLine: true,`
`29`	`30`	`}`
`30`	`31`
Original file line number	Diff line number	Diff line change
`@@ -29,6 +29,7 @@ func newInspectCommand(dockerCLI command.Cli) *cobra.Command {`
`29`	`29`	`opts.pluginNames = args`
`30`	`30`	`return runInspect(cmd.Context(), dockerCLI, opts)`
`31`	`31`	`},`
	`32`	`+ ValidArgsFunction: completeNames(dockerCLI, stateAny),`
`32`	`33`	`DisableFlagsInUseLine: true,`
`33`	`34`	`}`
`34`	`35`
Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@ func newInstallCommand(dockerCLI command.Cli) *cobra.Command {`
`36`	`36`	`}`
`37`	`37`	`return runInstall(cmd.Context(), dockerCLI, options)`
`38`	`38`	`},`
	`39`	`+ ValidArgsFunction: cobra.NoFileCompletions,`
`39`	`40`	`DisableFlagsInUseLine: true,`
`40`	`41`	`}`
`41`	`42`
Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,7 @@ func newPushCommand(dockerCLI command.Cli) *cobra.Command {`
`20`	`20`	`name := args[0]`
`21`	`21`	`return runPush(cmd.Context(), dockerCLI, name)`
`22`	`22`	`},`
	`23`	`+ ValidArgsFunction: completeNames(dockerCLI, stateAny),`
`23`	`24`	`DisableFlagsInUseLine: true,`
`24`	`25`	`}`
`25`	`26`
Original file line number	Diff line number	Diff line change
`@@ -29,6 +29,7 @@ func newRemoveCommand(dockerCLI command.Cli) *cobra.Command {`
`29`	`29`	`opts.plugins = args`
`30`	`30`	`return runRemove(cmd.Context(), dockerCLI, &opts)`
`31`	`31`	`},`
	`32`	`+ ValidArgsFunction: completeNames(dockerCLI, stateAny),`
`32`	`33`	`DisableFlagsInUseLine: true,`
`33`	`34`	`}`
`34`	`35`
Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,7 @@ func newSetCommand(dockerCLI command.Cli) *cobra.Command {`
`14`	`14`	`RunE: func(cmd *cobra.Command, args []string) error {`
`15`	`15`	`return dockerCLI.Client().PluginSet(cmd.Context(), args[0], args[1:])`
`16`	`16`	`},`
	`17`	`+ ValidArgsFunction: completeNames(dockerCLI, stateAny), // TODO(thaJeztah): should only complete for the first arg`
`17`	`18`	`DisableFlagsInUseLine: true,`
`18`	`19`	`}`
`19`	`20`	`}`
Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,7 @@ func newUpgradeCommand(dockerCLI command.Cli) *cobra.Command {`
`27`	`27`	`return runUpgrade(cmd.Context(), dockerCLI, options)`
`28`	`28`	`},`
`29`	`29`	`Annotations: map[string]string{"version": "1.26"},`
	`30`	`+ ValidArgsFunction: completeNames(dockerCLI, stateAny), // TODO(thaJeztah): should only complete for the first arg`
`30`	`31`	`DisableFlagsInUseLine: true,`
`31`	`32`	`}`
`32`	`33`