-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
inject_epub_toc.rb
119 lines (94 loc) · 2.99 KB
/
inject_epub_toc.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
require 'nokogiri'
require 'zip/zip'
class EpubContentsBuilder
def initialize(filename)
@filename = filename
@opf_filename = 'content.opf'
end
def inject_contents
Zip::ZipFile.open(@filename) do |zipfile|
@zipfile = zipfile
add_contents_file
insert_opf_reference
end
end
private
def add_contents_file
add_zipfile_entry("contents.xhtml", contents_xhtml)
end
def insert_opf_reference
insertions = {
'<item id="title_page" href="title_page.xhtml" media-type="application/xhtml+xml" />' =>
'<item id="contents" href="contents.xhtml" media-type="application/xhtml+xml" />',
'<itemref idref="title_page" />' =>
'<itemref idref="contents" />'
}
insert_lines(@opf_filename, insertions)
end
def add_zipfile_entry(filename, contents)
@zipfile.get_output_stream(filename) do |file|
file.puts contents
end
end
def insert_lines(filename, insertions)
opf_xml = @zipfile.read(filename)
@zipfile.get_output_stream(filename) do |file|
opf_xml.split("\n").each do |original_line|
file.puts original_line
insertions.each do |marker, line_to_insert|
if original_line.include?(marker)
file.puts line_to_insert
end
end
end
end
end
def contents_xhtml
<<-XML
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<title class="title">Table of Contents</title>
<link href="stylesheet.css" type="text/css" rel="stylesheet" />
</head>
<body>
#{contents_xhtml_links}
</body>
</html>
XML
end
def contents_xhtml_links
toc_xhtml = ""
ncx = Nokogiri::XML.parse(@zipfile.read('toc.ncx'))
chapter_files = ncx.css('content').map { |node| node['src'] }.select {|chapter_name| chapter_name =~ /ch\d/ };nil
chapter_files.each do |chapter_file|
chapter_xhtml = Nokogiri::XML.parse(@zipfile.read(chapter_file))
h1 = chapter_xhtml.css('h1').first
subheads = chapter_xhtml.css('h2, h3')
toc_xhtml << %Q{<p><a href="#{chapter_file}">#{h1.text}</a></p>\n}
toc_xhtml << "<ul>\n"
indented = false
subheads.each do |subhead|
if subhead.name == 'h3' && indented == false
indented = true
toc_xhtml << "<!-- begin h3s -->\n<ul>\n"
end
if subhead.name == 'h2' && indented == true
indented == false
toc_xhtml << "</ul>\n<!-- end h3s -->\n"
end
toc_xhtml << %Q{ <li><a href="#{chapter_file}##{subhead['id']}">#{subhead.text}</a></li>\n}
end
if indented == true
toc_xhtml << "</ul></li>\n<!-- end h3s at end of chapter -->\n"
end
toc_xhtml << "</ul>\n"
end
toc_xhtml
end
end
unless epub_filename = ARGV[0]
raise "Usage: ruby inject_epub_toc.rb epub_filename"
end
EpubContentsBuilder.new(epub_filename).inject_contents