1from __future__ import annotations
 2
 3from typing import Any
 4
 5from jinja2.runtime import Context
 6
 7from plain.templates import register_template_extension
 8from plain.templates.jinja.extensions import InclusionTagExtension
 9
10from .toolbar import Toolbar
11
12
13@register_template_extension
14class ToolbarExtension(InclusionTagExtension):
15    tags = {"toolbar"}
16    template_name = "toolbar/toolbar.html"
17
18    def get_context(self, context: Context, *args: Any, **kwargs: Any) -> Context:
19        context.vars["toolbar"] = Toolbar(context=context)
20        return context