Not atomic commit

app.rb

@@ -5,4 +5,6 @@
 require 'fetch'
 require 'convert'
 
-File.write('out.html', Fetch.run())
+content, toc = Fetch.run
+File.write('out.html', content)
+Convert.run toc

lib/convert.rb

@@ -1,20 +1,52 @@
 require 'eeepub'
+require 'nokogiri'
 
-epub = EeePub.make do
-  title       'Vala Tutorial'
-  creator     'Maciej Piechotka'
-  publisher   'Samuel Vasko'
-  date        '2014-01-25'
-  identifier  'http://example.com/book/foo', :scheme => 'URL'
-  uid         'http://example.com/book/foo'
+class Convert
+    def self.run toc
+        # Join the forces!
+        doc = (['header', 'out', 'footer'].map {|f| File.read(f + '.html')}).join
+        toc = self.parse_toc toc
+        epub = EeePub.make do
+            title       'Vala Tutorial'
+            creator     'Gnome'
+            publisher   'Samuel Vasko'
+            date        '2014-01-25'
+            identifier  'https://github.com/bliker/vala-tutorial-book', :scheme => 'URL'
+            uid         'https://github.com/bliker/vala-tutorial-book'
 
+            files ['out.html']
+            nav toc
+        end
+        epub.save('sample.epub')
+    end
 
-  files ['/path/to/foo.html', '/path/to/bar.html'] # or files [{'/path/to/foo.html' => 'dest/dir'}, {'/path/to/bar.html' => 'dest/dir'}]
-  nav [
-    {:label => '1. foo', :content => 'foo.html', :nav => [
-      {:label => '1.1 foo-1', :content => 'foo.html#foo-1'}
-    ]},
-    {:label => '1. bar', :content => 'bar.html'}
-  ]
-end
-epub.save('sample.epub')
+    def self.parse_toc doc
+        doc = Nokogiri::HTML(doc)
+        self.mush_ol(doc.css('body > ol'))
+    end
+
+    def self.mush_li li
+        data = {
+            :label => li.css('> a').text,
+            :content => 'out.html' + li.css('a').attr('href')
+        }
+        unless li.css('ol').empty?
+            data[:nav] = self.mush_ol(li.css('ol'))
+            return data
+        else
+            return data
+        end
+    end
+
+    def self.mush_ol ol
+        out = []
+        ol.children.map do |li|
+            if li.name == 'li'
+                out << self.mush_li(li)
+            end
+        end
+
+        return out
+    end
+
+end

lib/fetch.rb

@@ -7,8 +7,8 @@ class Fetch
 
     def self.run
         content = self.get_content
-
         toc = self.parse_toc content
+
         self.remove_elements content
         self.transform_code content
         self.absolute_links content
@@ -17,7 +17,7 @@ def self.run
 
         ['id', 'lang', 'dir'].each { |e| content[0].remove_attribute(e) }
 
-        return content
+        return content, toc
     end
 
     private
@@ -29,7 +29,7 @@ def self.get_content
     end
 
     def self.parse_toc content
-        content.css('.table-of-contents > ol > li > ol')
+        content.css('.table-of-contents > ol > li > ol').to_html
     end
 
     def self.cleanup_headings content
@@ -40,7 +40,7 @@ def self.cleanup_headings content
 
     def self.absolute_links content
         content.css('a').each do |a|
-            if a.attr('href').starts_with '/'
+            if a.attr('href')[0] == '/'
                 a.set_attribute('href', 'https://wiki.gnome.org' + a.attr('href'))
             end
         end