from sphinx.application import TemplateBridge, Sphinx

from sphinx.builders.html import StandaloneHTMLBuilder

from sphinx.builders.webapp.webapp import WebappHTMLTranslator

from sphinx.environment import BuildEnvironment

from sphinx.highlighting import PygmentsBridge

from sphinx.jinja2glue import BuiltinTemplateLoader

from sphinx.theming import Theme

from sphinx.util import relative_uri

from sphinx.writers.html import HTMLTranslator

# this class should be moved somewhere

class SinglefileHTMLBuilder(StandaloneHTMLBuilder, Sphinx):

    """Render a file when requested using data from a pickled archive.

    Handle theming."""

    def __init__(self, www_dir, repodir):

        self.outdir = www_dir

        self.srcdir = repodir

        self.cwd = os.getcwd()

        self.confdir = self.cwd

        self.doctreedir = path.join(self.cwd, '../doctrees')

        self.env = None

        self.freshenv = None

        confdir = path.join(self.cwd, '../../') # <cwd>/_build/webapp/../../

        self.app = Sphinx(self.srcdir, confdir, self.outdir, self.doctreedir,

                          'html', {}, None)

        self.info = self.app.info

        self.warn = self.app.warn

        self.config = self.app.config

        self.config.html_theme_path = 'themes/' + self.config.html_theme

        self.env = BuildEnvironment(self.srcdir, self.doctreedir, self.config)

        self.env.find_files(self.config)

        self.env.set_warnfunc(self.warn)

        self.load_i18n()

        self.load_env()

        StandaloneHTMLBuilder.init(self)

        self.app.add_javascript('jquery.form.js')

        self.app.add_javascript('comments.js')

        self.prepare_writing((None,))

        self.style = self.theme.get_confstr('theme', 'stylesheet')

        self.config.html_additional_pages = {}

        self.images = {}

    def init_translator_class(self):

        self.translator_class = WebappHTMLTranslator

    def render_page(self, docname):

        # rst file has changed and thus we need to build new doctree file

        # and then build new html file

        self.env.read_doc(docname)

        doctree = self.env.get_and_resolve_doctree(docname, self)

        StandaloneHTMLBuilder.write_doc(self, docname, doctree)

    # unused

    def __render_file(self, filename, ctx):

        ctx.update(self.globalcontext.copy())

        # current_page_name is backwards compatibility

        ctx['pagename'] = ctx['current_page_name'] = filename

        def pathto(otheruri, resource=False,

                   baseuri=self.get_target_uri(filename)):

            if not resource:

                otheruri = self.get_target_uri(otheruri)

            return relative_uri(baseuri, otheruri)

        ctx['pathto'] = pathto

        ctx['hasdoc'] = lambda name: name in self.env.all_docs

        pgname = path.basename(filename).split('.')[0]

        ctx['customsidebar'] = self.config.html_sidebars.get(pgname)

        ctx['encoding'] = encoding = self.config.html_output_encoding

        ctx['toctree'] = lambda **kw: self._get_local_toctree(filename, **kw)

        ctx['style'] = self.style

        return self.templates.render_string(tmp_template % (ctx), ctx)