feat: context aware completion

fix some stuff

Author: haorenW1025

lua/completion/complete.lua

@@ -31,7 +31,7 @@ M.clearCache = function()
 end
 
 -- perform completion
-M.performComplete = function(complete_source, complete_items_map, manager, bufnr, prefix, textMatch)
+M.performComplete = function(complete_source, complete_items_map, manager, opt)
 
   manager.insertChar = false
   if vim.fn.has_key(complete_source, "mode") > 0 then
@@ -50,7 +50,7 @@ M.performComplete = function(complete_source, complete_items_map, manager, bufnr
             table.insert(callback_array, true)
           else
             table.insert(callback_array, complete_items.callback)
-            complete_items.trigger(prefix, textMatch, bufnr, manager)
+            complete_items.trigger(manager, opt)
           end
           table.insert(items_array, complete_items.item)
         end
@@ -65,14 +65,14 @@ M.performComplete = function(complete_source, complete_items_map, manager, bufnr
         -- only perform complete when callback_array are all true
         if checkCallback(callback_array) == true and timer:is_closing() == false then
           if api.nvim_get_mode()['mode'] == 'i' or api.nvim_get_mode()['mode'] == 'ic' then
-            local items = getCompletionItems(items_array, prefix)
+            local items = getCompletionItems(items_array, opt.prefix)
             if vim.g.completion_sorting ~= "none" then
               util.sort_completion_items(items)
             end
             if #items ~= 0 then
               -- reset insertChar and handle auto changing source
               cache_complete_items = items
-              vim.fn.complete(textMatch+1, items)
+              vim.fn.complete(opt.textMatch+1, items)
               manager.changeSource = false
             else
               manager.changeSource = true
@@ -86,15 +86,15 @@ M.performComplete = function(complete_source, complete_items_map, manager, bufnr
       if api.nvim_get_mode()['mode'] == 'i' or api.nvim_get_mode()['mode'] == 'ic' then
         local items = {}
         for _, item in ipairs(cache_complete_items) do
-          match.matching(items, prefix, item)
+          match.matching(items, opt.prefix, item)
         end
         if vim.g.completion_sorting ~= "none" then
           util.sort_completion_items(items)
         end
         if #items ~= 0 then
           -- reset insertChar and handle auto changing source
           cache_complete_items = items
-          vim.fn.complete(textMatch+1, items)
+          vim.fn.complete(opt.textMatch+1, items)
           manager.changeSource = false
         else
           cache_complete_items = {}

lua/completion/matching.lua

@@ -52,6 +52,9 @@ M.matching = function(complete_items, prefix, item)
   for _, method in ipairs(matcher_list) do
     local is_match, score = matching_strategy[method](prefix, item.word)
     if is_match then
+      if item.abbr == nil then
+        item.abbr = item.word
+      end
       item.score = score
       if item.priority ~= nil then
         item.priority = item.priority + 10*matching_priority

lua/completion/source.lua

@@ -60,7 +60,7 @@ local getTriggerCharacter = function()
   return triggerCharacter
 end
 
-local triggerCurrentCompletion = function(manager, bufnr, line_to_cursor, prefix, textMatch, force)
+local triggerCurrentCompletion = function(manager, bufnr, line_to_cursor, prefix, textMatch, suffix, force)
   -- avoid rebundant calling of completion
   if manager.insertChar == false then return end
 
@@ -111,15 +111,16 @@ local triggerCurrentCompletion = function(manager, bufnr, line_to_cursor, prefix
     M.chain_complete_index = 1
   end
 
-  complete.performComplete(complete_source, complete_items_map, manager, bufnr, prefix, textMatch)
+  complete.performComplete(complete_source, complete_items_map, manager, {bufnr=bufnr, prefix=prefix, textMatch=textMatch, suffix=suffix, line_to_cursor=line_to_cursor})
 end
 
 local getPositionParam = function()
   local bufnr = api.nvim_get_current_buf()
   local pos = api.nvim_win_get_cursor(0)
   local line = api.nvim_get_current_line()
   local line_to_cursor = line:sub(1, pos[2])
-  return bufnr, line_to_cursor
+  local cursor_to_end = line:sub(pos[2]+1, #line)
+  return bufnr, line_to_cursor, cursor_to_end
 end
 
 ------------------------------------------------------------------------
@@ -132,19 +133,23 @@ function M.triggerCompletion(force, manager)
   if force then
     M.chain_complete_index = 1
   end
-  local bufnr, line_to_cursor = getPositionParam()
+  local bufnr, line_to_cursor, cursor_to_end = getPositionParam()
   local textMatch = vim.fn.match(line_to_cursor, '\\k*$')
   local prefix = line_to_cursor:sub(textMatch+1)
+  local rev_textMatch = #cursor_to_end - vim.fn.match(cursor_to_end:reverse(), '\\k*$')
+  local suffix = cursor_to_end:sub(1, rev_textMatch)
   manager.insertChar = true
   -- force is used when manually trigger, so it doesn't repect the trigger word length
-  triggerCurrentCompletion(manager, bufnr, line_to_cursor, prefix, textMatch, force)
+  triggerCurrentCompletion(manager, bufnr, line_to_cursor, prefix, textMatch, suffix, force)
 end
 
 -- Handler for auto completion
 function M.autoCompletion(manager)
-  local bufnr, line_to_cursor = getPositionParam()
+  local bufnr, line_to_cursor, cursor_to_end = getPositionParam()
   local textMatch = vim.fn.match(line_to_cursor, '\\k*$')
   local prefix = line_to_cursor:sub(textMatch+1)
+  local rev_textMatch = #cursor_to_end - vim.fn.match(cursor_to_end:reverse(), '\\k*$')
+  local suffix = cursor_to_end:sub(1, rev_textMatch)
   local length = vim.g.completion_trigger_keyword_length
 
   -- reset completion when deleting character in insert mode
@@ -171,7 +176,7 @@ function M.autoCompletion(manager)
   -- stop auto completion when all sources return no complete-items
   if M.stop_complete == true then return end
 
-  triggerCurrentCompletion(manager, bufnr, line_to_cursor, prefix, textMatch)
+  triggerCurrentCompletion(manager, bufnr, line_to_cursor, prefix, textMatch, suffix)
 
 end
 

lua/completion/source/lsp.lua

@@ -10,7 +10,7 @@ M.getCompletionItems = function(_, _)
   return M.items
 end
 
-local function get_completion_word(item, prefix)
+local function get_completion_word(item, prefix, suffix)
   if item.textEdit ~= nil and item.textEdit ~= vim.NIL
     and item.textEdit.newText ~= nil and (item.insertTextFormat ~= 2 or vim.fn.exists('g:loaded_vsnip_integ')) then
       local start_range = item.textEdit.range["start"]
@@ -36,7 +36,7 @@ local function get_completion_word(item, prefix)
   return item.label
 end
 
-local function text_document_completion_list_to_complete_items(result, prefix)
+local function text_document_completion_list_to_complete_items(result, opt)
   local items = vim.lsp.util.extract_completion_items(result)
   if vim.tbl_isempty(items) then
     return {}
@@ -63,19 +63,28 @@ local function text_document_completion_list_to_complete_items(result, prefix)
       end
       item.info = info
 
-      item.word = get_completion_word(completion_item, prefix)
+      item.word = get_completion_word(completion_item, opt.prefix, opt.suffix)
+      if opt.suffix ~= nil and #opt.suffix ~= 0 then
+        local index = item.word:find(opt.suffix)
+        if index ~= nil then
+          local newWord = item.word
+          newWord = newWord:sub(1, index-1)
+          item.word = newWord
+          print(item.word, index, newWord)
+        end
+      end
       item.user_data = {
         lsp = {
-          completion_item = completion_item
+          completion_item = completion_item,
         }
       }
       local kind = protocol.CompletionItemKind[completion_item.kind]
       item.kind = customize_label[kind] or kind
       item.abbr = completion_item.label
+      -- get_context_aware_snippets(item, completion_item, opt.textMatch)
       item.priority = vim.g.completion_items_priority[item.kind]
       item.menu = completion_item.detail or ''
-
-      match.matching(matches, prefix, item)
+      match.matching(matches, opt.prefix, item)
     end
   end
 
@@ -86,20 +95,20 @@ M.getCallback = function()
   return M.callback
 end
 
-M.triggerFunction = function(prefix, _, bufnr, _)
+M.triggerFunction = function(manager, opt)
   local params = vim.lsp.util.make_position_params()
   M.callback = false
   M.items = {}
   if #vim.lsp.buf_get_clients() == 0 then
     M.callback = true
     return
   end
-  vim.lsp.buf_request(bufnr, 'textDocument/completion', params, function(err, _, result)
+  vim.lsp.buf_request(opt.bufnr, 'textDocument/completion', params, function(err, _, result)
     if err or not result then
       M.callback = true
       return
     end
-    local matches = text_document_completion_list_to_complete_items(result, prefix)
+    local matches = text_document_completion_list_to_complete_items(result, opt)
     M.items = matches
     M.callback = true
   end)

lua/completion/source/path.lua

@@ -57,17 +57,14 @@ M.getCallback = function()
   return M.callback
 end
 
-M.triggerFunction = function()
-  local pos = api.nvim_win_get_cursor(0)
-  local line = api.nvim_get_current_line()
-  local line_to_cursor = line:sub(1, pos[2])
+M.triggerFunction = function(_, opt)
   local keyword
   if vim.v.completed_item ~= nil and vim.v.completed_item.kind == 'Path' and
-    line_to_cursor:find(vim.v.completed_item.word) then
+    opt.line_to_cursor:find(vim.v.completed_item.word) then
     keyword = M.keyword..vim.v.completed_item.word..'/'
   else
     M.keyword = nil
-    keyword = line_to_cursor:match("[^%s\"\']+%S*/?$")
+    keyword = opt.line_to_cursor:match("[^%s\"\']+%S*/?$")
   end
 
   if keyword ~= nil and keyword ~= '/' then