render.py

#!/usr/bin/env python
# coding: utf8
#
# This renders the template files from 'src' into 'www'
#
# Author: Harald Schilly <harald@schil.ly>
# License: Apache 2.0

import os
import sys
from os.path import join, normpath, exists, islink, basename, splitext
# import multiprocessing as mp
from glob import glob
import jinja2 as j2
import markdown

from scripts import log
from conf import config, mirrors

# go to where the script is to get relative paths right
os.chdir(os.path.dirname(os.path.abspath(sys.argv[0])))

SRC = "src"
TARG = "www"  # assumption: completely empty www
TARG_FILES = join(TARG, "files")
j2env = None


def index_changelogs():
    log("indexing changelogs")
    logs = [basename(_) for _ in glob(join("changelogs", "*.txt"))]

    def sortlogs(fn):
        name, version = splitext(basename(fn))[0].split("-")
        key = [0 if name == "pre" else 1]
        key.extend(int(_) for _ in version.split("."))
        return key

    return reversed(sorted(logs, key=sortlogs))


def copy_aux_files():
    """
    This copies auxiliary files into the output file tree.
    For example, the *.bib files from the publications subdirectory.
    """
    log("copying auxiliary files")

    # publication files
    PUB = "publications"
    os.chdir(PUB)
    for bib in glob("*.bib"):
        dst = normpath(join("..", TARG_FILES, bib))
        if exists(dst):
            os.remove(dst)
        os.link(bib, dst)
    os.chdir("..")

    # contributors for the devmap
    for xml in [join("conf", "geocode.xml"), join("conf", "contributors.xml")]:
        dst = join(TARG, "res", basename(xml))
        if exists(dst):
            os.remove(dst)
        os.link(xml, dst)

    # mirror_manager.py files
    for mm in ["metalink.helper", "torrent.helper", "mirror_list"]:
        dst = join(TARG, mm)
        if exists(dst):
            os.remove(dst)
        os.link(join("scripts", mm), dst)


def render_task(arg):
    """
    This is the worker task run on a sub-process,
    it needs TARG and j2env set properly (done inside render())
    """
    fn, root = arg
    src = join(root, fn)
    dst = normpath(join("..", TARG, src))
    lvl = root.count(os.sep)
    # log("processing/f: %s" % src, nl=False)
    if fn.endswith(".html"):
        # we ignore html files starting with "_" (e.g. language specific templates)
        # except the ones in doc, which might be __init__.html
        if fn.startswith("_") and fn != "__init__.html":
            return
        # assume it's a template and process it
        tmpl = j2env.get_template(src)
        c = fn.rsplit(".", 1)[0].split("-", 1)[0]
        content = tmpl.render(level=lvl, filename=fn, category=c)
        with open(dst, "wb") as output:
            output.write(content.encode("utf-8"))
            output.write(b"\n")
    elif islink(src):
        # we have a symlink, copy it
        # log("SYMLINK/files %s" % src)
        if islink(dst):
            os.remove(dst)
        os.symlink(os.readlink(src), dst)
    else:
        # all other files, hardlink them
        # log("hardlink %s -> %s" % (src, dst))
        if exists(dst):
            os.remove(dst)
        os.link(src, dst)


@j2.pass_context
def filter_prefix(ctx, link):
    """
    Prepend level-times "../" to the given string.
    Used to go up in the directory hierarchy.
    Yes, one could also do absolute paths, but then it is harder to debug locally!
    """

    level = ctx.get("level", 0)
    if level == 0:
        return link
    path = ['..'] * level
    path.append(link)
    return '/'.join(path)


md = markdown.Markdown()


@j2.pass_eval_context
def filter_markdown(eval_ctx, text):
    if eval_ctx.autoescape:
        return md.convert(j2.escape(text))
    return md.convert(text)


def render():
    if not exists("www"):
        os.mkdir("www")

    log("config: {} version {} @ {}, {} mirrors and".format(
        config["sage"], config["version"], config["release_date"],
        len(mirrors)))
    # for line in yaml.dump(config, indent=True, default_flow_style=False).splitlines():
    #    log("    %s" % line)

    # everything is now rooted in the src directory
    os.chdir(SRC)

    global j2env
    tmpl_dirs = [join("..", _) for _ in ["publications", "templates", "src"]]
    j2loader = j2.FileSystemLoader(tmpl_dirs)
    j2env = j2.Environment(loader=j2loader, undefined=j2.StrictUndefined)
    j2env.globals.update(config)
    j2env.globals["changelogs"] = index_changelogs()

    j2env.filters["prefix"] = filter_prefix
    j2env.filters["markdown"] = filter_markdown

    IGNORE_PATHS = ["old"]

    # pool must be created *after* global vars are set
    # it forks the main process, it's a "copy-on-write" memory architecture
    # pool = mp.Pool()

    nb_walks = 0
    len_walks = float(len(list(os.walk('.'))))
    for root, paths, filenames in os.walk("."):
        # check if we ignore a branch in a sub-tree
        root_split = root.split(os.sep)
        if len(root_split) > 1 and root_split[1] in IGNORE_PATHS:
            continue

        # path need to exist in the target before we copy and process the files
        for path in paths:
            src = join(root, path)
            dst = normpath(join("..", TARG, src))

            # log("processing/d: %s" % src, nl=False)

            # we have to take care of symlinks here, too!
            if islink(src):
                # log("SYMLINK/paths: %s" % src)
                os.symlink(os.readlink(src), dst)

            elif not exists(dst):
                # log("mkdir %s" % dst)
                os.makedirs(dst)

        # bad error handling, disabled parallelization
        # pool.map(render_task, [(_, root) for _ in filenames])
        for task in [(_, root) for _ in filenames]:
            render_task(task)

        if nb_walks % (len_walks // 27) == 0:
            log("processing: %5.1f%% of %d documents" %
                (100. * nb_walks / len_walks, len_walks),
                nl=False)
        nb_walks += 1

    log("processing: done", nl=False)
    os.chdir("..")
    copy_aux_files()


def reload():
    """
    This little helper will shift-reload the open tab in your web-browser
    via ctrl-shift-F5 -- but only if you have xdotool installed!
    """
    from subprocess import check_output, call, CalledProcessError
    try:
        check_output(["which", "xdotool"])
    except CalledProcessError:
        print("You seem to not have installed 'xdotool'")
        print("$ sudo apt-get install xdotool")
    else:
        # firefox or chrome?
        for browser in ["Chrome", "Mozilla Firefox", "Chromium"]:
            try:
                check_output(["xdotool", "search", "--name", "%s" % browser])
            except CalledProcessError:
                continue
            print("RELOAD ==> detected '%s' and sending Shift+Ctrl+R" %
                  browser)
            call([
                'xdotool', 'search', "--name",
                "%s" % browser, 'key', '--clearmodifiers', 'ctrl+shift+r'
            ])
            break
        else:
            print("==> sorry, could not find your browser?")


if __name__ == '__main__':
    render()
    if len(sys.argv) >= 2 and sys.argv[-1] == "reload":
        reload()
    log('Finished')