Fix toctree handling with single-file builder (#96)

src/hieroglyph/builder.py

@@ -210,6 +210,9 @@ class SingleFileSlideBuilder(AbstractSlideBuilder, SingleFileHTMLBuilder):
 
     name = 'singlefile-slides'
 
+    def init_translator_class(self):
+        self.translator_class = writer.SingleFileSlideTranslator
+
 
 class AbstractInlineSlideBuilder(object):
 

src/hieroglyph/tests/singlefile/index.txt

@@ -7,10 +7,9 @@
 some text
 
 
-Slide 2
-=======
+.. slide:: Slide 2
 
-TESTING_SENTINEL
+  TESTING_SENTINEL
 
 .. toctree::
 

src/hieroglyph/tests/test_builder.py

@@ -139,3 +139,21 @@ def test_adjusts_section_levels_to_account_for_toctree(self, app, *args):
             contents = tree.find_all('article')
 
             self.assertEqual(len(contents), 4)
+
+    @with_app(
+        buildername='singlefile-slides',
+        srcdir=util.test_root.parent/'singlefile',
+    )
+    def test_slide_directive_closes_correctly_at_end_of_source_file(self, app, *args):
+        app.build()
+
+        with open(app.builddir/'singlefile-slides'/'index.html') as html_file:
+            tree = BeautifulSoup(html_file.read())
+
+            # test to see that all the slides are siblings
+            slides = [
+                s for s in tree.section.children
+                if s.name == 'article'
+            ]
+
+            self.assertEqual(len(slides), 4)

src/hieroglyph/writer.py

@@ -349,3 +349,14 @@ def visit_start_of_file(self, node):
         self.section_level -= 1
 
         BaseSlideTranslator.visit_start_of_file(self, node)
+
+
+class SingleFileSlideTranslator(SlideTranslator):
+
+    def visit_compound(self, node):
+        if not 'toctree-wrapper' in node['classes']:
+            SlideTranslator.visit_compound(self, node)
+
+    def depart_compound(self, node):
+        if not 'toctree-wrapper' in node['classes']:
+            SlideTranslator.depart_compound(self, node)