#!/usr/bin/env python3
import glob
import itertools
from jinja2 import Environment, FileSystemLoader, select_autoescape, pass_context
from markdown import markdown
from markupsafe import Markup
import os
import os.path
import re
import shutil
import yaml

env = Environment(
    loader=FileSystemLoader('templates'),
    autoescape=select_autoescape(['html.j2']),
)


def load_yml_file(fn):
    with open(fn) as f:
        return yaml.safe_load(f)


env.filters['search_id'] = lambda input: input.replace(r'`', '')
env.filters['section_id'] = lambda input: re.sub(r"[^a-zA-Z0-9_]", '', input)
env.filters['entry_id'] = lambda input: re.sub(r"[ `]", '', input)
env.filters['markdownify'] = lambda input: Markup(markdown(input))
env.filters['no_paragraph'] = lambda input: Markup(re.sub(r"</?p>", '', input))

env.globals['unique_id'] = pass_context(
    lambda ctx: str(next(ctx['unique_ctr'])))

env.globals.update(load_yml_file('site.yml'))

env.globals['navigation'] = ['tutorial', 'download', 'manual']


def generate_file(env, fname='content/1.tutorial/default.yml'):
    path, base = os.path.split(fname)
    path = os.path.relpath(path, 'content')
    if path == '.':
        path = ''
        slug = 'index'
        permalink = ''
    else:
        slug = os.path.basename(path)
        permalink = path + '/'

    output_dir = os.path.join('output', path)
    output_path = os.path.join(output_dir, 'index.html')

    template_name = re.sub(r".yml$", '.html.j2', base)

    ctx = load_yml_file(fname)
    ctx.update(unique_ctr=itertools.count(1),
               permalink=permalink,
               slug=slug,
               navitem=path)
    os.makedirs(output_dir, exist_ok=True)
    env.get_template(template_name).stream(ctx).dump(output_path,
                                                     encoding='utf-8')


def copy_public_files(root=''):
    for f in os.scandir(os.path.join('public', root)):
        src = os.path.join(root, f.name)
        dst = os.path.join('output', src)
        if f.is_dir():
            os.makedirs(dst, exist_ok=True)
            copy_public_files(src)
        else:
            shutil.copyfile(f.path, dst)


copy_public_files()

for fn in glob.glob('content/**/*.yml', recursive=True):
    generate_file(env, fn)
