nixos/lua-lsp/script/core/diagnostics/lowercase-global.lua

64 lines
1.8 KiB
Lua

local files = require 'files'
local guide = require 'parser.guide'
local lang = require 'language'
local config = require 'config'
local vm = require 'vm'
local util = require 'utility'
local function isDocClass(source)
if not source.bindDocs then
return false
end
for _, doc in ipairs(source.bindDocs) do
if doc.type == 'doc.class' then
return true
end
end
return false
end
-- 不允许定义首字母小写的全局变量(很可能是拼错或者漏删)
return function (uri, callback)
local ast = files.getState(uri)
if not ast then
return
end
local definedGlobal = util.arrayToHash(config.get(uri, 'Lua.diagnostics.globals'))
guide.eachSourceType(ast.ast, 'setglobal', function (source)
local name = guide.getKeyName(source)
if not name or definedGlobal[name] then
return
end
local first = name:match '%w'
if not first then
return
end
if not first:match '%l' then
return
end
-- 如果赋值被标记为 doc.class ,则认为是允许的
if isDocClass(source) then
return
end
if definedGlobal[name] == nil then
definedGlobal[name] = false
local global = vm.getGlobal('variable', name)
if global then
for _, set in ipairs(global:getSets(uri)) do
if vm.isMetaFile(guide.getUri(set)) then
definedGlobal[name] = true
return
end
end
end
end
callback {
start = source.start,
finish = source.finish,
message = lang.script.DIAG_LOWERCASE_GLOBAL,
}
end)
end