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
|
local M = {}
---Find a command in node_modules
---@param cmd string
---@return fun(ctx: conform.Context): string
M.from_node_modules = function(cmd)
return M.find_executable({ "node_modules/.bin/" .. cmd }, cmd)
end
---Search parent directories for a relative path to a command
---@param paths string[]
---@param default string
---@return fun(ctx: conform.Context): string
---@example
--- local cmd = require("conform.util").find_executable({ "node_modules/.bin/prettier" }, "prettier")
M.find_executable = function(paths, default)
return function(ctx)
for _, path in ipairs(paths) do
local normpath = vim.fs.normalize(path)
local is_absolute = vim.startswith(normpath, "/")
if is_absolute and vim.fn.executable(normpath) then
return normpath
end
local idx = normpath:find("/", 1, true)
local dir, subpath
if idx then
dir = normpath:sub(1, idx - 1)
subpath = normpath:sub(idx)
else
-- This is a bare relative-path executable
dir = normpath
subpath = ""
end
local results = vim.fs.find(dir, { upward = true, path = ctx.dirname, limit = math.huge })
for _, result in ipairs(results) do
local fullpath = result .. subpath
if vim.fn.executable(fullpath) == 1 then
return fullpath
end
end
end
return default
end
end
---@param files string|string[]
---@return fun(ctx: conform.Context): nil|string
M.root_file = function(files)
return function(ctx)
local found = vim.fs.find(files, { upward = true, path = ctx.dirname })[1]
if found then
return vim.fs.dirname(found)
end
end
end
---@param bufnr integer
---@param range conform.Range
---@return integer start_offset
---@return integer end_offset
M.get_offsets_from_range = function(bufnr, range)
local row = range.start[1] - 1
local end_row = range["end"][1] - 1
local col = range.start[2]
local end_col = range["end"][2]
local start_offset = vim.api.nvim_buf_get_offset(bufnr, row) + col
local end_offset = vim.api.nvim_buf_get_offset(bufnr, end_row) + end_col
return start_offset, end_offset
end
---@generic T : any
---@param tbl T[]
---@param start_idx? number
---@param end_idx? number
---@return T[]
M.tbl_slice = function(tbl, start_idx, end_idx)
local ret = {}
if not start_idx then
start_idx = 1
end
if not end_idx then
end_idx = #tbl
end
for i = start_idx, end_idx do
table.insert(ret, tbl[i])
end
return ret
end
---@generic T : fun()
---@param cb T
---@param wrapper T
---@return T
M.wrap_callback = function(cb, wrapper)
return function(...)
wrapper(...)
cb(...)
end
end
---Helper function to add to the default args of a formatter.
---@param args string|string[]|fun(ctx: conform.Context): string|string[]
---@param extra_args string|string[]|fun(ctx: conform.Context): string|string[]
---@param opts? { append?: boolean }
---@example
--- local util = require("conform.util")
--- local prettier = require("conform.formatters.prettier")
--- require("conform").formatters.prettier = vim.tbl_deep_extend("force", prettier, {
--- args = util.extend_args(prettier.args, { "--tab", "--indent", "2" }),
--- range_args = util.extend_args(prettier.range_args, { "--tab", "--indent", "2" }),
--- })
M.extend_args = function(args, extra_args, opts)
opts = opts or {}
return function(ctx)
if type(args) == "function" then
args = args(ctx)
end
if type(extra_args) == "function" then
extra_args = extra_args(ctx)
end
if type(args) == "string" then
if type(extra_args) ~= "string" then
extra_args = table.concat(extra_args, " ")
end
if opts.append then
return args .. " " .. extra_args
else
return extra_args .. " " .. args
end
else
if type(extra_args) == "string" then
error("extra_args must be a table when args is a table")
end
if opts.append then
return vim.tbl_flatten({ args, extra_args })
else
return vim.tbl_flatten({ extra_args, args })
end
end
end
end
---@param formatter conform.FormatterConfig
---@param extra_args string|string[]|fun(ctx: conform.Context): string|string[]
---@param opts? { append?: boolean }
---@example
--- local util = require("conform.util")
--- local prettier = require("conform.formatters.prettier")
--- util.add_formatter_args(prettier, { "--tab", "--indent", "2" })
M.add_formatter_args = function(formatter, extra_args, opts)
formatter.args = M.extend_args(formatter.args, extra_args, opts)
if formatter.range_args then
formatter.range_args = M.extend_args(formatter.range_args, extra_args, opts)
end
end
---@param config conform.FormatterConfig
---@param override conform.FormatterConfigOverride
---@return conform.FormatterConfig
M.merge_formatter_configs = function(config, override)
local ret = vim.tbl_deep_extend("force", config, override)
if override.prepend_args then
M.add_formatter_args(ret, override.prepend_args, { append = false })
end
return ret
end
---@param bufnr integer
---@return integer
M.buf_get_changedtick = function(bufnr)
if not vim.api.nvim_buf_is_valid(bufnr) then
return -2
end
local changedtick = vim.b[bufnr].changedtick
-- changedtick gets set to -1 when vim is exiting. We have an autocmd that should store it in
-- last_changedtick before it is set to -1.
if changedtick == -1 then
return vim.b[bufnr].last_changedtick or -1
else
return changedtick
end
end
return M
|