feat: tool prefix name

Signed-off-by: hungran <[email protected]>

Author: hungran

cmd/github-mcp-server/main.go

@@ -52,6 +52,7 @@ var (
 				EnabledToolsets:      enabledToolsets,
 				DynamicToolsets:      viper.GetBool("dynamic_toolsets"),
 				ReadOnly:             viper.GetBool("read-only"),
+				ToolPrefix:           viper.GetString("tool-prefix"),
 				ExportTranslations:   viper.GetBool("export-translations"),
 				EnableCommandLogging: viper.GetBool("enable-command-logging"),
 				LogFilePath:          viper.GetString("log-file"),
@@ -72,6 +73,7 @@ func init() {
 	rootCmd.PersistentFlags().StringSlice("toolsets", github.DefaultTools, "An optional comma separated list of groups of tools to allow, defaults to enabling all")
 	rootCmd.PersistentFlags().Bool("dynamic-toolsets", false, "Enable dynamic toolsets")
 	rootCmd.PersistentFlags().Bool("read-only", false, "Restrict the server to read-only operations")
+	rootCmd.PersistentFlags().String("tool-prefix", "", "Optional prefix to add to all tool names (e.g. 'github_')")
 	rootCmd.PersistentFlags().String("log-file", "", "Path to log file")
 	rootCmd.PersistentFlags().Bool("enable-command-logging", false, "When enabled, the server will log all command requests and responses to the log file")
 	rootCmd.PersistentFlags().Bool("export-translations", false, "Save translations to a JSON file")
@@ -82,6 +84,7 @@ func init() {
 	_ = viper.BindPFlag("toolsets", rootCmd.PersistentFlags().Lookup("toolsets"))
 	_ = viper.BindPFlag("dynamic_toolsets", rootCmd.PersistentFlags().Lookup("dynamic-toolsets"))
 	_ = viper.BindPFlag("read-only", rootCmd.PersistentFlags().Lookup("read-only"))
+	_ = viper.BindPFlag("tool-prefix", rootCmd.PersistentFlags().Lookup("tool-prefix"))
 	_ = viper.BindPFlag("log-file", rootCmd.PersistentFlags().Lookup("log-file"))
 	_ = viper.BindPFlag("enable-command-logging", rootCmd.PersistentFlags().Lookup("enable-command-logging"))
 	_ = viper.BindPFlag("export-translations", rootCmd.PersistentFlags().Lookup("export-translations"))

internal/ghmcp/server.go

@@ -45,6 +45,9 @@ type MCPServerConfig struct {
 	// ReadOnly indicates if we should only offer read-only tools
 	ReadOnly bool
 
+	// ToolPrefix is an optional prefix to add to all tool names (e.g. "github_")
+	ToolPrefix string
+
 	// Translator provides translated text for the server tooling
 	Translator translations.TranslationHelperFunc
 
@@ -143,11 +146,11 @@ func NewMCPServer(cfg MCPServerConfig) (*server.MCPServer, error) {
 	}
 
 	// Register all mcp functionality with the server
-	tsg.RegisterAll(ghServer)
+	tsg.RegisterAllWithPrefix(ghServer, cfg.ToolPrefix)
 
 	if cfg.DynamicToolsets {
 		dynamic := github.InitDynamicToolset(ghServer, tsg, cfg.Translator)
-		dynamic.RegisterTools(ghServer)
+		dynamic.RegisterToolsWithPrefix(ghServer, cfg.ToolPrefix)
 	}
 
 	return ghServer, nil
@@ -174,6 +177,9 @@ type StdioServerConfig struct {
 	// ReadOnly indicates if we should only register read-only tools
 	ReadOnly bool
 
+	// ToolPrefix is an optional prefix to add to all tool names (e.g. "github_")
+	ToolPrefix string
+
 	// ExportTranslations indicates if we should export translations
 	// See: https://github.com/github/github-mcp-server?tab=readme-ov-file#i18n--overriding-descriptions
 	ExportTranslations bool
@@ -203,6 +209,7 @@ func RunStdioServer(cfg StdioServerConfig) error {
 		EnabledToolsets:   cfg.EnabledToolsets,
 		DynamicToolsets:   cfg.DynamicToolsets,
 		ReadOnly:          cfg.ReadOnly,
+		ToolPrefix:        cfg.ToolPrefix,
 		Translator:        t,
 		ContentWindowSize: cfg.ContentWindowSize,
 	})

pkg/toolsets/toolsets.go

@@ -29,6 +29,18 @@ func NewToolsetDoesNotExistError(name string) *ToolsetDoesNotExistError {
 	return &ToolsetDoesNotExistError{Name: name}
 }
 
+// createToolWithPrefixedName creates a new tool with the same properties as the original but with a prefixed name
+func createToolWithPrefixedName(original mcp.Tool, prefix string) mcp.Tool {
+	// Create a new tool with the prefixed name and copy all properties
+	newTool := mcp.Tool{
+		Name:        prefix + original.Name,
+		Description: original.Description,
+		InputSchema: original.InputSchema,
+		Annotations: original.Annotations,
+	}
+	return newTool
+}
+
 func NewServerTool(tool mcp.Tool, handler server.ToolHandlerFunc) server.ServerTool {
 	return server.ServerTool{Tool: tool, Handler: handler}
 }
@@ -80,15 +92,29 @@ func (t *Toolset) GetAvailableTools() []server.ServerTool {
 }
 
 func (t *Toolset) RegisterTools(s *server.MCPServer) {
+	t.RegisterToolsWithPrefix(s, "")
+}
+
+func (t *Toolset) RegisterToolsWithPrefix(s *server.MCPServer, prefix string) {
 	if !t.Enabled {
 		return
 	}
 	for _, tool := range t.readTools {
-		s.AddTool(tool.Tool, tool.Handler)
+		toolToRegister := tool.Tool
+		if prefix != "" {
+			// Create a new tool with the prefixed name
+			toolToRegister = createToolWithPrefixedName(tool.Tool, prefix)
+		}
+		s.AddTool(toolToRegister, tool.Handler)
 	}
 	if !t.readOnly {
 		for _, tool := range t.writeTools {
-			s.AddTool(tool.Tool, tool.Handler)
+			toolToRegister := tool.Tool
+			if prefix != "" {
+				// Create a new tool with the prefixed name
+				toolToRegister = createToolWithPrefixedName(tool.Tool, prefix)
+			}
+			s.AddTool(toolToRegister, tool.Handler)
 		}
 	}
 }
@@ -239,8 +265,12 @@ func (tg *ToolsetGroup) EnableToolset(name string) error {
 }
 
 func (tg *ToolsetGroup) RegisterAll(s *server.MCPServer) {
+	tg.RegisterAllWithPrefix(s, "")
+}
+
+func (tg *ToolsetGroup) RegisterAllWithPrefix(s *server.MCPServer, prefix string) {
 	for _, toolset := range tg.Toolsets {
-		toolset.RegisterTools(s)
+		toolset.RegisterToolsWithPrefix(s, prefix)
 		toolset.RegisterResourcesTemplates(s)
 		toolset.RegisterPrompts(s)
 	}