43 lines
1.4 KiB
Go
43 lines
1.4 KiB
Go
|
package tools
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"code.nochebuena.dev/einherjar/mcp/internal/index"
|
||
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||
|
|
)
|
||
|
|
|
||
|
|
type getSymbolInput struct {
|
||
|
|
Module string `json:"module" jsonschema:"the module name, e.g. core"`
|
||
|
|
Name string `json:"name" jsonschema:"the symbol name; for methods use Type.Method"`
|
||
|
|
SubPackage string `json:"subPackage,omitempty" jsonschema:"narrow to a specific sub-package, e.g. launcher"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type getSymbolOutput struct {
|
||
|
|
Matches []index.Symbol `json:"matches"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func registerGetSymbol(s *mcp.Server, idx *index.Index) {
|
||
|
|
mcp.AddTool(s, &mcp.Tool{
|
||
|
|
Name: "get_symbol",
|
||
|
|
Description: "Fetch full signature, doc comment, and source location for one symbol. Returns every match across sub-packages — use the subPackage filter when ambiguous.",
|
||
|
|
}, func(ctx context.Context, req *mcp.CallToolRequest, args getSymbolInput) (*mcp.CallToolResult, getSymbolOutput, error) {
|
||
|
|
m := idx.FindModule(args.Module)
|
||
|
|
if m == nil {
|
||
|
|
return errorResult("module not found: " + args.Module), getSymbolOutput{}, nil
|
||
|
|
}
|
||
|
|
out := getSymbolOutput{Matches: []index.Symbol{}}
|
||
|
|
for _, sym := range m.Symbols {
|
||
|
|
if !strings.EqualFold(sym.Name, args.Name) {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if args.SubPackage != "" && sym.SubPackage != args.SubPackage {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
out.Matches = append(out.Matches, sym)
|
||
|
|
}
|
||
|
|
return jsonText(out), out, nil
|
||
|
|
})
|
||
|
|
}
|