commit 71458d9766e5811351be474773a943b4666b0a03
parent 6d5267e532317de70384c645157ba4ed0feba35e
Author: Daniel Moch <daniel@danielmoch.com>
Date: Sat, 14 Dec 2019 07:49:35 -0500
Vis: Add ctags plugin
Diffstat:
2 files changed, 274 insertions(+), 0 deletions(-)
diff --git a/.config/vis/ctags.lua b/.config/vis/ctags.lua
@@ -0,0 +1,273 @@
+-- Code taken from https://github.com/kupospelov/vis-ctags
+-- MIT license
+require('vis')
+
+local positions = {}
+
+local function get_path(prefix, path)
+ if string.find(path, '^./') ~= nil then
+ path = path:sub(3)
+ end
+
+ return prefix .. path, path
+end
+
+local function find_tags(path)
+ for i = #path, 1, -1 do
+ if path:sub(i, i) == '/' then
+ local prefix = path:sub(1, i)
+ local filename = prefix .. 'tags'
+ local file = io.open(filename, 'r')
+
+ if file ~= nil then
+ return file, prefix
+ end
+ end
+ end
+end
+
+local function bsearch(file, word)
+ local buffer_size = 8096
+ local format = '\n(.-)\t(.-)\t(.-);\"\t'
+
+ local from = 0
+ local to = file:seek('end')
+ local startpos = nil
+
+ while from <= to do
+ local mid = from + math.floor((to - from) / 2)
+ file:seek('set', mid)
+
+ local content = file:read(buffer_size, '*line')
+ if content ~= nil then
+ local key, filename, excmd = string.match(content, format)
+ if key == nil then
+ break
+ end
+
+ if key == word then
+ startpos = mid
+ end
+
+ if key >= word then
+ to = mid - 1
+ else
+ from = mid + 1
+ end
+ else
+ to = mid - 1
+ end
+ end
+
+ if startpos ~= nil then
+ file:seek('set', startpos)
+
+ local result = {}
+ while true do
+ local content = file:read(buffer_size, '*line')
+ if content == nil then
+ break
+ end
+
+ for key, filename, excmd in string.gmatch(content, format) do
+ if key == word then
+ result[#result + 1] = {name = filename, excmd = excmd}
+ else
+ return result
+ end
+ end
+ end
+
+ return result
+ end
+end
+
+local function get_query()
+ local line = vis.win.selection.line
+ local pos = vis.win.selection.col
+ local str = vis.win.file.lines[line]
+
+ local from, to = 0, 0
+ while pos > to do
+ from, to = str:find('[%a_]+[%a%d_]*', to + 1)
+ if from == nil or from > pos then
+ return nil
+ end
+ end
+
+ return string.sub(str, from, to)
+end
+
+local function get_matches(word, path)
+ local file, prefix = find_tags(path)
+
+ if file ~= nil then
+ local results = bsearch(file, word)
+ file:close()
+
+ if results ~= nil then
+ local matches = {}
+ for i = 1, #results do
+ local result = results[i]
+ local path, name = get_path(prefix, result.name)
+ local desc = string.format('%s%s', name, tonumber(result.excmd) and ":"..result.excmd or "")
+
+ matches[#matches + 1] = {desc = desc, path = path, excmd = result.excmd}
+ end
+
+ return matches
+ end
+ end
+end
+
+local function get_match(word, path)
+ local matches = get_matches(word, path)
+ if matches ~= nil then
+ for i = 1, #matches do
+ if matches[i].path == path then
+ return matches[i]
+ end
+ end
+
+ return matches[1]
+ end
+end
+
+local function escape(text)
+ return text:gsub("[][)(}{|+?*.]", "\\%0")
+ :gsub("%^", "\\^"):gsub("^/\\%^", "/^")
+ :gsub("%$", "\\$"):gsub("\\%$/$", "$/")
+ :gsub("\\\\%$%$/$", "\\$$")
+end
+
+--[[
+- Can't test vis:command() as it will still return true if the edit command fails.
+- Can't test File.modified as the edit command can succeed if the current file is
+ modified but open in another window and this behavior is useful.
+- Instead just check the path again after trying the edit command.
+]]
+local function goto_pos(pos)
+ if pos.path ~= vis.win.file.path then
+ vis:command(string.format('e %s', pos.path))
+ if pos.path ~= vis.win.file.path then
+ return false
+ end
+ end
+ if tonumber(pos.excmd) then
+ vis.win.selection:to(pos.excmd, pos.col)
+ else
+ vis.win.selection:to(1, 1)
+ vis:command(escape(pos.excmd))
+ vis.win.selection.pos = vis.win.selection.range.start
+ vis.mode = vis.modes.NORMAL
+ end
+ return true
+end
+
+local function goto_tag(path, excmd)
+ local old = {
+ path = vis.win.file.path,
+ excmd = vis.win.selection.line,
+ col = vis.win.selection.col,
+ }
+
+ local last_search = vis.registers['/']
+ if goto_pos({ path = path, excmd = excmd, col = 1 }) then
+ positions[#positions + 1] = old
+ vis.registers['/'] = last_search
+ end
+end
+
+local function pop_pos()
+ if #positions < 1 then
+ return
+ end
+ if goto_pos(positions[#positions]) then
+ table.remove(positions, #positions)
+ end
+end
+
+local function get_path()
+ if vis.win.file.path == nil then
+ return os.getenv('PWD') .. '/'
+ end
+ return vis.win.file.path
+end
+
+local function tag_cmd(tag)
+ local match = get_match(tag, get_path())
+ if match == nil then
+ vis:info(string.format('Tag not found: %s', tag))
+ else
+ goto_tag(match.path, match.excmd)
+ end
+end
+
+local function tselect_cmd(tag)
+ local matches = get_matches(tag, get_path())
+ if matches == nil then
+ vis:info(string.format('Tag not found: %s', tag))
+ else
+ local keys = {}
+ for i = 1, #matches do
+ table.insert(keys, matches[i].desc)
+ end
+
+ local command = string.format(
+ [[echo -e "%s" | vis-menu -p "Choose tag:"]], table.concat(keys, [[\n]]))
+
+ local status, output =
+ vis:pipe(vis.win.file, {start = 0, finish = 0}, command)
+
+ if status ~= 0 then
+ vis:info('Command failed')
+ return
+ end
+
+ local choice = string.match(output, '(.*)\n')
+ for i = 1, #matches do
+ local match = matches[i]
+ if match.desc == choice then
+ goto_tag(match.path, match.excmd)
+ break
+ end
+ end
+ end
+end
+
+vis:command_register("tag", function(argv, force, win, selection, range)
+ if #argv == 1 then
+ tag_cmd(argv[1])
+ end
+end)
+
+vis:command_register("tselect", function(argv, force, win, selection, range)
+ if #argv == 1 then
+ tselect_cmd(argv[1])
+ end
+end)
+
+vis:command_register("pop", function(argv, force, win, selection, range)
+ pop_pos()
+end)
+
+vis:map(vis.modes.NORMAL, '<C-]>', function(keys)
+ local query = get_query()
+ if query ~= nil then
+ tag_cmd(query)
+ end
+ return 0
+end)
+
+vis:map(vis.modes.NORMAL, 'g<C-]>', function(keys)
+ local query = get_query()
+ if query ~= nil then
+ tselect_cmd(query)
+ end
+ return 0
+end)
+
+vis:map(vis.modes.NORMAL, '<C-t>', function(keys)
+ pop_pos()
+ return 0
+end)
diff --git a/.config/vis/visrc.lua b/.config/vis/visrc.lua
@@ -1,5 +1,6 @@
-- load standard vis module, providing parts of the Lua API
require('vis')
+require('ctags')
vis.events.subscribe(vis.events.INIT, function()
-- Your global configuration options
