summaryrefslogtreecommitdiffstats
path: root/.config/nvim/plugin/50-session.lua
blob: b4aadda8d75caf796ff49f48a8e893714ab12660 (plain)
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
--
-- Session plugin
--

local session_dir = vim.fn.stdpath("state") .. "/sessions"
local session_default = session_dir .. "/default.vim"
if not vim.uv.fs_stat(session_dir) then
	vim.uv.fs_mkdir(session_dir, tonumber("755", 8))
	vim.notify("Sessions save directory created at " .. session_dir)
end

local function save_session(path)
	vim.cmd.mksession({ path, bang = true })
end

local function load_session(path)
	vim.cmd.source(path)
end

local function reload_session(path)
	save_session(path)
	vim.cmd.restart({ args = { "+qall", "SessionLoad", path } })
end

local function delete_session(path)
	vim.fs.rm(path)
end

local function session_completefunc(arg_lead, _, _)
	local completions = {}
	for path in vim.fs.dir(session_dir) do
		if string.match(path, "^" .. arg_lead) and string.match(path, ".vim$") then
			completions[#completions + 1] = path:sub(1, -5)
		end
	end
	return completions
end

local function session_op(base, op)
	local path = #base > 0 and base or session_default
	if not string.match(path, "^" .. session_dir) then
		path = session_dir .. "/" .. path
	end
	if not string.match(path, "%.vim$") then
		path = path .. ".vim"
	end
	op(path)
end

vim.api.nvim_create_user_command("SessionSave", function(ev)
	session_op(ev.args, save_session)
end, { desc = "Save session", nargs = "?", complete = session_completefunc })
vim.api.nvim_create_user_command("SessionLoad", function(ev)
	session_op(ev.args, load_session)
end, { desc = "Load session", nargs = "?", complete = session_completefunc })
vim.api.nvim_create_user_command("SessionDelete", function(ev)
	session_op(ev.args, delete_session)
end, { desc = "Delete session", nargs = "?", complete = session_completefunc })
vim.api.nvim_create_user_command("SessionRestart", function(ev)
	session_op(ev.args, reload_session)
end, { desc = "Reload session", nargs = "?", complete = session_completefunc })
vim.api.nvim_create_user_command("SessionExitSave", function(ev)
	session_op(ev.args, save_session)
	vim.cmd.qall()
end, { desc = "Save and exit session", nargs = "?", complete = session_completefunc })
vim.api.nvim_create_user_command("SessionExitNoSave", function(ev)
	session_op(ev.args, save_session)
	vim.cmd.qall({ bang = true })
end, { desc = "Save and exit session", nargs = "?", complete = session_completefunc })