126 lines
3.3 KiB
Lua
126 lines
3.3 KiB
Lua
|
local brave = require 'brave.brave'
|
||
|
|
|
||
|
|
brave.on('loadProtoByStdio', function ()
|
||
|
|
local jsonrpc = require 'jsonrpc'
|
||
|
|
while true do
|
||
|
|
local proto, err = jsonrpc.decode(io.read)
|
||
|
|
--log.debug('loaded proto', proto.method)
|
||
|
|
if not proto then
|
||
|
|
brave.push('protoerror', err)
|
||
|
|
return
|
||
|
|
end
|
||
|
|
brave.push('proto', proto)
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
brave.on('loadProtoBySocket', function (param)
|
||
|
|
local jsonrpc = require 'jsonrpc'
|
||
|
|
local socket = require 'bee.socket'
|
||
|
|
local util = require 'utility'
|
||
|
|
local rfd = socket.fd(param.rfd)
|
||
|
|
local wfd = socket.fd(param.wfd)
|
||
|
|
local buf = ''
|
||
|
|
|
||
|
|
---@async
|
||
|
|
local parser = coroutine.create(function ()
|
||
|
|
while true do
|
||
|
|
---@async
|
||
|
|
local proto, err = jsonrpc.decode(function (len)
|
||
|
|
while true do
|
||
|
|
if #buf >= len then
|
||
|
|
local res = buf:sub(1, len)
|
||
|
|
buf = buf:sub(len + 1)
|
||
|
|
return res
|
||
|
|
end
|
||
|
|
coroutine.yield()
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
--log.debug('loaded proto', proto.method)
|
||
|
|
if not proto then
|
||
|
|
brave.push('protoerror', err)
|
||
|
|
return
|
||
|
|
end
|
||
|
|
brave.push('proto', proto)
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
while true do
|
||
|
|
local rd = socket.select({rfd, wfd}, nil, 10)
|
||
|
|
if not rd or #rd == 0 then
|
||
|
|
goto continue
|
||
|
|
end
|
||
|
|
if util.arrayHas(rd, wfd) then
|
||
|
|
local needSend = wfd:recv()
|
||
|
|
if needSend then
|
||
|
|
rfd:send(needSend)
|
||
|
|
elseif needSend == nil then
|
||
|
|
error('socket closed!')
|
||
|
|
end
|
||
|
|
end
|
||
|
|
if util.arrayHas(rd, rfd) then
|
||
|
|
local recved = rfd:recv()
|
||
|
|
if recved then
|
||
|
|
buf = buf .. recved
|
||
|
|
elseif recved == nil then
|
||
|
|
error('socket closed!')
|
||
|
|
end
|
||
|
|
coroutine.resume(parser)
|
||
|
|
end
|
||
|
|
::continue::
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
brave.on('timer', function (time)
|
||
|
|
local thread = require 'bee.thread'
|
||
|
|
while true do
|
||
|
|
thread.sleep(time)
|
||
|
|
brave.push('wakeup')
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
brave.on('loadFile', function (path)
|
||
|
|
local util = require 'utility'
|
||
|
|
return util.loadFile(path)
|
||
|
|
end)
|
||
|
|
|
||
|
|
brave.on('removeCaches', function (path)
|
||
|
|
local fs = require 'bee.filesystem'
|
||
|
|
local fsu = require 'fs-utility'
|
||
|
|
for dir in fs.pairs(fs.path(path)) do
|
||
|
|
local lockFile = dir / '.lock'
|
||
|
|
local f = io.open(lockFile:string(), 'wb')
|
||
|
|
if f then
|
||
|
|
f:close()
|
||
|
|
fsu.fileRemove(dir)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
---@class brave.param.compile
|
||
|
|
---@field uri uri
|
||
|
|
---@field text string
|
||
|
|
---@field mode string
|
||
|
|
---@field version string
|
||
|
|
---@field options brave.param.compile.options
|
||
|
|
|
||
|
|
---@class brave.param.compile.options
|
||
|
|
---@field special table<string, string>
|
||
|
|
---@field unicodeName boolean
|
||
|
|
---@field nonstandardSymbol table<string, true>
|
||
|
|
|
||
|
|
---@param param brave.param.compile
|
||
|
|
brave.on('compile', function (param)
|
||
|
|
local parser = require 'parser'
|
||
|
|
local clock = os.clock()
|
||
|
|
local state, err = parser.compile(param.text
|
||
|
|
, param.mode
|
||
|
|
, param.version
|
||
|
|
, param.options
|
||
|
|
)
|
||
|
|
log.debug('Async compile', param.uri, 'takes:', os.clock() - clock)
|
||
|
|
return {
|
||
|
|
state = state,
|
||
|
|
err = err,
|
||
|
|
}
|
||
|
|
end)
|