39 lines
958 B
Lua
39 lines
958 B
Lua
---@alias avante.ui.LineSection string[]
|
|
---
|
|
---@class avante.ui.Line
|
|
---@field sections avante.ui.LineSection[]
|
|
local M = {}
|
|
M.__index = M
|
|
|
|
---@param sections avante.ui.LineSection[]
|
|
function M:new(sections)
|
|
local this = setmetatable({}, M)
|
|
this.sections = sections
|
|
return this
|
|
end
|
|
|
|
---@param ns_id number
|
|
---@param bufnr number
|
|
---@param line number
|
|
function M:set_highlights(ns_id, bufnr, line)
|
|
if not vim.api.nvim_buf_is_valid(bufnr) then return end
|
|
local col_start = 0
|
|
for _, section in ipairs(self.sections) do
|
|
local text = section[1]
|
|
local highlight = section[2]
|
|
if highlight then vim.api.nvim_buf_add_highlight(bufnr, ns_id, highlight, line, col_start, col_start + #text) end
|
|
col_start = col_start + #text
|
|
end
|
|
end
|
|
|
|
function M:__tostring()
|
|
local content = {}
|
|
for _, section in ipairs(self.sections) do
|
|
local text = section[1]
|
|
table.insert(content, text)
|
|
end
|
|
return table.concat(content, "")
|
|
end
|
|
|
|
return M
|