nixos/lua-lsp/script/plugin.lua

136 lines
3.7 KiB
Lua
Raw Normal View History

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