diff options
Diffstat (limited to 'script/method/textDocument/semanticTokens/full.lua')
| -rw-r--r-- | script/method/textDocument/semanticTokens/full.lua | 206 |
1 files changed, 206 insertions, 0 deletions
diff --git a/script/method/textDocument/semanticTokens/full.lua b/script/method/textDocument/semanticTokens/full.lua new file mode 100644 index 00000000..bc6d2bcb --- /dev/null +++ b/script/method/textDocument/semanticTokens/full.lua @@ -0,0 +1,206 @@ +local TokenTypes = require 'constant.TokenTypes' +local TokenModifiers = require 'constant.TokenModifiers' +local findLib = require 'core.find_lib' + +local timerCache = {} +local constLib = { + ['_G'] = true, + ['_VERSION'] = true, + ['math.pi'] = true, + ['math.huge'] = true, + ['math.maxinteger'] = true, + ['math.mininteger'] = true, + ['utf8.charpattern'] = true, + ['io.stdin'] = true, + ['io.stdout'] = true, + ['io.stderr'] = true, + ['package.config'] = true, + ['package.cpath'] = true, + ['package.loaded'] = true, + ['package.loaders'] = true, + ['package.path'] = true, + ['package.preload'] = true, + ['package.searchers'] = true +} + +local Care = { + ['name'] = function (source, sources) + if source[1] == '' then + return + end + if source:get 'global' then + if findLib(source) then + if source[1] == '_G' then + return + end + sources[#sources+1] = { + start = source.start, + finish = source.finish, + type = TokenTypes.namespace, + modifieres = TokenModifiers.static, + } + return + end + sources[#sources+1] = { + start = source.start, + finish = source.finish, + type = TokenTypes.namespace, + modifieres = TokenModifiers.deprecated, + } + elseif source:get 'table index' then + sources[#sources+1] = { + start = source.start, + finish = source.finish, + type = TokenTypes.property, + modifieres = TokenModifiers.declaration, + } + elseif source:bindLocal() then + if source:get 'arg' then + sources[#sources+1] = { + start = source.start, + finish = source.finish, + type = TokenTypes.parameter, + modifieres = TokenModifiers.declaration, + } + end + if source[1] == '_ENV' + or source[1] == 'self' then + return + end + local value = source:bindValue() + local func = value:getFunction() + if func and func:getSource().name == source then + sources[#sources+1] = { + start = source.start, + finish = source.finish, + type = TokenTypes.interface, + modifieres = TokenModifiers.declaration, + } + return + end + sources[#sources+1] = { + start = source.start, + finish = source.finish, + type = TokenTypes.variable, + } + end + end, +} + +local function buildTokens(sources, lines) + local tokens = {} + local lastLine = 0 + local lastStartChar = 0 + for i, source in ipairs(sources) do + local row, col = lines:rowcol(source.start) + local line = row - 1 + local startChar = col - 1 + local deltaLine = line - lastLine + local deltaStartChar + if deltaLine == 0 then + deltaStartChar = startChar - lastStartChar + else + deltaStartChar = startChar + end + lastLine = line + lastStartChar = startChar + local len = i * 5 - 5 + tokens[len + 1] = deltaLine + tokens[len + 2] = deltaStartChar + tokens[len + 3] = source.finish - source.start + 1 -- length + tokens[len + 4] = source.type + tokens[len + 5] = source.modifieres or 0 + end + return tokens +end + +local function resolveTokens(vm, lines) + local sources = {} + for _, source in ipairs(vm.sources) do + if Care[source.type] then + Care[source.type](source, sources) + end + end + + -- 先进行排序 + table.sort(sources, function (a, b) + return a.start < b.start + end) + + local tokens = buildTokens(sources, lines) + + return tokens +end + +local function toArray(map) + local array = {} + for k in pairs(map) do + array[#array+1] = k + end + table.sort(array, function (a, b) + return map[a] < map[b] + end) + return array +end + +local function testTokens(vm, lines) + local text = vm.text + local sources = {} + local init = 1 + while true do + local start, finish = text:find('[%w_%.]+', init) + if not start then + break + end + init = finish + 1 + local token = text:sub(start, finish) + local type = token:match '[%w_]+' + local mod = token:match '%.([%w_]+)' + sources[#sources+1] = { + start = start, + finish = finish, + type = TokenTypes[type], + modifieres = TokenModifiers[mod] or 0, + } + end + local tokens = buildTokens(sources, lines) + log.debug(table.dump(sources)) + log.debug(table.dump(tokens)) + return tokens +end + +--- @param lsp LSP +--- @param params table +--- @return function +return function (lsp, params) + local uri = params.textDocument.uri + + if timerCache[uri] then + timerCache[uri]:remove() + timerCache[uri] = nil + end + + return function (response) + local clock = os.clock() + timerCache[uri] = ac.loop(0.1, function (t) + local vm, lines = lsp:getVM(uri) + if not vm then + if os.clock() - clock > 10 then + t:remove() + timerCache[uri] = nil + response(nil) + end + return + end + + t:remove() + timerCache[uri] = nil + + local tokens = resolveTokens(vm, lines) + --local tokens = testTokens(vm, lines) + response { + data = tokens, + } + end) + end +end |