mirror of
https://github.com/kristoferssolo/telescope-frecency.nvim.git
synced 2025-10-21 20:10:38 +00:00
59 lines
1.3 KiB
Lua
59 lines
1.3 KiB
Lua
local sorters = require "telescope.sorters"
|
|
local util = require("telescope._extensions.frecency.util")
|
|
|
|
local my_sorters = {}
|
|
|
|
-- local function split(s, delimiter)
|
|
-- local result = {}
|
|
-- for match in (s .. delimiter):gmatch("(.-)" .. delimiter) do
|
|
-- table.insert(result, match)
|
|
-- end
|
|
-- return result
|
|
-- end
|
|
|
|
local substr_highlighter = function(_, prompt, display)
|
|
local highlights = {}
|
|
display = display:lower()
|
|
|
|
local search_terms = util.split(prompt, " ")
|
|
local hl_start, hl_end
|
|
|
|
for _, word in pairs(search_terms) do
|
|
hl_start, hl_end = display:find(word, 1, true)
|
|
if hl_start then
|
|
table.insert(highlights, {start = hl_start, finish = hl_end})
|
|
end
|
|
end
|
|
|
|
return highlights
|
|
end
|
|
|
|
my_sorters.get_substr_matcher = function(opts)
|
|
opts = opts or {}
|
|
|
|
local substr = sorters:new()
|
|
substr.highlighter = substr_highlighter
|
|
substr.scoring_function = function(_, prompt, _, entry)
|
|
local display = entry.name:lower()
|
|
|
|
local search_terms = util.split(prompt, " ")
|
|
local matched
|
|
for _, word in pairs(search_terms) do
|
|
matched = display:find(word, 1, true) and 1 or -1
|
|
if matched == -1 then goto continue end
|
|
end
|
|
|
|
::continue::
|
|
|
|
if matched == -1 then
|
|
return -1
|
|
else
|
|
return entry.index
|
|
end
|
|
end
|
|
|
|
return substr
|
|
end
|
|
|
|
return my_sorters
|