nvim-plugins/nvim-ufo
1
0
Fork 0
mirror of https://github.com/kevinhwang91/nvim-ufo synced 2024-09-16 14:14:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(decorator): export fold kind in fold_virt_text_handler (#207)

Browse source
This commit is contained in:
kevinhwang91 2024-03-23 12:41:43 +08:00
parent b22a65207b
commit 458aa4451b
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
lua/ufo/decorator.lua
View file

@ -239,7 +239,8 @@ function Decorator:getVirtTextAndCloseFold(winid, lnum, endLnum, doRender)
local getFoldVirtText
if self.enableGetFoldVirtText then
getFoldVirtText = function(l)
assert(type(l) == 'number', 'expected a number, got ' .. type(l))
local t = type(l)
assert(t == 'number', 'expected a number, got ' .. t)
assert(lnum <= l and l <= endLnum,
('expected lnum range from %d to %d, got %d'):format(lnum, endLnum, l))
local line = fb:lines(l)[1]
@ -250,6 +251,12 @@ function Decorator:getVirtTextAndCloseFold(winid, lnum, endLnum, doRender)
bufnr = bufnr,
winid = winid,
text = text,
get_fold_kind = function(l)
l = l == nil and lnum or l
local t = type(l)
assert(t == 'number', 'expected a number, got ' .. t)
return fb:lineKind(winid, l)
end,
get_fold_virt_text = getFoldVirtText
})
wses.foldedTextMaps[lnum] = res