diff --git a/src/wagtail_personalisation/wagtail_hooks.py b/src/wagtail_personalisation/wagtail_hooks.py index 725bed9..3933ad5 100644 --- a/src/wagtail_personalisation/wagtail_hooks.py +++ b/src/wagtail_personalisation/wagtail_hooks.py @@ -63,6 +63,24 @@ def segment_user(page, request, serve_args, serve_kwargs): adapter.set_segments([segment]) +class UserbarSegmentedLinkItem: + def __init__(self, segment): + self.segment = segment + + def render(self, request): + return f"""
""" + + +@hooks.register('construct_wagtail_userbar') +def add_segment_link_items(request, items): + for item in Segment.objects.enabled(): + items.append(UserbarSegmentedLinkItem(item)) + return items + + @hooks.register('before_serve_page') def serve_variant(page, request, serve_args, serve_kwargs): """Apply a segment to a visitor before serving the page.