support some missing Jinja blocks (#25)

g-plane · Jun 11, 2024 · 5bc7f82 · 5bc7f82
1 parent 28da3ac
commit 5bc7f82
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 0 deletions.
diff --git a/markup_fmt/src/parser.rs b/markup_fmt/src/parser.rs
@@ -726,6 +726,9 @@ impl<'s> Parser<'s> {
  | "autoescape"
  | "embed"
  | "with"
+ | "set"
+ | "trans"
+ | "raw"
  ) {
  let mut body = vec![JinjaTagOrChildren::Tag(first_tag)];
 

diff --git a/markup_fmt/tests/fmt/jinja/control-structure/fixture.jinja b/markup_fmt/tests/fmt/jinja/control-structure/fixture.jinja
@@ -50,3 +50,29 @@ Sold-out!
 {%- endmacro %}
 
 <title>{% block title %}{% endblock %}</title>
+
+{% set reply | wordwrap %}
+ You wrote:
+ {{ message }}
+{% endset %}
+
+{% trans %}Hello, {{ user }}!{% endtrans %}
+{% trans user=user.username %}Hello, {{ user }}!{% endtrans %}
+
+{% trans book_title=book.title, author=author.name %}
+This is {{ book_title }} by {{ author }}
+{% endtrans %}
+
+{% trans count=list|length %}
+There is {{ count }} {{ name }} object.
+{% pluralize %}
+There are {{ count }} {{ name }} objects.
+{% endtrans %}
+
+{% raw %}
+<ul>
+{% for item in seq %}
+ <li>{{ item }}</li>
+{% endfor %}
+</ul>
+{% endraw %}
diff --git a/markup_fmt/tests/fmt/jinja/control-structure/fixture.snap b/markup_fmt/tests/fmt/jinja/control-structure/fixture.snap
@@ -58,3 +58,31 @@ source: markup_fmt/tests/fmt.rs
 {%- endmacro %}
 
 <title>{% block title %}{% endblock %}</title>
+
+{% set reply | wordwrap %}
+ You wrote:
+ {{ message }}
+{% endset %}
+
+{% trans %}Hello, {{ user }}!{% endtrans %}
+{% trans user=user.username %}Hello, {{ user }}!{% endtrans %}
+
+{% trans book_title=book.title, author=author.name %}
+ This is {{ book_title }} by {{ author }}
+{% endtrans %}
+
+{% trans count=list|length %}
+ There is {{ count }}
+ {{ name }} object.
+ {% pluralize %}
+ There are {{ count }}
+ {{ name }} objects.
+{% endtrans %}
+
+{% raw %}
+ <ul>
+ {% for item in seq %}
+ <li>{{ item }}</li>
+ {% endfor %}
+ </ul>
+{% endraw %}