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
|
vim.api.nvim_create_autocmd('TextYankPost', {
desc = 'Highlight when yanking text',
group = vim.api.nvim_create_augroup('highlight-yank', { clear = true }),
callback = function()
vim.highlight.on_yank()
end,
})
vim.api.nvim_create_autocmd('BufEnter', {
desc = 'Disable newlines on commented lines from continuing the comment',
group = vim.api.nvim_create_augroup('disable-comments-continuation', { clear = true }),
callback = function()
-- vim.opt_local.formatoptions:remove('r') -- no comments on enter
vim.opt_local.formatoptions:remove('o') -- no comments on `o` or `O`
end,
})
vim.api.nvim_create_user_command('Format', function(args)
local range = nil
if args.count ~= -1 then
local end_line = vim.api.nvim_buf_get_lines(0, args.line2 - 1, args.line2, true)[1]
range = {
start = { args.line1, 0 },
['end'] = { args.line2, end_line:len() },
}
end
require('conform').format({ async = true, lsp_format = 'fallback', range = range })
end, { range = true })
vim.api.nvim_create_user_command('FormatToggle', function(args)
local is_global = not args.bang
if is_global then
vim.g.disable_autoformat = not vim.g.disable_autoformat
if vim.g.disable_autoformat then
Notify('Autoformat on save disabled globally')
else
Notify('Autoformat on save enabled globally')
end
else
vim.b.disable_autoformat = not vim.b.disable_autoformat
if vim.b.disable_autoformat then
Notify('Autoformat on save disabled for this buffer')
else
Notify('Autoformat on save enabled for this buffer')
end
end
end, {
desc = 'Toggle autoformat on save',
bang = true,
})
-- NOTE: highlight %v, %s etc in go string literals
vim.api.nvim_create_autocmd({ 'BufEnter', 'TextChanged', 'InsertLeave' }, {
pattern = '*.go',
callback = function()
local query = vim.treesitter.query.parse('go', '(interpreted_string_literal) @string')
local parser = vim.treesitter.get_parser(0, 'go')
local tree = parser:parse()[1]
local root = tree:root()
local bufnr = vim.api.nvim_get_current_buf()
for id, node in query:iter_captures(root, bufnr, 0, -1) do
if query.captures[id] == 'string' then
local start_row, start_col, end_row, end_col = node:range()
-- Get the text of the string literal
local text = vim.api.nvim_buf_get_text(bufnr, start_row, start_col, end_row, end_col, {})[1]
-- Highlight only the parts matching `%[a-z]`
for match_start, match_end in text:gmatch('()%%[a-z]()') do
vim.api.nvim_buf_add_highlight(
bufnr,
-1,
'@lsp.type.formatSpecifier.go', -- Higlight group
start_row,
start_col + match_start - 1,
start_col + match_end - 1
)
end
end
end
end,
})
-- NOTE: highlight zig's escape sequences
vim.api.nvim_create_autocmd({ 'BufEnter', 'TextChanged', 'InsertLeave' }, {
pattern = '*.go',
callback = function()
local query = vim.treesitter.query.parse('zig', '(interpreted_string_literal) @string')
local parser = vim.treesitter.get_parser(0, 'zig')
local tree = parser:parse()[1]
local root = tree:root()
local bufnr = vim.api.nvim_get_current_buf()
for id, node in query:iter_captures(root, bufnr, 0, -1) do
if query.captures[id] == 'string' then
local start_row, start_col, end_row, end_col = node:range()
-- Get the text of the string literal
local text = vim.api.nvim_buf_get_text(bufnr, start_row, start_col, end_row, end_col, {})[1]
-- Highlight only the parts matching `%[a-z]`
for match_start, match_end in text:gmatch('()%%[a-z]()') do
vim.api.nvim_buf_add_highlight(
bufnr,
-1,
'@lsp.type.formatSpecifier.go', -- Higlight group
start_row,
start_col + match_start - 1,
start_col + match_end - 1
)
end
end
end
end,
})
-- NOTE: show unused zig variables as "DiagnosticUnnecessary" instead of error
local orig_underline_show = vim.diagnostic.handlers.underline.show
local custom_ns = vim.api.nvim_create_namespace('custom_unused_ns')
vim.diagnostic.handlers.underline.show = function(namespace, bufnr, diagnostics, opts)
local normal_diags = {}
local custom_diags = {}
for _, diag in ipairs(diagnostics) do
if diag.message:find('unused', 1, true) then
table.insert(custom_diags, diag)
else
table.insert(normal_diags, diag)
end
end
if #normal_diags > 0 and orig_underline_show ~= nil then
orig_underline_show(namespace, bufnr, normal_diags, opts)
end
for _, diag in ipairs(custom_diags) do
vim.highlight.range(
bufnr,
custom_ns,
'DiagnosticUnnecessary',
{ diag.lnum, diag.col },
{ diag.end_lnum, diag.end_col },
{ inclusive = false }
)
end
end
vim.diagnostic.handlers.underline.hide = function(namespace, bufnr)
vim.api.nvim_buf_clear_namespace(bufnr, namespace, 0, -1)
vim.api.nvim_buf_clear_namespace(bufnr, custom_ns, 0, -1)
end
|
