src/service/init.lua


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

local sleep = require 'ffi.sleep'
local ext   = require 'process.ext'
local lsp   = require 'lsp'

local Method = {}

function Method.initialize()
    return {
        capabilities = {
            definitionProvider = true,
        }
    }
end

local function listen(self, input, output)
    if input then
        log.info('指定输入文件，路径为：', input)
        fs.create_directories(input:parent_path())
        io.input(io.open(input:string(), 'rb'))
    else
        ext.set_filemode(io.stdin, 'b')
    end
    if output then
        log.info('指定输出文件，路径为：', output)
        fs.create_directories(output:parent_path())
        io.output(io.open(output:string(), 'wb'))
    else
        ext.set_filemode(io.stdout, 'b')
        io.stdout:setvbuf 'no'
    end

    local session = lsp()
    session:setInput(function (mode)
        return io.read(mode)
    end)
    session:setOutput(function (buf)
        io.write(buf)
        log.debug(buf)
    end)
    session:start(function (method, params)
        local f = Method[method]
        if f then
            return f(params)
        end
        return nil
    end)
end

local mt = {
    definition = require 'service.definition',
    listen     = listen,
}
mt.__index = mt

return function ()
    local session = setmetatable({}, mt)
    return session
end