-
Notifications
You must be signed in to change notification settings - Fork 1
/
navigation.py
106 lines (82 loc) · 2.92 KB
/
navigation.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
from __future__ import annotations
import typing
from django import template
from django.contrib.auth import REDIRECT_FIELD_NAME
from django.urls import reverse
from django.utils.http import urlencode
from django.utils.translation import gettext_lazy as _
if typing.TYPE_CHECKING:
from apps.plugins.middleware.plugin import HttpRequest
from apps.plugins.models import Plugin
from apps.plugins.plugin import BasePluginAppConfig
from apps.sections.models import Section, SectionMembership
register = template.Library()
class NavigationItemSpec(typing.NamedTuple):
title: str
url: str
children: list[NavigationItemSpec] = []
active: bool = False
@register.simple_tag(takes_context=True)
def get_navigation_items(context):
from apps.plugins.utils import all_plugins_to_order
from apps.sections.models import SectionMembership
request: HttpRequest = context["request"]
membership: SectionMembership | None = request.membership
section: Section | None = request.in_space_of_section
if not membership:
if request.all_memberships.filter(
role__in=(SectionMembership.Role.ADMIN, SectionMembership.Role.EDITOR)
).exists():
# TODO: consider better way
return [
NavigationItemSpec(
_("Docs"),
reverse("wiki:index"),
[],
request.path.startswith(reverse("wiki:index")),
),
]
return []
plugins: list[Plugin] = (
section.enabled_plugins_for_privileged if membership.is_privileged else section.enabled_plugins
)
orders = all_plugins_to_order()
def _all_items():
for plugin in sorted(plugins, key=lambda p: (orders[p.app_label], p.app_label)):
apps: BasePluginAppConfig = plugin.app_config
item = apps.as_navigation_item(request=request, bound_plugin=plugin)
if item:
yield item
return list(_all_items())
@register.simple_tag(takes_context=True)
def get_home_url(context):
"""Called also from error pages, checks everything potentially set by middlewares."""
url = reverse("public:home")
try:
request: HttpRequest = context["request"]
except KeyError:
return url
try:
membership: SectionMembership | None = request.membership
except AttributeError:
membership = None
try:
section: Section | None = request.in_space_of_section
except AttributeError:
section = None
if section:
url = request.in_space_of_section.section_home_url(membership)
if url:
return url
return (
reverse("account_login")
+ "?"
+ (
urlencode(
{
**request.GET,
REDIRECT_FIELD_NAME: request.GET.get(REDIRECT_FIELD_NAME) or request.path,
}
)
)
)