1// Copyright 2018 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
4
5package protocol
6
7import (
8	"context"
9
10	"golang.org/x/tools/internal/jsonrpc2"
11	"golang.org/x/tools/internal/lsp/xlog"
12)
13
14const defaultMessageBufferSize = 20
15const defaultRejectIfOverloaded = false
16
17func canceller(ctx context.Context, conn *jsonrpc2.Conn, id jsonrpc2.ID) {
18	conn.Notify(context.Background(), "$/cancelRequest", &CancelParams{ID: id})
19}
20
21func NewClient(stream jsonrpc2.Stream, client Client) (*jsonrpc2.Conn, Server, xlog.Logger) {
22	log := xlog.New(NewLogger(client))
23	conn := jsonrpc2.NewConn(stream)
24	conn.Capacity = defaultMessageBufferSize
25	conn.RejectIfOverloaded = defaultRejectIfOverloaded
26	conn.Handler = clientHandler(log, client)
27	conn.Canceler = jsonrpc2.Canceler(canceller)
28	return conn, &serverDispatcher{Conn: conn}, log
29}
30
31func NewServer(stream jsonrpc2.Stream, server Server) (*jsonrpc2.Conn, Client, xlog.Logger) {
32	conn := jsonrpc2.NewConn(stream)
33	client := &clientDispatcher{Conn: conn}
34	log := xlog.New(NewLogger(client))
35	conn.Capacity = defaultMessageBufferSize
36	conn.RejectIfOverloaded = defaultRejectIfOverloaded
37	conn.Handler = serverHandler(log, server)
38	conn.Canceler = jsonrpc2.Canceler(canceller)
39	return conn, client, log
40}
41
42func sendParseError(ctx context.Context, log xlog.Logger, req *jsonrpc2.Request, err error) {
43	if _, ok := err.(*jsonrpc2.Error); !ok {
44		err = jsonrpc2.NewErrorf(jsonrpc2.CodeParseError, "%v", err)
45	}
46	if err := req.Reply(ctx, nil, err); err != nil {
47		log.Errorf(ctx, "%v", err)
48	}
49}
50