dict/grep

lua/cmp_dictionary/dict/grep.lua

@@ -0,0 +1,40 @@
+local util = require("cmp_dictionary.util")
+
+---@class CmpDictionaryDictGrep: CmpDictionaryDict
+---@field command string[]
+---@field paths string[]
+local M = {}
+
+---@param command string[]
+---@return CmpDictionaryDictGrep
+function M.new(command)
+  return setmetatable({
+    command = command,
+    paths = {},
+  }, { __index = M })
+end
+
+---@param paths string[]
+function M:update(paths)
+  self.paths = paths
+end
+
+---@param prefix string
+function M:search(prefix)
+  local items = {}
+  for _, path in ipairs(self.paths) do
+    local command = vim.tbl_map(function(c)
+      return c:gsub("${prefix}", prefix):gsub("${path}", path)
+    end, self.command)
+    local info = string.format("belong to `%s`", vim.fn.fnamemodify(path, ":t"))
+    items = vim.list_extend(
+      items,
+      vim.tbl_map(function(word)
+        return { label = word, info = info }
+      end, vim.split(util.system(command), "\n"))
+    )
+  end
+  return items
+end
+
+return M

lua/cmp_dictionary/dict/init.lua

@@ -1,4 +1,5 @@
 local trie = require("cmp_dictionary.dict.trie")
+local grep = require("cmp_dictionary.dict.grep")
 
 local M = {}
 
@@ -9,7 +10,11 @@ local M = {}
 ---@param opts CmpDictionaryOptions
 ---@return CmpDictionaryDict
 function M.new(opts)
-  return trie.new()
+  if #opts.grep_command > 0 then
+    return grep.new(opts.grep_command)
+  else
+    return trie.new()
+  end
 end
 
 return M