1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Summary: LSP transport utilities to read and write JSON-RPC messages with Content-Length framing.
package lsp
import (
"encoding/json"
"fmt"
"io"
"net/textproto"
"strconv"
"strings"
"codeberg.org/snonux/hexai/internal/logging"
)
func (s *Server) readMessage() ([]byte, error) {
tp := textproto.NewReader(s.in)
var contentLength int
for {
line, err := tp.ReadLine()
if err != nil {
return nil, err
}
if line == "" { // end of headers
break
}
parts := strings.SplitN(line, ":", 2)
if len(parts) != 2 {
continue
}
key := strings.TrimSpace(strings.ToLower(parts[0]))
val := strings.TrimSpace(parts[1])
switch key {
case "content-length":
n, err := strconv.Atoi(val)
if err != nil {
return nil, fmt.Errorf("invalid Content-Length: %v", err)
}
contentLength = n
}
}
if contentLength <= 0 {
return nil, fmt.Errorf("missing or invalid Content-Length")
}
buf := make([]byte, contentLength)
if _, err := io.ReadFull(s.in, buf); err != nil {
return nil, err
}
return buf, nil
}
func (s *Server) writeMessage(v any) {
s.outMu.Lock()
defer s.outMu.Unlock()
data, err := json.Marshal(v)
if err != nil {
logging.Logf("lsp ", "marshal error: %v", err)
return
}
header := fmt.Sprintf("Content-Length: %d\r\n\r\n", len(data))
if _, err := io.WriteString(s.out, header); err != nil {
logging.Logf("lsp ", "write header error: %v", err)
return
}
if _, err := s.out.Write(data); err != nil {
logging.Logf("lsp ", "write body error: %v", err)
return
}
}
|
