Files
2026-07-13 13:00:08 +08:00

96 lines
3.4 KiB
Go

package lsp
import (
"context"
"encoding/json"
"fmt"
"reasonix/internal/tool"
)
// Tools adapts the manager's read-only queries to the tool.Tool interface. They
// report ReadOnly=true, so a batch of them rides the agent's parallel dispatch
// and shares one server per language.
func Tools(m *Manager) []tool.Tool {
return []tool.Tool{
posTool{m, "lsp_definition", "Jump to where a symbol is defined. Give the file, the 1-based line the symbol appears on, and the symbol text itself.", m.Definition},
posTool{m, "lsp_references", "List every reference to a symbol across the workspace. Give the file, the 1-based line, and the symbol text.", m.References},
posTool{m, "lsp_hover", "Show the type signature and documentation for a symbol. Give the file, the 1-based line, and the symbol text.", m.Hover},
diagTool{m},
}
}
type posArgs struct {
File string `json:"file"`
Line int `json:"line"`
Symbol string `json:"symbol"`
}
type posTool struct {
m *Manager
name string
desc string
fn func(context.Context, string, int, string) (string, error)
}
func (t posTool) Name() string { return t.name }
func (t posTool) Description() string { return t.desc }
func (t posTool) ReadOnly() bool { return true }
// SnipHint keeps a long head of definitions/references and a short tail: the
// first results are the ones the model acts on.
func (posTool) SnipHint() tool.SnipHint {
return tool.SnipHint{Head: 60, Tail: 10, HeadChars: 10000, TailChars: 1500}
}
func (t posTool) Schema() json.RawMessage {
return json.RawMessage(`{
"type":"object",
"properties":{
"file":{"type":"string","description":"Path to the source file, relative to the workspace root or absolute."},
"line":{"type":"integer","description":"1-based line number the symbol appears on."},
"symbol":{"type":"string","description":"The exact symbol text on that line, e.g. \"executeBatch\". Used to locate the column."}
},
"required":["file","line","symbol"]
}`)
}
func (t posTool) Execute(ctx context.Context, args json.RawMessage) (string, error) {
var p posArgs
if err := json.Unmarshal(args, &p); err != nil {
return "", fmt.Errorf("invalid args: %w", err)
}
if p.File == "" || p.Symbol == "" || p.Line < 1 {
return "", fmt.Errorf("file, line (>=1) and symbol are required")
}
return t.fn(ctx, p.File, p.Line, p.Symbol)
}
type diagTool struct{ m *Manager }
func (diagTool) Name() string { return "lsp_diagnostics" }
func (diagTool) ReadOnly() bool { return true }
// SnipHint keeps a long head of diagnostics and a short tail.
func (diagTool) SnipHint() tool.SnipHint {
return tool.SnipHint{Head: 60, Tail: 10, HeadChars: 10000, TailChars: 1500}
}
func (diagTool) Description() string {
return "Report compiler/linter diagnostics (errors, warnings) for a file from its language server. Use after editing to check the change compiles."
}
func (diagTool) Schema() json.RawMessage {
return json.RawMessage(`{"type":"object","properties":{"file":{"type":"string","description":"Path to the source file, relative to the workspace root or absolute."}},"required":["file"]}`)
}
func (t diagTool) Execute(ctx context.Context, args json.RawMessage) (string, error) {
var p struct {
File string `json:"file"`
}
if err := json.Unmarshal(args, &p); err != nil {
return "", fmt.Errorf("invalid args: %w", err)
}
if p.File == "" {
return "", fmt.Errorf("file is required")
}
return t.m.Diagnostics(ctx, p.File)
}