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
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
|
local M = {}
---@class (exact) conform.FormatterInfo
---@field name string
---@field command string
---@field cwd? string
---@field available boolean
---@field available_msg? string
---@class (exact) conform.StaticFormatterConfig
---@field command string|fun(ctx: conform.Context): string
---@field args? string[]|fun(ctx: conform.Context): string[]
---@field cwd? fun(ctx: conform.Context): nil|string
---@field require_cwd? boolean When cwd is not found, don't run the formatter (default false)
---@field stdin? boolean Send buffer contents to stdin (default true)
---@field condition? fun(ctx: conform.Context): boolean
---@field exit_codes? integer[] Exit codes that indicate success (default {0})
---@class (exact) conform.FormatterConfig : conform.StaticFormatterConfig
---@field meta conform.FormatterMeta
---@class (exact) conform.FormatterMeta
---@field url string
---@field description string
---
---@class (exact) conform.Context
---@field buf integer
---@field filename string
---@field dirname string
---@class (exact) conform.RunOptions
---@field run_all_formatters nil|boolean Run all listed formatters instead of stopping at the first one.
---@field format_on_save nil|boolean Run these formatters in the built-in format_on_save autocmd.
---@class (exact) conform.FormatterList : conform.RunOptions
---@field formatters string[]
---@type table<string, string[]|conform.FormatterList>
M.formatters_by_ft = {}
---@type table<string, conform.StaticFormatterConfig|fun(): conform.StaticFormatterConfig>
M.formatters = {}
M.setup = function(opts)
opts = opts or {}
M.formatters = vim.tbl_extend("force", M.formatters, opts.formatters or {})
M.formatters_by_ft = vim.tbl_extend("force", M.formatters_by_ft, opts.formatters_by_ft or {})
if opts.log_level then
require("conform.log").level = opts.log_level
end
if opts.format_on_save then
if type(opts.format_on_save) == "boolean" then
opts.format_on_save = {}
end
local aug = vim.api.nvim_create_augroup("Conform", { clear = true })
vim.api.nvim_create_autocmd("BufWritePre", {
pattern = "*",
group = aug,
callback = function(args)
local format_opts = vim.tbl_deep_extend("keep", opts.format_on_save, {
buf = args.buf,
})
local filetypes = vim.split(vim.bo[args.buf].filetype, ".", { plain = true })
for _, ft in ipairs(filetypes) do
local ft_formatters = M.formatters_by_ft[ft]
if ft_formatters and ft_formatters.format_on_save == false then
return
end
end
M.format(format_opts)
end,
})
end
---@diagnostic disable-next-line: duplicate-set-field
vim.lsp.handlers["textDocument/formatting"] = function(_, result, ctx, _)
if not result then
return
end
local client = vim.lsp.get_client_by_id(ctx.client_id)
local restore = require("conform.util").save_win_positions(ctx.bufnr)
vim.lsp.util.apply_text_edits(result, ctx.bufnr, client.offset_encoding)
restore()
end
end
---@param bufnr integer
---@return boolean
local function supports_lsp_format(bufnr)
for _, client in ipairs(vim.lsp.get_active_clients({ bufnr = bufnr })) do
if client.supports_method("textDocument/formatting", { bufnr = bufnr }) then
return true
end
end
return false
end
---@param bufnr? integer
---@return conform.FormatterInfo[]
---@return conform.RunOptions
local function list_formatters_for_buffer(bufnr)
if not bufnr or bufnr == 0 then
bufnr = vim.api.nvim_get_current_buf()
end
local formatters = {}
local run_options = {
run_all_formatters = false,
format_on_save = true,
}
local filetypes = vim.split(vim.bo[bufnr].filetype, ".", { plain = true })
table.insert(filetypes, "*")
for _, filetype in ipairs(filetypes) do
local ft_formatters = M.formatters_by_ft[filetype]
if ft_formatters then
if not vim.tbl_islist(ft_formatters) then
for k, v in pairs(ft_formatters) do
if k ~= "formatters" then
run_options[k] = v
end
end
ft_formatters = ft_formatters.formatters
end
for _, formatter in ipairs(ft_formatters) do
formatters[formatter] = true
end
end
end
---@type conform.FormatterInfo[]
local all_info = {}
for formatter in pairs(formatters) do
local info = M.get_formatter_info(formatter)
table.insert(all_info, info)
end
return all_info, run_options
end
---@param formatters conform.FormatterInfo[]
---@param run_options conform.RunOptions
---@return conform.FormatterInfo[]
local function filter_formatters(formatters, run_options)
---@type conform.FormatterInfo[]
local all_info = {}
for _, info in ipairs(formatters) do
if info.available then
table.insert(all_info, info)
if not run_options.run_all_formatters then
break
end
end
end
return all_info
end
---Format a buffer
---@param opts? table
--- timeout_ms nil|integer Time in milliseconds to block for formatting. Defaults to 1000. No effect if async = true.
--- bufnr nil|integer Format this buffer (default 0)
--- async nil|boolean If true the method won't block. Defaults to false.
--- formatters nil|string[] List of formatters to run. Defaults to all formatters for the buffer filetype.
--- lsp_fallback nil|boolean Attempt LSP formatting if no formatters are available. Defaults to false.
--- quiet nil|boolean Don't show any notifications for warnings or failures. Defaults to false.
---@return boolean True if any formatters were attempted
M.format = function(opts)
---@type {timeout_ms: integer, bufnr: integer, async: boolean, lsp_fallback: boolean, quiet: boolean, formatters?: string[]}
opts = vim.tbl_extend("keep", opts or {}, {
timeout_ms = 1000,
bufnr = 0,
async = false,
lsp_fallback = false,
quiet = false,
})
local log = require("conform.log")
local formatters = {}
local any_formatters_configured
if opts.formatters then
any_formatters_configured = true
for _, formatter in ipairs(opts.formatters) do
local info = M.get_formatter_info(formatter)
if info.available then
table.insert(formatters, info)
else
if opts.quiet then
log.warn("Formatter '%s' unavailable: %s", info.name, info.available_msg)
else
vim.notify(
string.format("Formatter '%s' unavailable: %s", info.name, info.available_msg),
vim.log.levels.WARN
)
end
end
end
else
local run_info
formatters, run_info = list_formatters_for_buffer(opts.bufnr)
any_formatters_configured = not vim.tbl_isempty(formatters)
formatters = filter_formatters(formatters, run_info)
end
local formatter_names = vim.tbl_map(function(f)
return f.name
end, formatters)
log.debug("Running formatters on %s: %s", vim.api.nvim_buf_get_name(opts.bufnr), formatter_names)
local any_formatters = not vim.tbl_isempty(formatters)
if any_formatters then
if opts.async then
require("conform.runner").format_async(opts.bufnr, formatters)
else
require("conform.runner").format_sync(opts.bufnr, formatters, opts.timeout_ms, opts.quiet)
end
elseif opts.lsp_fallback and supports_lsp_format(opts.bufnr) then
log.debug("Running LSP formatter on %s", vim.api.nvim_buf_get_name(opts.bufnr))
local restore = require("conform.util").save_win_positions(opts.bufnr)
vim.lsp.buf.format(opts)
if not opts.async then
restore()
end
elseif any_formatters_configured and not opts.quiet then
vim.notify("No formatters found for buffer. See :checkhealth conform", vim.log.levels.WARN)
else
log.debug("No formatters found for %s", vim.api.nvim_buf_get_name(opts.bufnr))
end
return any_formatters
end
---Retrieve the available formatters for a buffer
---@param bufnr? integer
---@return conform.FormatterInfo[]
M.list_formatters = function(bufnr)
local formatters, run_options = list_formatters_for_buffer(bufnr)
return filter_formatters(formatters, run_options)
end
---List information about all filetype-configured formatters
---@return conform.FormatterInfo[]
M.list_all_formatters = function()
local formatters = {}
for _, ft_formatters in pairs(M.formatters_by_ft) do
if not vim.tbl_islist(ft_formatters) then
ft_formatters = ft_formatters.formatters
end
for _, formatter in ipairs(ft_formatters) do
formatters[formatter] = true
end
end
---@type conform.FormatterInfo[]
local all_info = {}
for formatter in pairs(formatters) do
local info = M.get_formatter_info(formatter)
table.insert(all_info, info)
end
table.sort(all_info, function(a, b)
return a.name < b.name
end)
return all_info
end
---@private
---@param formatter string
---@return nil|conform.StaticFormatterConfig
M.get_formatter_config = function(formatter)
local config = M.formatters[formatter]
if not config then
local ok
ok, config = pcall(require, "conform.formatters." .. formatter)
if not ok then
return nil
end
end
if type(config) == "function" then
config = config()
end
if config.stdin == nil then
config.stdin = true
end
return config
end
---@private
---@param formatter string
---@param bufnr? integer
---@return conform.FormatterInfo
M.get_formatter_info = function(formatter, bufnr)
if not bufnr or bufnr == 0 then
bufnr = vim.api.nvim_get_current_buf()
end
local config = M.get_formatter_config(formatter)
if not config then
return {
name = formatter,
command = formatter,
available = false,
available_msg = "No config found",
}
end
local ctx = require("conform.runner").build_context(bufnr, config)
local command = config.command
if type(command) == "function" then
command = command(ctx)
end
local available = true
local available_msg = nil
if vim.fn.executable(command) == 0 then
available = false
available_msg = "Command not found"
elseif config.condition and not config.condition(ctx) then
available = false
available_msg = "Condition failed"
end
local cwd = nil
if config.cwd then
cwd = config.cwd(ctx)
if available and not cwd and config.require_cwd then
available = false
available_msg = "Root directory not found"
end
end
---@type conform.FormatterInfo
return {
name = formatter,
command = command,
cwd = cwd,
available = available,
available_msg = available_msg,
}
end
return M
|