.

files/.config/vis/plugins/vis-sneak/init.lua

@@ -0,0 +1,83 @@
+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)