Function extractPublishedDiagnostics

pkg/lsp/handler_functional_test.go:278–321

(t *testing.T, output string)

}

func extractPublishedDiagnostics(t *testing.T, output string) []Diagnostic {
	t.Helper()
	// Output may contain multiple LSP messages; find the publishDiagnostics one.
	// Each message is prefixed with Content-Length header.
	parts := strings.Split(output, "Content-Length:")
	for _, part := range parts {
		if !strings.Contains(part, "publishDiagnostics") {
			continue
		}
		// Find the JSON body
		idx := strings.Index(part, "{")
		if idx < 0 {
			continue
		}
		body := part[idx:]
		// May have trailing data; find balanced braces
		var msg struct {
			Params struct {
				Diagnostics []Diagnostic `json:"diagnostics"`
			} `json:"params"`
		}
		if err := json.Unmarshal([]byte(body), &msg); err != nil {
			// Try to find end of JSON
			depth := 0
			end := 0
			for i, c := range body {
				if c == '{' {
					depth++
				} else if c == '}' {
					depth--
					if depth == 0 {
						end = i + 1
						break
					}
				}
			}
			if end > 0 {
				_ = json.Unmarshal([]byte(body[:end]), &msg)
			}
		}
		return msg.Params.Diagnostics
	}
	return nil
}