Go back to searchingfor treesitter

Signed-off-by: Tomas Slusny <[email protected]>

Author: deathbeam

lua/autocomplete/buffer.lua

@@ -37,14 +37,19 @@ local function complete_treesitter(bufnr, prefix, cmp_start)
     local defs = locals.get_definitions_lookup_table(bufnr)
     local ft = vim.bo[bufnr].filetype
     local items = {}
-    local kind_map = vim.tbl_map(string.lower, vim.tbl_keys(vim.lsp.protocol.CompletionItemKind))
 
     for id, entry in pairs(defs) do
         local name = id:match('k_(.+)_%d+_%d+_%d+_%d+$')
         local node = entry.node
         local kind = entry.kind
         if node and kind then
-            kind = kind_map[kind:lower()] or kind
+            for _, k in ipairs(vim.lsp.protocol.CompletionItemKind) do
+                if string.find(k:lower(), kind:lower()) then
+                    kind = k
+                    break
+                end
+            end
+
             local start_line_node, _, _ = node:start()
             local end_line_node, _, _ = node:end_()