local listchars = vim.opt_local.listchars:get()
local ids = setmetatable({}, {
__index = function(t, k)
rawset(t, k, register_hl(state, k))
return rawget(t, k)
end,
})
if listchars.eol then
for row = 1, state.buflen do
local style_line = state.style[row]
style_line_insert_overlay_char(
style_line,
#vim.fn.getline(row) + 1,
{ listchars.eol, ids.NonText }
)
end
end
if listchars.tab and state.tabstop then
for _, match in
ipairs(vim.fn.matchbufline(state.bufnr, '\t', 1, '$') --[[@as (table[])]])
do
--- @type integer
local tablen = #state.tabstop
- ((vim.fn.virtcol({ match.lnum, match.byteidx }, false, state.winid)) % #state.tabstop)
--- @type string?
local text
if len(listchars.tab) == 3 then
if tablen == 1 then
text = utf8_sub(listchars.tab, 3, 3)
else
text = utf8_sub(listchars.tab, 1, 1)
.. utf8_sub(listchars.tab, 2, 2):rep(tablen - 2)
.. utf8_sub(listchars.tab, 3, 3)
end
else
text = utf8_sub(listchars.tab, 1, 1) .. utf8_sub(listchars.tab, 2, 2):rep(tablen - 1)
end
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + 1,
{ text, ids.Whitespace }
)
end
end
if listchars.space then
for _, match in
ipairs(vim.fn.matchbufline(state.bufnr, ' ', 1, '$') --[[@as (table[])]])
do
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + 1,
{ listchars.space, ids.Whitespace }
)
end
end
if listchars.multispace then
for _, match in
ipairs(vim.fn.matchbufline(state.bufnr, [[ \+]], 1, '$') --[[@as (table[])]])
do
local text = utf8_sub(listchars.multispace:rep(len(match.text)), 1, len(match.text))
for i = 1, len(text) do
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + i,
{ utf8_sub(text, i, i), ids.Whitespace }
)
end
end
end
if listchars.lead or listchars.leadmultispace then
for _, match in
ipairs(vim.fn.matchbufline(state.bufnr, [[^ \+]], 1, '$') --[[@as (table[])]])
do
local text = ''
if len(match.text) == 1 or not listchars.leadmultispace then
if listchars.lead then
text = listchars.lead:rep(len(match.text))
end
elseif listchars.leadmultispace then
text = utf8_sub(listchars.leadmultispace:rep(len(match.text)), 1, len(match.text))
end
for i = 1, len(text) do
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + i,
{ utf8_sub(text, i, i), ids.Whitespace }
)
end
end
end
if listchars.trail then
for _, match in
ipairs(vim.fn.matchbufline(state.bufnr, [[ \+$]], 1, '$') --[[@as (table[])]])
do
local text = listchars.trail:rep(len(match.text))
for i = 1, len(text) do
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + i,
{ utf8_sub(text, i, i), ids.Whitespace }
)
end
end
end
if listchars.nbsp then
for _, match in
ipairs(
vim.fn.matchbufline(state.bufnr, '\226\128\175\\|\194\160', 1, '$') --[[@as (table[])]]
)
do
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + 1,
{ listchars.nbsp, ids.Whitespace }
)
for i = 2, #match.text do
style_line_insert_overlay_char(
state.style[match.lnum],
match.byteidx + i,
{ '', ids.Whitespace }
)
end
end
end
end
--- @param name string
--- @return string
local function highlight_name_to_class_name(name)
return (name:gsub('%.', '-'):gsub('@', '-'))
end
--- @param name string
--- @return string
local function name_to_tag(name)
return ''
end
--- @param _ string
--- @return string
local function name_to_closetag(_)
return ''
end
--- @param str string
--- @param tabstop string|false?
--- @return string
local function html_escape(str, tabstop)
str = str:gsub('&', '&'):gsub('<', '<'):gsub('>', '>'):gsub('"', '"')
if tabstop then
--- @type string
str = str:gsub('\t', tabstop)
end
return str
end
--- @param out string[]
--- @param state vim.tohtml.state.global
local function extend_style(out, state)
table.insert(out, '')
end
--- @param out string[]
--- @param state vim.tohtml.state.global
local function extend_head(out, state)
table.insert(out, '')
table.insert(out, '')
if state.title ~= false then
table.insert(out, ('%s'):format(state.title))
end
local colorscheme = vim.api.nvim_exec2('colorscheme', { output = true }).output
table.insert(
out,
(''):format(html_escape(colorscheme))
)
extend_style(out, state)
table.insert(out, '')
end
--- @param out string[]
--- @param state vim.tohtml.state
--- @param row integer
local function _extend_virt_lines(out, state, row)
local style_line = state.style[row]
for _, virt_line in ipairs(style_line.virt_lines) do
local virt_s = ''
for _, v in ipairs(virt_line) do
if v[2] then
virt_s = virt_s .. (name_to_tag(state.highlights_name[v[2]]))
end
virt_s = virt_s .. v[1]
if v[2] then
--- @type string
virt_s = virt_s .. (name_to_closetag(state.highlights_name[v[2]]))
end
end
table.insert(out, virt_s)
end
end
--- @param state vim.tohtml.state
--- @param row integer
--- @return string
local function _pre_text_to_html(state, row)
local style_line = state.style[row]
local s = ''
for _, pre_text in ipairs(style_line.pre_text) do
if pre_text[2] then
s = s .. (name_to_tag(state.highlights_name[pre_text[2]]))
end
s = s .. (html_escape(pre_text[1], state.tabstop))
if pre_text[2] then
--- @type string
s = s .. (name_to_closetag(state.highlights_name[pre_text[2]]))
end
end
return s
end
--- @param state vim.tohtml.state
--- @param char table
--- @return string
local function _char_to_html(state, char)
local s = ''
if char[2] then
s = s .. name_to_tag(state.highlights_name[char[2]])
end
s = s .. html_escape(char[1], state.tabstop)
if char[2] then
s = s .. name_to_closetag(state.highlights_name[char[2]])
end
return s
end
--- @param state vim.tohtml.state
--- @param cell vim.tohtml.cell
--- @return string
local function _virt_text_to_html(state, cell)
local s = ''
for _, v in ipairs(cell[3]) do
if v[2] then
s = s .. (name_to_tag(state.highlights_name[v[2]]))
end
--- @type string
s = s .. html_escape(v[1], state.tabstop)
if v[2] then
s = s .. name_to_closetag(state.highlights_name[v[2]])
end
end
return s
end
--- @param out string[]
--- @param state vim.tohtml.state
local function extend_pre(out, state)
local styletable = state.style
table.insert(out, '')
local hide_count = 0
--- @type integer[]
local stack = {}
local function loop(row)
local style_line = styletable[row]
if style_line.hide and (styletable[row - 1] or {}).hide then
return
end
_extend_virt_lines(out, state, row)
--Possible improvement (altermo):
--Instead of looping over all the buffer characters per line,
--why not loop over all the style_line cells,
--and then calculating the amount of text.
if style_line.hide then
return
end
local line = vim.api.nvim_buf_get_lines(state.bufnr, row - 1, row, false)[1] or ''
local s = ''
s = s .. _pre_text_to_html(state, row)
local true_line_len = #line + 1
for k in pairs(style_line) do
if type(k) == 'number' and k > true_line_len then
true_line_len = k
end
end
for col = 1, true_line_len do
local cell = style_line[col]
--- @type table?
local char
if cell then
for i = #cell[2], 1, -1 do
local hlid = cell[2][i]
if hlid < 0 then
if hlid == HIDE_ID then
hide_count = hide_count - 1
end
else
--- @type integer?
local index
for idx = #stack, 1, -1 do
s = s .. (name_to_closetag(state.highlights_name[stack[idx]]))
if stack[idx] == hlid then
index = idx
break
end
end
assert(index, 'a coles tag which has no corresponding open tag')
for idx = index + 1, #stack do
s = s .. (name_to_tag(state.highlights_name[stack[idx]]))
end
table.remove(stack, index)
end
end
for _, hlid in ipairs(cell[1]) do
if hlid < 0 then
if hlid == HIDE_ID then
hide_count = hide_count + 1
end
else
table.insert(stack, hlid)
s = s .. (name_to_tag(state.highlights_name[hlid]))
end
end
if cell[3] then
s = s .. _virt_text_to_html(state, cell)
end
char = cell[4][#cell[4]]
end
if col == true_line_len and not char then
break
end
if hide_count == 0 then
s = s
.. _char_to_html(
state,
char
or { vim.api.nvim_buf_get_text(state.bufnr, row - 1, col - 1, row - 1, col, {})[1] }
)
end
end
table.insert(out, s)
end
for row = 1, state.buflen + 1 do
loop(row)
end
assert(#stack == 0, 'an open HTML tag was never closed')
table.insert(out, '')
end
--- @param out string[]
--- @param fn fun()
local function extend_body(out, fn)
table.insert(out, '')
fn()
table.insert(out, '')
end
--- @param out string[]
--- @param fn fun()
local function extend_html(out, fn)
table.insert(out, '')
table.insert(out, '')
fn()
table.insert(out, '')
end
--- @param winid integer
--- @param global_state vim.tohtml.state.global
--- @return vim.tohtml.state
local function global_state_to_state(winid, global_state)
local bufnr = vim.api.nvim_win_get_buf(winid)
local opt = global_state.conf
local width = opt.width or vim.bo[bufnr].textwidth
if not width or width < 1 then
width = vim.api.nvim_win_get_width(winid)
end
local state = setmetatable({
winid = winid == 0 and vim.api.nvim_get_current_win() or winid,
opt = vim.wo[winid],
style = generate_styletable(bufnr),
bufnr = bufnr,
tabstop = (' '):rep(vim.bo[bufnr].tabstop),
width = width,
buflen = vim.api.nvim_buf_line_count(bufnr),
}, { __index = global_state })
return state --[[@as vim.tohtml.state]]
end
--- @param opt vim.tohtml.opt
--- @param title? string
--- @return vim.tohtml.state.global
local function opt_to_global_state(opt, title)
local fonts = {}
if opt.font then
fonts = type(opt.font) == 'string' and { opt.font } or opt.font --[[@as (string[])]]
elseif vim.o.guifont:match('^[^:]+') then
table.insert(fonts, vim.o.guifont:match('^[^:]+'))
end
table.insert(fonts, 'monospace')
--- @type vim.tohtml.state.global
local state = {
background = get_background_color(),
foreground = get_foreground_color(),
title = opt.title or title or false,
font = table.concat(fonts, ','),
highlights_name = {},
conf = opt,
}
return state
end
--- @type fun(state: vim.tohtml.state)[]
local styletable_funcs = {
styletable_syntax,
styletable_diff,
styletable_treesitter,
styletable_match,
styletable_extmarks,
styletable_conceal,
styletable_listchars,
styletable_folds,
styletable_statuscolumn,
}
--- @param state vim.tohtml.state
local function state_generate_style(state)
vim._with({ win = state.winid }, function()
for _, fn in ipairs(styletable_funcs) do
--- @type string?
local cond
if type(fn) == 'table' then
cond = fn[2] --[[@as string]]
--- @type function
fn = fn[1]
end
if not cond or cond(state) then
fn(state)
end
end
end)
end
--- @param winid integer[]|integer
--- @param opt? vim.tohtml.opt
--- @return string[]
local function win_to_html(winid, opt)
if type(winid) == 'number' then
winid = { winid }
end
--- @cast winid integer[]
assert(#winid > 0, 'no window specified')
opt = opt or {}
local title = table.concat(
vim.tbl_map(vim.api.nvim_buf_get_name, vim.tbl_map(vim.api.nvim_win_get_buf, winid)),
','
)
local global_state = opt_to_global_state(opt, title)
--- @type vim.tohtml.state[]
local states = {}
for _, i in ipairs(winid) do
local state = global_state_to_state(i, global_state)
state_generate_style(state)
table.insert(states, state)
end
local html = {}
extend_html(html, function()
extend_head(html, global_state)
extend_body(html, function()
for _, state in ipairs(states) do
extend_pre(html, state)
end
end)
end)
return html
end
local M = {}
--- @class vim.tohtml.opt
--- @inlinedoc
---
--- Title tag to set in the generated HTML code.
--- (default: buffer name)
--- @field title? string|false
---
--- Show line numbers.
--- (default: `false`)
--- @field number_lines? boolean
---
--- Fonts to use.
--- (default: `guifont`)
--- @field font? string[]|string
---
--- Width used for items which are either right aligned or repeat a character
--- infinitely.
--- (default: 'textwidth' if non-zero or window width otherwise)
--- @field width? integer
--- Converts the buffer shown in the window {winid} to HTML and returns the output as a list of string.
--- @param winid? integer Window to convert (defaults to current window)
--- @param opt? vim.tohtml.opt Optional parameters.
--- @return string[]
function M.tohtml(winid, opt)
return win_to_html(winid or 0, opt)
end
return M