# -*- coding: utf-8 -*-

# See LICENSE.txt for licensing terms

# Some fragments of code are copied from Reportlab under this license:
#
#####################################################################################
#
#       Copyright (c) 2000-2008, ReportLab Inc.
#       All rights reserved.
#
#       Redistribution and use in source and binary forms, with or without modification,
#       are permitted provided that the following conditions are met:
#
#               *       Redistributions of source code must retain the above copyright notice,
#                       this list of conditions and the following disclaimer.
#               *       Redistributions in binary form must reproduce the above copyright notice,
#                       this list of conditions and the following disclaimer in the documentation
#                       and/or other materials provided with the distribution.
#               *       Neither the name of the company nor the names of its contributors may be
#                       used to endorse or promote products derived from this software without
#                       specific prior written permission.
#
#       THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
#       ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
#       WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
#       IN NO EVENT SHALL THE OFFICERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
#       INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
#       TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
#       OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
#       IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
#       IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
#       SUCH DAMAGE.
#
#####################################################################################

__docformat__ = 'reStructuredText'

from importlib import import_module

import sys
import os
import re
import logging

from urllib.parse import urlunparse
from os.path import abspath, dirname, expanduser, join
from copy import copy, deepcopy
from optparse import OptionParser

import docutils.readers.doctree
import docutils.core
import docutils.nodes
from docutils.parsers.rst import directives
from docutils.readers import standalone
from docutils.transforms import Transform

try:
    from roman import toRoman
except ImportError:
    from docutils.utils.roman import toRoman

import reportlab
from reportlab.lib.units import cm
from reportlab.platypus import doctemplate
from reportlab.platypus.doctemplate import (
    ActionFlowable,
    BaseDocTemplate,
    FrameActionFlowable,
    IndexingFlowable,
    LayoutError,
    PageTemplate,
)
from reportlab.platypus.flowables import (
    _listWrapOn,
    _Container,
    Flowable,
    ImageAndFlowables,
    PageBreak,
    SlowPageBreak,
)
from reportlab.platypus.paragraph import Paragraph
from reportlab.platypus.tables import TableStyle

from . import config

from rst2pdf.directives import code_block
from rst2pdf import flowables
from rst2pdf.flowables import (
    BoundByWidth,
    DelayedTable,
    Heading,
    MyPageBreak,
    MySpacer,
    OddEven,
    Separation,
    SmartFrame,
    XXPreformatted,
)
from rst2pdf.sinker import Sinker
from rst2pdf.image import MyImage, missing
from rst2pdf.log import log, nodeid
from smartypants import smartypants
from rst2pdf import styles as sty
from rst2pdf.nodehandlers import nodehandlers
from rst2pdf.languages import get_language_available

# Template engine for covers
import jinja2

# Side effects
from rst2pdf.directives import aafigure  # noqa
from rst2pdf.directives import contents  # noqa
from rst2pdf.directives import oddeven  # noqa
from rst2pdf.roles import counter as counter_role  # noqa
from rst2pdf.roles import package as package_role  # noqa

try:
    import sphinx as sphinx_module
except ImportError:
    sphinx_module = None


numberingstyles = {
    'arabic': 'ARABIC',
    'roman': 'ROMAN_UPPER',
    'lowerroman': 'ROMAN_LOWER',
    'alpha': 'LETTERS_UPPER',
    'loweralpha': 'LETTERS_LOWER',
}


class RstToPdf(object):
    def __init__(
        self,
        stylesheets=[],
        language='en_US',
        header=None,
        footer=None,
        inlinelinks=False,
        breaklevel=1,
        font_path=[],
        style_path=[],
        fit_mode='shrink',
        background_fit_mode='center',
        sphinx=False,
        smarty='0',
        baseurl=None,
        repeat_table_rows=False,
        footnote_backlinks=True,
        inline_footnotes=False,
        real_footnotes=False,
        def_dpi=300,
        show_frame=False,
        highlightlang='python',  # this one is only used by Sphinx
        basedir=os.getcwd(),
        splittables=False,
        blank_first_page=False,
        first_page_on_right=False,
        breakside='odd',
        custom_cover='cover.tmpl',
        floating_images=False,
        numbered_links=False,
        section_header_depth=2,
        toc_depth=0,
        raw_html=False,
        strip_elements_with_classes=[],
    ):
        self.debugLinesPdf = False
        self.depth = 0
        self.breakside = breakside
        self.first_page_on_right = first_page_on_right
        self.blank_first_page = blank_first_page
        self.splittables = splittables
        self.basedir = basedir
        self.language, self.docutils_language = get_language_available(language)[:2]
        self.doc_title = ""
        self.doc_title_clean = ""
        self.doc_subtitle = ""
        self.doc_author = ""
        self.header = header
        self.footer = footer
        self.custom_cover = custom_cover
        self.floating_images = floating_images
        self.decoration = {
            'header': header,
            'footer': footer,
            'endnotes': [],
            'extraflowables': [],
        }
        # find base path
        if hasattr(sys, 'frozen'):
            self.PATH = abspath(dirname(sys.executable))
        else:
            self.PATH = abspath(dirname(__file__))

        self.font_path = font_path
        self.style_path = style_path
        self.def_dpi = def_dpi
        self.loadStyles(stylesheets)

        self.docutils_languages = {}
        self.inlinelinks = inlinelinks
        self.breaklevel = breaklevel
        self.fit_mode = fit_mode
        self.background_fit_mode = background_fit_mode
        self.to_unlink = []

        # See https://pythonhosted.org/smartypants/reference.html#smartypants-module
        self.smartypants_attributes = 0
        if smarty == '1':
            self.smartypants_attributes = 1 | 6 | 8 | 64 | 512
        elif smarty == '2':
            self.smartypants_attributes = 1 | 6 | 24 | 64 | 512
        elif smarty == '3':
            self.smartypants_attributes = 1 | 6 | 40 | 64 | 512

        self.baseurl = baseurl
        self.repeat_table_rows = repeat_table_rows
        self.footnote_backlinks = footnote_backlinks
        self.inline_footnotes = inline_footnotes
        self.real_footnotes = real_footnotes
        # Real footnotes are always a two-pass thing.
        if self.real_footnotes:
            self.mustMultiBuild = True
        self.def_dpi = def_dpi
        self.show_frame = show_frame
        self.numbered_links = numbered_links
        self.section_header_depth = section_header_depth
        self.toc_depth = toc_depth
        self.img_dir = os.path.join(self.PATH, 'images')
        self.raw_html = raw_html
        self.strip_elements_with_classes = strip_elements_with_classes

        # Sorry about this, but importing sphinx.roles makes some
        # ordinary documents fail (demo.txt specifically) so
        # I can' t just try to import it outside. I need
        # to do it only if it's requested
        if sphinx and sphinx_module:
            import sphinx.roles
            from rst2pdf.sphinxnodes import sphinxhandlers

            self.highlightlang = highlightlang
            self.gen_pdftext, self.gen_elements = sphinxhandlers(self)
        else:
            # These rst2pdf extensions conflict with sphinx
            directives.register_directive('code-block', code_block.code_block_directive)
            directives.register_directive('code', code_block.code_block_directive)
            self.gen_pdftext, self.gen_elements = nodehandlers(self)

        self.sphinx = sphinx

        if not self.styles.languages:
            self.styles.languages = []
            if self.language:
                self.styles.languages.append(self.language)
                self.styles['bodytext'].language = self.language
            else:
                self.styles.languages.append('en_US')
                self.styles['bodytext'].language = 'en_US'
        # Load the docutils language modules for all required languages
        for lang in self.styles.languages:
            self.docutils_languages[lang] = get_language_available(lang)[2]

        self.pending_targets = []
        self.targets = []

    def loadStyles(self, styleSheets=None):

        if styleSheets is None:
            styleSheets = []

        self.styles = sty.StyleSheet(
            styleSheets, self.font_path, self.style_path, def_dpi=self.def_dpi
        )

    def style_language(self, style):
        """Return language corresponding to this style."""
        try:
            return style.language
        except AttributeError:
            pass
        try:
            return self.styles['bodytext'].language
        except AttributeError:
            return os.environ['LANG'] or 'en'

    def text_for_label(self, label, style):
        """Translate text for label."""
        try:
            text = self.docutils_languages[self.style_language(style)].labels[label]
        except KeyError:
            text = label.capitalize()
        return text

    def text_for_bib_field(self, field, style):
        """Translate text for bibliographic fields."""
        try:
            text = self.docutils_languages[
                self.style_language(style)
            ].bibliographic_fields[field]
        except KeyError:
            text = field
        return text + ":"

    def author_separator(self, style):
        """Return separator string for authors."""
        try:
            sep = self.docutils_languages[self.style_language(style)].author_separators[
                0
            ]
        except KeyError:
            sep = ';'
        return sep + " "

    def styleToTags(self, style):
        """Takes a style name, returns a pair of opening/closing tags for it, like
        "<font face=helvetica size=14 color=red>". Used for inline
        nodes (custom interpreted roles)"""

        try:
            s = self.styles[style]
            r1 = [
                '<font face="%s" color="#%s" ' % (s.fontName, s.textColor.hexval()[2:])
            ]
            bc = s.backColor
            if bc:
                r1.append('backColor="#%s"' % bc.hexval()[2:])
            if s.trueFontSize:
                r1.append('size="%d"' % s.fontSize)
            r1.append('>')
            r2 = ['</font>']

            if s.strike:
                r1.append('<strike>')
                r2.insert(0, '</strike>')
            if s.underline:
                r1.append('<u>')
                r2.insert(0, '</u>')

            return [''.join(r1), ''.join(r2)]
        except KeyError:
            log.warning('Unknown class %s', style)
            return None

    def styleToFont(self, style):
        """Takes a style name, returns a font tag for it, like
        "<font face=helvetica size=14 color=red>". Used for inline
        nodes (custom interpreted roles)"""

        try:
            s = self.styles[style]
            r = [
                '<font face="%s" color="#%s" ' % (s.fontName, s.textColor.hexval()[2:])
            ]
            bc = s.backColor
            if bc:
                r.append('backColor="#%s"' % bc.hexval()[2:])
            if s.trueFontSize:
                r.append('size="%d"' % s.fontSize)
            r.append('>')
            return ''.join(r)
        except KeyError:
            log.warning('Unknown class %s', style)
            return None

    def gather_pdftext(self, node, replaceEnt=True):
        return ''.join([self.gen_pdftext(n, replaceEnt) for n in node.children])

    def gather_elements(self, node, style=None):
        if style is None:
            style = self.styles.styleForNode(node)
        r = []
        if 'float' in style.__dict__:
            style = None  # Don't pass floating styles to children!
        for n in node.children:
            r.extend(self.gen_elements(n, style=style))
        return r

    def bullet_for_node(self, node):
        """Takes a node, assumes it's some sort of
        item whose parent is a list, and
        returns the bullet text it should have"""
        b = ""
        t = 'item'
        if node.parent.get('start'):
            start = int(node.parent.get('start'))
        else:
            start = 1

        if node.parent.get('bullet') or isinstance(
            node.parent, docutils.nodes.bullet_list
        ):
            b = node.parent.get('bullet', '*')
            if b == "None":
                b = ""
            t = 'bullet'

        elif node.parent.get('enumtype') == 'arabic':
            b = str(node.parent.children.index(node) + start) + '.'

        elif node.parent.get('enumtype') == 'lowerroman':
            b = toRoman(node.parent.children.index(node) + start).lower() + '.'
        elif node.parent.get('enumtype') == 'upperroman':
            b = toRoman(node.parent.children.index(node) + start).upper() + '.'
        elif node.parent.get('enumtype') == 'loweralpha':
            b = (
                'abcdefghijklmnopqrstuvwxyz'[
                    node.parent.children.index(node) + start - 1
                ]
                + '.'
            )
        elif node.parent.get('enumtype') == 'upperalpha':
            b = (
                'ABCDEFGHIJKLMNOPQRSTUVWXYZ'[
                    node.parent.children.index(node) + start - 1
                ]
                + '.'
            )
        else:
            log.critical("Unknown kind of list_item %s [%s]", node.parent, nodeid(node))
        return b, t

    def filltable(self, rows):
        """
        Takes a list of rows, consisting of cells and performs the following fixes:

        * For multicolumn cells, add continuation cells, to make all rows the same
        size. These cells have to be multirow if the original cell is multirow.

        * For multirow cell, insert continuation cells, to make all columns the
        same size.

        * If there are still shorter rows, add empty cells at the end (ReST quirk)

        * Once the table is *normalized*, create spans list, fitting for reportlab's
        Table class.

        """

        # If there is a multicol cell, we need to insert Continuation Cells
        # to make all rows the same length

        for y in range(0, len(rows)):
            for x in range(len(rows[y]) - 1, -1, -1):
                cell = rows[y][x]
                if isinstance(cell, str):
                    continue
                if cell.get("morecols"):
                    for i in range(0, cell.get("morecols")):
                        e = docutils.nodes.entry("")
                        e["morerows"] = cell.get("morerows", 0)
                        rows[y].insert(x + 1, e)

        for y in range(0, len(rows)):
            for x in range(0, len(rows[y])):
                cell = rows[y][x]
                if isinstance(cell, str):
                    continue
                if cell.get("morerows"):
                    for i in range(0, cell.get("morerows")):
                        rows[y + i + 1].insert(x, "")

        # If a row is shorter, add empty cells at the right end
        maxw = max([len(r) for r in rows])
        for r in rows:
            while len(r) < maxw:
                r.append("")

        # Create spans list for reportlab's table style
        spans = []
        for y in range(0, len(rows)):
            for x in range(0, len(rows[y])):
                cell = rows[y][x]
                if isinstance(cell, str):
                    continue
                if cell.get("morecols"):
                    mc = cell.get("morecols")
                else:
                    mc = 0
                if cell.get("morerows"):
                    mr = cell.get("morerows")
                else:
                    mr = 0
                if mc or mr:
                    spans.append(('SPAN', (x, y), (x + mc, y + mr)))
        return spans

    def PreformattedFit(self, text, style):
        """Preformatted section that gets horizontally compressed if needed."""
        # Pass a ridiculous size, then it will shrink to what's available
        # in the frame
        return BoundByWidth(
            2000 * cm,
            content=[XXPreformatted(text, style)],
            mode=self.fit_mode,
            style=style,
        )

    def createPdf(
        self,
        text=None,
        source_path=None,
        output=None,
        doctree=None,
        compressed=False,
        # This adds entries to the PDF TOC
        # matching the rst source lines
        debugLinesPdf=False,
    ):
        """Create a PDF from text (ReST input),
        or doctree (docutil nodes) and save it in outfile.

        If outfile is a string, it's a filename.
        If it's something with a write method, (like a StringIO,
        or a file object), the data is saved there.

        """
        self.decoration = {
            'header': self.header,
            'footer': self.footer,
            'endnotes': [],
            'extraflowables': [],
        }

        self.pending_targets = []
        self.targets = []

        self.debugLinesPdf = debugLinesPdf

        if doctree is None:
            if text is not None:
                if self.language:
                    settings_overrides = {'language_code': self.docutils_language}
                else:
                    settings_overrides = {}
                settings_overrides[
                    'strip_elements_with_classes'
                ] = self.strip_elements_with_classes
                settings_overrides['exit_status_level'] = 3

                try:
                    self.doctree = docutils.core.publish_doctree(
                        text,
                        source_path=source_path,
                        settings_overrides=settings_overrides,
                    )
                    log.debug(self.doctree)
                except Exception as e:
                    if log.isEnabledFor(logging.INFO):
                        # Log exception with traceback if more detailed logging has been set
                        log.exception('Error generating doctree')
                    else:
                        log.error(f"Error generating doctree: {e}")
                    log.error("Cannot generate PDF, exiting")
                    return 1
            else:
                log.error('Error: createPdf needs a text or a doctree')
                return 1
        else:
            self.doctree = doctree

        if self.numbered_links:
            # Transform all links to sections so they show numbers
            from .sectnumlinks import SectNumFolder, SectRefExpander

            snf = SectNumFolder(self.doctree)
            self.doctree.walk(snf)
            srf = SectRefExpander(self.doctree, snf.sectnums)
            self.doctree.walk(srf)

        if self.strip_elements_with_classes:
            from docutils.transforms.universal import StripClassesAndElements

            sce = StripClassesAndElements(self.doctree)
            sce.apply()

        if self.toc_depth == 0:
            # use the `:depth:` option from `.. contents::`
            self.toc_depth = contents.Contents.depth

        try:
            elements = self.gen_elements(self.doctree)
        except Exception as e:
            if log.isEnabledFor(logging.INFO):
                # Log exception with traceback if more detailed logging has been set
                log.exception('Error generating document elements')
            else:
                log.error(f"Error generating document elements: {e}")
            log.error("Cannot generate PDF, exiting")
            return 1

        # Find cover template, save it in cover_file
        jinja_env = jinja2.Environment(
            loader=jinja2.FileSystemLoader(
                [
                    self.basedir,
                    os.path.expanduser('~/.rst2pdf'),
                    os.path.join(self.PATH, 'templates'),
                ]
            ),
            autoescape=jinja2.select_autoescape(['html', 'xml']),
        )

        try:
            template = jinja_env.get_template(self.custom_cover)
        except jinja2.TemplateNotFound:
            log.error("Can't find cover template %s, using default" % self.custom_cover)
            template = jinja_env.get_template('cover.tmpl')

        # Feed data to the template, get restructured text.
        cover_text = template.render(title=self.doc_title, subtitle=self.doc_subtitle)

        # This crashes sphinx because .. class:: in sphinx is
        # something else. Ergo, pdfbuilder does it in its own way.
        if not self.sphinx:
            elements = (
                self.gen_elements(
                    publish_secondary_doctree(cover_text, self.doctree, source_path)
                )
                + elements
            )

        if self.blank_first_page:
            elements.insert(0, PageBreak())

        # Put the endnotes at the end ;-)
        endnotes = self.decoration['endnotes']
        if endnotes:
            elements.append(MySpacer(1, 2 * cm))
            elements.append(Separation())
            for n in self.decoration['endnotes']:
                t_style = TableStyle(self.styles['endnote'].commands)
                colWidths = self.styles['endnote'].colWidths
                elements.append(
                    DelayedTable([[n[0], n[1]]], style=t_style, colWidths=colWidths)
                )

        if self.floating_images:
            # Handle images with alignment more like in HTML
            new_elem = []
            for i, e in enumerate(elements[::-1]):
                if isinstance(e, MyImage) and e.image.hAlign != 'CENTER' and new_elem:
                    # This is an image where flowables should wrap
                    # around it
                    popped = new_elem.pop()
                    new_elem.append(
                        ImageAndFlowables(e, popped, imageSide=e.image.hAlign.lower())
                    )
                else:
                    new_elem.append(e)

            elements = new_elem
            elements.reverse()

        head = self.decoration['header']
        foot = self.decoration['footer']

        # So, now, create the FancyPage with the right sizes and elements
        FP = FancyPage("fancypage", head, foot, self)

        def cleantags(s):
            re.sub(r'<[^>]*?>', '', str(s).strip())

        pdfdoc = FancyDocTemplate(
            output,
            pageTemplates=[FP],
            showBoundary=0,
            pagesize=self.styles.ps,
            title=self.doc_title_clean,
            author=self.doc_author,
            pageCompression=compressed,
        )
        pdfdoc.client = self

        # Handle totally empty documents (Issue #547)
        if not elements:
            elements.append(Paragraph("", style=self.styles['base']))
        if getattr(self, 'mustMultiBuild', False):
            # Force a multibuild pass
            if not isinstance(elements[-1], UnhappyOnce):
                log.info('Forcing second pass so Total pages work')
                elements.append(UnhappyOnce())
        while True:
            try:
                log.info("Starting build")
                self.elements = elements
                # See if this *must* be multipass
                pdfdoc.multiBuild(elements)
                # Force a multibuild pass

                # FIXME: since mustMultiBuild is set by the
                # first pass in the case of ###Total###, then we
                # make a new forced two-pass build. This is broken.
                # conceptually.

                if getattr(self, 'mustMultiBuild', False):
                    # Force a multibuild pass
                    if not isinstance(elements[-1], UnhappyOnce):
                        log.info('Forcing second pass so Total pages work')
                        elements.append(UnhappyOnce())
                        continue
                # Rearrange footnotes if needed
                if self.real_footnotes:
                    newStory = []
                    fnPile = []
                    for e in elements:
                        if getattr(e, 'isFootnote', False):
                            # Add it to the pile
                            # if not isinstance (e, MySpacer):
                            fnPile.append(e)
                        elif getattr(e, '_atTop', False) or isinstance(
                            e, (UnhappyOnce, MyPageBreak)
                        ):
                            if fnPile:
                                fnPile.insert(0, Separation())
                                newStory.append(Sinker(fnPile))
                            newStory.append(e)
                            fnPile = []
                        else:
                            newStory.append(e)
                    elements = newStory + fnPile
                    for e in elements:
                        if hasattr(e, '_postponed'):
                            delattr(e, '_postponed')
                    self.real_footnotes = False
                    continue

                break
            except ValueError:
                # FIXME: cross-document links come through here, which means
                # an extra pass per cross-document reference. Which sucks.
                # if v.args and str(v.args[0]).startswith('format not resolved'):
                # missing=str(v.args[0]).split(' ')[-1]
                # log.error('Adding missing reference to %s and rebuilding. This is slow!'%missing)
                # elements.append(Reference(missing))
                # for e in elements:
                # if hasattr(e,'_postponed'):
                # delattr(e,'_postponed')
                # else:
                # raise
                raise

        # doc = SimpleDocTemplate("phello.pdf")
        # doc.build(elements)
        for fn in self.to_unlink:
            try:
                os.unlink(fn)
            except OSError:
                pass

        return 0


class FancyDocTemplate(BaseDocTemplate):
    def onProgress(self, typ, value):
        global _counter
        message = ''
        if typ == 'SIZE_EST':
            log.debug(f'Number of flowables: {value}')
        elif typ == 'PROGRESS':
            message = f'Flowable {value}'
            # add class name for this flowable if we can
            if hasattr(self.client, 'elements'):
                if 0 <= value < len(self.client.elements):
                    element = self.client.elements[value]
                    message += f" {type(element)}"
            log.debug(f'Page {_counter}: {message}')

    def afterInit(self):
        self.setProgressCallBack(self.onProgress)

    def afterFlowable(self, flowable):

        if isinstance(flowable, Heading):
            # Notify TOC entry for headings/abstracts/dedications.
            level, text = flowable.level, flowable.text
            parent_id = flowable.parent_id
            node = flowable.node
            pagenum = setPageCounter()
            self.notify('TOCEntry', (level, text, pagenum, parent_id, node))

    def handle_flowable(self, flowables):
        '''try to handle one flowable from the front of list flowables.'''

        # this method is copied from reportlab

        # allow document a chance to look at, modify or ignore
        # the object(s) about to be processed
        self.filterFlowables(flowables)

        self.handle_breakBefore(flowables)
        self.handle_keepWithNext(flowables)
        f = flowables[0]
        del flowables[0]
        if f is None:
            return

        if isinstance(f, PageBreak):
            if isinstance(f, SlowPageBreak):
                self.handle_pageBreak(slow=1)
            else:
                self.handle_pageBreak()
            self.afterFlowable(f)
        elif isinstance(f, ActionFlowable):
            f.apply(self)
            self.afterFlowable(f)
        else:
            frame = self.frame
            canv = self.canv
            # try to fit it then draw it
            if frame.add(f, canv, trySplit=self.allowSplitting):
                if not isinstance(f, FrameActionFlowable):
                    self._curPageFlowableCount += 1
                    self.afterFlowable(f)
                doctemplate._addGeneratedContent(flowables, frame)
            else:
                if self.allowSplitting:
                    # see if this is a splittable thing
                    S = frame.split(f, canv)
                    n = len(S)
                else:
                    n = 0
                if n:
                    if not isinstance(S[0], (PageBreak, SlowPageBreak, ActionFlowable)):
                        if frame.add(S[0], canv, trySplit=0):
                            self._curPageFlowableCount += 1
                            self.afterFlowable(S[0])
                            doctemplate._addGeneratedContent(flowables, frame)
                        else:
                            ident = "Splitting error(n==%d) on page %d in\n%s" % (
                                n,
                                self.page,
                                self._fIdent(f, 60, frame),
                            )
                            # leave to keep apart from the raise
                            raise LayoutError(ident)
                        del S[0]
                    for i, f in enumerate(S):
                        flowables.insert(i, f)  # put split flowables back on the list
                else:
                    if hasattr(f, '_postponed') and f._postponed > 4:
                        ident = "Flowable %s%s too large on page %d in frame %r%s of template %r" % (
                            self._fIdent(f, 60, frame),
                            doctemplate._fSizeString(f),
                            self.page,
                            self.frame.id,
                            self.frame._aSpaceString(),
                            self.pageTemplate.id,
                        )
                        # leave to keep apart from the raise
                        raise LayoutError(ident)
                    # this ought to be cleared when they are finally drawn!
                    f._postponed = 1
                    mbe = getattr(self, '_multiBuildEdits', None)
                    if mbe:
                        mbe((delattr, f, '_postponed'))
                    flowables.insert(0, f)  # put the flowable back
                    self.handle_frameEnd()


_counter = 0
_counterStyle = 'arabic'


class PageCounter(Flowable):
    def __init__(self, number=0, style='arabic'):
        self.style = str(style).lower()
        self.number = int(number)
        Flowable.__init__(self)

    def wrap(self, availWidth, availHeight):
        global _counter, _counterStyle
        _counterStyle = self.style
        _counter = self.number
        return (self.width, self.height)

    def drawOn(self, canvas, x, y, _sW):
        pass


flowables.PageCounter = PageCounter


def setPageCounter(counter=None, style=None):

    global _counter, _counterStyle

    if counter is not None:
        _counter = counter
    if style is not None:
        _counterStyle = style

    if _counterStyle == 'lowerroman':
        ptext = toRoman(_counter).lower()
    elif _counterStyle == 'roman':
        ptext = toRoman(_counter).upper()
    elif _counterStyle == 'alpha':
        ptext = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'[_counter % 26]
    elif _counterStyle == 'loweralpha':
        ptext = 'abcdefghijklmnopqrstuvwxyz'[_counter % 26]
    else:
        ptext = str(_counter)
    return ptext


class MyContainer(_Container, Flowable):
    pass


class UnhappyOnce(IndexingFlowable):
    """An indexing flowable that is only unsatisfied once.
    If added to a story, it will make multiBuild run
    at least two passes. Useful for ###Total###"""

    _unhappy = True

    def isSatisfied(self):
        if self._unhappy:
            self._unhappy = False
            return False
        return True

    def draw(self):
        pass


class HeaderOrFooter(object):
    """A helper object for FancyPage (below)
    HeaderOrFooter handles operations which are common
    to both headers and footers
    """

    def __init__(self, items=None, isfooter=False, client=None):
        self.items = items
        if isfooter:
            locinfo = 'footer showFooter defaultFooter footerSeparator'
        else:
            locinfo = 'header showHeader defaultHeader headerSeparator'
        self.isfooter = isfooter
        self.loc, self.showloc, self.defaultloc, self.addsep = locinfo.split()
        self.totalpages = 0
        self.client = client

    def prepare(self, pageobj, canv, doc):
        showloc = pageobj.template.get(self.showloc, True)
        height = 0
        items = self.items
        if showloc:
            if not items:
                items = pageobj.template.get(self.defaultloc)
                if items:
                    items = self.client.gen_elements(
                        publish_secondary_doctree(items, self.client.doctree, None)
                    )
            if items:
                if isinstance(items, list):
                    items = items[:]
                else:
                    items = [Paragraph(items, pageobj.styles[self.loc])]
                addsep = pageobj.template.get(self.addsep, False)
                if addsep:
                    if self.isfooter:
                        items.insert(0, Separation())
                    else:
                        items.append(Separation())
                _, height = _listWrapOn(items, pageobj.tw, canv)
        self.prepared = height and items
        return height

    def replaceTokens(self, elems, canv, doc, smarty):
        """Put doc_title/page number/etc in text of header/footer."""

        # Make sure page counter is up to date
        pnum = setPageCounter()

        def replace(text):
            # Ensure text is unicode
            if isinstance(text, bytes):
                try:
                    text = text.decode(e.encoding)
                except (AttributeError, TypeError):
                    text = text.decode('utf-8')

            text = text.replace(u'###Page###', pnum)
            if '###Total###' in text:
                text = text.replace(u'###Total###', str(self.totalpages))
                self.client.mustMultiBuild = True
            text = text.replace(u"###Title###", doc.title)
            text = text.replace(u"###Section###", getattr(canv, 'sectName', ''))
            text = text.replace(u"###SectNum###", getattr(canv, 'sectNum', ''))
            text = smartypants(text, smarty)
            return text

        for i, e in enumerate(elems):
            # TODO: implement a search/replace for arbitrary things
            if isinstance(e, Paragraph):
                text = replace(e.text)
                elems[i] = Paragraph(text, e.style)
            elif isinstance(e, DelayedTable):
                data = deepcopy(e.data)
                for r, row in enumerate(data):
                    for c, cell in enumerate(row):
                        if isinstance(cell, list):
                            data[r][c] = self.replaceTokens(cell, canv, doc, smarty)
                        else:
                            row[c] = self.replaceTokens([cell], canv, doc, smarty)[0]
                elems[i] = DelayedTable(data, e._colWidths, e.style)
            elif isinstance(e, BoundByWidth):
                for index, item in enumerate(e.content):
                    if isinstance(item, Paragraph):
                        e.content[index] = Paragraph(replace(item.text), item.style)
                elems[i] = e
            elif isinstance(e, OddEven):
                odd = self.replaceTokens([e.odd], canv, doc, smarty)[0]
                even = self.replaceTokens([e.even], canv, doc, smarty)[0]
                elems[i] = OddEven(odd, even)

        return elems

    def draw(self, pageobj, canv, doc, x, y, width, height):
        self.totalpages = max(self.totalpages, doc.page)
        items = self.prepared
        if items:
            self.replaceTokens(items, canv, doc, pageobj.smartypants_attributes)
            container = MyContainer()
            container._content = items
            container.width = width
            container.height = height
            container.drawOn(canv, x, y)


class FancyPage(PageTemplate):
    """A page template that handles changing layouts."""

    def __init__(self, _id, _head, _foot, client):
        self.client = client
        self.styles = client.styles
        self._head = HeaderOrFooter(_head, client=client)
        self._foot = HeaderOrFooter(_foot, True, client)
        self.smartypants_attributes = client.smartypants_attributes
        self.show_frame = client.show_frame
        self.image_cache = {}
        PageTemplate.__init__(self, _id, [])

    def draw_background(self, which, canv):
        """Draws a background and/or foreground image
        on each page which uses the template.

        Calculates the image one time, and caches
        it for reuse on every page in the template.

        How the background is drawn depends on the
        --fit-background-mode option.

        If desired, we could add code to push it around
        on the page, using stylesheets to align and/or
        set the offset.
        """
        uri = self.template[which]
        background_fit_mode = self.client.background_fit_mode
        if 'background_fit_mode' in self.template:
            background_fit_mode = self.template['background_fit_mode']

        info = self.image_cache.get(f"{uri}{background_fit_mode}")
        if info is None:
            fname, _, _ = MyImage.split_uri(uri)
            if not os.path.exists(fname):
                del self.template[which]
                log.error("Missing %s image file: %s", which, uri)
                return
            try:
                w, h, _ = MyImage.size_for_node(
                    dict(
                        uri=uri,
                    ),
                    self.client,
                )
            except ValueError:
                # Broken image, return arbitrary stuff
                uri = missing
                w, h, = (
                    100,
                    100,
                )

            pw, ph = self.styles.pw, self.styles.ph
            if background_fit_mode == 'center':
                scale = min(1.0, 1.0 * pw / w, 1.0 * ph / h)
                sw, sh = w * scale, h * scale
                x, y = (pw - sw) / 2.0, (ph - sh) / 2.0
            elif background_fit_mode == 'scale':
                x, y = 0, 0
                sw, sh = pw, ph
            elif background_fit_mode == 'scale_width':
                x, y = 0, 0
                sw, sh = pw, h
            else:
                log.error('Unknown background fit mode: %s' % background_fit_mode)
                # Do scale anyway
                x, y = 0, 0
                sw, sh = pw, ph

            bg = MyImage(uri, sw, sh, client=self.client)
            self.image_cache[uri] = info = bg, x, y
        bg, x, y = info
        bg.drawOn(canv, x, y)

    def is_left(self, page_num):
        """Default behavior is that the first page is on the left.

        If the user has --first_page_on_right, the calculation is reversed.
        """
        val = page_num % 2 == 1
        if self.client.first_page_on_right:
            val = not val
        return val

    def beforeDrawPage(self, canv, doc):
        """Do adjustments to the page according to where we are in the document.

        * Gutter margins on left or right as needed

        """

        global _counter, _counterStyle

        styles = self.styles
        self.tw = styles.pw - styles.lm - styles.rm - styles.gm

        # What page template to use?
        tname = canv.__dict__.get('templateName', self.styles.firstTemplate)
        if tname not in self.styles.pageTemplates:
            log.error(f"Template '{tname}' is not defined")
            sys.exit(1)
        self.template = self.styles.pageTemplates[tname]
        canv.templateName = tname

        # Any background with fit mode?
        background = canv.__dict__.get('background', None)
        if background:
            self.template = self.template.copy()
            self.template['background'] = background
            background_fit_mode = canv.__dict__.get('background_fit_mode', None)
            if background_fit_mode:
                self.template['background_fit_mode'] = background_fit_mode

        doct = getattr(canv, '_doctemplate', None)
        canv._doctemplate = None  # to make _listWrapOn work

        if doc.page == 1:
            _counter = 0
            _counterStyle = 'arabic'
        _counter += 1

        # Adjust text space accounting for header/footer

        self.hh = self._head.prepare(self, canv, doc)
        self.fh = self._foot.prepare(self, canv, doc)

        canv._doctemplate = doct

        self.hx = styles.lm
        self.hy = styles.ph - styles.tm - self.hh

        self.fx = styles.lm
        self.fy = styles.bm
        self.th = (
            styles.ph
            - styles.tm
            - styles.bm
            - self.hh
            - self.fh
            - styles.ts
            - styles.bs
        )
        # Adjust gutter margins
        if self.is_left(doc.page):  # Left page
            x1 = styles.lm
        else:  # Right page
            x1 = styles.lm + styles.gm
        y1 = styles.bm + self.fh + styles.bs

        # If there is a background parameter for this page Template, draw it
        if 'background' in self.template:
            self.draw_background('background', canv)

        self.frames = []
        if 'frames' not in self.template:
            log.error('No frames in template')
            sys.exit(1)

        for frame in self.template['frames']:
            frame = frame[:]
            while len(frame) < 8:
                # This is the default in SmartFrame. At some point in the future we
                # may want to change this to 0.
                frame.append(6)
            self.frames.append(
                SmartFrame(
                    self,
                    styles.adjustUnits(frame[0], self.tw) + x1,
                    styles.adjustUnits(frame[1], self.th) + y1,
                    styles.adjustUnits(frame[2], self.tw),
                    styles.adjustUnits(frame[3], self.th),
                    leftPadding=styles.adjustUnits(frame[4], self.tw),
                    bottomPadding=styles.adjustUnits(frame[5], self.th),
                    rightPadding=styles.adjustUnits(frame[6], self.tw),
                    topPadding=styles.adjustUnits(frame[7], self.th),
                    showBoundary=self.show_frame,
                )
            )
        canv.firstSect = True
        canv._pagenum = doc.page
        for frame in self.frames:
            frame._pagenum = doc.page

    def afterDrawPage(self, canv, doc):
        """Draw header/footer."""
        # Adjust for gutter margin
        canv.addPageLabel(
            canv._pageNumber - 1, numberingstyles[_counterStyle], _counter
        )

        log.info('Page %s [%s]' % (_counter, doc.page))
        if self.is_left(doc.page):  # Left page
            hx = self.hx
            fx = self.fx
        else:  # Right Page
            hx = self.hx + self.styles.gm
            fx = self.fx + self.styles.gm

        self._head.draw(self, canv, doc, hx, self.hy, self.tw, self.hh)
        self._foot.draw(self, canv, doc, fx, self.fy, self.tw, self.fh)

        # If there is a foreground parameter for this page Template, draw it
        if 'foreground' in self.template:
            self.draw_background('foreground', canv)


def parse_commandline():

    parser = OptionParser()

    parser.add_option(
        '--config',
        dest='configfile',
        metavar='FILE',
        help='Config file to use. Default=~/.rst2pdf/config',
    )

    parser.add_option(
        '-o', '--output', dest='output', metavar='FILE', help='Write the PDF to FILE'
    )

    def_ssheets = ','.join(
        [
            expanduser(p)
            for p in config.getValue("general", "stylesheets", "").split(',')
        ]
    )
    parser.add_option(
        '-s',
        '--stylesheets',
        dest='style',
        type='string',
        action='append',
        metavar='STYLESHEETS',
        default=[def_ssheets],
        help='A comma-separated list of custom stylesheets. Default="%s"' % def_ssheets,
    )

    def_sheetpath = os.pathsep.join(
        [
            expanduser(p)
            for p in config.getValue("general", "stylesheet_path", "").split(os.pathsep)
        ]
    )
    parser.add_option(
        '--stylesheet-path',
        dest='stylepath',
        metavar='FOLDER%sFOLDER%s...%sFOLDER' % ((os.pathsep,) * 3),
        default=def_sheetpath,
        help='A list of folders to search for stylesheets,'
        ' separated using "%s". Default="%s"' % (os.pathsep, def_sheetpath),
    )

    def_compressed = config.getValue("general", "compressed", False)
    parser.add_option(
        '-c',
        '--compressed',
        dest='compressed',
        action="store_true",
        default=def_compressed,
        help='Create a compressed PDF. Default=%s' % def_compressed,
    )

    parser.add_option(
        '--print-stylesheet',
        dest='printssheet',
        action="store_true",
        default=False,
        help='Print the default stylesheet and exit',
    )

    parser.add_option(
        '--font-folder',
        dest='ffolder',
        metavar='FOLDER',
        help='Search this folder for fonts. (Deprecated)',
    )

    def_fontpath = os.pathsep.join(
        [
            expanduser(p)
            for p in config.getValue("general", "font_path", "").split(os.pathsep)
        ]
    )
    parser.add_option(
        '--font-path',
        dest='fpath',
        metavar='FOLDER%sFOLDER%s...%sFOLDER' % ((os.pathsep,) * 3),
        default=def_fontpath,
        help='A list of folders to search for fonts, separated using "%s".'
        ' Default="%s"' % (os.pathsep, def_fontpath),
    )

    def_baseurl = urlunparse(['file', os.getcwd() + os.sep, '', '', '', ''])
    parser.add_option(
        '--baseurl',
        dest='baseurl',
        metavar='URL',
        default=def_baseurl,
        help='The base URL for relative URLs. Default="%s"' % def_baseurl,
    )

    def_lang = config.getValue("general", "language", 'en_US')
    parser.add_option(
        '-l',
        '--language',
        metavar='LANG',
        default=def_lang,
        dest='language',
        help='Language to be used for hyphenation'
        ' and docutils localizations. Default="%s"' % def_lang,
    )

    def_header = config.getValue("general", "header")
    parser.add_option(
        '--header',
        metavar='HEADER',
        default=def_header,
        dest='header',
        help='Page header if not specified in the document.'
        ' Default="%s"' % def_header,
    )

    def_footer = config.getValue("general", "footer")
    parser.add_option(
        '--footer',
        metavar='FOOTER',
        default=def_footer,
        dest='footer',
        help='Page footer if not specified in the document.'
        ' Default="%s"' % def_footer,
    )

    def_section_header_depth = config.getValue("general", "section_header_depth", 2)
    parser.add_option(
        '--section-header-depth',
        metavar='N',
        default=def_section_header_depth,
        dest='section_header_depth',
        help='''Sections up to this depth will be used in the header and footer's replacement of ###Section###. Default=%s'''
        % def_section_header_depth,
    )

    def_smartquotes = config.getValue("general", "smartquotes", "0")
    parser.add_option(
        "--smart-quotes",
        metavar="VALUE",
        default=def_smartquotes,
        dest="smarty",
        help='Try to convert ASCII quotes, ellipses and dashes'
        ' to the typographically correct equivalent. For details,'
        ' read the man page or the manual. Default="%s"' % def_smartquotes,
    )

    def_fit = config.getValue("general", "fit_mode", "shrink")
    parser.add_option(
        '--fit-literal-mode',
        metavar='MODE',
        default=def_fit,
        dest='fit_mode',
        help='What to do when a literal is too wide. One of error,'
        ' overflow,shrink,truncate. Default="%s"' % def_fit,
    )

    def_fit_background = config.getValue("general", "background_fit_mode", "center")
    parser.add_option(
        '--fit-background-mode',
        metavar='MODE',
        default=def_fit_background,
        dest='background_fit_mode',
        help='How to fit the background image to the page.'
        ' One of scale, scale_width or center. Default="%s"' % def_fit_background,
    )

    parser.add_option(
        '--inline-links',
        action="store_true",
        dest='inlinelinks',
        default=False,
        help='Shows target between parentheses instead of active link.',
    )

    parser.add_option(
        '--repeat-table-rows',
        action="store_true",
        dest='repeattablerows',
        default=False,
        help='Repeats header row for each split table.',
    )

    def_raw_html = config.getValue("general", "raw_html", False)
    parser.add_option(
        '--raw-html',
        action="store_true",
        dest='raw_html',
        default=def_raw_html,
        help='Support embeddig raw HTML. Default=%s' % def_raw_html,
    )

    parser.add_option(
        '-q',
        '--quiet',
        action="store_true",
        dest='quiet',
        default=False,
        help='Print less information.',
    )

    parser.add_option(
        '-v',
        '--verbose',
        action="store_true",
        dest='verbose',
        default=False,
        help='Print debug information.',
    )

    parser.add_option(
        '--very-verbose',
        action="store_true",
        dest='vverbose',
        default=False,
        help='Print even more debug information.',
    )

    parser.add_option(
        '--version',
        action="store_true",
        dest='version',
        default=False,
        help='Print version number and exit.',
    )

    def_footnote_backlinks = config.getValue("general", "footnote_backlinks", True)
    parser.add_option(
        '--no-footnote-backlinks',
        action='store_false',
        dest='footnote_backlinks',
        default=def_footnote_backlinks,
        help='Disable footnote backlinks.'
        ' Default=%s' % str(not def_footnote_backlinks),
    )

    def_inline_footnotes = config.getValue("general", "inline_footnotes", False)
    parser.add_option(
        '--inline-footnotes',
        action='store_true',
        dest='inline_footnotes',
        default=def_inline_footnotes,
        help='Show footnotes inline.' ' Default=%s' % str(not def_inline_footnotes),
    )

    def_real_footnotes = config.getValue("general", "real_footnotes", False)
    parser.add_option(
        '--real-footnotes',
        action='store_true',
        dest='real_footnotes',
        default=def_real_footnotes,
        help='Show footnotes at the bottom of the page where they are defined.'
        ' Default=%s' % str(def_real_footnotes),
    )

    def_dpi = config.getValue("general", "default_dpi", 300)
    parser.add_option(
        '--default-dpi',
        dest='def_dpi',
        metavar='NUMBER',
        default=def_dpi,
        help='DPI for objects sized in pixels. Default=%d' % def_dpi,
    )

    parser.add_option(
        '--show-frame-boundary',
        dest='show_frame',
        action='store_true',
        default=False,
        help='Show frame borders (only useful for debugging). Default=False',
    )

    parser.add_option(
        '--disable-splittables',
        dest='splittables',
        action='store_false',
        default=True,
        help="Don't use splittable flowables in some elements."
        " Only try this if you can't process a document any other way.",
    )

    def_break = config.getValue("general", "break_level", 0)
    parser.add_option(
        '-b',
        '--break-level',
        dest='breaklevel',
        metavar='LEVEL',
        default=def_break,
        help='Maximum section level that starts in a new page.'
        ' Default: %d' % def_break,
    )

    def_blankfirst = config.getValue("general", "blank_first_page", False)
    parser.add_option(
        '--blank-first-page',
        dest='blank_first_page',
        action='store_true',
        default=def_blankfirst,
        help='Add a blank page at the beginning of the document.',
    )

    def_first_page_on_right = config.getValue("general", "first_page_on_right", False)
    parser.add_option(
        '--first-page-on-right',
        dest='first_page_on_right',
        action='store_true',
        default=def_first_page_on_right,
        help='Two-sided book style (where first page starts on the right side)',
    )

    def_breakside = config.getValue("general", "break_side", 'any')
    parser.add_option(
        '--break-side',
        dest='breakside',
        metavar='VALUE',
        default=def_breakside,
        help='How section breaks work. Can be "even", and sections start'
        ' in an even page, "odd", and sections start in odd pages,'
        ' or "any" and sections start in the next page, be it even or odd.'
        ' See also the -b option.',
    )

    parser.add_option(
        '--date-invariant',
        dest='invariant',
        action='store_true',
        default=False,
        help="Don't store the current date in the PDF."
        " Useful mainly for the test suite,"
        " where we don't want the PDFs to change.",
    )

    parser.add_option(
        '-e',
        '--extension-module',
        dest='extensions',
        action="append",
        type="string",
        default=[],
        help="Add a helper extension module to this invocation of rst2pdf "
        "(module must end in .py and be on the python path)",
    )

    def_cover = config.getValue("general", "custom_cover", 'cover.tmpl')
    parser.add_option(
        '--custom-cover',
        dest='custom_cover',
        metavar='FILE',
        default=def_cover,
        help='Template file used for the cover page. Default: %s' % def_cover,
    )

    def_floating_images = config.getValue("general", "floating_images", False)
    parser.add_option(
        '--use-floating-images',
        action='store_true',
        default=def_floating_images,
        help='Makes images with :align: attribute work more like in rst2html. Default: %s'
        % def_floating_images,
        dest='floating_images',
    )

    def_numbered_links = config.getValue("general", "numbered_links", False)
    parser.add_option(
        '--use-numbered-links',
        action='store_true',
        default=def_numbered_links,
        help='When using numbered sections, adds the numbers to all links referring to the section headers. Default: %s'
        % def_numbered_links,
        dest='numbered_links',
    )

    parser.add_option(
        '--strip-elements-with-class',
        action='append',
        dest='strip_elements_with_classes',
        metavar='CLASS',
        help='Remove elements with this CLASS from the output. Can be used multiple times.',
    )

    return parser


def main(_args=None):
    """Parse command line and call createPdf with the correct data."""

    parser = parse_commandline()
    # Fix issue 430: don't overwrite args
    # need to parse_args to see i we have a custom config file
    options, args = parser.parse_args(copy(_args))

    if options.configfile:
        # If there is a config file, we need to reparse
        # the command line because we have different defaults
        config.parseConfig(options.configfile)
        parser = parse_commandline()
        options, args = parser.parse_args(copy(_args))

    if options.version:
        from rst2pdf import version

        print(version)
        sys.exit(0)

    if options.quiet:
        log.setLevel(logging.CRITICAL)

    if options.verbose:
        log.setLevel(logging.INFO)

    if options.vverbose:
        log.setLevel(logging.DEBUG)

    if options.printssheet:
        # find base path
        if hasattr(sys, 'frozen'):
            PATH = abspath(dirname(sys.executable))
        else:
            PATH = abspath(dirname(__file__))
        with open(join(PATH, 'styles', 'styles.yaml')) as fh:
            print(fh.read())
        sys.exit(0)

    filename = False

    if len(args) == 0:
        args = [
            '-',
        ]
    elif len(args) > 2:
        log.critical('Usage: %s [ file.txt [ file.pdf ] ]', sys.argv[0])
        sys.exit(1)
    elif len(args) == 2:
        if options.output:
            log.critical('You may not give both "-o/--output" and second argument')
            sys.exit(1)
        options.output = args.pop()

    close_infile = False
    if args[0] == '-':
        infile = sys.stdin
        options.basedir = os.getcwd()
    elif len(args) > 1:
        log.critical('Usage: %s file.txt [ -o file.pdf ]', sys.argv[0])
        sys.exit(1)
    else:
        filename = args[0]
        options.basedir = os.path.dirname(os.path.abspath(filename))
        try:
            infile = open(filename, 'rb')
            close_infile = True
        except IOError as e:
            log.error(e)
            sys.exit(1)
    options.infile = infile

    if options.output:
        outfile = options.output
        if outfile == '-':
            outfile = sys.stdout.buffer
            options.compressed = False
            # we must stay quiet
            log.setLevel(logging.CRITICAL)
    else:
        if filename:
            if filename.endswith('.txt') or filename.endswith('.rst'):
                outfile = filename[:-4] + '.pdf'
            else:
                outfile = filename + '.pdf'
        else:
            outfile = sys.stdout.buffer
            options.compressed = False
            # we must stay quiet
            log.setLevel(logging.CRITICAL)
            # /reportlab/pdfbase/pdfdoc.py output can
            # be a callable (stringio, stdout ...)
    options.outfile = outfile

    ssheet = []
    if options.style:
        for l in options.style:
            ssheet += l.split(',')
    else:
        ssheet = []
    options.style = [x for x in ssheet if x]

    fpath = []
    if options.fpath:
        fpath = options.fpath.split(os.pathsep)
    if options.ffolder:
        fpath.append(options.ffolder)
    options.fpath = fpath

    spath = []
    if options.stylepath:
        spath = options.stylepath.split(os.pathsep)
    options.stylepath = spath

    if options.real_footnotes:
        options.inline_footnotes = True

    if reportlab.Version < '3.0':
        log.warning(
            'You are using Reportlab version %s.'
            ' The suggested version is 3.0 or higher' % reportlab.Version
        )

    if options.invariant:
        patch_PDFDate()
        patch_digester()

    add_extensions(options)

    return_code = RstToPdf(
        stylesheets=options.style,
        language=options.language,
        header=options.header,
        footer=options.footer,
        inlinelinks=options.inlinelinks,
        breaklevel=int(options.breaklevel),
        baseurl=options.baseurl,
        fit_mode=options.fit_mode,
        background_fit_mode=options.background_fit_mode,
        smarty=str(options.smarty),
        font_path=options.fpath,
        style_path=options.stylepath,
        repeat_table_rows=options.repeattablerows,
        footnote_backlinks=options.footnote_backlinks,
        inline_footnotes=options.inline_footnotes,
        real_footnotes=options.real_footnotes,
        def_dpi=int(options.def_dpi),
        basedir=options.basedir,
        show_frame=options.show_frame,
        splittables=options.splittables,
        blank_first_page=options.blank_first_page,
        first_page_on_right=options.first_page_on_right,
        breakside=options.breakside,
        custom_cover=options.custom_cover,
        floating_images=options.floating_images,
        numbered_links=options.numbered_links,
        raw_html=options.raw_html,
        section_header_depth=int(options.section_header_depth),
        strip_elements_with_classes=options.strip_elements_with_classes,
    ).createPdf(
        text=options.infile.read(),
        source_path=options.infile.name,
        output=options.outfile,
        compressed=options.compressed,
    )

    if close_infile:
        infile.close()

    sys.exit(return_code)


# Ugly hack that fixes Issue 335
reportlab.lib.utils.ImageReader.__deepcopy__ = lambda self, *x: copy(self)


def patch_digester():
    """Patch digester so that we can get the same results when image
    filenames change"""
    import reportlab.pdfgen.canvas as canvas

    cache = {}

    def _digester(s):
        index = cache.setdefault(s, len(cache))
        return 'rst2pdf_image_%s' % index

    canvas._digester = _digester


def patch_PDFDate():
    '''Patch reportlab.pdfdoc.PDFDate so the invariant dates work correctly'''
    from reportlab.pdfbase import pdfdoc
    import reportlab

    class PDFDate(pdfdoc.PDFObject):
        __PDFObject__ = True

        # gmt offset now suppported
        def __init__(self, invariant=True, ts=None, dateFormatter=None):
            now = (2000, 0o1, 0o1, 00, 00, 00, 0)
            self.date = now[:6]
            self.dateFormatter = dateFormatter

        def format(self, doc):
            dfmt = self.dateFormatter or (
                lambda yyyy, mm, dd, hh, m, s: "D:%04d%02d%02d%02d%02d%02d%+03d'%02d'"
                % (yyyy, mm, dd, hh, m, s, 0, 0)
            )
            return pdfdoc.format(pdfdoc.PDFString(dfmt(*self.date)), doc)

    pdfdoc.PDFDate = PDFDate
    reportlab.rl_config.invariant = 1


def add_extensions(options):

    extensions = []
    for ext in options.extensions:
        if not ext.startswith('!'):
            extensions.append(ext)
            continue
        ext = ext[1:]
        try:
            extensions.remove(ext)
        except ValueError:
            log.warning(
                'Could not remove extension %s -- no such extension installed' % ext
            )
        else:
            log.info('Removed extension %s' % ext)

    options.extensions[:] = extensions
    if not extensions:
        return

    class ModuleProxy(object):
        def __init__(self):
            self.__dict__ = globals()

    createpdf = ModuleProxy()
    for modname in options.extensions:
        prefix, modname = os.path.split(modname)
        path_given = prefix
        if modname.endswith('.py'):
            modname = modname[:-3]
            path_given = True
        if not prefix:
            prefix = os.path.join(os.path.dirname(__file__), 'extensions')
            if prefix not in sys.path:
                sys.path.append(prefix)
            prefix = os.getcwd()
        if prefix not in sys.path:
            sys.path.insert(0, prefix)

        log.info('Importing extension module %s', repr(modname))

        firstname = path_given and modname or (modname + '_r2p')
        _names = [firstname, modname]
        import_exc = None
        for _name in _names:
            try:
                module = import_module(_name, 'rst2pdf')
                break
            except ImportError as e:
                import_exc = e
        else:
            if not import_exc:
                continue

            if str(import_exc).split()[-1].replace("'", '') not in [firstname, modname]:
                raise import_exc

            raise SystemExit(
                '\nError: Could not find module %s in sys.path [\n    %s\n]\n'
                'Exiting...\n' % (modname, ',\n    '.join(sys.path))
            )

        if hasattr(module, 'install'):
            module.install(createpdf, options)


def publish_secondary_doctree(text, main_tree, source_path):
    # This is a hack so the text substitutions defined
    # in the document are available when we process the cover
    # page. See Issue 322
    dt = main_tree

    # Add substitutions from the main doctree
    class addSubsts(Transform):
        default_priority = 219

        def apply(self):
            self.document.substitution_defs.update(dt.substitution_defs)
            self.document.substitution_names.update(dt.substitution_names)

    # Use an own reader to modify transformations done.
    class Reader(standalone.Reader):
        def get_transforms(self):
            default = standalone.Reader.get_transforms(self)
            return default + [
                addSubsts,
            ]

    # End of Issue 322 hack

    return docutils.core.publish_doctree(text, reader=Reader(), source_path=source_path)


if __name__ == "__main__":
    main(sys.argv[1:])
