hledger/tools/pandoc-toc.lua

66 lines
1.9 KiB
Lua
Raw Normal View History

-- Replace a $TOC$ marker with a table of contents generated from the document's headings.
local headers = {}
function Header(h)
table.insert(headers, h)
return h
end
function markupLink(hAttr, headerText)
local headerId = hAttr.identifier
local headerProperties = hAttr.attributes
return pandoc.Link(headerText, "#" .. headerId, headerId)
end
function markupElement(elem)
local hAttr = elem.attr
local hText = elem.label
local hNested = elem.contents
local link = pandoc.Plain(markupLink(hAttr, hText))
if not hNested[1] then return {link} end
return {link, markupElements(hNested)}
end
function markupElements(elems)
local newElems = {}
for _,e in pairs(elems) do
table.insert(newElems, markupElement(e))
end
return pandoc.OrderedList(newElems, {"1", "Decimal", "Period"})
end
function createTable(elems)
local navBegin = pandoc.RawBlock("html", "<nav id=\"toc\" class=\"right-toc\">")
local navEnd = pandoc.RawBlock("html", "</nav>")
local contentsP = pandoc.Para(pandoc.Str("Contents"))
return {navBegin, contentsP, markupElements(elems), navEnd}
end
function isTocMarker(blk)
local tocMarker = "$TOC$"
if not (blk.t == "Para") then return false end
if not blk.content[1] then return false end
if not (blk.content[1].t == "Str") then return false end
return blk.content[1].text == tocMarker
end
function Pandoc(doc)
newBlocks = {}
tocBlocks = createTable(pandoc.utils.hierarchicalize(headers))
for _,blk in pairs(doc.blocks) do
-- replace a TOC placeholder with the table of contents,
-- or nothing if there's less than two headings
if isTocMarker(blk) then
if #headers > 1 then
for _,tocBlk in pairs(tocBlocks) do
table.insert(newBlocks, tocBlk)
end
end
else
table.insert(newBlocks, blk)
end
end
return pandoc.Doc(newBlocks, doc.meta)
end