ollie/tools/lsp/cmd/lsp_definition/main.go

61 lines
1.3 KiB
Go

// lsp_definition — Go to definition of symbol at position.
package main
import (
"encoding/json"
"fmt"
"ollie/tools/lsp/internal/lsp"
"ollie/tools/lsp/internal/cmdutil"
)
func main() {
cmdutil.Run("definition", func(result json.RawMessage) {
var locations []struct {
URI string `json:"uri"`
TargetURI string `json:"targetUri"`
Range *rang `json:"range"`
TargetRange *rang `json:"targetRange"`
}
if err := json.Unmarshal(result, &locations); err != nil {
var single struct {
URI string `json:"uri"`
TargetURI string `json:"targetUri"`
Range *rang `json:"range"`
TargetRange *rang `json:"targetRange"`
}
if err := json.Unmarshal(result, &single); err != nil {
fmt.Println("(no definition found)")
return
}
locations = append(locations, single)
}
for _, loc := range locations {
uri := loc.URI
if uri == "" {
uri = loc.TargetURI
}
path := lsp.URIToPath(uri)
r := loc.Range
if r == nil {
r = loc.TargetRange
}
if r == nil {
fmt.Println(path)
continue
}
fmt.Printf("%s:%d:%d\n", path, r.Start.Line+1, r.Start.Character+1)
}
})
}
type rang struct {
Start position `json:"start"`
}
type position struct {
Line int `json:"line"`
Character int `json:"character"`
}