new-horizons/docs/generate.py
2022-03-04 00:29:32 -05:00

104 lines
3.1 KiB
Python

import os
from pathlib import Path
from shutil import copytree, rmtree
from htmlmin import minify
from jinja2 import Environment, select_autoescape, FileSystemLoader
from markupsafe import Markup
from json_schema_for_humans.generate import GenerationConfiguration
from markdown import Markdown
from lib.BootstrapExtension import BootstrapExtension
from lib.Schema import Schema
from lib.Page import Page
OUT_DIR = os.getenv("OUT_DIR", "/")
BASE_URL = os.getenv("BASE_URL", "")
if Path("out/").exists():
rmtree("out/", ignore_errors=True)
copytree("content/static", "out")
os.makedirs("out/schemas", exist_ok=True)
env = Environment(
loader=FileSystemLoader("content"),
autoescape=select_autoescape(['jinja2'])
)
markdown_settings = {
'extensions': ['extra', 'toc', 'meta', BootstrapExtension()]
}
schema_settings = GenerationConfiguration(custom_template_path="content/base/schema_base.jinja2")
schema_settings.link_to_reused_ref = False
schema_settings.minify = True
minify_settings = {
'remove_empty_space': True,
'keep_pre': True,
'remove_optional_attribute_quotes': False
}
env.minify_settings = minify_settings
md = Markdown(**markdown_settings)
env.filters["upper_first"] = lambda x: x[0].upper() + x[1:]
env.filters["markdown"] = lambda text: Markup(md.convert(text))
env.filters["static"] = lambda path: str(Path(OUT_DIR, path).as_posix())
pages_paths = Path("content/pages").glob("**/*.md")
schemas_paths = Path("content/schemas").glob("**/*.json")
router = {}
env.filters['route'] = lambda title: router.get(title.lower(), "#")
env.filters['full_url'] = lambda relative: BASE_URL + (relative[1:] if relative[0] == "/" else relative)
pages = []
schemas = []
for page_path in pages_paths:
new_page = Page(page_path, env, markdown_settings)
router[new_page.title.lower()] = OUT_DIR + str(new_page.out_path.relative_to('out/'))
pages.append(new_page)
for schema_path in schemas_paths:
new_schema = Schema(schema_path, env, schema_settings)
router[new_schema.title.lower()] = OUT_DIR + "schemas/" + str(new_schema.out_path.relative_to("out/schemas/"))
schemas.append(new_schema)
content = pages + schemas
if OUT_DIR != "":
router['home'] = OUT_DIR
pages.sort(key=lambda p: p.sort_priority, reverse=True)
schemas.sort(key=lambda s: s.title)
def log_build(in_path, out_path):
print("Building:", str(in_path), "->", str(out_path))
def build_meta(in_path, out_path, do_minify=False):
log_build(in_path, out_path)
meta_template = env.get_template(str(in_path.relative_to("content/")))
with Path("out/", out_path).open(mode="w+", encoding="utf-8") as file:
out = meta_template.render(content=content)
if do_minify:
out = minify(out, **minify_settings)
file.write(out)
print("Building Meta Files")
build_meta(Path("content/sitemap.jinja2"), Path("sitemap.xml"), do_minify=True)
build_meta(Path("content/robots.jinja2"), Path("robots.txt"))
build_meta(Path("content/browserconfig.jinja2"), Path("fav/browserconfig.xml"), do_minify=True)
print("Building Pages")
for item in content:
log_build(item.in_path, item.out_path)
item.render(page=item, pages=pages, schemas=schemas)