vim-config/lua/user.lua at master · ioreilly/vim-config · GitHub

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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
-- rafi user functions
-- https://github.com/rafi/vim-config

local legacy_api = vim.fn.has('nvim-0.6') == 0
local diagnostic = legacy_api and vim.lsp.diagnostic or vim.diagnostic

local lists = {
	qf = {
		qf_isLoc = 0,
		cmd_open = 'botright copen',
		cmd_close = 'cclose',
	},
	location = {
		qf_isLoc = 1,
		cmd_open = 'botright lopen',
		cmd_close = 'lclose',
	},
}

local user = {}

-- QFLT_QUICKFIX Quickfix list - global list
----

user.qflist = {}

-- Open quickfix list window
user.qflist.open = function()
	vim.api.nvim_command(lists.qf.cmd_open)
end

-- Toggle quickfix list window
user.qflist.toggle = function()
	user._toggle_list('qf')
end

-- QFLT_LOCATION Location list - per window list
----

user.loclist = {}

-- Open location list window
user.loclist.open = function()
	vim.api.nvim_command(lists.location.cmd_open)
end

-- Toggle location list window
user.loclist.toggle = function()
	user._toggle_list('location')
end

-- Diagnostics
----

user.diagnostic = {}

-- Set locations with diagnostics and open the list.
user.diagnostic.publish_loclist = function(toggle)
	if vim.api.nvim_buf_get_option(0, 'filetype') ~= 'qf' then
		if legacy_api then
			diagnostic.set_loclist({ open = false })
		else
			diagnostic.setloclist({ open = false })
		end
	end
	if toggle then
		user.loclist.toggle()
	else
		user.loclist.open()
	end
end

user.diagnostic.disable = function(bufnr, namespace)
	diagnostic.disable(bufnr, namespace)
end

user.diagnostic.goto_prev = function()
	diagnostic.goto_prev()
end

user.diagnostic.goto_next = function()
	diagnostic.goto_next()
end

user.diagnostic.show_line_diagnostics = function(opts)
	if legacy_api then
		diagnostic.show_line_diagnostics(opts)
	else
		vim.diagnostic.open_float(opts)
	end
end

-- Git
----

user.githunk = {}

user.githunk.publish_loclist = function(toggle)
	if vim.api.nvim_buf_get_option(0, 'filetype') ~= 'qf' then
		require('gitsigns').setloclist()
	end
	if toggle then
		user.loclist.toggle()
	else
		user.loclist.open()
	end
end

-- Private
----

-- Toggle list window
user._toggle_list = function(key)
	local win_bufs = user._get_tabpage_win_bufs(0)
	local was_opened = false
	for win, buf in pairs(win_bufs) do
		if vim.api.nvim_buf_get_option(buf, 'filetype') == 'qf' then
			local qf_isLoc = vim.api.nvim_buf_get_var(buf, 'qf_isLoc')
			if qf_isLoc == lists[key].qf_isLoc then
				was_opened = true
				vim.api.nvim_win_close(win, false)
				-- vim.api.nvim_buf_call(buf, function()
				-- 	vim.api.nvim_command(lists[key].cmd_close)
				-- end)
			end
		end
	end

	if not was_opened then
		vim.api.nvim_command(lists[key].cmd_open)
	end
end

-- Return a table with all window buffers from a tabpage.
user._get_tabpage_win_bufs = function(tabpage)
	local bufs = {}
	for _, win in pairs(vim.api.nvim_tabpage_list_wins(tabpage)) do
		if win ~= nil and vim.api.nvim_win_is_valid(win) then
			local buf = vim.api.nvim_win_get_buf(win)
			if buf ~= nil and vim.api.nvim_buf_is_valid(buf) then
				bufs[win] = buf
			end
		end
	end
	return bufs
end

return user

-- vim: set ts=2 sw=2 tw=80 noet :