local core = require 'core.completion' local files = require 'files' local define = require 'proto.define' local config = require 'config' local EXISTS = {'EXISTS'} local function eq(a, b) if a == EXISTS and b ~= nil then return true end local tp1, tp2 = type(a), type(b) if tp1 ~= tp2 then return false end if tp1 == 'table' then local mark = {} for k in pairs(a) do if not eq(a[k], b[k]) then return false end mark[k] = true end for k in pairs(b) do if not mark[k] then return false end end return true end return a == b end local function include(a, b) if a == EXISTS and b ~= nil then return true end local tp1, tp2 = type(a), type(b) if tp1 ~= tp2 then return false end if tp1 == 'table' then for k in pairs(a) do if not eq(a[k], b[k]) then return false end end return true end return a == b end rawset(_G, 'TEST', true) local Cared = { ['label'] = true, ['kind'] = true, ['textEdit'] = true, ['additionalTextEdits'] = true, ['deprecated'] = true, } function TEST(script) return function (expect) files.removeAll() local pos = script:find('$', 1, true) - 1 local new_script = script:gsub('%$', '') files.setText('', new_script) core.clearCache() local result = core.completion('', pos) if not expect then assert(result == nil) return end assert(result ~= nil) result.enableCommon = nil for _, item in ipairs(result) do local r = core.resolve(item.id) for k, v in pairs(r or {}) do item[k] = v end for k in pairs(item) do if not Cared[k] then item[k] = nil end end end assert(result) if expect.include then expect.include = nil assert(include(expect, result)) else assert(eq(expect, result)) end end end config.config.completion.callSnippet = 'Both' config.config.completion.keywordSnippet = 'Both' config.config.completion.workspaceWord = false TEST [[ local zabcde za$ ]] { { label = 'zabcde', kind = define.CompletionItemKind.Variable, } } TEST [[ local zabcdefg local zabcde zabcde$ ]] { { label = 'zabcdefg', kind = define.CompletionItemKind.Variable, }, { label = 'zabcde', kind = define.CompletionItemKind.Variable, }, } TEST [[ local zabcdefg za$ local zabcde ]] { { label = 'zabcdefg', kind = define.CompletionItemKind.Variable, }, { label = 'zabcde', kind = define.CompletionItemKind.Text, }, } TEST [[ local zabcde zace$ ]] { { label = 'zabcde', kind = define.CompletionItemKind.Variable, } } TEST [[ ZABC = x local zabc zac$ ]] { { label = 'zabc', kind = define.CompletionItemKind.Variable, }, { label = 'ZABC', kind = define.CompletionItemKind.Field, }, } TEST [[ ass$ ]] { { label = 'assert', kind = define.CompletionItemKind.Function, }, { label = 'assert()', kind = define.CompletionItemKind.Snippet, }, } TEST [[ local assert = 1 ass$ ]] { { label = 'assert', kind = define.CompletionItemKind.Variable, }, } TEST [[ local assert = 1 _G.ass$ ]] { { label = 'assert', kind = define.CompletionItemKind.Function, }, { label = 'assert()', kind = define.CompletionItemKind.Snippet, }, } TEST [[ local function ffff(a, b) end ff$ ]] { { label = 'ffff', kind = define.CompletionItemKind.Function, }, { label = 'ffff()', kind = define.CompletionItemKind.Snippet, } } TEST [[ local zabc = 1 z$ ]] { { label = 'zabc', kind = define.CompletionItemKind.Variable, } } TEST [[ local zabc = 1.0 z$ ]] { { label = 'zabc', kind = define.CompletionItemKind.Variable, } } TEST [[ local t = { abc = 1, } t.ab$ ]] { { label = 'abc', kind = define.CompletionItemKind.Enum, } } TEST [[ local t = { abc = 1, } local n = t.abc t.ab$ ]] { { label = 'abc', kind = define.CompletionItemKind.Enum, } } TEST [[ local mt = {} mt.ggg = 1 function mt:get(a, b) return 1 end mt:g$ ]] { { label = 'get', kind = define.CompletionItemKind.Method, }, { label = 'get()', kind = define.CompletionItemKind.Snippet, }, { label = 'ggg', kind = define.CompletionItemKind.Text, } } TEST [[ loc$ ]] { { label = 'local', kind = define.CompletionItemKind.Keyword, }, { label = 'local function', kind = define.CompletionItemKind.Snippet, }, { label = 'collectgarbage', kind = define.CompletionItemKind.Function, }, { label = 'collectgarbage()', kind = define.CompletionItemKind.Snippet, }, } TEST [[ do$ ]] { { label = 'do', kind = define.CompletionItemKind.Keyword, }, { label = 'do .. end', kind = define.CompletionItemKind.Snippet, }, { label = 'dofile', kind = define.CompletionItemKind.Function, }, { label = 'dofile()', kind = define.CompletionItemKind.Snippet, }, { label = 'load', kind = define.CompletionItemKind.Function, }, { label = 'load()', kind = define.CompletionItemKind.Snippet, }, { label = 'loadfile', kind = define.CompletionItemKind.Function, }, { label = 'loadfile()', kind = define.CompletionItemKind.Snippet, }, { label = 'loadstring', kind = define.CompletionItemKind.Function, deprecated = true, }, { label = 'loadstring()', kind = define.CompletionItemKind.Snippet, deprecated = true, }, { label = 'module', kind = define.CompletionItemKind.Function, deprecated = true, }, { label = 'module()', kind = define.CompletionItemKind.Snippet, deprecated = true, }, } TEST [[ while true d$ ]] { { label = 'do', kind = define.CompletionItemKind.Keyword, }, { label = 'do .. end', kind = define.CompletionItemKind.Snippet, }, } TEST [[ results$ ]] (nil) TEST [[ result$ local results ]] (EXISTS) TEST [[ local a$ local function f(fff) fff = ast fff = ast fff = ast fff = ast fff = ast fff = ast end ]] { { label = 'ast', kind = define.CompletionItemKind.Variable, } } TEST [[ t.a = {} t.b = {} t.$ ]] { { label = 'a', kind = define.CompletionItemKind.Field, }, { label = 'b', kind = define.CompletionItemKind.Field, }, } TEST [[ t.a = {} t.b = {} t. $ ]] { { label = 'a', kind = define.CompletionItemKind.Field, }, { label = 'b', kind = define.CompletionItemKind.Field, }, } TEST [[ t.a = {} function t:b() end t:$ ]] { { label = 'b', kind = define.CompletionItemKind.Method, }, { label = 'b()', kind = define.CompletionItemKind.Snippet, }, } TEST [[ local t = { a = {}, } t.$ xxx() ]] { { label = 'a', kind = define.CompletionItemKind.Field, }, } TEST [[ (''):$ ]] (EXISTS) TEST [[ local zzz return 'aa' .. zz$ ]] { { label = 'zzz', kind = define.CompletionItemKind.Variable, }, } TEST 'local s = "a:$"' (nil) TEST 'debug.$' (EXISTS) TEST [[ local xxxx = { xxyy = 1, xxzz = 2, } local t = { x$ } ]] { { label = 'xxxx', kind = define.CompletionItemKind.Variable, }, { label = 'xxyy', kind = define.CompletionItemKind.Property, }, { label = 'xxzz', kind = define.CompletionItemKind.Property, }, { label = 'next', kind = define.CompletionItemKind.Function, }, { label = 'next()', kind = define.CompletionItemKind.Snippet, }, { label = 'xpcall', kind = define.CompletionItemKind.Function, }, { label = 'xpcall()', kind = define.CompletionItemKind.Snippet, }, } TEST [[ print(ff2) local faa local f$ print(fff) ]] { { label = 'function', kind = define.CompletionItemKind.Keyword, }, { label = 'function ()', kind = define.CompletionItemKind.Snippet, }, { label = 'fff', kind = define.CompletionItemKind.Variable, }, { label = 'ff2', kind = define.CompletionItemKind.Text, }, { label = 'faa', kind = define.CompletionItemKind.Text, }, } TEST [[ local function f(ff$) print(fff) end ]] { { label = 'fff', kind = define.CompletionItemKind.Variable, }, } TEST [[ collectgarbage($) ]] (EXISTS) TEST [[ collectgarbage('$') ]] { { label = "'collect'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'collect'", }, }, { label = "'stop'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'stop'", }, }, { label = "'restart'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'restart'", }, }, { label = "'count'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'count'", }, }, { label = "'step'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'step'", }, }, { label = "'isrunning'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'isrunning'", }, }, { label = "'incremental'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'incremental'", }, }, { label = "'generational'", kind = define.CompletionItemKind.EnumMember, textEdit = { start = 16, finish = 17, newText = "'generational'", }, }, } TEST [[ io.read($) ]] { { label = '"n"', kind = define.CompletionItemKind.EnumMember, }, { label = '"a"', kind = define.CompletionItemKind.EnumMember, }, { label = '"l"', kind = define.CompletionItemKind.EnumMember, }, { label = '"L"', kind = define.CompletionItemKind.EnumMember, }, } TEST [[ io.open('', $) ]] (EXISTS) TEST [[ local function f(a, $) end ]] (nil) TEST [[ self.results.list[#$] ]] { { label = '#self.results.list+1', kind = define.CompletionItemKind.Snippet, textEdit = { start = 19, finish = 20, newText = '#self.results.list+1] = ', }, }, } TEST [[ self.results.list[#$] local n = 1 ]] { { label = '#self.results.list+1', kind = define.CompletionItemKind.Snippet, textEdit = { start = 19, finish = 20, newText = '#self.results.list+1] = ', }, }, } TEST [[ self.results.list[#$] = 1 ]] { { label = '#self.results.list+1', kind = define.CompletionItemKind.Snippet, textEdit = { start = 19, finish = 20, newText = '#self.results.list+1]', }, }, } TEST [[ self.results.list[#self.re$] ]] { { label = '#self.results.list+1', kind = define.CompletionItemKind.Snippet, textEdit = { start = 19, finish = 27, newText = '#self.results.list+1] = ', }, }, { label = 'results', kind = define.CompletionItemKind.Field, }, } TEST [[ fff[#ff$] ]] { { label = '#fff+1', kind = define.CompletionItemKind.Snippet, textEdit = { start = 5, finish = 8, newText = '#fff+1] = ', }, }, { label = 'fff', kind = define.CompletionItemKind.Text, } } TEST [[ local _ = fff.kkk[#$] ]] { { label = '#fff.kkk', kind = define.CompletionItemKind.Snippet, textEdit = { start = 19, finish = 20, newText = '#fff.kkk]', }, }, } TEST [[ fff.kkk[#$].yy ]] { { label = '#fff.kkk', kind = define.CompletionItemKind.Snippet, textEdit = { start = 9, finish = 10, newText = '#fff.kkk]', }, }, } TEST [[ local t = { a = 1, } t . $ ]] (EXISTS) TEST [[ local t = { a = 1, } t . $ b ]] (EXISTS) TEST [[ local t = { a = 1, } t $ ]] (nil) TEST [[ local t = { a = 1, } t $. ]] (nil) TEST [[ local xxxx xxxx$ ]] { { label = 'xxxx', kind = define.CompletionItemKind.Variable, }, } TEST [[ local xxxx local XXXX xxxx$ ]] { { label = 'xxxx', kind = define.CompletionItemKind.Variable, }, { label = 'XXXX', kind = define.CompletionItemKind.Variable, }, } TEST [[ local t = { xxxxx = 1, } xx$ ]] { { label = 'xxxxx', kind = define.CompletionItemKind.Text, }, } TEST [[ local index tbl[inde$] ]] { { label = 'index', kind = define.CompletionItemKind.Variable, }, } TEST [[ return function () local t = { a = {}, b = {}, } t.$ end ]] { { label = 'a', kind = define.CompletionItemKind.Field, }, { label = 'b', kind = define.CompletionItemKind.Field, }, } TEST [[ local ast = 1 local t = 'as$' local ask = 1 ]] (EXISTS) TEST [[ local add function f(ad$) local _ = add end ]] { { label = 'add', kind = define.CompletionItemKind.Variable, }, } TEST [[ function table.i$ ]] (EXISTS) TEST [[ do xx.$ end ]] (nil) TEST [[ print(io.$) ]] (EXISTS) require 'config' .config.runtime.version = 'Lua 5.4' --TEST [[ --local $ --]] --{ -- { -- label = '', -- kind = define.CompletionItemKind.Keyword, -- }, -- { -- label = '', -- kind = define.CompletionItemKind.Keyword, -- }, --} -- --TEST [[ --local local x x.a = $ ]] { { label = "'a'", kind = define.CompletionItemKind.EnumMember, }, { label = "'b'", kind = define.CompletionItemKind.EnumMember, }, { label = "'c'", kind = define.CompletionItemKind.EnumMember, }, } TEST [[ ---@type table local x x['a'] = $ ]] { { label = "'a'", kind = define.CompletionItemKind.EnumMember, }, { label = "'b'", kind = define.CompletionItemKind.EnumMember, }, { label = "'c'", kind = define.CompletionItemKind.EnumMember, }, } TEST [[ ---@type table local x = { a = $ } ]] { { label = "'a'", kind = define.CompletionItemKind.EnumMember, }, { label = "'b'", kind = define.CompletionItemKind.EnumMember, }, { label = "'c'", kind = define.CompletionItemKind.EnumMember, }, } TEST [[ ---@type table local x = { ['a'] = $ } ]] { { label = "'a'", kind = define.CompletionItemKind.EnumMember, }, { label = "'b'", kind = define.CompletionItemKind.EnumMember, }, { label = "'c'", kind = define.CompletionItemKind.EnumMember, }, } Cared['insertText'] = true TEST [[ ---@class A.B.C local m function m.f() end m.f$ ]]{ { label = "f", kind = define.CompletionItemKind.Function, }, { label = "f()", kind = define.CompletionItemKind.Snippet, insertText = 'f()', }, } Cared['insertText'] = nil TEST [[ if true then$ ]] { { label = 'then', kind = define.CompletionItemKind.Keyword, }, { label = 'then .. end', kind = define.CompletionItemKind.Snippet, } } TEST [[ if true then$ end ]] { { label = 'then', kind = define.CompletionItemKind.Keyword, }, } TEST [[ if true then$ else ]] { { label = 'then', kind = define.CompletionItemKind.Keyword, }, } TEST [[ if true then$ elseif ]] { { label = 'then', kind = define.CompletionItemKind.Keyword, }, } TEST [[ do if true then$ end ]] { { label = 'then', kind = define.CompletionItemKind.Keyword, }, { label = 'then .. end', kind = define.CompletionItemKind.Snippet, } } TEST [[ ---@class C ---@field x number ---@field y number ---@vararg C local function f(x, ...) end f(1, { $ }) ]] { { label = 'x', kind = define.CompletionItemKind.Property, }, { label = 'y', kind = define.CompletionItemKind.Property, } } TEST [[ ---@class C ---@field x number ---@field y number ---@vararg C local function f(x, ...) end f(1, {}, {}, { $ }) ]] { { label = 'x', kind = define.CompletionItemKind.Property, }, { label = 'y', kind = define.CompletionItemKind.Property, } } TEST [[ ---@class C ---@field x number ---@field y number ---@type C local t = { $ } ]] { { label = 'x', kind = define.CompletionItemKind.Property, }, { label = 'y', kind = define.CompletionItemKind.Property, } } TEST [[ ---@class C ---@field x number ---@field y number ---@type C local t = { x$ } ]] { include = true, { label = 'x', kind = define.CompletionItemKind.Property, }, } TEST [[ if $ then ]] (nil) TEST [[ elseif $ then ]] (nil)