google-sre-ebook/generate.rb

65 lines
1.6 KiB
Ruby

require 'nokogiri'
require 'pp'
require 'fileutils'
# First we get the list of all the book sections:
chapter_links = Nokogiri::HTML(open("html/index.html"))
.css('#drop-down a')
.map {|l| l.attribute('href').value}
html = ''
chapter_links.each do |chapter_link|
chapter_file = File.basename chapter_link
html += "<span class=\"hidden\" name=\"#{chapter_file}\"></span>"
content = Nokogiri::HTML(open("html/#{chapter_link}")).css('.content')
content.css('.cont').each do |e|
e.remove
end
# Ensure that all links are to the same file
content.css('a').each do |a|
link = a.attribute('href')
if link
matches = link.value.scan /^([\w-]+.html)#([\w-]+)$/
if matches.length == 1
a['href'] = '#' + matches[0][1]
end
end
end
chapter_header = content.at('.heading')
headers = (1..6).map {|x| "h#{x}"}
# headers.each_with_index
content.css(headers.join(',')).each do |e|
# If chapter heading
if e == chapter_header
puts "Chapter Header"
e.name = 'h1'
else
# Reduce everything by 1
i = headers.index e.name
new_name = headers[i+1] ? headers[i+1] : 'h6'
e.name = new_name
end
end
content.css('a').each do |a|
link = a.attribute('href')
if link
# Link to a direct chapter
matches = link.value.scan /^([\w-]+.html)$/
if matches.length == 1
a['href'] = '#' + matches[0][0]
end
end
end
html += content.inner_html
end
File.open("html/chapters/sre.html", 'w') { |file| file.write(html) }
puts "[html] Generated HTML file"