Some prelim work on builder api.

[librarian.git] / src / librarian / builders / html.py

diff --git a/src/librarian/builders/html.py b/src/librarian/builders/html.py
index 8015c6a..66db675 100644 (file)
--- a/src/librarian/builders/html.py
+++ b/src/librarian/builders/html.py
@@ -1,49 +1,118 @@
+# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
+#
+from collections import defaultdict
+import os
+from urllib.request import urlopen

 from lxml import etree
 from lxml import etree

+from librarian.html import add_table_of_contents, add_table_of_themes, add_image_sizes

 from librarian import OutputFile
 
 
 from librarian import OutputFile
 
 

-class HtmlBuilder:
+class TreeBuilder:
+    @property
+    def cursor(self):
+        return self.current_cursors[-1]
+
+    def enter_fragment(self, fragment):
+        cursor = self.cursors.get(fragment, self.cursor)
+        self.current_cursors.append(cursor)
+
+    def exit_fragment(self):
+        self.current_cursors.pop()
+
+    def create_fragment(self, name, element):
+        assert name not in self.cursors
+        self.cursors[name] = element
+
+    def forget_fragment(self, name):
+        del self.cursors[name]
+
+    def start_element(self, tag, attrib=None):
+        self.current_cursors.append(etree.SubElement(
+            self.cursor,
+            tag,
+            **(attrib or {})
+        ))
+
+    def end_element(self):
+        self.current_cursors.pop()
+
+    def push_text(self, text):
+        cursor = self.cursor
+        if len(cursor):
+            cursor[-1].tail = (cursor[-1].tail or '') + text
+        else:
+            cursor.text = (cursor.text or '') + text
+
+
+class HtmlBuilder(TreeBuilder):
+    build_method_fn = 'html_build'

     file_extension = "html"
     file_extension = "html"

-    identifier = "html"
+    with_themes = True
+    with_toc = True
+    with_footnotes = True
+    with_nota_red = True
+    with_ids = True
+    with_numbering = True
+    no_externalities = False
+    orphans = True

 
 

-    def __init__(self, image_location='https://wolnelektury.pl/media/book/pictures/marcos-historia-kolorow/'):
-        self.image_location = image_location
+    root_tag = 'div'
+    root_attrib = {'id': 'book-text'}

 
 

-        #self.tree = etree.Element('html')
-        #body = etree.SubElement(self.tree, 'body')
-        #text = etree.SubElement(body, 'div', **{'id': 'book-text'})
-        self.tree = text = etree.Element('div', **{'id': 'book-text'})
-        toc = etree.SubElement(text, 'div', id='toc')
-        themes = etree.SubElement(text, 'div', id='themes')
-        h1 = etree.SubElement(text, 'h1')
+    def __init__(self, gallery_path=None, gallery_url=None, base_url=None):
+        self._base_url = base_url
+        self.gallery_path = gallery_path
+        self.gallery_url = gallery_url
+
+        self.tree = text = etree.Element(self.root_tag, **self.root_attrib)
+        self.header = etree.Element('h1')
+
+        self.footnotes = etree.Element('div', id='footnotes')
+
+        self.nota_red = etree.Element('div', id='nota_red')

 
         self.cursors = {
             None: text,
 
         self.cursors = {
             None: text,

-            'toc': toc,
-            'themes': themes,
-            'header': h1,
+            'header': self.header,
+            'footnotes': self.footnotes,
+            'nota_red': self.nota_red,

         }
         }

-        self.current_cursors = [None]
+        self.current_cursors = [text]

 
 

-    def enter_fragment(self, fragment):
-        self.current_cursors.append(fragment)
+    @property
+    def base_url(self):
+        if self._base_url is not None:
+            return self._base_url
+        else:
+            return 'https://wolnelektury.pl/media/book/pictures/{}/'.format(self.document.meta.url.slug)

 
 

-    def exit_fragment(self):
-        self.current_cursors.pop()
-        
-    def build(self, document):
-        document.tree.getroot().html_build(self)
+    def build(self, document, element=None, **kwargs):
+        self.document = document
+        self.document.assign_ids()
+        self.prepare_images()

 
 

-        head = etree.Element('head')
-        self.tree.insert(0, head)
-        etree.SubElement(
-            head,
-            'link',
-            href="https://static.wolnelektury.pl/css/compressed/book_text.b15153e56c0a.css",
-            rel="stylesheet",
-            type="text/css",
-        )
-        
+        if element is None:
+            element = document.tree.getroot()
+
+        element.html_build(self)
+        self.postprocess(document)
+        return self.output()
+
+    def prepare_images(self):
+        # Temporarily use the legacy method, before transitioning to external generators.
+        if self.gallery_path is None:
+            return
+        try:
+            os.makedirs(self.gallery_path)
+        except OSError:
+            pass
+        add_image_sizes(self.document.tree, self.gallery_path, self.gallery_url, self.base_url)
+
+    def output(self):
+        if not len(self.tree):
+            return None

         return OutputFile.from_bytes(
             etree.tostring(
                 self.tree,
         return OutputFile.from_bytes(
             etree.tostring(
                 self.tree,


@@ -53,20 +122,145 @@ class HtmlBuilder:
             )
         )
 
             )
         )
 

-    def start_element(self, tag, attrib):
-        self.cursors[self.current_cursors[-1]] = etree.SubElement(
-            self.cursors[self.current_cursors[-1]],
-            tag,
-            **attrib
-        )
-        print(self.cursors)
+    def postprocess(self, document):
+        _ = document.tree.getroot().gettext

 
 

-    def end_element(self):
-        self.cursors[self.current_cursors[-1]] = self.cursors[self.current_cursors[-1]].getparent()
+        if document.meta.translators:
+            self.enter_fragment('header')
+            self.start_element('span', {'class': 'translator'})
+            self.push_text(_("translated by") + " ")
+            self.push_text(
+                ", ".join(
+                    translator.readable()
+                    for translator in document.meta.translators
+                )
+            )
+            self.exit_fragment()

 
 

-    def push_text(self, text):
-        cursor = self.cursors[self.current_cursors[-1]]
-        if len(cursor):
-            cursor.tail = (cursor[-1].tail or '') + text
+        if len(self.header):
+            self.tree.insert(0, self.header)
+            
+        if self.with_nota_red and len(self.nota_red):
+            self.tree.append(self.nota_red)
+        if self.with_themes:
+            add_table_of_themes(self.tree)
+        if self.with_toc:
+            add_table_of_contents(self.tree)
+
+        if len(self.footnotes):
+            fnheader = etree.Element("h3")
+            fnheader.text = _("Footnotes")
+            self.footnotes.insert(0, fnheader)
+            self.tree.append(self.footnotes)
+
+    def add_visible_number(self, element):
+        assert '_id' in element.attrib, etree.tostring(element)
+        self.start_element('a', {
+            'href': f'#{element.attrib["_id"]}',
+            'class': 'wl-num',
+        })
+        self.push_text(element.attrib['_visible_numbering'])
+        self.end_element()
+
+
+class StandaloneHtmlBuilder(HtmlBuilder):
+    css_url = "https://static.wolnelektury.pl/css/compressed/book_text.css"
+
+    def postprocess(self, document):
+        super(StandaloneHtmlBuilder, self).postprocess(document)
+
+        tree = etree.Element('html')
+        body = etree.SubElement(tree, 'body')
+        body.append(self.tree)
+        self.tree = tree
+
+        head = etree.Element('head')
+        tree.insert(0, head)
+
+        etree.SubElement(head, 'meta', charset='utf-8')
+        etree.SubElement(head, 'title').text = document.meta.title
+
+        etree.SubElement(
+            head,
+            'meta',
+            name="viewport",
+            content="width=device-width, initial-scale=1, maximum-scale=1"
+        )
+
+        if self.no_externalities:
+            etree.SubElement(
+                head, 'style',
+            ).text = urlopen(self.css_url).read().decode('utf-8')

         else:
         else:

-            cursor.text = (cursor.text or '') + text
+            etree.SubElement(
+                head,
+                'link',
+                href=self.css_url,
+                rel="stylesheet",
+                type="text/css",
+            )
+
+            etree.SubElement(
+                body, 'script',
+                src="https://ajax.googleapis.com/ajax/libs/jquery/1/jquery.min.js"
+            )
+
+            etree.SubElement(
+                body,
+                "script",
+                src="http://malsup.github.io/min/jquery.cycle2.min.js"
+            )
+
+
+class SnippetHtmlBuilder(HtmlBuilder):
+    with_themes = False
+    with_toc = False
+    with_footnotes = False
+    with_nota_red = False
+    with_ids = False
+    with_numbering = False
+
+
+class AbstraktHtmlBuilder(HtmlBuilder):
+    with_themes = False
+    with_toc = False
+    with_footnotes = False
+    with_nota_red = False
+    with_ids = False
+    with_numbering = False
+
+    root_tag = 'blockquote'
+    root_attrib = {}
+
+    def build(self, document, element=None, **kwargs):
+        if element is None:
+            element = document.tree.find('//abstrakt')
+        if element is None:
+            return OutputFile.from_bytes(b'')
+        element.attrib['_force'] = '1'
+        return super().build(document, element, **kwargs)
+
+            
+class DaisyHtmlBuilder(StandaloneHtmlBuilder):
+    file_extension = 'xhtml'
+    with_themes = False
+    with_toc = False
+    with_footnotes = False
+    with_nota_red = False
+    with_deep_identifiers = False
+    no_externalities = True
+    with_numbering = False
+
+    def output(self):
+        tree = etree.ElementTree(self.tree)
+        tree.docinfo.public_id = '-//W3C//DTD XHTML 1.0 Transitional//EN'
+        tree.docinfo.system_url = 'http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd'
+        return OutputFile.from_bytes(
+            etree.tostring(
+                tree,
+                encoding='utf-8',
+                pretty_print=True,
+                xml_declaration=True
+            )
+        )
+