83 lines
2 KiB
Lua
83 lines
2 KiB
Lua
local pattern = nil
|
|
local matches = {}
|
|
|
|
-- iterater for doing find in pattern until nil
|
|
local pattern_iterator = function(content, pattern)
|
|
local offset = 1
|
|
return function()
|
|
local starts, ends = string.find(content, pattern, offset)
|
|
if starts == nil then return nil end
|
|
offset = ends + 1
|
|
return starts, ends
|
|
end
|
|
end
|
|
|
|
-- is cursor pos is in one of the matches?
|
|
local cursor_on_match = function(win)
|
|
local pos = win.selection.pos
|
|
for _, range in ipairs(matches) do
|
|
if pos >= range.start and pos <= range.finish then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- check if range is in viewport range
|
|
local range_in_viewport = function(viewport, range)
|
|
return range.start >= viewport.bytes.start and range.finish <= viewport.bytes.finish
|
|
end
|
|
|
|
-- highlihght current matches
|
|
local highlight = function(win)
|
|
|
|
-- clear matches if cursor is not on a match
|
|
if not cursor_on_match(win) then
|
|
matches = {}
|
|
return
|
|
end
|
|
|
|
-- style matches in viewport
|
|
for _, range in ipairs(matches) do
|
|
if range_in_viewport(win.viewport, range) then
|
|
win:style(win.STYLE_CURSOR, range.start, range.finish)
|
|
end
|
|
end
|
|
end
|
|
|
|
-- collect matches (ranges) for pattern in file
|
|
local collect_matches = function()
|
|
local file = vis.win.file
|
|
local content = file:content(0, file.size)
|
|
matches = {}
|
|
for starts, ends in pattern_iterator(content, pattern) do
|
|
table.insert(matches, { start = starts - 1, finish = ends - 1 })
|
|
end
|
|
end
|
|
|
|
-- highlight matches on WIN_HIGHLIGHT
|
|
vis.events.subscribe(vis.events.WIN_HIGHLIGHT, function(win)
|
|
highlight(win)
|
|
end)
|
|
|
|
-- create search for two chars and collect matches for highlighting
|
|
local sneak = function(keys, search_char)
|
|
if #keys < 2 then
|
|
pattern = nil
|
|
return -1
|
|
end
|
|
vis:feedkeys(search_char .. keys .. '<Enter>')
|
|
pattern = keys
|
|
collect_matches()
|
|
return 2
|
|
end
|
|
|
|
-- sneak forward on 's'
|
|
vis:map(vis.modes.NORMAL, 's', function(keys)
|
|
return sneak(keys, '/')
|
|
end)
|
|
|
|
-- sneak backwards on 'S'
|
|
vis:map(vis.modes.NORMAL, 'S', function(keys)
|
|
return sneak(keys, '?')
|
|
end)
|