import html
import os
import secrets
from datetime import datetime, timezone

from jinja2 import Environment, PackageLoader, select_autoescape

env = Environment(
    loader=PackageLoader("cloudflare_error_page"),
    autoescape=select_autoescape(),
    trim_blocks=True,
    lstrip_blocks=True,
)


def get_resources_folder() -> str:
    """
    Get resources folder that contains stylesheet and icons for the error page.
    If you pass resources_use_cdn=True to render(), local resources are not used.
    """
    return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'resources')


def render(params: dict, allow_html: bool=True, use_cdn: bool=True) -> str:
    """
    Render a customized Cloudflare error page.
    """
    params = {**params}
    if not params.get('time'):
        utc_now = datetime.now(timezone.utc)
        params['time'] = utc_now.strftime("%Y-%m-%d %H:%M:%S UTC")
    if not params.get('ray_id'):
        params['ray_id'] = secrets.token_hex(8)
    if not allow_html:
        params['what_happened'] = html.escape(params.get('what_happened', ''))
        params['what_can_i_do'] = html.escape(params.get('what_can_i_do', ''))

    template = env.get_template("error.html")
    return template.render(params=params, resources_use_cdn=use_cdn)