nixos/lua-lsp/script/vm/doc.lua

417 lines
11 KiB
Lua
Raw Normal View History

local files = require 'files'
local guide = require 'parser.guide'
---@class vm
local vm = require 'vm.vm'
local config = require 'config'
---获取class与alias
---@param suri uri
---@param name? string
---@return parser.object[]
function vm.getDocSets(suri, name)
if name then
local global = vm.getGlobal('type', name)
if not global then
return {}
end
return global:getSets(suri)
else
return vm.getGlobalSets(suri, 'type')
end
end
---@param uri uri
---@return boolean
function vm.isMetaFile(uri)
local status = files.getState(uri)
if not status then
return false
end
local cache = files.getCache(uri)
if not cache then
return false
end
if cache.isMeta ~= nil then
return cache.isMeta
end
cache.isMeta = false
if not status.ast.docs then
return false
end
for _, doc in ipairs(status.ast.docs) do
if doc.type == 'doc.meta' then
cache.isMeta = true
return true
end
end
return false
end
---@param doc parser.object
---@return table<string, boolean>?
function vm.getValidVersions(doc)
if doc.type ~= 'doc.version' then
return
end
if doc._validVersions then
return doc._validVersions
end
local valids = {
['Lua 5.1'] = false,
['Lua 5.2'] = false,
['Lua 5.3'] = false,
['Lua 5.4'] = false,
['LuaJIT'] = false,
}
for _, version in ipairs(doc.versions) do
if version.ge and type(version.version) == 'number' then
for ver in pairs(valids) do
local verNumber = tonumber(ver:sub(-3))
if verNumber and verNumber >= version.version then
valids[ver] = true
end
end
elseif version.le and type(version.version) == 'number' then
for ver in pairs(valids) do
local verNumber = tonumber(ver:sub(-3))
if verNumber and verNumber <= version.version then
valids[ver] = true
end
end
elseif type(version.version) == 'number' then
valids[('Lua %.1f'):format(version.version)] = true
elseif 'JIT' == version.version then
valids['LuaJIT'] = true
end
end
if valids['Lua 5.1'] then
valids['LuaJIT'] = true
end
doc._validVersions = valids
return valids
end
---@param value parser.object
---@return parser.object?
local function getDeprecated(value)
if not value.bindDocs then
return nil
end
if value._deprecated ~= nil then
return value._deprecated or nil
end
for _, doc in ipairs(value.bindDocs) do
if doc.type == 'doc.deprecated' then
value._deprecated = doc
return doc
elseif doc.type == 'doc.version' then
local valids = vm.getValidVersions(doc)
if valids and not valids[config.get(guide.getUri(value), 'Lua.runtime.version')] then
value._deprecated = doc
return doc
end
end
end
if value.type == 'function' then
local doc = getDeprecated(value.parent)
if doc then
value._deprecated = doc
return doc
end
end
value._deprecated = false
return nil
end
---@param value parser.object
---@param deep boolean?
---@return parser.object?
function vm.getDeprecated(value, deep)
if deep then
local defs = vm.getDefs(value)
if #defs == 0 then
return nil
end
local deprecated
for _, def in ipairs(defs) do
if def.type == 'setglobal'
or def.type == 'setfield'
or def.type == 'setmethod'
or def.type == 'setindex'
or def.type == 'tablefield'
or def.type == 'tableindex' then
deprecated = getDeprecated(def)
if not deprecated then
return nil
end
end
end
return deprecated
else
return getDeprecated(value)
end
end
---@param value parser.object
---@return boolean
local function isAsync(value)
if value.type == 'function' then
if not value.bindDocs then
return false
end
if value._async ~= nil then
return value._async
end
for _, doc in ipairs(value.bindDocs) do
if doc.type == 'doc.async' then
value._async = true
return true
end
end
value._async = false
return false
end
if value.type == 'main' then
return true
end
return value.async == true
end
---@param value parser.object
---@param deep boolean?
---@return boolean
function vm.isAsync(value, deep)
if isAsync(value) then
return true
end
if deep then
local defs = vm.getDefs(value)
if #defs == 0 then
return false
end
for _, def in ipairs(defs) do
if isAsync(def) then
return true
end
end
end
return false
end
---@param value parser.object
---@return boolean
local function isNoDiscard(value)
if value.type == 'function' then
if not value.bindDocs then
return false
end
if value._nodiscard ~= nil then
return value._nodiscard
end
for _, doc in ipairs(value.bindDocs) do
if doc.type == 'doc.nodiscard' then
value._nodiscard = true
return true
end
end
value._nodiscard = false
return false
end
return false
end
---@param value parser.object
---@param deep boolean?
---@return boolean
function vm.isNoDiscard(value, deep)
if isNoDiscard(value) then
return true
end
if deep then
local defs = vm.getDefs(value)
if #defs == 0 then
return false
end
for _, def in ipairs(defs) do
if isNoDiscard(def) then
return true
end
end
end
return false
end
---@param param parser.object
---@return boolean
local function isCalledInFunction(param)
if not param.ref then
return false
end
local func = guide.getParentFunction(param)
for _, ref in ipairs(param.ref) do
if ref.type == 'getlocal' then
if ref.parent.type == 'call'
and guide.getParentFunction(ref) == func then
return true
end
if ref.parent.type == 'callargs'
and ref.parent[1] == ref
and guide.getParentFunction(ref) == func then
if ref.parent.parent.node.special == 'pcall'
or ref.parent.parent.node.special == 'xpcall' then
return true
end
end
end
end
return false
end
---@param node parser.object
---@param index integer
---@return boolean
local function isLinkedCall(node, index)
for _, def in ipairs(vm.getDefs(node)) do
if def.type == 'function' then
local param = def.args and def.args[index]
if param then
if isCalledInFunction(param) then
return true
end
end
end
end
return false
end
---@param node parser.object
---@param index integer
---@return boolean
function vm.isLinkedCall(node, index)
return isLinkedCall(node, index)
end
---@param call parser.object
---@return boolean
function vm.isAsyncCall(call)
if vm.isAsync(call.node, true) then
return true
end
if not call.args then
return false
end
for i, arg in ipairs(call.args) do
if vm.isAsync(arg, true)
and isLinkedCall(call.node, i) then
return true
end
end
return false
end
---@param uri uri
---@param doc parser.object
---@param results table[]
local function makeDiagRange(uri, doc, results)
local names
if doc.names then
names = {}
for i, nameUnit in ipairs(doc.names) do
local name = nameUnit[1]
names[name] = true
end
end
local row = guide.rowColOf(doc.start)
if doc.mode == 'disable-next-line' then
results[#results+1] = {
mode = 'disable',
names = names,
row = row + 1,
source = doc,
}
results[#results+1] = {
mode = 'enable',
names = names,
row = row + 2,
source = doc,
}
elseif doc.mode == 'disable-line' then
results[#results+1] = {
mode = 'disable',
names = names,
row = row,
source = doc,
}
results[#results+1] = {
mode = 'enable',
names = names,
row = row + 1,
source = doc,
}
elseif doc.mode == 'disable' then
results[#results+1] = {
mode = 'disable',
names = names,
row = row + 1,
source = doc,
}
elseif doc.mode == 'enable' then
results[#results+1] = {
mode = 'enable',
names = names,
row = row + 1,
source = doc,
}
end
end
---@param uri uri
---@param position integer
---@param name string
---@param err? boolean
---@return boolean
function vm.isDiagDisabledAt(uri, position, name, err)
local status = files.getState(uri)
if not status then
return false
end
if not status.ast.docs then
return false
end
local cache = files.getCache(uri)
if not cache then
return false
end
if not cache.diagnosticRanges then
cache.diagnosticRanges = {}
for _, doc in ipairs(status.ast.docs) do
if doc.type == 'doc.diagnostic' then
makeDiagRange(uri, doc, cache.diagnosticRanges)
end
end
table.sort(cache.diagnosticRanges, function (a, b)
return a.row < b.row
end)
end
if #cache.diagnosticRanges == 0 then
return false
end
local myRow = guide.rowColOf(position)
local count = 0
for _, range in ipairs(cache.diagnosticRanges) do
if range.row <= myRow then
if (range.names and range.names[name])
or (not range.names and not err) then
if range.mode == 'disable' then
count = count + 1
elseif range.mode == 'enable' then
count = count - 1
end
end
else
break
end
end
return count > 0
end