正路目录

4 files changed, 181 insertions, 0 deletions

diff --git a/script-beta/brave/brave.lua b/script-beta/brave/brave.lua
new file mode 100644
index 00000000..08909074
--- /dev/null
+++ b/script-beta/brave/brave.lua
@@ -0,0 +1,70 @@
+local thread = require 'bee.thread'
+
+---@class pub_brave
+local m = {}
+m.type = 'brave'
+m.ability = {}
+m.queue = {}
+
+--- 注册成为勇者
+function m.register(id)
+    m.taskpad = thread.channel('taskpad' .. id)
+    m.waiter  = thread.channel('waiter'  .. id)
+    m.id      = id
+
+    if #m.queue > 0 then
+        for _, info in ipairs(m.queue) do
+            m.waiter:push(info.name, info.params)
+        end
+    end
+    m.queue = nil
+
+    m.start()
+end
+
+--- 注册能力
+function m.on(name, callback)
+    m.ability[name] = callback
+end
+
+--- 报告
+function m.push(name, params)
+    if m.waiter then
+        m.waiter:push(name, params)
+    else
+        m.queue[#m.queue+1] = {
+            name   = name,
+            params = params,
+        }
+    end
+end
+
+--- 开始找工作
+function m.start()
+    m.push('mem', collectgarbage 'count')
+    while true do
+        local suc, name, id, params = m.taskpad:pop()
+        if not suc then
+            -- 找不到工作的勇者，只好睡觉
+            thread.sleep(0.001)
+            goto CONTINUE
+        end
+        local ability = m.ability[name]
+        -- TODO
+        if not ability then
+            m.waiter:push(id)
+            log.error('Brave can not handle this work: ' .. name)
+            goto CONTINUE
+        end
+        local ok, res = xpcall(ability, log.error, params)
+        if ok then
+            m.waiter:push(id, res)
+        else
+            m.waiter:push(id)
+        end
+        m.push('mem', collectgarbage 'count')
+        ::CONTINUE::
+    end
+end
+
+return m
diff --git a/script-beta/brave/init.lua b/script-beta/brave/init.lua
new file mode 100644
index 00000000..24c2e412
--- /dev/null
+++ b/script-beta/brave/init.lua
@@ -0,0 +1,4 @@
+local brave   = require 'brave.brave'
+require 'brave.work'
+
+return brave
diff --git a/script-beta/brave/log.lua b/script-beta/brave/log.lua
new file mode 100644
index 00000000..cd27cd55
--- /dev/null
+++ b/script-beta/brave/log.lua
@@ -0,0 +1,52 @@
+local brave          = require 'brave'
+
+local tablePack      = table.pack
+local tostring       = tostring
+local tableConcat    = table.concat
+local debugTraceBack = debug.traceback
+local debugGetInfo   = debug.getinfo
+
+_ENV = nil
+
+local function pushLog(level, ...)
+    local t = tablePack(...)
+    for i = 1, t.n do
+        t[i] = tostring(t[i])
+    end
+    local str = tableConcat(t, '\t', 1, t.n)
+    if level == 'error' then
+        str = str .. '\n' .. debugTraceBack(nil, 3)
+    end
+    local info = debugGetInfo(3, 'Sl')
+    brave.push('log', {
+        level = level,
+        msg   = str,
+        src   = info.source,
+        line  = info.currentline,
+    })
+    return str
+end
+
+local m = {}
+
+function m.info(...)
+    pushLog('info', ...)
+end
+
+function m.debug(...)
+    pushLog('debug', ...)
+end
+
+function m.trace(...)
+    pushLog('trace', ...)
+end
+
+function m.warn(...)
+    pushLog('warn', ...)
+end
+
+function m.error(...)
+    pushLog('error', ...)
+end
+
+return m
diff --git a/script-beta/brave/work.lua b/script-beta/brave/work.lua
new file mode 100644
index 00000000..dba27808
--- /dev/null
+++ b/script-beta/brave/work.lua
@@ -0,0 +1,55 @@
+local brave   = require 'brave.brave'
+local jsonrpc = require 'jsonrpc'
+local parser  = require 'parser'
+local fs      = require 'bee.filesystem'
+local furi    = require 'file-uri'
+local util    = require 'utility'
+
+brave.on('loadProto', function ()
+    while true do
+        local proto = jsonrpc.decode(io.read, log.error)
+        if proto then
+            brave.push('proto', proto)
+        end
+    end
+end)
+
+brave.on('compile', function (text)
+    local state, err = parser:compile(text, 'lua', 'Lua 5.4')
+    if not state then
+        log.error(err)
+        return
+    end
+    local lines = parser:lines(text)
+    return {
+        root  = state.root,
+        value = state.value,
+        errs  = state.errs,
+        lines = lines,
+    }
+end)
+
+brave.on('listDirectory', function (uri)
+    local path = fs.path(furi.decode(uri))
+    local uris = {}
+    for child in path:list_directory() do
+        local childUri = furi.encode(child:string())
+        uris[#uris+1] = childUri
+    end
+    return uris
+end)
+
+brave.on('isDirectory', function (uri)
+    local path = fs.path(furi.decode(uri))
+    return fs.is_directory(path)
+end)
+
+brave.on('loadFile', function (uri)
+    local filename = furi.decode(uri)
+    return util.loadFile(filename)
+end)
+
+brave.on('saveFile', function (params)
+    local filename = furi.decode(params.uri)
+    return util.saveFile(filename, params.text)
+end)