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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
local json = require 'json'
local Method = require 'lsp.method'
local ErrorCodes = {
-- Defined by JSON RPC
ParseError = -32700,
InvalidRequest = -32600,
MethodNotFound = -32601,
InvalidParams = -32602,
InternalError = -32603,
serverErrorStart = -32099,
serverErrorEnd = -32000,
ServerNotInitialized = -32002,
UnknownErrorCode = -32001,
-- Defined by the protocol.
RequestCancelled = -32800,
}
local mt = {}
mt.__index = mt
mt._input = nil
mt._output = nil
mt._inited = false
mt._method = nil
function mt:_callback(method, params)
local f = self._method
if f then
return f(method, params)
end
return nil
end
function mt:_send(data)
local f = self._output
if not f then
return
end
local content = json.encode(data)
local buf = ('Content-Length: %d\r\n\r\n%s'):format(#content + 2, content)
f(buf)
end
function mt:_readAsContent(header)
local len = tonumber(header:match('%d+'))
if not len then
log.error('错误的协议头：', header)
return
end
local buf = self:read(len)
local suc, res = pcall(json.decode, buf)
if not suc then
log.error('错误的协议：', buf)
return
end
log.debug(table.dump(res))
local id = res.id
local method = res.method
local params = res.params
local f = Method[method]
if not f then
return
end
local response, err = f(self, params)
if response then
self._send {
id = id,
result = response,
}
else
self._send {
id = id,
error = {
code = ErrorCodes.UnknownErrorCode,
message = err,
},
}
end
end
function mt:setInput(input)
self._input = input
end
function mt:setOutput(output)
self._output = output
end
function mt:read(mode)
if not self._input then
return nil
end
return self._input(mode)
end
function mt:start(method)
self._method = method
while true do
local header = self:read 'L'
if not header then
return
end
if header:sub(1, #'Content-Length') == 'Content-Length' then
self:_readAsContent(header)
elseif header:sub(1, #'Content-Type') == 'Content-Type' then
else
log.error('错误的协议头：', header)
end
end
return true
end
function mt:stop()
self._input = nil
self._output = nil
end
return function ()
return setmetatable({}, mt)
end
|
