144 lines
3.5 KiB
Lua
144 lines
3.5 KiB
Lua
local M = {}
|
|
local type_check = vim.validate
|
|
M.__index = M
|
|
M.lines = {}
|
|
M.highlight = {}
|
|
M.height = 0
|
|
M.width = 0
|
|
M.interval = ' '
|
|
M.opts = {}
|
|
|
|
|
|
|
|
function M:new(opts)
|
|
if opts then
|
|
self.opts = opts
|
|
end
|
|
|
|
local content = {}
|
|
setmetatable(content, self)
|
|
return content
|
|
end
|
|
|
|
-- NOTE :
|
|
-- local items = {
|
|
-- -- style1: string 不需要单独设置高亮的情况
|
|
-- 'text',
|
|
-- -- style2: string[] 需要设置高亮，第二个名称为高亮组
|
|
-- {'text2', 'highlight name'},
|
|
-- }
|
|
|
|
-- local opts = {
|
|
-- -- 可选的参数
|
|
-- highlight = 'highlight name' -- string 该行的高亮
|
|
-- indent = 4 -- integer 该行的应该在开头的缩进
|
|
-- interval = 4 -- integer 该行组件的间隔
|
|
-- }
|
|
function M:insert(items)
|
|
type_check {
|
|
items = { items, 'table' },
|
|
}
|
|
|
|
self.height = self.height + 1 -- line数加一
|
|
|
|
local line = {
|
|
indent = items.indent,
|
|
highlight = items.highlight,
|
|
}
|
|
|
|
local highlight = {}
|
|
|
|
|
|
for i, item in ipairs(items) do
|
|
if type(item) == 'string' then
|
|
item = { item }
|
|
end
|
|
line[i] = item[1]
|
|
if item[2] then
|
|
highlight[i] = item[2]
|
|
end
|
|
end
|
|
|
|
self.highlight[self.height] = highlight
|
|
self.lines[self.height] = line
|
|
end
|
|
|
|
---Usage:
|
|
--- local buffer_id
|
|
--- local lines, highlights = M:lines()
|
|
--- vim.api.nvim_buf_set_lines(buffer_id, 0, -1, false,lines)
|
|
--- for i, hl in ipairs(highlights) do
|
|
--- vim.api.nvim_buf_add_highlight(buffer_id, 0, hl.name, i, hl._start, hl._end)
|
|
--- end
|
|
---@return table line
|
|
---@return table highlight
|
|
function M:data()
|
|
-- NOTE 返回格式化的行，如果需要高亮，则第二个参数返回高亮
|
|
local lines = {}
|
|
local highlights = {}
|
|
|
|
for index = 1, #self.lines do
|
|
local line = ''
|
|
local highlight = {}
|
|
local l = self.lines[index]
|
|
local hl = self.highlight[index]
|
|
if l.indent then
|
|
line = (' '):rep(l.indent)
|
|
end
|
|
|
|
if l.highlight then
|
|
line = line .. table.concat(l, self.interval)
|
|
highlight[1] = { name = l.highlight, _start = 1, _end = -1 }
|
|
|
|
else
|
|
line = line .. l[1]
|
|
|
|
if hl[1] then
|
|
-- WARN :可能需要设置成字符串宽度!!!
|
|
table.insert(highlight, { name = hl[1], _start = #line - #l[1], _end = #line })
|
|
end
|
|
|
|
for i = 2, #l do
|
|
line = line .. self.interval .. l[i]
|
|
if hl[i] then
|
|
local _end = #line
|
|
table.insert(highlight, { name = hl[i], _start = _end - #l[i], _end = _end })
|
|
end
|
|
end
|
|
end
|
|
|
|
lines[index] = line
|
|
local len = #line
|
|
if self.width < len then
|
|
self.width = len
|
|
end
|
|
highlights[index] = highlight
|
|
end
|
|
return lines, highlights
|
|
end
|
|
|
|
|
|
function M:attach()
|
|
local height = self.opts.win.height
|
|
local width = self.opts.win.width
|
|
|
|
local lines, hls = self:data()
|
|
vim.api.nvim_buf_set_lines(self.opts.bufnr, 0, -1, false, lines)
|
|
|
|
for line, l_hl in ipairs(hls) do
|
|
for _, hl in ipairs(l_hl) do
|
|
vim.api.nvim_buf_add_highlight(self.opts.bufnr, -1, hl.name, line - 1, hl._start, hl._end)
|
|
end
|
|
end
|
|
|
|
if self.height < height then
|
|
vim.api.nvim_win_set_height(self.opts.winid, self.height)
|
|
end
|
|
|
|
if self.width < width then
|
|
vim.api.nvim_win_set_width(self.opts.winid, self.width)
|
|
end
|
|
end
|
|
|
|
return M
|