475 lines
12 KiB
Lua
475 lines
12 KiB
Lua
|
local fs = require 'bee.filesystem'
|
||
|
|
local nonil = require 'without-check-nil'
|
||
|
|
local util = require 'utility'
|
||
|
|
local lang = require 'language'
|
||
|
|
local proto = require 'proto'
|
||
|
|
local define = require 'proto.define'
|
||
|
|
local config = require 'config'
|
||
|
|
local converter = require 'proto.converter'
|
||
|
|
local json = require 'json-beautify'
|
||
|
|
local await = require 'await'
|
||
|
|
local scope = require 'workspace.scope'
|
||
|
|
local inspect = require 'inspect'
|
||
|
|
|
||
|
|
local m = {}
|
||
|
|
m._eventList = {}
|
||
|
|
|
||
|
|
function m.client(newClient)
|
||
|
|
if newClient then
|
||
|
|
m._client = newClient
|
||
|
|
else
|
||
|
|
return m._client
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.isVSCode()
|
||
|
|
if not m._client then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
if m._isvscode == nil then
|
||
|
|
local lname = m._client:lower()
|
||
|
|
if lname:find 'vscode'
|
||
|
|
or lname:find 'visual studio code' then
|
||
|
|
m._isvscode = true
|
||
|
|
else
|
||
|
|
m._isvscode = false
|
||
|
|
end
|
||
|
|
end
|
||
|
|
return m._isvscode
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.getOption(name)
|
||
|
|
nonil.enable()
|
||
|
|
local option = m.info.initializationOptions[name]
|
||
|
|
nonil.disable()
|
||
|
|
return option
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.getAbility(name)
|
||
|
|
if not m.info
|
||
|
|
or not m.info.capabilities then
|
||
|
|
return nil
|
||
|
|
end
|
||
|
|
local current = m.info.capabilities
|
||
|
|
while true do
|
||
|
|
local parent, nextPos = name:match '^([^%.]+)()'
|
||
|
|
if not parent then
|
||
|
|
break
|
||
|
|
end
|
||
|
|
current = current[parent]
|
||
|
|
if not current then
|
||
|
|
return nil
|
||
|
|
end
|
||
|
|
if nextPos > #name then
|
||
|
|
break
|
||
|
|
else
|
||
|
|
name = name:sub(nextPos + 1)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
return current
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.getOffsetEncoding()
|
||
|
|
if m._offsetEncoding then
|
||
|
|
return m._offsetEncoding
|
||
|
|
end
|
||
|
|
local clientEncodings = m.getAbility 'offsetEncoding'
|
||
|
|
if type(clientEncodings) == 'table' then
|
||
|
|
for _, encoding in ipairs(clientEncodings) do
|
||
|
|
if encoding == 'utf-8' then
|
||
|
|
m._offsetEncoding = 'utf-8'
|
||
|
|
return m._offsetEncoding
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
m._offsetEncoding = 'utf-16'
|
||
|
|
return m._offsetEncoding
|
||
|
|
end
|
||
|
|
|
||
|
|
local function packMessage(...)
|
||
|
|
local strs = table.pack(...)
|
||
|
|
for i = 1, strs.n do
|
||
|
|
strs[i] = tostring(strs[i])
|
||
|
|
end
|
||
|
|
return table.concat(strs, '\t')
|
||
|
|
end
|
||
|
|
|
||
|
|
---@alias message.type '"Error"'|'"Warning"'|'"Info"'|'"Log"'
|
||
|
|
|
||
|
|
---show message to client
|
||
|
|
---@param type message.type
|
||
|
|
function m.showMessage(type, ...)
|
||
|
|
local message = packMessage(...)
|
||
|
|
proto.notify('window/showMessage', {
|
||
|
|
type = define.MessageType[type] or 3,
|
||
|
|
message = message,
|
||
|
|
})
|
||
|
|
proto.notify('window/logMessage', {
|
||
|
|
type = define.MessageType[type] or 3,
|
||
|
|
message = message,
|
||
|
|
})
|
||
|
|
end
|
||
|
|
|
||
|
|
---@param type message.type
|
||
|
|
---@param message string
|
||
|
|
---@param titles string[]
|
||
|
|
---@param callback fun(action?: string, index?: integer)
|
||
|
|
function m.requestMessage(type, message, titles, callback)
|
||
|
|
proto.notify('window/logMessage', {
|
||
|
|
type = define.MessageType[type] or 3,
|
||
|
|
message = message,
|
||
|
|
})
|
||
|
|
local map = {}
|
||
|
|
local actions = {}
|
||
|
|
for i, title in ipairs(titles) do
|
||
|
|
actions[i] = {
|
||
|
|
title = title,
|
||
|
|
}
|
||
|
|
map[title] = i
|
||
|
|
end
|
||
|
|
proto.request('window/showMessageRequest', {
|
||
|
|
type = define.MessageType[type] or 3,
|
||
|
|
message = message,
|
||
|
|
actions = actions,
|
||
|
|
}, function (item)
|
||
|
|
if item then
|
||
|
|
callback(item.title, map[item.title])
|
||
|
|
else
|
||
|
|
callback(nil, nil)
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
end
|
||
|
|
|
||
|
|
---@param type message.type
|
||
|
|
---@param message string
|
||
|
|
---@param titles string[]
|
||
|
|
---@return string action
|
||
|
|
---@return integer index
|
||
|
|
---@async
|
||
|
|
function m.awaitRequestMessage(type, message, titles)
|
||
|
|
return await.wait(function (waker)
|
||
|
|
m.requestMessage(type, message, titles, waker)
|
||
|
|
end)
|
||
|
|
end
|
||
|
|
|
||
|
|
---@param type message.type
|
||
|
|
function m.logMessage(type, ...)
|
||
|
|
local message = packMessage(...)
|
||
|
|
proto.notify('window/logMessage', {
|
||
|
|
type = define.MessageType[type] or 4,
|
||
|
|
message = message,
|
||
|
|
})
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.watchFiles(path)
|
||
|
|
path = path:gsub('\\', '/')
|
||
|
|
:gsub('[%[%]%{%}%*%?]', '\\%1')
|
||
|
|
local registration = {
|
||
|
|
id = path,
|
||
|
|
method = 'workspace/didChangeWatchedFiles',
|
||
|
|
registerOptions = {
|
||
|
|
watchers = {
|
||
|
|
{
|
||
|
|
globPattern = path .. '/**',
|
||
|
|
kind = 1 | 2 | 4,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
proto.request('client/registerCapability', {
|
||
|
|
registrations = {
|
||
|
|
registration,
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
return function ()
|
||
|
|
local unregisteration = {
|
||
|
|
id = path,
|
||
|
|
method = 'workspace/didChangeWatchedFiles',
|
||
|
|
}
|
||
|
|
proto.request('client/registerCapability', {
|
||
|
|
unregisterations = {
|
||
|
|
unregisteration,
|
||
|
|
}
|
||
|
|
})
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
---@class config.change
|
||
|
|
---@field key string
|
||
|
|
---@field prop? string
|
||
|
|
---@field value any
|
||
|
|
---@field action '"add"'|'"set"'|'"prop"'
|
||
|
|
---@field global? boolean
|
||
|
|
---@field uri? uri
|
||
|
|
|
||
|
|
---@param cfg table
|
||
|
|
---@param uri uri
|
||
|
|
---@param changes config.change[]
|
||
|
|
---@return boolean
|
||
|
|
local function applyConfig(cfg, uri, changes)
|
||
|
|
local scp = scope.getScope(uri)
|
||
|
|
local ok = false
|
||
|
|
for _, change in ipairs(changes) do
|
||
|
|
if scp:isChildUri(change.uri)
|
||
|
|
or scp:isLinkedUri(change.uri) then
|
||
|
|
local value = config.getRaw(change.uri, change.key)
|
||
|
|
local key = change.key:match('^Lua%.(.+)$')
|
||
|
|
if cfg[key] then
|
||
|
|
cfg[key] = value
|
||
|
|
else
|
||
|
|
cfg[change.key] = value
|
||
|
|
end
|
||
|
|
ok = true
|
||
|
|
end
|
||
|
|
end
|
||
|
|
return ok
|
||
|
|
end
|
||
|
|
|
||
|
|
local function tryModifySpecifiedConfig(uri, finalChanges)
|
||
|
|
if #finalChanges == 0 then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local workspace = require 'workspace'
|
||
|
|
local scp = scope.getScope(uri)
|
||
|
|
if scp:get('lastLocalType') ~= 'json' then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local suc = applyConfig(scp:get('lastLocalConfig'), uri, finalChanges)
|
||
|
|
if not suc then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local path = workspace.getAbsolutePath(uri, CONFIGPATH)
|
||
|
|
if not path then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
util.saveFile(path, json.beautify(scp:get('lastLocalConfig'), { indent = ' ' }))
|
||
|
|
return true
|
||
|
|
end
|
||
|
|
|
||
|
|
local function tryModifyRC(uri, finalChanges, create)
|
||
|
|
if #finalChanges == 0 then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local workspace = require 'workspace'
|
||
|
|
local path = workspace.getAbsolutePath(uri, '.luarc.jsonc')
|
||
|
|
if not path then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
path = fs.exists(fs.path(path)) and path or workspace.getAbsolutePath(uri, '.luarc.json')
|
||
|
|
if not path then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local buf = util.loadFile(path)
|
||
|
|
if not buf and not create then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local loader = require 'config.loader'
|
||
|
|
local rc = loader.loadRCConfig(uri, path) or {
|
||
|
|
['$schema'] = lang.id == 'zh-cn'
|
||
|
|
and [[https://raw.githubusercontent.com/sumneko/vscode-lua/master/setting/schema-zh-cn.json]]
|
||
|
|
or [[https://raw.githubusercontent.com/sumneko/vscode-lua/master/setting/schema.json]]
|
||
|
|
}
|
||
|
|
local suc = applyConfig(rc, uri, finalChanges)
|
||
|
|
if not suc then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
util.saveFile(path, json.beautify(rc, { indent = ' ' }))
|
||
|
|
return true
|
||
|
|
end
|
||
|
|
|
||
|
|
local function tryModifyClient(uri, finalChanges)
|
||
|
|
if #finalChanges == 0 then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
if not m.getOption 'changeConfiguration' then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
local scp = scope.getScope(uri)
|
||
|
|
local scpChanges = {}
|
||
|
|
for _, change in ipairs(finalChanges) do
|
||
|
|
if change.uri
|
||
|
|
and (scp:isChildUri(change.uri) or scp:isLinkedUri(change.uri)) then
|
||
|
|
scpChanges[#scpChanges+1] = change
|
||
|
|
end
|
||
|
|
end
|
||
|
|
if #scpChanges == 0 then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
proto.notify('$/command', {
|
||
|
|
command = 'lua.config',
|
||
|
|
data = scpChanges,
|
||
|
|
})
|
||
|
|
return true
|
||
|
|
end
|
||
|
|
|
||
|
|
---@param finalChanges config.change[]
|
||
|
|
local function tryModifyClientGlobal(finalChanges)
|
||
|
|
if #finalChanges == 0 then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
if not m.getOption 'changeConfiguration' then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
local changes = {}
|
||
|
|
for i = #finalChanges, 1, -1 do
|
||
|
|
local change = finalChanges[i]
|
||
|
|
if change.global then
|
||
|
|
changes[#changes+1] = change
|
||
|
|
finalChanges[i] = finalChanges[#finalChanges]
|
||
|
|
finalChanges[#finalChanges] = nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
proto.notify('$/command', {
|
||
|
|
command = 'lua.config',
|
||
|
|
data = changes,
|
||
|
|
})
|
||
|
|
end
|
||
|
|
|
||
|
|
---@param changes config.change[]
|
||
|
|
---@param onlyMemory? boolean
|
||
|
|
function m.setConfig(changes, onlyMemory)
|
||
|
|
local finalChanges = {}
|
||
|
|
for _, change in ipairs(changes) do
|
||
|
|
if change.action == 'add' then
|
||
|
|
local suc = config.add(change.uri, change.key, change.value)
|
||
|
|
if suc then
|
||
|
|
finalChanges[#finalChanges+1] = change
|
||
|
|
end
|
||
|
|
elseif change.action == 'set' then
|
||
|
|
local suc = config.set(change.uri, change.key, change.value)
|
||
|
|
if suc then
|
||
|
|
finalChanges[#finalChanges+1] = change
|
||
|
|
end
|
||
|
|
elseif change.action == 'prop' then
|
||
|
|
local suc = config.prop(change.uri, change.key, change.prop, change.value)
|
||
|
|
if suc then
|
||
|
|
finalChanges[#finalChanges+1] = change
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
if onlyMemory then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
if #finalChanges == 0 then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
xpcall(function ()
|
||
|
|
local ws = require 'workspace'
|
||
|
|
if #ws.folders == 0 then
|
||
|
|
tryModifyClient(nil, finalChanges)
|
||
|
|
return
|
||
|
|
end
|
||
|
|
tryModifyClientGlobal(finalChanges)
|
||
|
|
for _, scp in ipairs(ws.folders) do
|
||
|
|
if tryModifySpecifiedConfig(scp.uri, finalChanges) then
|
||
|
|
goto CONTINUE
|
||
|
|
end
|
||
|
|
if tryModifyRC(scp.uri, finalChanges, false) then
|
||
|
|
goto CONTINUE
|
||
|
|
end
|
||
|
|
if tryModifyClient(scp.uri, finalChanges) then
|
||
|
|
goto CONTINUE
|
||
|
|
end
|
||
|
|
tryModifyRC(scp.uri, finalChanges, true)
|
||
|
|
::CONTINUE::
|
||
|
|
end
|
||
|
|
end, log.error)
|
||
|
|
end
|
||
|
|
|
||
|
|
---@alias textEditor {start: integer, finish: integer, text: string}
|
||
|
|
|
||
|
|
---@param uri uri
|
||
|
|
---@param edits textEditor[]
|
||
|
|
function m.editText(uri, edits)
|
||
|
|
local files = require 'files'
|
||
|
|
local state = files.getState(uri)
|
||
|
|
if not state then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
local textEdits = {}
|
||
|
|
for i, edit in ipairs(edits) do
|
||
|
|
textEdits[i] = converter.textEdit(converter.packRange(state, edit.start, edit.finish), edit.text)
|
||
|
|
end
|
||
|
|
local params = {
|
||
|
|
edit = {
|
||
|
|
changes = {
|
||
|
|
[uri] = textEdits,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
proto.request('workspace/applyEdit', params)
|
||
|
|
log.info('workspace/applyEdit', inspect(params))
|
||
|
|
end
|
||
|
|
|
||
|
|
---@alias textMultiEditor {uri: uri, start: integer, finish: integer, text: string}
|
||
|
|
|
||
|
|
---@param editors textMultiEditor[]
|
||
|
|
function m.editMultiText(editors)
|
||
|
|
local files = require 'files'
|
||
|
|
local changes = {}
|
||
|
|
for _, editor in ipairs(editors) do
|
||
|
|
local uri = editor.uri
|
||
|
|
local state = files.getState(uri)
|
||
|
|
if state then
|
||
|
|
if not changes[uri] then
|
||
|
|
changes[uri] = {}
|
||
|
|
end
|
||
|
|
local edit = converter.textEdit(converter.packRange(state, editor.start, editor.finish), editor.text)
|
||
|
|
table.insert(changes[uri], edit)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
local params = {
|
||
|
|
edit = {
|
||
|
|
changes = changes,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
proto.request('workspace/applyEdit', params)
|
||
|
|
log.info('workspace/applyEdit', inspect(params))
|
||
|
|
end
|
||
|
|
|
||
|
|
---@param callback async fun(ev: string)
|
||
|
|
function m.event(callback)
|
||
|
|
m._eventList[#m._eventList+1] = callback
|
||
|
|
end
|
||
|
|
|
||
|
|
function m._callEvent(ev)
|
||
|
|
for _, callback in ipairs(m._eventList) do
|
||
|
|
await.call(function ()
|
||
|
|
callback(ev)
|
||
|
|
end)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.setReady()
|
||
|
|
m._ready = true
|
||
|
|
m._callEvent('ready')
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.isReady()
|
||
|
|
return m._ready == true
|
||
|
|
end
|
||
|
|
|
||
|
|
local function hookPrint()
|
||
|
|
if TEST or CLI then
|
||
|
|
return
|
||
|
|
end
|
||
|
|
print = function (...)
|
||
|
|
m.logMessage('Log', ...)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
function m.init(t)
|
||
|
|
log.debug('Client init', inspect(t))
|
||
|
|
m.info = t
|
||
|
|
nonil.enable()
|
||
|
|
m.client(t.clientInfo.name)
|
||
|
|
nonil.disable()
|
||
|
|
lang(LOCALE or t.locale)
|
||
|
|
converter.setOffsetEncoding(m.getOffsetEncoding())
|
||
|
|
hookPrint()
|
||
|
|
m._callEvent('init')
|
||
|
|
end
|
||
|
|
|
||
|
|
return m
|