42 lines
908 B
Ruby
42 lines
908 B
Ruby
# table of contents generation
|
|
|
|
# add toc to current @item
|
|
class Nanoc::Item
|
|
def toc
|
|
require 'nokogiri'
|
|
|
|
compiled_content = self.rep_named(:default).instance_eval { @content[:pre] }
|
|
doc = Nokogiri::HTML(compiled_content)
|
|
|
|
max_levels = 3
|
|
|
|
# begin toc
|
|
res = '<h2>Content</h2>'
|
|
|
|
# iterate through the body, find headers and build toc as we go along
|
|
level = 0
|
|
doc.css("body").children.each do |node|
|
|
if m = node.name.match(/^h(\d)$/)
|
|
hlevel = m[1].to_i
|
|
|
|
# nest lists
|
|
diff = hlevel - level
|
|
if diff > 0
|
|
res << '<ol>' * diff
|
|
elsif diff < 0
|
|
res << "</ol>" * diff.abs
|
|
end
|
|
level = hlevel
|
|
|
|
res << '<li>'
|
|
res << '<a class="toc" href="' + '#' + node[:id] + '">' + node.text + '</a>'
|
|
res << '</li>'
|
|
end
|
|
end
|
|
|
|
# end toc
|
|
res << '</ol>'
|
|
res
|
|
end
|
|
end
|