99 "strings"
1010
1111 "github.com/docker/cli/cli-plugins/plugin"
12+ "github.com/docker/cli/cli/command"
1213 "github.com/spf13/cobra"
1314
1415 "github.com/docker/mcp-gateway/cmd/docker-mcp/backup"
@@ -43,7 +44,7 @@ Examples:
4344`
4445
4546// rootCommand returns the root command for the init plugin
46- func rootCommand (ctx context.Context , cwd string , docker docker. Client ) * cobra.Command {
47+ func rootCommand (ctx context.Context , cwd string , dockerCli command. Cli ) * cobra.Command {
4748 cmd := & cobra.Command {
4849 Use : "mcp [OPTIONS]" ,
4950 TraverseChildren : true ,
@@ -58,7 +59,14 @@ func rootCommand(ctx context.Context, cwd string, docker docker.Client) *cobra.C
5859 }
5960
6061 if os .Getenv ("DOCKER_MCP_IN_CONTAINER" ) != "1" {
61- return desktop .CheckFeatureIsEnabled (ctx , "enableDockerMCPToolkit" , "Docker MCP Toolkit" )
62+ runningInDockerCE , err := docker .RunningInDockerCE (ctx , dockerCli )
63+ if err != nil {
64+ return err
65+ }
66+
67+ if ! runningInDockerCE {
68+ return desktop .CheckFeatureIsEnabled (ctx , "enableDockerMCPToolkit" , "Docker MCP Toolkit" )
69+ }
6270 }
6371
6472 return nil
@@ -73,15 +81,17 @@ func rootCommand(ctx context.Context, cwd string, docker docker.Client) *cobra.C
7381 return []string {"--help" }, cobra .ShellCompDirectiveNoFileComp
7482 })
7583
76- cmd .AddCommand (secret .NewSecretsCmd (docker ))
84+ dockerClient := docker .NewClient (dockerCli )
85+
86+ cmd .AddCommand (secret .NewSecretsCmd (dockerClient ))
7787 cmd .AddCommand (policy .NewPolicyCmd ())
7888 cmd .AddCommand (oauth .NewOAuthCmd ())
7989 cmd .AddCommand (client .NewClientCmd (cwd ))
8090 cmd .AddCommand (catalog .NewCatalogCmd ())
8191 cmd .AddCommand (versionCommand ())
82- cmd .AddCommand (gatewayCommand (docker ))
83- cmd .AddCommand (configCommand (docker ))
84- cmd .AddCommand (serverCommand (docker ))
92+ cmd .AddCommand (gatewayCommand (dockerClient ))
93+ cmd .AddCommand (configCommand (dockerClient ))
94+ cmd .AddCommand (serverCommand (dockerClient ))
8595 cmd .AddCommand (toolsCommand ())
8696
8797 if os .Getenv ("DOCKER_MCP_SHOW_HIDDEN" ) == "1" {
0 commit comments