From 4ea133608350e6e99008f46cb4c45dd18989b836 Mon Sep 17 00:00:00 2001
From: Asko Nõmm <asko@nmm.ee>
Date: Mon, 6 Jan 2025 01:16:33 +0200
Subject: Implement `Iterate` parser.

---
 htmtl/htmtl.py            | 17 +++++++++---
 htmtl/parsers/__init__.py | 10 ++++++-
 htmtl/parsers/iterate.py  | 67 +++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 89 insertions(+), 5 deletions(-)
 create mode 100644 htmtl/parsers/iterate.py

(limited to 'htmtl')

diff --git a/htmtl/htmtl.py b/htmtl/htmtl.py
index 7732f06..6570e11 100644
--- a/htmtl/htmtl.py
+++ b/htmtl/htmtl.py
@@ -1,18 +1,21 @@
 from typing import Any
 from dompa import Dompa
+from dompa.actions import ToHtml
+from dompa.nodes import Node
 from .parser import Parser
 from .parsers.generic_value import GenericValue
 from .parsers.inner_html import InnerHtml
 from .parsers.inner_partial import InnerPartial
 from .parsers.inner_text import InnerText
+from .parsers.iterate import Iterate
 from .parsers.outer_html import OuterHtml
 from .parsers.outer_partial import OuterPartial
 from .parsers.outer_text import OuterText
+from .parsers.when import When
+from .parsers.when_not import WhenNot
 from .modifier import Modifier
 from .modifiers.truncate import Truncate
 from .expression_parser import ExpressionParser
-from .parsers.when import When
-from .parsers.when_not import WhenNot
 
 
 class Htmtl:
@@ -27,6 +30,7 @@ class Htmtl:
 
         # set default attribute parsers
         self.__attribute_parsers = [
+            Iterate,
             InnerText,
             InnerHtml,
             InnerPartial,
@@ -64,7 +68,12 @@ class Htmtl:
             parser_instance = parser(self.__data, expression_parser)
             self.__dom.traverse(parser_instance.traverse)
 
-    def html(self) -> str:
+    def to_html(self) -> str:
+        self.__parse()
+
+        return self.__dom.action(ToHtml)
+
+    def nodes(self) -> list[Node]:
         self.__parse()
 
-        return self.__dom.html()
\ No newline at end of file
+        return self.__dom.get_nodes()
\ No newline at end of file
diff --git a/htmtl/parsers/__init__.py b/htmtl/parsers/__init__.py
index 4ad8185..5c98cba 100644
--- a/htmtl/parsers/__init__.py
+++ b/htmtl/parsers/__init__.py
@@ -1,2 +1,10 @@
 from .inner_text import InnerText
-from .outer_text import OuterText
\ No newline at end of file
+from .inner_html import InnerHtml
+from .inner_partial import InnerPartial
+from .outer_text import OuterText
+from .outer_html import OuterHtml
+from .outer_partial import OuterPartial
+from .generic_value import GenericValue
+from .when import When
+from .when_not import WhenNot
+from .iterate import Iterate
\ No newline at end of file
diff --git a/htmtl/parsers/iterate.py b/htmtl/parsers/iterate.py
new file mode 100644
index 0000000..a70d026
--- /dev/null
+++ b/htmtl/parsers/iterate.py
@@ -0,0 +1,67 @@
+from collections.abc import Iterable
+from typing import Optional
+
+from dompa.nodes.actions import ToHtml
+from dompa.nodes import Node, FragmentNode
+from ..parser import Parser
+import htmtl
+
+class IterateOp:
+    var: str
+    iter_var_as: Optional[str]
+    iter_index_as: Optional[str]
+
+    def __init__(self, var: str, iter_var_as: Optional[str] = None, iter_index_as: Optional[str] = None):
+        self.var = var
+        self.iter_var_as = iter_var_as or None
+        self.iter_index_as = iter_index_as or None
+
+
+class Iterate(Parser):
+    def traverse(self, node: Node) -> Optional[Node]:
+        if "iterate" in node.attributes:
+            replacement_nodes = []
+            iterate_op = self.__parse_exp(node.attributes["iterate"])
+            collection = self.expression(iterate_op.var)
+            node.attributes.pop("iterate")
+
+            if isinstance(collection, Iterable):
+                data = self.data()
+
+                for idx, item in enumerate(collection):
+                    if iterate_op.iter_var_as:
+                        data[iterate_op.iter_var_as] = item
+
+                    if iterate_op.iter_index_as:
+                        data[iterate_op.iter_index_as] = idx
+
+                    template = htmtl.Htmtl(node.action(ToHtml), data)
+                    template_nodes = template.nodes()
+
+                    if len(template_nodes) > 0:
+                        replacement_nodes.append(template_nodes[0])
+
+            return FragmentNode(children=replacement_nodes)
+
+        return node
+
+    @staticmethod
+    def __parse_exp(exp: str) -> IterateOp:
+        parts = exp.split(" ")
+
+        if len(parts) == 3:
+            if parts[2].count(":") == 1:
+                return IterateOp(
+                    var=parts[0].strip(),
+                    iter_index_as=parts[2].split(":")[0].strip(),
+                    iter_var_as=parts[2].split(":")[1].strip()
+                )
+
+            return IterateOp(
+                var=parts[0].strip(),
+                iter_var_as=parts[2].strip()
+            )
+
+        return IterateOp(
+            var=parts[0]
+        )
-- 
cgit v1.2.3