from flask import session, current_app
from flask_login import current_user
from flask_wtf.csrf import generate_csrf


def init_template_directives(app):
    @app.template_global()
    def authorize(power):
        if current_user.username != current_app.config.get("SUPERADMIN"):
            return bool(power in session.get('permissions'))
        else:
            return True

    @app.template_global()
    def csrf_input():
        return f'<input type="hidden" name="csrf_token" value="{generate_csrf()}">'

    @app.template_global()
    def nav_categories():
        """前台左侧侧栏 / 移动端抽屉共用的分类列表（带 anchor 锚点）。

        以函数形式提供而非 context_processor，后台模板不调用就不会查库。
        """
        from applications.view.public.nav import get_nav_categories
        return get_nav_categories()