add mcp list-tools to list available tool calls

Author: burmudar

cmd/src/mcp.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+func init() {
+	flagSet := flag.NewFlagSet("mcp", flag.ExitOnError)
+	handler := func(args []string) error {
+		return mcpMain(args)
+	}
+
+	commands = append(commands, &command{
+		flagSet: flagSet,
+		handler: handler,
+	})
+}
+func mcpMain(args []string) error {
+	fmt.Println("NOTE: This command is still experimental")
+	tools, err := LoadMCPToolDefinitions(mcpToolListJSON)
+	if err != nil {
+		return err
+	}
+
+	subcmd := args[0]
+	if subcmd == "list-tools" {
+		fmt.Println("Available tools")
+		for name := range tools {
+			fmt.Printf("- %s\n", name)
+		}
+		return nil
+	}
+
+	tool, ok := tools[subcmd]
+	if !ok {
+		return fmt.Errorf("tool definition for %q not found - run src mcp list-tools to see a list of available tools", subcmd)
+	}
+	return handleMcpTool(tool, args[1:])
+}
+
+func handleMcpTool(tool *MCPToolDef, args []string) error {
+	fmt.Printf("handling tool %q args: %+v", tool.Name, args)
+	return nil
+}

internal/mcp/mcp_parse.go

@@ -5,6 +5,7 @@ import (
 	_ "embed"
 	"encoding/json"
 	"fmt"
+	"strings"
 
 	"github.com/sourcegraph/sourcegraph/lib/errors"
 )
@@ -85,7 +86,8 @@ func LoadToolDefinitions(data []byte) (map[string]*ToolDef, error) {
 	parser := &parser{}
 
 	for _, t := range defs.Tools {
-		tools[t.Name] = &ToolDef{
+		name := normalizeToolName(t.Name)
+		tools[name] = &ToolDef{
 			Name:         t.Name,
 			Description:  t.Description,
 			InputSchema:  parser.parseRootSchema(t.InputSchema),
@@ -164,3 +166,9 @@ func (p *parser) parseProperties(props map[string]json.RawMessage) map[string]Sc
 	}
 	return res
 }
+
+// normalizeToolName takes mcp tool names like 'sg_keyword_search' and normalizes it to 'keyword-search"
+func normalizeToolName(toolName string) string {
+	toolName, _ = strings.CutPrefix(toolName, "sg_")
+	return strings.ReplaceAll(toolName, "_", "-")
+}