102 lines
2.1 KiB
Go
102 lines
2.1 KiB
Go
// lsp_definition — Go to definition of symbol at position.
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
|
|
"ollie/tools/lsp"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) > 1 && os.Args[1] == "--bridge" {
|
|
if err := lsp.RunBridge(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "bridge: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
return
|
|
}
|
|
|
|
var args struct {
|
|
Path string `json:"path"`
|
|
Line string `json:"line"`
|
|
Col string `json:"col"`
|
|
}
|
|
if err := json.NewDecoder(os.Stdin).Decode(&args); err != nil {
|
|
fmt.Fprintf(os.Stderr, "error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
file, _ := filepath.Abs(args.Path)
|
|
line, _ := strconv.Atoi(args.Line)
|
|
col, _ := strconv.Atoi(args.Col)
|
|
|
|
result, err := lsp.ClientRequest(&lsp.BridgeRequest{
|
|
Method: "definition",
|
|
File: file,
|
|
Line: line - 1, // 1-indexed → 0-indexed
|
|
Col: col - 1, // 1-indexed → 0-indexed
|
|
})
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if result == nil || string(result) == "null" {
|
|
fmt.Println("(no definition found)")
|
|
return
|
|
}
|
|
|
|
// Parse location(s)
|
|
var locations []struct {
|
|
URI string `json:"uri"`
|
|
TargetURI string `json:"targetUri"`
|
|
Range *rang `json:"range"`
|
|
TargetRange *rang `json:"targetRange"`
|
|
}
|
|
// Try as array first
|
|
if err := json.Unmarshal(result, &locations); err != nil {
|
|
// Try as single location
|
|
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"`
|
|
}
|