feat(toolsets): add support for multiple toolsets in configuration (#323)

Users can now enable or disable different toolsets either by providing
a command-line flag or by setting the toolsets array field in the TOML
configuration.

Downstream Kubernetes API developers can declare toolsets for their
APIs by creating a new nested package in pkg/toolsets and registering
it in pkg/mcp/modules.go

Signed-off-by: Marc Nuri <marc@marcnuri.com>

internal/test/test.go

@@ -0,0 +1,8 @@
+package test
+
+func Must[T any](v T, err error) T {
+	if err != nil {
+		panic(err)
+	}
+	return v
+}

internal/tools/update-readme/main.go

@@ -0,0 +1,99 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"maps"
+	"os"
+	"slices"
+	"strings"
+
+	internalk8s "github.com/containers/kubernetes-mcp-server/pkg/kubernetes"
+	"github.com/containers/kubernetes-mcp-server/pkg/toolsets"
+
+	_ "github.com/containers/kubernetes-mcp-server/pkg/toolsets/config"
+	_ "github.com/containers/kubernetes-mcp-server/pkg/toolsets/core"
+	_ "github.com/containers/kubernetes-mcp-server/pkg/toolsets/helm"
+)
+
+type OpenShift struct{}
+
+func (o *OpenShift) IsOpenShift(ctx context.Context) bool {
+	return true
+}
+
+var _ internalk8s.Openshift = (*OpenShift)(nil)
+
+func main() {
+	readme, err := os.ReadFile(os.Args[1])
+	if err != nil {
+		panic(err)
+	}
+	// Available Toolsets
+	toolsetsList := toolsets.Toolsets()
+	maxNameLen, maxDescLen := len("Toolset"), len("Description")
+	for _, toolset := range toolsetsList {
+		nameLen := len(toolset.GetName())
+		descLen := len(toolset.GetDescription())
+		if nameLen > maxNameLen {
+			maxNameLen = nameLen
+		}
+		if descLen > maxDescLen {
+			maxDescLen = descLen
+		}
+	}
+	availableToolsets := strings.Builder{}
+	availableToolsets.WriteString(fmt.Sprintf("| %-*s | %-*s |\n", maxNameLen, "Toolset", maxDescLen, "Description"))
+	availableToolsets.WriteString(fmt.Sprintf("|-%s-|-%s-|\n", strings.Repeat("-", maxNameLen), strings.Repeat("-", maxDescLen)))
+	for _, toolset := range toolsetsList {
+		availableToolsets.WriteString(fmt.Sprintf("| %-*s | %-*s |\n", maxNameLen, toolset.GetName(), maxDescLen, toolset.GetDescription()))
+	}
+	updated := replaceBetweenMarkers(
+		string(readme),
+		"<!-- AVAILABLE-TOOLSETS-START -->",
+		"<!-- AVAILABLE-TOOLSETS-END -->",
+		availableToolsets.String(),
+	)
+
+	// Available Toolset Tools
+	toolsetTools := strings.Builder{}
+	for _, toolset := range toolsetsList {
+		toolsetTools.WriteString("<details>\n\n<summary>" + toolset.GetName() + "</summary>\n\n")
+		tools := toolset.GetTools(&OpenShift{})
+		for _, tool := range tools {
+			toolsetTools.WriteString(fmt.Sprintf("- **%s** - %s\n", tool.Tool.Name, tool.Tool.Description))
+			for _, propName := range slices.Sorted(maps.Keys(tool.Tool.InputSchema.Properties)) {
+				property := tool.Tool.InputSchema.Properties[propName]
+				toolsetTools.WriteString(fmt.Sprintf("  - `%s` (`%s`)", propName, property.Type))
+				if slices.Contains(tool.Tool.InputSchema.Required, propName) {
+					toolsetTools.WriteString(" **(required)**")
+				}
+				toolsetTools.WriteString(fmt.Sprintf(" - %s\n", property.Description))
+			}
+			toolsetTools.WriteString("\n")
+		}
+		toolsetTools.WriteString("</details>\n\n")
+	}
+	updated = replaceBetweenMarkers(
+		updated,
+		"<!-- AVAILABLE-TOOLSETS-TOOLS-START -->",
+		"<!-- AVAILABLE-TOOLSETS-TOOLS-END -->",
+		toolsetTools.String(),
+	)
+
+	if err := os.WriteFile(os.Args[1], []byte(updated), 0o644); err != nil {
+		panic(err)
+	}
+}
+
+func replaceBetweenMarkers(content, startMarker, endMarker, replacement string) string {
+	startIdx := strings.Index(content, startMarker)
+	if startIdx == -1 {
+		return content
+	}
+	endIdx := strings.Index(content, endMarker)
+	if endIdx == -1 || endIdx <= startIdx {
+		return content
+	}
+	return content[:startIdx+len(startMarker)] + "\n\n" + replacement + "\n" + content[endIdx:]
+}