karkas/src/karkas_blocks/karkas_blocks/standard/miniapp/lib.py

import asyncio
from collections import OrderedDict
from typing import TYPE_CHECKING

import dash
import dash_bootstrap_components as dbc
import flask
from dash_extensions.enrich import DashBlueprint, DashProxy, Input, Output, dcc, html
from dash_extensions.pages import setup_page_components

from karkas_core.modules_system.public_api import get_module, log

from .dash_telegram_auth import get_auth_server, setup_auth_clientcallbacks

if TYPE_CHECKING:
    from karkas_blocks.standard.config import IConfig
    from karkas_blocks.standard.roles import Roles as IRoles

pages = OrderedDict()


def register_page(name, path, blueprint, prefix="", role="USER"):
    pages[path] = {
        "name": name,
        "blueprint": blueprint,
        "prefix": prefix,
        "role": role,
    }


def register_home_page():
    page = DashBlueprint()
    page.layout = html.Div([html.H1("Главная")])
    register_page("Главная", path="/", blueprint=page)


register_home_page()

config: "IConfig" = get_module("standard.config", "config")
Roles: "type[IRoles]" = get_module("standard.roles", "Roles")


def create_dash_app(requests_pathname_prefix: str = None) -> dash.Dash:
    log(requests_pathname_prefix)

    real_prefix = f"{requests_pathname_prefix}_internal/"

    server = get_auth_server(config.get("core::token"))

    app = DashProxy(
        pages_folder="",
        use_pages=True,
        suppress_callback_exceptions=True,
        external_stylesheets=[
            dbc.themes.BOOTSTRAP,
            dbc.icons.BOOTSTRAP,
        ],
        external_scripts=[
            #
            "https://telegram.org/js/telegram-web-app.js"
        ],
        server=server,
        requests_pathname_prefix=real_prefix,
        routes_pathname_prefix="/_internal/",
        # requests_pathname_prefix=requests_pathname_prefix,
        meta_tags=[
            {"name": "viewport", "content": "width=device-width, initial-scale=1"},
        ],
    )

    # app.enable_dev_tools(
    #     dev_tools_ui=True,
    #     dev_tools_serve_dev_bundles=True,
    # )

    # Register pages
    for path, page in pages.items():
        page["blueprint"].register(app, path, prefix=page["prefix"])

    # Create navbar
    navbar = dbc.Navbar(
        dbc.Container(
            [
                dbc.Button(
                    html.I(className="bi bi-list"),
                    id="open-offcanvas",
                    color="light",
                    className="me-2",
                ),
                dbc.NavbarBrand("Karkas"),
            ]
        ),
        color="primary",
        dark=True,
    )

    roles = Roles()

    def create_layout():
        user = getattr(flask.g, "user", None)

        if not user:
            return html.Div()

        user_id = user["id"]
        user_permission = asyncio.run(roles.get_user_permission(user_id)) or "USER"

        available_pages = {
            path: page
            for path, page in pages.items()
            if (isinstance(page["role"], list) and user_permission in page["role"])
            or page["role"] == user_permission
            or page["role"] == "USER"
        }

        # Create sidebar
        sidebar = dbc.Offcanvas(
            id="offcanvas",
            title="Меню",
            is_open=False,
            children=[
                dbc.Nav(
                    [
                        dbc.NavLink(
                            page["name"],
                            href=f"{real_prefix}/{path.lstrip('/')}",
                            id={"type": "nav-link", "index": path},
                        )
                        for path, page in available_pages.items()
                    ],
                    vertical=True,
                    pills=True,
                ),
            ],
        )

        layout = html.Div(
            [
                dcc.Location(id="url", refresh=False),
                dcc.Interval(
                    id="init-telegram-interval",
                    interval=100,
                    n_intervals=0,
                    max_intervals=1,
                ),
                navbar,
                sidebar,
                dash.page_container,
                setup_page_components(),
            ]
        )

        return layout

    app.layout = create_layout

    setup_auth_clientcallbacks(app)

    # Открытие на кнопку меню
    app.clientside_callback(
        """
        function(n_clicks) {
            if (n_clicks == null) {
                return false;
            }
            return true;
        }
        """,
        Output(
            "offcanvas",
            "is_open",
        ),
        Input("open-offcanvas", "n_clicks"),
    )

    # Закрываем offcanvas при клике на ссылку в меню
    app.clientside_callback(
        """
        function(n_clicks) {
            if (n_clicks == null) {
                return true;
            }
            return false;
        }
        """,
        Output("offcanvas", "is_open", allow_duplicate=True),
        Input({"type": "nav-link", "index": dash.dependencies.ALL}, "n_clicks"),
        prevent_initial_call="initial_duplicate",
    )

    return app