|
|
|
|
@@ -1,75 +1,81 @@
|
|
|
|
|
require('vis')
|
|
|
|
|
|
|
|
|
|
local lpeg = vis.lpeg
|
|
|
|
|
|
|
|
|
|
local P = lpeg.P
|
|
|
|
|
local Cp = lpeg.Cp
|
|
|
|
|
local Ct = lpeg.Ct
|
|
|
|
|
P = lpeg.P
|
|
|
|
|
R = lpeg.R
|
|
|
|
|
S = lpeg.S
|
|
|
|
|
Cp = lpeg.Cp
|
|
|
|
|
Ct = lpeg.Ct
|
|
|
|
|
space = S(' \t\n\r')
|
|
|
|
|
alnum = R('az', 'AZ', '09')
|
|
|
|
|
|
|
|
|
|
local function search(p)
|
|
|
|
|
return Ct(((1 - p)^0 * Cp() * p)^0)
|
|
|
|
|
return Ct(((1 - p)^0 * Cp() * p * Cp())^0)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
local up = {
|
|
|
|
|
'a', 's', 'd', 'f' ,
|
|
|
|
|
'q', 'w', 'e', 'r', 't', 'g',
|
|
|
|
|
'z', 'x', 'c', 'v', 'b',
|
|
|
|
|
'1', '2', '3', '4', '5',
|
|
|
|
|
'A', 'S', 'D', 'F' ,
|
|
|
|
|
'Q', 'W', 'E', 'R', 'T', 'G',
|
|
|
|
|
'Z', 'X', 'C', 'V', 'B',
|
|
|
|
|
'!', '@', '#', '$', '%',
|
|
|
|
|
'`', '~',
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
local down = {
|
|
|
|
|
'j', 'k', 'l', ';',
|
|
|
|
|
'y', 'u', 'i', 'o', 'p',
|
|
|
|
|
'n', 'm', ',', '.', '/',
|
|
|
|
|
'6', '7', '8', '9', '0',
|
|
|
|
|
'J', 'K', 'L', ':',
|
|
|
|
|
'Y', 'U', 'I', 'O', 'P',
|
|
|
|
|
'N', 'M', '<', '>', '?',
|
|
|
|
|
'^', '&', '*', '(', ')',
|
|
|
|
|
"'", '"', '\\', '|',
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
local jumps = {}
|
|
|
|
|
|
|
|
|
|
local function jump(keys)
|
|
|
|
|
vis:map(vis.modes.NORMAL, '\\', function(keys)
|
|
|
|
|
if #keys > 1 then
|
|
|
|
|
vis.win:draw()
|
|
|
|
|
end
|
|
|
|
|
local _, esc = keys:find('<Escape>')
|
|
|
|
|
if esc then
|
|
|
|
|
return esc
|
|
|
|
|
end
|
|
|
|
|
local ret = #keys
|
|
|
|
|
keys = keys:gsub('<Enter>', '\n')
|
|
|
|
|
if #keys < 2 then
|
|
|
|
|
local _, enter = keys:find('<Enter>')
|
|
|
|
|
if enter then
|
|
|
|
|
keys = keys:gsub('<Enter>', '')
|
|
|
|
|
end
|
|
|
|
|
keys = keys:gsub('<Tab>', 'J')
|
|
|
|
|
keys = keys:gsub('<S[-]Tab>', 'K')
|
|
|
|
|
vis:info(keys)
|
|
|
|
|
local n = 0
|
|
|
|
|
keys = keys:gsub('[KJ]', function(s)
|
|
|
|
|
if s == 'K' then
|
|
|
|
|
n = n - 1
|
|
|
|
|
elseif s == 'J' then
|
|
|
|
|
n = n + 1
|
|
|
|
|
end
|
|
|
|
|
return ''
|
|
|
|
|
end)
|
|
|
|
|
keys = keys:gsub('.<Backspace>', '')
|
|
|
|
|
if #keys < 1 then
|
|
|
|
|
return -1
|
|
|
|
|
end
|
|
|
|
|
if #keys > 2 then
|
|
|
|
|
vis:info('not found')
|
|
|
|
|
return ret
|
|
|
|
|
end
|
|
|
|
|
local up = n < 0
|
|
|
|
|
local v = vis.win.viewport.bytes
|
|
|
|
|
local data = vis.win.file:content(v)
|
|
|
|
|
|
|
|
|
|
local p = search(P(keys))
|
|
|
|
|
local p = P(false)
|
|
|
|
|
local prefix = nil
|
|
|
|
|
for i = 1, #keys do
|
|
|
|
|
local x = keys:sub(i, i)
|
|
|
|
|
if x == 'A' then
|
|
|
|
|
prefix = alnum
|
|
|
|
|
elseif x == 'F' then
|
|
|
|
|
prefix = P(1) - space
|
|
|
|
|
else
|
|
|
|
|
local y = S(x:lower() .. x:upper())
|
|
|
|
|
if i == 1 then
|
|
|
|
|
p = y
|
|
|
|
|
elseif prefix then
|
|
|
|
|
p = p * (prefix - y)^0 * y
|
|
|
|
|
elseif x == ' ' then
|
|
|
|
|
p = p * space^1
|
|
|
|
|
else
|
|
|
|
|
p = p * y
|
|
|
|
|
end
|
|
|
|
|
prefix = false
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
p = search(p)
|
|
|
|
|
local lst = p:match(data)
|
|
|
|
|
if not lst or #lst == 0 then
|
|
|
|
|
if not lst then
|
|
|
|
|
vis:info('not found')
|
|
|
|
|
return ret
|
|
|
|
|
return
|
|
|
|
|
end
|
|
|
|
|
if #lst == 1 then
|
|
|
|
|
vis.win.selection.pos = v.start + lst[1] - 1
|
|
|
|
|
return ret
|
|
|
|
|
end
|
|
|
|
|
local pos = vis.win.selection.pos - v.start
|
|
|
|
|
local pos = vis.win.selection.pos
|
|
|
|
|
local prev = 0
|
|
|
|
|
local next = #lst + 1
|
|
|
|
|
local t = {}
|
|
|
|
|
jumps = {}
|
|
|
|
|
for i = 1, #lst do
|
|
|
|
|
local a = lst[i]
|
|
|
|
|
local next = 0
|
|
|
|
|
for i = 2, #lst, 2 do
|
|
|
|
|
local a = lst[i-1] + v.start - 1
|
|
|
|
|
if a < pos then
|
|
|
|
|
prev = i
|
|
|
|
|
end
|
|
|
|
|
@@ -78,53 +84,27 @@ local function jump(keys)
|
|
|
|
|
break
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
local j = 1
|
|
|
|
|
for i = prev, 1, -1 do
|
|
|
|
|
if not up[j] then
|
|
|
|
|
break
|
|
|
|
|
end
|
|
|
|
|
t[i] = up[j]
|
|
|
|
|
jumps[t[i]] = v.start + lst[i] - 1
|
|
|
|
|
j = j + 1
|
|
|
|
|
end
|
|
|
|
|
j = 1
|
|
|
|
|
for i = next, #lst do
|
|
|
|
|
if not down[j] then
|
|
|
|
|
break
|
|
|
|
|
end
|
|
|
|
|
t[i] = down[j]
|
|
|
|
|
jumps[t[i]] = v.start + lst[i] - 1
|
|
|
|
|
j = j + 1
|
|
|
|
|
end
|
|
|
|
|
local input = ''
|
|
|
|
|
local k = 1
|
|
|
|
|
for i = 1, #lst do
|
|
|
|
|
input = input .. data:sub(k, lst[i] - 1)
|
|
|
|
|
if t[i] then
|
|
|
|
|
input = input .. '\x1b[1;37;44m' .. keys .. '\x1b[42m' .. t[i] .. '\x1b[0m'
|
|
|
|
|
k = lst[i] + 3
|
|
|
|
|
if data:sub(k - 1, k - 1) == '\n' then
|
|
|
|
|
k = k - 1
|
|
|
|
|
end
|
|
|
|
|
else
|
|
|
|
|
input = input .. '\x1b[1;37;44m' .. keys .. '\x1b[0m'
|
|
|
|
|
k = lst[i] + 2
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
input = input .. data:sub(k)
|
|
|
|
|
|
|
|
|
|
local code, out, err = vis:pipe(input, 'fzf --ansi --layout=reverse-list --no-info --no-separator --color gutter:-1 --marker="" --padding 0,0,0,3 --print-query --bind change:accept')
|
|
|
|
|
if code ~= 0 then
|
|
|
|
|
vis:info(err or ('fzf exit code ' .. code))
|
|
|
|
|
local j = 0
|
|
|
|
|
if up then
|
|
|
|
|
j = prev + 1 + 2 * n
|
|
|
|
|
else
|
|
|
|
|
local p = jumps[out:sub(1, 1)]
|
|
|
|
|
if p then
|
|
|
|
|
vis.win.selection.pos = p
|
|
|
|
|
j = next - 1 + 2 * n
|
|
|
|
|
end
|
|
|
|
|
if j > 0 and j < #lst then
|
|
|
|
|
pos = lst[j] + v.start - 1
|
|
|
|
|
end
|
|
|
|
|
if enter then
|
|
|
|
|
vis.win.selection.pos = pos
|
|
|
|
|
return enter
|
|
|
|
|
end
|
|
|
|
|
for i = 2, #lst, 2 do
|
|
|
|
|
local a = lst[i-1] + v.start - 1
|
|
|
|
|
local b = lst[i] + v.start - 2
|
|
|
|
|
if a == pos then
|
|
|
|
|
vis.win:style(vis.win.STYLE_COLOR_COLUMN, a, b)
|
|
|
|
|
else
|
|
|
|
|
vis:info('not found')
|
|
|
|
|
vis.win:style(vis.win.STYLE_SELECTION, a, b)
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
return ret
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
vis:map(vis.modes.NORMAL, '\\', jump)
|
|
|
|
|
return -1
|
|
|
|
|
end, 'colorize token')
|
|
|
|
|
|