137 lines
3.5 KiB
Lua
137 lines
3.5 KiB
Lua
---@class ow.dap.hover.content.Capture
|
|
---@field start_col integer
|
|
---@field end_col integer
|
|
---@field text string
|
|
---@field group string
|
|
---@field priority integer
|
|
|
|
---@class ow.dap.hover.Highlight
|
|
---@field group string
|
|
---@field start_row integer 0-indexed
|
|
---@field start_col integer 0-indexed
|
|
---@field end_row integer 0-indexed
|
|
---@field end_col integer 0-indexed
|
|
|
|
---@class ow.dap.hover.Content
|
|
---@field text string
|
|
---@field highlights ow.dap.hover.Highlight[]
|
|
---@field _current_row integer
|
|
---@field _current_col integer
|
|
local Content = {}
|
|
Content.__index = Content
|
|
|
|
---@return ow.dap.hover.Content
|
|
function Content.new()
|
|
return setmetatable({
|
|
text = "",
|
|
highlights = {},
|
|
_current_row = 0,
|
|
_current_col = 0,
|
|
}, Content)
|
|
end
|
|
|
|
---@return integer
|
|
function Content:current_line()
|
|
return self._current_row + 1
|
|
end
|
|
|
|
---@param text string May not contain line breaks
|
|
---@param highlight_group? string
|
|
function Content:add(text, highlight_group)
|
|
local start_row = self._current_row
|
|
local start_col = self._current_col
|
|
local end_row = self._current_row
|
|
local end_col = start_col + #text
|
|
|
|
self.text = self.text .. text
|
|
self._current_col = end_col
|
|
|
|
if highlight_group then
|
|
table.insert(self.highlights, {
|
|
group = highlight_group,
|
|
start_row = start_row,
|
|
start_col = start_col,
|
|
end_row = end_row,
|
|
end_col = end_col,
|
|
})
|
|
end
|
|
end
|
|
|
|
---@param text string May not contain line breaks
|
|
---@param lang string
|
|
function Content:add_with_treesitter(text, lang)
|
|
local start_row = self._current_row
|
|
local start_col = self._current_col
|
|
|
|
self:add(text)
|
|
|
|
local ok, parser = pcall(vim.treesitter.get_string_parser, text, lang)
|
|
if not ok or not parser then
|
|
return
|
|
end
|
|
---@cast parser vim.treesitter.LanguageTree
|
|
|
|
local trees = parser:parse()
|
|
if not trees then
|
|
return
|
|
end
|
|
|
|
local tree = trees[1]
|
|
if not tree then
|
|
return
|
|
end
|
|
|
|
local query = vim.treesitter.query.get(lang, "highlights")
|
|
if not query then
|
|
return
|
|
end
|
|
|
|
for id, node in query:iter_captures(tree:root(), text, 0, -1) do
|
|
local capture_name = query.captures[id]
|
|
local start_row_rel, start_col_rel, end_row_rel, end_col_rel =
|
|
node:range()
|
|
|
|
local abs_start_row = start_row + start_row_rel
|
|
local abs_end_row = start_row + end_row_rel
|
|
local abs_start_col = start_col + start_col_rel
|
|
local abs_end_col = start_col + end_col_rel
|
|
|
|
table.insert(self.highlights, {
|
|
group = "@" .. capture_name,
|
|
start_row = abs_start_row,
|
|
start_col = abs_start_col,
|
|
end_row = abs_end_row,
|
|
end_col = abs_end_col,
|
|
})
|
|
end
|
|
end
|
|
|
|
function Content:newline()
|
|
self:add("\n")
|
|
self._current_col = 0
|
|
self._current_row = self._current_row + 1
|
|
end
|
|
|
|
---@return string[]
|
|
function Content:get_lines()
|
|
local text = self.text:gsub("%s+$", "")
|
|
return vim.split(text, "\n")
|
|
end
|
|
|
|
---@param ns_id integer
|
|
---@param buf integer
|
|
---@param row_offset integer 0-indexed
|
|
function Content:apply_highlights(ns_id, buf, row_offset)
|
|
for _, highlight in ipairs(self.highlights) do
|
|
vim.hl.range(
|
|
buf,
|
|
ns_id,
|
|
highlight.group,
|
|
{ row_offset + highlight.start_row, highlight.start_col },
|
|
{ row_offset + highlight.end_row, highlight.end_col }
|
|
)
|
|
end
|
|
end
|
|
|
|
return Content
|