aboutsummaryrefslogtreecommitdiffstats
path: root/init.lua
blob: 9494575e23d7d3ca2a6adeddd4eb0d02ad490a92 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
local M = {}
local cursors = {}

local get_default_cache_path = function()
	local HOME = os.getenv('HOME')
	local XDG_CACHE_HOME = os.getenv('XDG_CACHE_HOME')
	local BASE = XDG_CACHE_HOME or HOME
	return BASE .. '/.vis-cursors.csv'
end

M.path = get_default_cache_path()

local apply_cursor_pos = function(win)
	if win.file == nil or win.file.path == nil then
		return
	end
	local pos = cursors[win.file.path]
	if pos == nil then
		return
	end
	win.selection.pos = tonumber(pos)
	vis:feedkeys("zz")
end

local file_exists = function(path)
	local f = io.open(path)
	if f == nil then
		return false
	end
	f:close()
	return true
end

local read_cursors = function()
	cursors = {}
	local f = io.open(M.path)
	if f == nil then
		return
	end
	-- read positions per file path
	local prev_dir
	for line in f:lines() do
		for path, pos in string.gmatch(line, '(.+)[,%s](%d+)') do
			-- append prev dir if '@' (compressed)
			local repeat_dir = string.match(path, '^%@')
			if repeat_dir then
				local filename = string.match(path, '^.*/(.*)')
				path = prev_dir .. filename
			else
				local dir = string.match(path, '(.*/)')
				prev_dir = dir
			end
			cursors[path] = pos
		end
	end
	f:close()
end

local write_cursors = function()
	local f = io.open(M.path, 'w+')
	if f == nil then return end
	-- sort paths
	local paths = {}
	for path in pairs(cursors) do
		table.insert(paths, path)
	end
	table.sort(paths)
	-- buffer cursors string
	local t = {}
	local prev_dir
	for i, path in ipairs(paths) do
		local dir = string.match(path, '(.*/)')
		-- simplify prev dirs to '@'
		if dir == prev_dir then
			local filename = string.match(path, '^.*/(.*)')
			table.insert(t, string.format('@/%s,%d', filename, cursors[path]))
		else
			prev_dir = dir
			table.insert(t, string.format('%s,%d', path, cursors[path]))
		end
	end
	local s = table.concat(t, '\n')
	f:write(s)
	f:close()
end

local set_cursor_pos = function(win)
	if win.file == nil or win.file.path == nil then
		return
	end
	if not file_exists(win.file.path) then
		return
	end
 	-- read cursors file in case other vis processes edited files
 	read_cursors()
 	-- set cursor pos for current file path
	cursors[win.file.path] = win.selection.pos
end

vis.events.subscribe(vis.events.INIT, read_cursors)
vis.events.subscribe(vis.events.WIN_OPEN, apply_cursor_pos)
vis.events.subscribe(vis.events.WIN_CLOSE, set_cursor_pos)
vis.events.subscribe(vis.events.QUIT, write_cursors)

return M