local core = require 'core.completion' local files = require 'files' local CompletionItemKind = require 'define.CompletionItemKind' 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 rawset(_G, 'TEST', 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) local result = core.completion('', pos) for _, item in ipairs(result) do for k in pairs(item) do if k ~= 'label' and k ~= 'kind' then item[k] = nil end end end if expect then assert(result) assert(eq(expect, result)) else assert(result == nil) end end end TEST [[ local zabcde za$ ]] { { label = 'zabcde', kind = CompletionItemKind.Variable, } } TEST [[ local zabcdefg local zabcde zabcde$ ]] { { label = 'zabcdefg', kind = CompletionItemKind.Variable, }, { label = 'zabcde', kind = CompletionItemKind.Variable, }, } TEST [[ local zabcdefg za$ local zabcde ]] { { label = 'zabcdefg', kind = CompletionItemKind.Variable, }, { label = 'zabcde', kind = CompletionItemKind.Text, }, } TEST [[ local zabcde zace$ ]] { { label = 'zabcde', kind = CompletionItemKind.Variable, } } TEST [[ ZABC local zabc zac$ ]] { { label = 'zabc', kind = CompletionItemKind.Variable, }, { label = 'ZABC', kind = CompletionItemKind.Field, }, } TEST [[ ass$ ]] { { label = 'assert', kind = CompletionItemKind.Function, }, { label = 'assert()', kind = CompletionItemKind.Snippet, }, } TEST [[ local function ffff(a, b) end ff$ ]] { { label = 'ffff', kind = CompletionItemKind.Function, }, { label = 'ffff()', kind = CompletionItemKind.Snippet, } } TEST [[ local zabc = 1 z$ ]] { { label = 'zabc', kind = CompletionItemKind.Variable, } } TEST [[ local zabc = 1.0 z$ ]] { { label = 'zabc', kind = CompletionItemKind.Variable, } } TEST [[ local t = { abc = 1, } t.ab$ ]] { { label = 'abc', kind = CompletionItemKind.Enum, } } TEST [[ local mt = {} mt.ggg = 1 function mt:get(a, b) return 1 end mt:g$ ]] { { label = 'get', kind = CompletionItemKind.Method, }, { label = 'get()', kind = CompletionItemKind.Snippet, }, { label = 'ggg', kind = CompletionItemKind.Text, } } TEST [[ loc$ ]] { { label = 'collectgarbage', kind = CompletionItemKind.Function, }, { label = 'collectgarbage()', kind = CompletionItemKind.Snippet, }, { label = 'local', kind = CompletionItemKind.Keyword, }, { label = 'local function', kind = CompletionItemKind.Snippet, } } TEST [[ do$ ]] { { label = 'dofile', kind = CompletionItemKind.Function, }, { label = 'dofile()', kind = CompletionItemKind.Snippet, }, { label = 'load', kind = CompletionItemKind.Function, }, { label = 'load()', kind = CompletionItemKind.Snippet, }, { label = 'loadfile', kind = CompletionItemKind.Function, }, { label = 'loadfile()', kind = CompletionItemKind.Snippet, }, { label = 'do', kind = CompletionItemKind.Keyword, }, { label = 'do .. end', kind = CompletionItemKind.Snippet, } } TEST [[ while true d$ ]] { { label = 'do', kind = CompletionItemKind.Keyword, }, } TEST [[ t.a = {} t.b = {} t.$ ]] { { label = 'a', kind = CompletionItemKind.Field, detail = EXISTS, }, { label = 'b', kind = CompletionItemKind.Field, detail = EXISTS, }, } TEST [[ t.a = {} t.b = {} t. $ ]] { { label = 'a', kind = CompletionItemKind.Field, detail = EXISTS, }, { label = 'b', kind = CompletionItemKind.Field, detail = EXISTS, }, } TEST [[ t.a = {} function t:b() end t:$ ]] { { label = 'b', kind = CompletionItemKind.Method, documentation = EXISTS, detail = EXISTS, }, { label = 'b()', kind = CompletionItemKind.Snippet, documentation = EXISTS, detail = EXISTS, insertText = EXISTS, }, } TEST [[ local t = { a = {}, } t.$ xxx() ]] { { label = 'a', kind = CompletionItemKind.Field, detail = EXISTS, }, { label = 'xxx', kind = CompletionItemKind.Function, documentation = EXISTS, detail = EXISTS, }, { label = 'xxx()', kind = CompletionItemKind.Snippet, documentation = EXISTS, detail = EXISTS, insertText = EXISTS, }, } TEST [[ (''):$ ]] (EXISTS) TEST 'local s = "a:$"' (nil) TEST 'debug.$' (EXISTS) TEST [[ local xxxx = { xxyy = 1, xxzz = 2, } local t = { x$ } ]] { { label = 'xxxx', kind = CompletionItemKind.Variable, detail = EXISTS, }, { label = 'xxyy', kind = CompletionItemKind.Property, }, { label = 'xxzz', kind = CompletionItemKind.Property, }, { label = 'next', kind = CompletionItemKind.Function, documentation = EXISTS, detail = EXISTS, }, { label = 'next()', kind = CompletionItemKind.Snippet, documentation = EXISTS, detail = EXISTS, insertText = EXISTS, }, { label = 'xpcall', kind = CompletionItemKind.Function, documentation = EXISTS, detail = EXISTS, }, { label = 'xpcall()', kind = CompletionItemKind.Snippet, documentation = EXISTS, detail = EXISTS, insertText = EXISTS, }, } TEST [[ print(ff2) local faa local f$ print(fff) ]] { { label = 'fff', kind = CompletionItemKind.Variable, }, { label = 'function', kind = CompletionItemKind.Keyword, }, { label = 'function name()', kind = CompletionItemKind.Snippet, insertText = EXISTS, }, { label = 'ff2', kind = CompletionItemKind.Text, }, { label = 'faa', kind = CompletionItemKind.Text, }, } TEST [[ local function f(ff$) print(fff) end ]] { { label = 'fff', kind = CompletionItemKind.Variable, }, } TEST [[ collectgarbage('$') ]] { { label = 'collect', filterText = 'collect', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'stop', filterText = 'stop', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'restart', filterText = 'restart', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'count', filterText = 'count', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'step', filterText = 'step', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'setpause', filterText = 'setpause', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'setstepmul', filterText = 'setstepmul', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, { label = 'isrunning', filterText = 'isrunning', kind = CompletionItemKind.EnumMember, documentation = EXISTS, textEdit = EXISTS, additionalTextEdits = EXISTS, }, } TEST [[ collectgarbage($) ]] (EXISTS) TEST [[ io.read($) ]] { { label = '"n"', kind = CompletionItemKind.EnumMember, documentation = EXISTS, }, { label = '"a"', kind = CompletionItemKind.EnumMember, documentation = EXISTS, }, { label = '"l"', kind = CompletionItemKind.EnumMember, documentation = EXISTS, }, { label = '"L"', kind = CompletionItemKind.EnumMember, documentation = EXISTS, }, } TEST [[ local function f(a, $) end ]] (nil) TEST [[ self.results.list[#$] ]] { { label = 'self.results.list+1', kind = CompletionItemKind.Snippet, textEdit = { start = 20, finish = 20, newText = 'self.results.list+1] = ', }, }, } TEST [[ self.results.list[#self.re$] ]] { { label = 'self.results.list+1', kind = CompletionItemKind.Snippet, textEdit = { start = 20, finish = 27, newText = 'self.results.list+1] = ', }, }, { label = 'results', kind = CompletionItemKind.Field, detail = EXISTS, }, } TEST [[ fff[#ff$] ]] { { label = 'fff+1', kind = CompletionItemKind.Snippet, textEdit = { start = 6, finish = 8, newText = 'fff+1] = ', }, }, { label = 'fff', kind = CompletionItemKind.Field, detail = EXISTS, } } TEST [[ local _ = fff.kkk[#$] ]] { { label = 'fff.kkk', kind = CompletionItemKind.Snippet, textEdit = { start = 20, finish = 20, 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 = CompletionItemKind.Variable, }, } TEST [[ local xxxx local XXXX xxxx$ ]] { { label = 'xxxx', kind = CompletionItemKind.Variable, }, { label = 'XXXX', kind = CompletionItemKind.Variable, }, } TEST [[ local t = { xxxxx = 1, } xx$ ]] { { label = 'xxxxx', kind = CompletionItemKind.Text, }, } TEST [[ local index tbl[ind$] ]] { { label = 'index', kind = CompletionItemKind.Variable, }, } TEST [[ return function () local t = { a = {}, b = {}, } t.$ end ]] { { label = 'a', kind = CompletionItemKind.Field, detail = EXISTS, }, { label = 'b', kind = CompletionItemKind.Field, detail = EXISTS, }, } TEST [[ local ast = 1 local t = 'as$' local ask = 1 ]] (nil) TEST [[ local add function f(a$) local _ = add end ]] { { label = 'add', kind = CompletionItemKind.Variable, }, } TEST [[ function table.i$ ]] (EXISTS) TEST [[ do xx.$ end ]] (nil) require 'config' .config.runtime.version = 'Lua 5.4' --TEST [[ --local $ --]] --{ -- { -- label = '', -- kind = CompletionItemKind.Keyword, -- }, -- { -- label = '', -- kind = CompletionItemKind.Keyword, -- }, --} -- --TEST [[ --local string ``` JustTest ```lua ``` ]] }, }, { label = 'zzzzz()', kind = CompletionItemKind.Snippet, detail = '(function)(4 prototypes)', insertText = EXISTS, documentation = EXISTS, } } TEST [[ --- abc zzz = 1 zz$ ]] { { label = 'zzz', kind = CompletionItemKind.Enum, detail = '(number) = 1', documentation = { kind = 'markdown', value = 'abc', } } } TEST [[ ---@param x string ---| "'选项1'" # 注释1 ---| "'选项2'" # 注释2 function f(x) end f($) ]] { { label = "'选项1'", kind = CompletionItemKind.EnumMember, documentation = '注释1', }, { label = "'选项2'", kind = CompletionItemKind.EnumMember, documentation = '注释2', }, }