summaryrefslogtreecommitdiff
path: root/internal/lsp/transport.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/lsp/transport.go')
-rw-r--r--internal/lsp/transport.go63
1 files changed, 63 insertions, 0 deletions
diff --git a/internal/lsp/transport.go b/internal/lsp/transport.go
new file mode 100644
index 0000000..671d69b
--- /dev/null
+++ b/internal/lsp/transport.go
@@ -0,0 +1,63 @@
+package lsp
+
+import (
+ "encoding/json"
+ "fmt"
+ "io"
+ "net/textproto"
+ "strconv"
+ "strings"
+)
+
+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) {
+ data, err := json.Marshal(v)
+ if err != nil {
+ s.logger.Printf("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 {
+ s.logger.Printf("write header error: %v", err)
+ return
+ }
+ if _, err := s.out.Write(data); err != nil {
+ s.logger.Printf("write body error: %v", err)
+ return
+ }
+}