正路目录

3 files changed, 276 insertions, 0 deletions

diff --git a/script-beta/proto/define.lua b/script-beta/proto/define.lua
new file mode 100644
index 00000000..61c4037c
--- /dev/null
+++ b/script-beta/proto/define.lua
@@ -0,0 +1,140 @@
+local guide = require 'parser.guide'
+
+local m = {}
+
+--- 获取 position 对应的光标位置
+---@param lines table
+---@param text string
+---@param position position
+---@return integer
+function m.offset(lines, text, position)
+    local row    = position.line + 1
+    local start  = guide.lineRange(lines, row)
+    local offset = utf8.offset(text, position.character + 1, start)
+    if text:sub(offset-1, offset):match '[%w_][^%w_]' then
+        offset = offset - 1
+    end
+    return offset
+end
+
+--- 将光标位置转化为 position
+---@alias position table
+---@param lines table
+---@param text string
+---@param offset integer
+---@return position
+function m.position(lines, text, offset)
+    local row, col = guide.positionOf(lines, offset)
+    local start    = guide.lineRange(lines, row)
+    if start < 1 then
+        start = 1
+    end
+    local ucol     = utf8.len(text, start, start + col - 1, true)
+    if row < 1 then
+        row = 1
+    end
+    return {
+        line      = row - 1,
+        character = ucol,
+    }
+end
+
+--- 将起点与终点位置转化为 range
+---@alias range table
+---@param lines table
+---@param text string
+---@param offset1 integer
+---@param offset2 integer
+function m.range(lines, text, offset1, offset2)
+    local range = {
+        start   = m.position(lines, text, offset1),
+        ['end'] = m.position(lines, text, offset2),
+    }
+    if range.start.character > 0 then
+        range.start.character = range.start.character - 1
+    end
+    return range
+end
+
+---@alias location table
+---@param uri string
+---@param range range
+---@return location
+function m.location(uri, range)
+    return {
+        uri   = uri,
+        range = range,
+    }
+end
+
+---@alias locationLink table
+---@param uri string
+---@param range range
+---@param selection range
+---@param origin range
+function m.locationLink(uri, range, selection, origin)
+    return {
+        targetUri            = uri,
+        targetRange          = range,
+        targetSelectionRange = selection,
+        originSelectionRange = origin,
+    }
+end
+
+function m.textEdit(range, newtext)
+    return {
+        range   = range,
+        newText = newtext,
+    }
+end
+
+--- 诊断等级
+m.DiagnosticSeverity = {
+    Error       = 1,
+    Warning     = 2,
+    Information = 3,
+    Hint        = 4,
+}
+
+--- 诊断类型与默认等级
+m.DiagnosticDefaultSeverity = {
+    ['unused-local']        = 'Hint',
+    ['unused-function']     = 'Hint',
+    ['undefined-global']    = 'Warning',
+    ['global-in-nil-env']   = 'Warning',
+    ['unused-label']        = 'Hint',
+    ['unused-vararg']       = 'Hint',
+    ['trailing-space']      = 'Hint',
+    ['redefined-local']     = 'Hint',
+    ['newline-call']        = 'Information',
+    ['newfield-call']       = 'Warning',
+    ['redundant-parameter'] = 'Hint',
+    ['ambiguity-1']         = 'Warning',
+    ['lowercase-global']    = 'Information',
+    ['undefined-env-child'] = 'Information',
+    ['duplicate-index']     = 'Warning',
+    ['empty-block']         = 'Hint',
+    ['redundant-value']     = 'Hint',
+    ['emmy-lua']            = 'Warning',
+}
+
+--- 诊断报告标签
+m.DiagnosticTag = {
+    Unnecessary = 1,
+    Deprecated  = 2,
+}
+
+m.DocumentHighlightKind = {
+    Text  = 1,
+    Read  = 2,
+    Write = 3,
+}
+
+m.MessageType = {
+    Error   = 1,
+    Warning = 2,
+    Info    = 3,
+    Log     = 4,
+}
+
+return m
diff --git a/script-beta/proto/init.lua b/script-beta/proto/init.lua
new file mode 100644
index 00000000..33e637f6
--- /dev/null
+++ b/script-beta/proto/init.lua
@@ -0,0 +1,3 @@
+local proto   = require 'proto.proto'
+
+return proto
diff --git a/script-beta/proto/proto.lua b/script-beta/proto/proto.lua
new file mode 100644
index 00000000..f04653d5
--- /dev/null
+++ b/script-beta/proto/proto.lua
@@ -0,0 +1,133 @@
+local subprocess = require 'bee.subprocess'
+local util       = require 'utility'
+local await      = require 'await'
+local pub        = require 'pub'
+local jsonrpc    = require 'jsonrpc'
+local ErrorCodes = require 'define.ErrorCodes'
+
+local reqCounter = util.counter()
+
+local m = {}
+
+m.ability = {}
+m.waiting = {}
+
+function m.getMethodName(proto)
+    if proto.method:sub(1, 2) == '$/' then
+        return proto.method:sub(3), true
+    else
+        return proto.method, false
+    end
+end
+
+function m.on(method, callback)
+    m.ability[method] = callback
+end
+
+function m.response(id, res)
+    if id == nil then
+        log.error('Response id is nil!', util.dump(res))
+        return
+    end
+    -- res 可能是nil，为了转成json时保留nil，使用 container 容器
+    local data = util.container()
+    data.id     = id
+    data.result = res
+    local buf = jsonrpc.encode(data)
+    log.debug('Response', id, #buf)
+    io.stdout:write(buf)
+end
+
+function m.responseErr(id, code, message)
+    if id == nil then
+        log.error('Response id is nil!', util.dump(message))
+        return
+    end
+    local buf = jsonrpc.encode {
+        id    = id,
+        error = {
+            code    = code,
+            message = message,
+        }
+    }
+    log.debug('ResponseErr', id, #buf)
+    io.stdout:write(buf)
+end
+
+function m.notify(name, params)
+    local buf = jsonrpc.encode {
+        method = name,
+        params = params,
+    }
+    log.debug('Notify', name, #buf)
+    io.stdout:write(buf)
+end
+
+function m.awaitRequest(name, params)
+    local id = reqCounter()
+    local buf = jsonrpc.encode {
+        id     = id,
+        method = name,
+        params = params,
+    }
+    log.debug('Request', name, #buf)
+    io.stdout:write(buf)
+    return await.wait(function (waker)
+        m.waiting[id] = waker
+    end)
+end
+
+function m.doMethod(proto)
+    local method, optional = m.getMethodName(proto)
+    local abil = m.ability[method]
+    if not abil then
+        if not optional then
+            log.warn('Recieved unknown proto: ' .. method)
+        end
+        if proto.id then
+            m.responseErr(proto.id, ErrorCodes.MethodNotFound, method)
+        end
+        return
+    end
+    await.create(function ()
+        local clock = os.clock()
+        local ok, res = xpcall(abil, log.error, proto.params)
+        local passed = os.clock() - clock
+        if passed > 0.2 then
+            log.debug(('Method [%s] takes [%.3f]sec.'):format(method, passed))
+        end
+        if not proto.id then
+            return
+        end
+        if ok then
+            m.response(proto.id, res)
+        else
+            m.responseErr(proto.id, ErrorCodes.InternalError, res)
+        end
+    end)
+end
+
+function m.doResponse(proto)
+    local id = proto.id
+    local waker = m.waiting[id]
+    if not waker then
+        log.warn('Response id not found: ' .. util.dump(proto))
+        return
+    end
+    m.waiting[id] = nil
+    if proto.error then
+        log.warn(('Response error [%d]: %s'):format(proto.error.code, proto.error.message))
+        return
+    end
+    waker(proto.result)
+end
+
+function m.listen()
+    subprocess.filemode(io.stdin,  'b')
+    subprocess.filemode(io.stdout, 'b')
+    io.stdin:setvbuf  'no'
+    io.stdout:setvbuf 'no'
+    pub.task('loadProto')
+end
+
+return m