diff options
| -rw-r--r-- | script/core/completion/completion.lua | 38 |
1 files changed, 37 insertions, 1 deletions
diff --git a/script/core/completion/completion.lua b/script/core/completion/completion.lua index d047dd56..cf2ec745 100644 --- a/script/core/completion/completion.lua +++ b/script/core/completion/completion.lua @@ -2256,6 +2256,37 @@ local function tryluaDocOfFunction(doc, results, pad) } end +---Checks for a lua symbol reference in comment and returns an async callback if found +local function trySymbolReference(state, position, results) + local doc = getLuaDoc(state, position) + if not doc then + return + end + + local line = doc.originalComment.text ---@type string + local col = select(2, guide.rowColOf(position)) - 2 ---@type integer + + -- User will ask for completion at end of symbol name so we need to perform a reverse match to see if they are in a symbol reference + -- Matching in reverse allows the symbol to be of any length and we can still match all the way back to `](lua://` from right to left + local symbol = string.match(string.reverse(line), "%)?(.*)//:aul%(%]", #line - col) + + if symbol then + -- flip it back the right way around + symbol = string.reverse(symbol) + + ---@async + return function () + for _, match in ipairs(wssymbol(symbol)) do + results[#results+1] = { + label = match.name, + kind = define.CompletionItemKind.Class, + insertText = match.name + } + end + end + end +end + ---@async local function tryLuaDoc(state, position, results) local doc = getLuaDoc(state, position) @@ -2327,8 +2358,13 @@ end ---@async local function tryCompletions(state, position, triggerCharacter, results) if getComment(state, position) then + local callback = trySymbolReference(state, position, results) + if callback then + callback() + else + tryComment(state, position, results) + end tryLuaDoc(state, position, results) - tryComment(state, position, results) return end if postfix(state, position, results) then |