feat: fuzzy match

Author: haorenW1025

lua/completion/util.lua

@@ -25,35 +25,28 @@ local function get_completion_word(item)
   return item.label
 end
 
-local function remove_unmatch_completion_items(items, prefix)
-  return vim.tbl_filter(function(item)
-    local word = get_completion_word(item)
-    return vim.startswith(word, prefix)
-  end, items)
-end
-
 function M.sort_completion_items(items)
   table.sort(items, function(a, b)
     if a.priority ~= b.priority and a.priority ~= nil and b.priority ~= nil then
       return a.priority > b.priority
-    elseif vim.g.completion_sorting == 'alphabet' then
-      return a.word < b.word
     elseif a.score ~= b.score and a.score ~= nil and b.score ~= nil then
       return a.score < b.score
+    elseif vim.g.completion_sorting == 'alphabet' then
+      return a.word < b.word
     else
       return string.len(a.word) < string.len(b.word)
     end
   end)
 end
 
-function M.text_document_completion_list_to_complete_items(result, prefix)
+function M.text_document_completion_list_to_complete_items(result, prefix, score_func)
   local items = vim.lsp.util.extract_completion_items(result)
   if vim.tbl_isempty(items) then
     return {}
   end
 
   local customize_label = vim.g.completion_customize_lsp_label
-  items = remove_unmatch_completion_items(items, prefix)
+  -- items = remove_unmatch_completion_items(items, prefix)
   -- sort_completion_items(items)
 
   local matches = {}
@@ -82,18 +75,39 @@ function M.text_document_completion_list_to_complete_items(result, prefix)
       }
       local kind = protocol.CompletionItemKind[completion_item.kind]
       local priority = vim.g.completion_items_priority[kind] or 1
-      table.insert(matches, {
-        word = word,
-        abbr = completion_item.label,
-        kind = customize_label[kind] or kind or '',
-        menu = completion_item.detail or '',
-        info = info,
-        priority = priority,
-        icase = 1,
-        user_data = user_data,
-        dup = 1,
-        empty = 1,
-      })
+      if vim.g.completion_fuzzy_match == 1 then
+        local score = score_func(prefix, word)
+        if score <= 1 then
+          table.insert(matches, {
+            word = word,
+            abbr = completion_item.label,
+            kind = customize_label[kind] or kind or '',
+            menu = completion_item.detail or '',
+            info = info,
+            priority = priority,
+            score = score,
+            icase = 1,
+            user_data = user_data,
+            dup = 1,
+            empty = 1,
+          })
+        end
+      else
+        if vim.startswith(word, prefix) then
+          table.insert(matches, {
+            word = word,
+            abbr = completion_item.label,
+            kind = customize_label[kind] or kind or '',
+            menu = completion_item.detail or '',
+            info = info,
+            priority = priority,
+            icase = 1,
+            user_data = user_data,
+            dup = 1,
+            empty = 1,
+          })
+        end
+      end
     end
   end
 

lua/source/lsp.lua

@@ -27,7 +27,7 @@ M.triggerFunction = function(prefix, _, bufnr, _)
       return
     end
     if api.nvim_get_mode()['mode'] == 'i' or api.nvim_get_mode()['mode'] == 'ic' then
-      local matches = util.text_document_completion_list_to_complete_items(result, prefix)
+      local matches = util.text_document_completion_list_to_complete_items(result, prefix, util.fuzzy_score)
       M.items = matches
     end
     M.callback = true

plugin/completion.vim

@@ -69,14 +69,14 @@ if ! exists('g:completion_sorting')
     let g:completion_sorting = 'alphabet'
 endif
 
-if ! exists('g:completion_sorted_alphabetically')
-    let g:completion_sorted_alphabetically = v:false
-endif
-
 if ! exists('g:completion_max_items')
     let g:completion_max_items = v:null
 endif
 
+if ! exists('g:completion_fuzzy_match')
+    let g:completion_fuzzy_match = 1
+endif
+
 if ! exists('g:completion_chain_complete_list')
     let g:completion_chain_complete_list = {
                 \ 'default' : {