Add all resource templates at once

Signed-off-by: David Gageot <[email protected]>

Author: dgageot

cmd/docker-mcp/internal/gateway/capabilitites.go

@@ -16,12 +16,7 @@ type Capabilities struct {
 	Tools             []server.ServerTool
 	Prompts           []server.ServerPrompt
 	Resources         []server.ServerResource
-	ResourceTemplates []ServerResourceTemplate
-}
-
-type ServerResourceTemplate struct {
-	ResourceTemplate mcp.ResourceTemplate
-	Handler          server.ResourceTemplateHandlerFunc
+	ResourceTemplates []server.ServerResourceTemplate
 }
 
 func (g *Gateway) listCapabilities(ctx context.Context, configuration Configuration, serverNames []string) (*Capabilities, error) {
@@ -89,9 +84,9 @@ func (g *Gateway) listCapabilities(ctx context.Context, configuration Configurat
 				resourceTemplates, err := client.ListResourceTemplates(ctx, mcp.ListResourceTemplatesRequest{})
 				if err == nil {
 					for _, resourceTemplate := range resourceTemplates.ResourceTemplates {
-						capabilities.ResourceTemplates = append(capabilities.ResourceTemplates, ServerResourceTemplate{
-							ResourceTemplate: resourceTemplate,
-							Handler:          g.mcpServerResourceTemplateHandler(*serverConfig),
+						capabilities.ResourceTemplates = append(capabilities.ResourceTemplates, server.ServerResourceTemplate{
+							Template: resourceTemplate,
+							Handler:  g.mcpServerResourceTemplateHandler(*serverConfig),
 						})
 					}
 				}
@@ -161,7 +156,7 @@ func (g *Gateway) listCapabilities(ctx context.Context, configuration Configurat
 	var serverTools []server.ServerTool
 	var serverPrompts []server.ServerPrompt
 	var serverResources []server.ServerResource
-	var serverResourceTemplates []ServerResourceTemplate
+	var serverResourceTemplates []server.ServerResourceTemplate
 	for _, capabilities := range allCapabilities {
 		serverTools = append(serverTools, capabilities.Tools...)
 		serverPrompts = append(serverPrompts, capabilities.Prompts...)

cmd/docker-mcp/internal/gateway/run.go

@@ -201,9 +201,7 @@ func (g *Gateway) reloadConfiguration(ctx context.Context, mcpServer *server.MCP
 	mcpServer.SetPrompts(capabilities.Prompts...)
 	mcpServer.SetResources(capabilities.Resources...)
 	mcpServer.RemoveAllResourceTemplates()
-	for _, v := range capabilities.ResourceTemplates {
-		mcpServer.AddResourceTemplate(v.ResourceTemplate, v.Handler)
-	}
+	mcpServer.AddResourceTemplates(capabilities.ResourceTemplates...)
 	g.health.SetHealthy()
 
 	return nil