144 lines
4.9 KiB
Ruby
144 lines
4.9 KiB
Ruby
# This code was originally written by Davide Ficano:
|
|
# https://github.com/dafi/jekyll-toc-generator
|
|
# Thank you for that!
|
|
|
|
require 'nokogiri'
|
|
require 'digest/md5'
|
|
|
|
module Jekyll
|
|
module TOCGenerator
|
|
TOGGLE_HTML = '<div id="toctitle"><h2>%1</h2>%2</div>'
|
|
TOC_CONTAINER_HTML = '<div id="toc-container">%1<ul>%2</ul></div>'
|
|
|
|
def toc_generate(html)
|
|
# No Toc can be specified on every single page
|
|
# For example the index page has no table of contents
|
|
no_toc = @context.environments.first["page"]["noToc"] || false;
|
|
|
|
if no_toc
|
|
return html
|
|
end
|
|
|
|
config = @context.registers[:site].config
|
|
# Minimum number of items needed to show TOC, default 0 (0 means no minimum)
|
|
min_items_to_show_toc = config["minItemsToShowToc"] || 0
|
|
|
|
anchor_prefix = config["anchorPrefix"] || 'tocAnchor-'
|
|
|
|
# better for traditional page seo, commonlly use h1 as title
|
|
toc_top_tag = config["tocTopTag"] || 'h1'
|
|
toc_top_tag = toc_top_tag.gsub(/h/, '').to_i
|
|
if toc_top_tag > 5
|
|
toc_top_tag = 5
|
|
end
|
|
toc_sec_tag = toc_top_tag + 1
|
|
toc_top_tag = "h#{toc_top_tag}"
|
|
toc_sec_tag = "h#{toc_sec_tag}"
|
|
|
|
|
|
# Text labels
|
|
contents_label = config["contentsLabel"] || 'Contents'
|
|
hide_label = config["hideLabel"] || 'hide'
|
|
show_label = config["showLabel"] || 'show'
|
|
show_toggle_button = config["showToggleButton"]
|
|
|
|
toc_html = ''
|
|
toc_level = 1
|
|
toc_section = 1
|
|
item_number = 1
|
|
level_html = ''
|
|
|
|
doc = Nokogiri::HTML(html)
|
|
|
|
# Find H1 tag and all its H2 siblings until next H1
|
|
doc.css(toc_top_tag).each do |tag|
|
|
# TODO This XPATH expression can greatly improved
|
|
ct = tag.xpath("count(following-sibling::#{toc_top_tag})")
|
|
sects = tag.xpath("following-sibling::#{toc_sec_tag}[count(following-sibling::#{toc_top_tag})=#{ct}]")
|
|
|
|
level_html = '';
|
|
inner_section = 0;
|
|
|
|
sects.map.each do |sect|
|
|
inner_section += 1;
|
|
#anchor_id = anchor_prefix + toc_level.to_s + '-' + toc_section.to_s + '-' + inner_section.to_s
|
|
anchor_id = anchor_prefix + Digest::SHA256.hexdigest(tag.text + sect.text);
|
|
anchor_id = anchor_id[0..7];
|
|
sect['id'] = "#{anchor_id}"
|
|
|
|
level_html += create_level_html(anchor_id,
|
|
toc_level + 1,
|
|
toc_section + inner_section,
|
|
item_number.to_s + '.' + inner_section.to_s,
|
|
sect.text,
|
|
'')
|
|
end
|
|
if level_html.length > 0
|
|
level_html = '<ul>' + level_html + '</ul>';
|
|
end
|
|
|
|
#anchor_id = anchor_prefix + toc_level.to_s + '-' + toc_section.to_s;
|
|
anchor_id = anchor_prefix + Digest::SHA256.hexdigest(tag.text);
|
|
anchor_id = anchor_id[0..7];
|
|
tag['id'] = "#{anchor_id}"
|
|
|
|
toc_html += create_level_html(anchor_id,
|
|
toc_level,
|
|
toc_section,
|
|
item_number,
|
|
tag.text,
|
|
level_html);
|
|
|
|
toc_section += 1 + inner_section;
|
|
item_number += 1;
|
|
end
|
|
|
|
#toc_html += create_level_html("ccba53",
|
|
# 1,
|
|
# toc_section,
|
|
# item_number,
|
|
# "Comments",
|
|
# "");
|
|
|
|
# for convenience item_number starts from 1
|
|
# so we decrement it to obtain the index count
|
|
toc_index_count = item_number - 1
|
|
|
|
if toc_html.length > 0
|
|
hide_html = '';
|
|
if (show_toggle_button)
|
|
hide_html = HIDE_HTML.gsub('%1', hide_label)
|
|
end
|
|
|
|
if min_items_to_show_toc <= toc_index_count
|
|
replaced_toggle_html = TOGGLE_HTML
|
|
.gsub('%1', contents_label)
|
|
.gsub('%2', hide_html);
|
|
toc_table = TOC_CONTAINER_HTML
|
|
.gsub('%1', replaced_toggle_html)
|
|
.gsub('%2', toc_html);
|
|
doc.css('body').children.before(toc_table)
|
|
end
|
|
doc.css('body').children.to_xhtml(indent:3, indent_text:" ")
|
|
else
|
|
return html
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def create_level_html(anchor_id, toc_level, toc_section, tocNumber, tocText, tocInner)
|
|
link = '<a href="#%1"><span class="tocnumber">%2</span> <span class="toctext">%3</span></a>%4'
|
|
.gsub('%1', anchor_id.to_s)
|
|
.gsub('%2', tocNumber.to_s)
|
|
.gsub('%3', tocText)
|
|
.gsub('%4', tocInner ? tocInner : '');
|
|
'<li class="toc_level-%1 toc_section-%2">%3</li>'
|
|
.gsub('%1', toc_level.to_s)
|
|
.gsub('%2', toc_section.to_s)
|
|
.gsub('%3', link)
|
|
end
|
|
end
|
|
end
|
|
|
|
Liquid::Template.register_filter(Jekyll::TOCGenerator)
|