136 lines
3.7 KiB
Lua
136 lines
3.7 KiB
Lua
local config = require 'config'
|
|
local util = require 'utility'
|
|
local client = require 'client'
|
|
local lang = require 'language'
|
|
local await = require 'await'
|
|
local scope = require 'workspace.scope'
|
|
local ws = require 'workspace'
|
|
local fs = require 'bee.filesystem'
|
|
|
|
---@class plugin
|
|
local m = {}
|
|
|
|
function m.showError(scp, err)
|
|
if m._hasShowedError then
|
|
return
|
|
end
|
|
m._hasShowedError = true
|
|
client.showMessage('Error', lang.script('PLUGIN_RUNTIME_ERROR', scp:get('pluginPath'), err))
|
|
end
|
|
|
|
function m.dispatch(event, uri, ...)
|
|
local scp = scope.getScope(uri)
|
|
local interface = scp:get('pluginInterface')
|
|
if not interface then
|
|
return false
|
|
end
|
|
local method = interface[event]
|
|
if type(method) ~= 'function' then
|
|
return false
|
|
end
|
|
local clock = os.clock()
|
|
tracy.ZoneBeginN('plugin dispatch:' .. event)
|
|
local suc, res1, res2 = xpcall(method, log.error, uri, ...)
|
|
tracy.ZoneEnd()
|
|
local passed = os.clock() - clock
|
|
if passed > 0.1 then
|
|
log.warn(('Call plugin event [%s] takes [%.3f] sec'):format(event, passed))
|
|
end
|
|
if suc then
|
|
return true, res1, res2
|
|
else
|
|
m.showError(scp, res1)
|
|
end
|
|
return false, res1
|
|
end
|
|
|
|
---@async
|
|
---@param scp scope
|
|
local function checkTrustLoad(scp)
|
|
local pluginPath = scp:get('pluginPath')
|
|
local filePath = LOGPATH .. '/trusted'
|
|
local trusted = util.loadFile(filePath)
|
|
local lines = {}
|
|
if trusted then
|
|
for line in util.eachLine(trusted) do
|
|
lines[#lines+1] = line
|
|
if line == pluginPath then
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
local _, index = client.awaitRequestMessage('Warning', lang.script('PLUGIN_TRUST_LOAD', pluginPath), {
|
|
lang.script('PLUGIN_TRUST_YES'),
|
|
lang.script('PLUGIN_TRUST_NO'),
|
|
})
|
|
if not index then
|
|
return false
|
|
end
|
|
lines[#lines+1] = pluginPath
|
|
util.saveFile(filePath, table.concat(lines, '\n'))
|
|
return true
|
|
end
|
|
|
|
---@param uri uri
|
|
local function initPlugin(uri)
|
|
await.call(function () ---@async
|
|
local scp = scope.getScope(uri)
|
|
local interface = {}
|
|
scp:set('pluginInterface', interface)
|
|
|
|
if not scp.uri then
|
|
return
|
|
end
|
|
|
|
local pluginPath = ws.getAbsolutePath(scp.uri, config.get(scp.uri, 'Lua.runtime.plugin'))
|
|
log.info('plugin path:', pluginPath)
|
|
if not pluginPath then
|
|
return
|
|
end
|
|
|
|
--Adding the plugins path to package.path allows for requires in files
|
|
--to find files relative to itself.
|
|
local oldPath = package.path
|
|
local path = fs.path(pluginPath):parent_path() / '?.lua'
|
|
if not package.path:find(path:string(), 1, true) then
|
|
package.path = package.path .. ';' .. path:string()
|
|
end
|
|
|
|
local pluginLua = util.loadFile(pluginPath)
|
|
if not pluginLua then
|
|
log.warn('plugin not found:', pluginPath)
|
|
package.path = oldPath
|
|
return
|
|
end
|
|
|
|
scp:set('pluginPath', pluginPath)
|
|
|
|
local env = setmetatable(interface, { __index = _ENV })
|
|
local f, err = load(pluginLua, '@'..pluginPath, "t", env)
|
|
if not f then
|
|
log.error(err)
|
|
m.showError(scp, err)
|
|
return
|
|
end
|
|
if not client.isVSCode() and not checkTrustLoad(scp) then
|
|
return
|
|
end
|
|
local pluginArgs = config.get(scp.uri, 'Lua.runtime.pluginArgs')
|
|
local suc, err = xpcall(f, log.error, f, uri, pluginArgs)
|
|
if not suc then
|
|
m.showError(scp, err)
|
|
return
|
|
end
|
|
|
|
ws.resetFiles(scp)
|
|
end)
|
|
end
|
|
|
|
ws.watch(function (ev, uri)
|
|
if ev == 'startReload' then
|
|
initPlugin(uri)
|
|
end
|
|
end)
|
|
|
|
return m
|