vis-config

ctags.lua

(7578B)

 local positions = {}
 local tags = { 'tags' }
 local ctags = { actions = {} }
 
 local function abs_path(prefix, path)
 	if string.find(path, '^/') ~= nil then
 		return path, path
 	end
 
 	if string.find(path, '^./') ~= nil then
 		path = path:sub(3)
 	end
 
 	return prefix .. path, path
 end
 
 local function is_directory(path)
 	local dir = io.open(path .. '/', 'r')
 	if dir then
 		dir:close()
 		return true
 	else
 		return false
 	end
 end
 
 local function find_tags(path)
 	for i = #path, 1, -1 do
 		if path:sub(i, i) == '/' then
 			local prefix = path:sub(1, i)
 			for j = 1, #tags do
 				local tagfile = tags[j]
 				local filename
 				if tagfile:sub(1, 1) == '/' then
 					filename = tagfile
 				else
 					filename = prefix .. tagfile
 				end
 				if not is_directory(filename) then
 					local file = io.open(filename, 'r')
 
 					if file ~= nil then
 						return file, prefix
 					end
 				end
 			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, _, _ = 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 abspath, name = abs_path(prefix, result.name)
 				local desc = string.format('%s%s', name, tonumber(result.excmd) and ':' .. result.excmd or '')
 
 				matches[#matches + 1] = { desc = desc, path = abspath, 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, force)
 	if pos.path ~= vis.win.file.path then
 		vis:command(string.format(force and 'e! "%s"' or '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, force)
 	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 }, force) then
 		positions[#positions + 1] = old
 		vis.registers['/'] = last_search
 	end
 end
 
 local function pop_pos(force)
 	if #positions < 1 then
 		return
 	end
 	if goto_pos(positions[#positions], force) then
 		table.remove(positions, #positions)
 	end
 end
 
 local function win_path()
 	if vis.win.file.path == nil then
 		return os.getenv('PWD') .. '/'
 	end
 	return vis.win.file.path
 end
 
 local function tag_cmd(tag, force)
 	local match = get_match(tag, win_path())
 	if match == nil then
 		vis:info(string.format('Tag not found: %s', tag))
 	else
 		goto_tag(match.path, match.excmd, force)
 	end
 end
 
 local function gen_vis_menu(matches)
 	local width = 0
 	for _, match in ipairs(matches) do
 		width = math.max(width, match.desc:len())
 	end
 	-- limit max width of desc field (filename) in menu
 	width = math.min(width, 40)
 	local fmt = '%' .. #tostring(#matches) .. 'd %-' .. width .. 's %s'
 
 	local lines = {}
 	for i, match in ipairs(matches) do
 		local desc = match.desc
 		if desc:len() > width then
 			desc = '...' .. desc:sub(desc:len() - width + 4)
 		end
 
 		-- work around bug displaying tabs in vis-menu and
 		-- provide a clearer context
 		local excmd = match.excmd:gsub('%s+', ' ')
 		excmd = excmd:gsub('^/^', '')
 		excmd = excmd:gsub('$/$', '')
 		table.insert(lines, fmt:format(i, desc, excmd))
 	end
 
 	-- limit vis-menu height to ~1/4 the window height
 	-- +1 gives an empty line at bottom to signify
 	-- that there are no more lines to scroll through
 	local nlines = math.min(math.floor(vis.win.height / 4), #lines)
 	if nlines == #lines then
 		nlines = nlines + 1
 	end
 	return 'vis-menu -l ' .. nlines .. " -p 'Choose tag:' << 'EOF'\n" .. table.concat(lines, '\n') .. '\n' .. 'EOF'
 end
 
 local function tselect_cmd(tag, force)
 	local matches = get_matches(tag, win_path())
 	if matches == nil then
 		vis:info(string.format('Tag not found: %s', tag))
 	else
 		local status, output = vis:pipe(vis.win.file, { start = 0, finish = 0 }, gen_vis_menu(matches))
 
 		if status ~= 0 then
 			vis:info('Command failed')
 			return
 		end
 
 		local choice = tonumber(string.match(output, '%d+'))
 		if choice == nil or choice < 1 or choice > #matches then
 			vis:info('Invalid choice')
 			return
 		end
 		goto_tag(matches[choice].path, matches[choice].excmd, force)
 	end
 end
 
 vis:command_register('tag', function(argv, force, win, selection, range)
 	if #argv == 1 then
 		tag_cmd(argv[1], force)
 	end
 end)
 
 vis:command_register('tselect', function(argv, force, win, selection, range)
 	if #argv == 1 then
 		tselect_cmd(argv[1], force)
 	end
 end)
 
 vis:command_register('pop', function(argv, force, win, selection, range)
 	pop_pos(force)
 end)
 
 vis:option_register('tags', 'string', function(value)
 	tags = {}
 	for str in value:gmatch('([^%s]+)') do
 		table.insert(tags, str)
 	end
 end, 'Paths to search for tags (separated by spaces)')
 
 ctags.actions.tag = function(keys)
 	local query = get_query()
 	local force = false
 	if query ~= nil then
 		tag_cmd(query, force)
 	end
 	return 0
 end
 
 ctags.actions.tselect = function(keys)
 	local query = get_query()
 	local force = false
 	if query ~= nil then
 		tselect_cmd(query, force)
 	end
 	return 0
 end
 
 ctags.actions.pop = function(keys)
 	pop_pos()
 	return 0
 end
 
 vis:map(vis.modes.NORMAL, '<C-]>', ctags.actions.tag)
 
 vis:map(vis.modes.NORMAL, 'g<C-]>', ctags.actions.tselect)
 
 vis:map(vis.modes.NORMAL, '<C-t>', ctags.actions.pop)
 
 return ctags