66 lines
1.6 KiB
Lua
66 lines
1.6 KiB
Lua
local api = vim.api
|
|
local add_hl = api.nvim_buf_add_highlight
|
|
|
|
local item_meta = {
|
|
render = function(self, bufnr, line, col)
|
|
if self[2] then
|
|
add_hl(bufnr, self.ns or -1, self[2], line, col, col + #self[1])
|
|
end
|
|
end,
|
|
}
|
|
|
|
local text_meta = {
|
|
render = function(self, bufnr, line, col)
|
|
local items = self.items
|
|
local step = self.step or ''
|
|
local len = #step
|
|
|
|
for i = 1, self.size do
|
|
local item = items[i]
|
|
item:render(bufnr, line, col)
|
|
col = col + #item[1] + len
|
|
end
|
|
end
|
|
}
|
|
|
|
item_meta.__index = item_meta
|
|
text_meta.__index = function(self, key)
|
|
return text_meta[key] or (key == 1 and table.concat(self.strs, self.step) or nil)
|
|
end
|
|
|
|
return {
|
|
item = function(text, highlight)
|
|
return setmetatable({
|
|
[1] = text,
|
|
[2] = highlight,
|
|
}, item_meta)
|
|
end,
|
|
text = function(items)
|
|
local strs = {}
|
|
local size = #items
|
|
assert(size > 1)
|
|
for i = 1, size do
|
|
strs[i] = items[i][1]
|
|
end
|
|
|
|
return setmetatable({
|
|
strs = strs,
|
|
size = size,
|
|
items = items,
|
|
}, text_meta)
|
|
end,
|
|
format = function(opts)
|
|
local text = opts.text
|
|
local size = text.size
|
|
local width = opts.width
|
|
local spin = opts.spin or ' '
|
|
|
|
local wid = text[1]:width()
|
|
local space = math.max(math.floor((width - wid) / (size - 1)), 0)
|
|
if space > 0 then
|
|
text.step = spin:rep(space)
|
|
end
|
|
return text
|
|
end,
|
|
}
|