create slots helper module (#51)

Author: joshuadavidthomas

src/django_bird/components/slots.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from django.template.base import NodeList
+from django.template.base import Template
+from django.template.context import Context
+
+DEFAULT_SLOT = "default"
+
+
+@dataclass
+class Slots:
+    slots: dict[str, list[str]]
+    context: Context
+
+    @classmethod
+    def collect(cls, nodelist: NodeList | None, context: Context):
+        from django_bird.templatetags.tags.slot import SlotNode
+
+        if nodelist is None:
+            return cls({}, context)
+
+        slots: dict[str, list[str]] = {DEFAULT_SLOT: []}
+        active_slot = DEFAULT_SLOT
+
+        for node in nodelist:
+            if isinstance(node, SlotNode):
+                active_slot = node.name
+                slots.setdefault(active_slot, [])
+            else:
+                active_slot = DEFAULT_SLOT
+
+            rendered_content = node.render(context)
+            slots[active_slot].append(rendered_content)
+
+        if all(not content for content in slots[DEFAULT_SLOT]) and "slot" in context:
+            slots[DEFAULT_SLOT] = [context["slot"]]
+
+        return cls(slots, context)
+
+    def render(self):
+        return {
+            slot: Template("".join(content)).render(self.context)
+            for slot, content in self.slots.items()
+        }

src/django_bird/templatetags/tags/bird.py

@@ -10,15 +10,14 @@
 from django.template.context import Context
 from django.template.loader import select_template
 from django.utils.safestring import SafeString
-from django.utils.safestring import mark_safe
 
 from django_bird._typing import TagBits
 from django_bird._typing import override
 from django_bird.components.attrs import Attrs
+from django_bird.components.slots import DEFAULT_SLOT
+from django_bird.components.slots import Slots
 from django_bird.components.templates import get_template_names
 
-from .slot import SlotNode
-
 TAG = "bird"
 END_TAG = "endbird"
 
@@ -62,7 +61,6 @@ def __init__(self, name: str, attrs: list[str], nodelist: NodeList | None) -> No
         self.name = name
         self.attrs = attrs
         self.nodelist = nodelist
-        self.default_slot = "default"
 
     @override
     def render(self, context: Context) -> SafeString:
@@ -81,40 +79,10 @@ def get_component_name(self, context: Context) -> str:
 
     def get_component_context_data(self, context: Context) -> dict[str, Any]:
         attrs = Attrs.parse(self.attrs, context)
-        rendered_slots = self.render_slots(context)
-        default_slot = rendered_slots.get(self.default_slot) or context.get("slot")
+        slots = Slots.collect(self.nodelist, context).render()
+        default_slot = slots.get(DEFAULT_SLOT) or context.get("slot")
         return {
             "attrs": attrs.flatten(),
-            "slot": mark_safe(default_slot),
-            "slots": {
-                name: mark_safe(content) for name, content in rendered_slots.items()
-            },
-        }
-
-    def render_slots(self, context: Context) -> dict[str, str]:
-        if self.nodelist is None:
-            return {}
-
-        contents: dict[str, list[str]] = {self.default_slot: []}
-        active_slot = self.default_slot
-
-        for node in self.nodelist:
-            if isinstance(node, SlotNode):
-                active_slot = node.name
-                contents.setdefault(active_slot, [])
-            else:
-                active_slot = self.default_slot
-
-            rendered_content = node.render(context)
-            contents[active_slot].append(rendered_content)
-
-        if (
-            all(not content for content in contents[self.default_slot])
-            and "slot" in context
-        ):
-            contents[self.default_slot] = [context["slot"]]
-
-        return {
-            slot: template.Template("".join(content)).render(context)
-            for slot, content in contents.items()
+            "slot": default_slot,
+            "slots": slots,
         }