Allow passing through server options

omgitsads · omgitsads · commit 06a028e32d8b · 2025-04-11T10:43:24.000+02:00
diff --git a/pkg/github/server.go b/pkg/github/server.go
@@ -17,13 +17,17 @@ import (
 type GetClientFn func(context.Context) (*github.Client, error)
 
 // NewServer creates a new GitHub MCP server with the specified GH client and logger.
-func NewServer(getClient GetClientFn, version string, readOnly bool, t translations.TranslationHelperFunc) *server.MCPServer {
+func NewServer(getClient GetClientFn, version string, readOnly bool, t translations.TranslationHelperFunc, opts ...server.ServerOption) *server.MCPServer {
+	// Add default options
+	opts = append(opts, server.WithResourceCapabilities(true, true))
+	opts = append(opts, server.WithLogging())
+
 	// Create a new MCP server
 	s := server.NewMCPServer(
 		"github-mcp-server",
 		version,
-		server.WithResourceCapabilities(true, true),
-		server.WithLogging())
+		opts...,
+	)
 
 	// Add GitHub Resources
 	s.AddResourceTemplate(GetRepositoryResourceContent(getClient, t))
