forked from pivotal-cf/bookbinder
-
Notifications
You must be signed in to change notification settings - Fork 0
/
quicklinks_renderer.rb
80 lines (64 loc) · 1.82 KB
/
quicklinks_renderer.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
require 'nokogiri'
require 'redcarpet'
class QuicklinksRenderer < Redcarpet::Render::Base
class BadHeadingLevelError < StandardError; end
attr_reader :vars
def initialize(template_variables)
super()
@vars = template_variables
end
def doc_header
@items = []
@items[1] = document.css('ul').first
nil
end
def doc_footer
document.css('.quick-links').to_html if any_headers?
end
def header(text, header_level)
return unless [2, 3].include?(header_level)
return unless anchor_for(text)
li = Nokogiri::XML::Node.new('li', document)
li.add_child anchor_for(text)
last_list_of_level(header_level-1).add_child(li)
@items[header_level] = li
nil
rescue BadHeadingLevelError => e
raise BadHeadingLevelError.new "The header \"#{text}\", which is at level #{e.message}, has no higher-level headers occurring before it."
end
private
def any_headers?
@items[2]
end
def anchor_for(text)
text = ERB.new(text).result(binding)
doc = Nokogiri::HTML(text)
target_anchor = doc.css('a').first
return unless target_anchor && target_anchor['id']
return if (target_anchor['class'] || '').match(/\bno-quick-link\b/)
anchor = Nokogiri::XML::Node.new('a', document)
anchor['href'] = "##{target_anchor['id']}"
anchor.content = doc.text.strip
anchor
end
def last_list_of_level(n)
item = @items[n]
raise BadHeadingLevelError.new("#{n+1}") unless item
return item if item.name == 'ul'
item.add_child('<ul>') unless item.css('ul').any?
@items[n] = item.css('ul').first
end
def document
builder.doc
end
def builder
@builder ||= Nokogiri::HTML::Builder.new(&base_quicklinks_doc)
end
def base_quicklinks_doc
Proc.new do |html|
html.div(class: 'quick-links') {
html.ul
}
end
end
end