Skip to content

Commit

Permalink
Add support for elif
Browse files Browse the repository at this point in the history
  • Loading branch information
GuillaumeGomez committed May 23, 2024
1 parent 627d58b commit 22b2a28
Show file tree
Hide file tree
Showing 6 changed files with 82 additions and 43 deletions.
59 changes: 33 additions & 26 deletions askama_parser/src/node.rs
Original file line number Diff line number Diff line change
Expand Up @@ -372,21 +372,28 @@ impl<'a> Cond<'a> {
let mut p = tuple((
|i| s.tag_block_start(i),
opt(Whitespace::parse),
ws(alt((keyword("else"), |i| {
let _ = keyword("elif")(i)?;
Err(nom::Err::Failure(ErrorContext::new(
"unknown `elif` keyword; did you mean `else if`?",
i,
)))
}))),
cut(tuple((
opt(|i| CondTest::parse(i, s)),
opt(Whitespace::parse),
|i| s.tag_block_end(i),
cut(|i| Node::many(i, s)),
))),
alt((
tuple((
ws(keyword("else")),
cut(tuple((
opt(|i| CondTest::parse(i, s)),
opt(Whitespace::parse),
|i| s.tag_block_end(i),
cut(|i| Node::many(i, s)),
))),
)),
tuple((
ws(keyword("elif")),
cut(tuple((
opt(|i| CondTest::parse_cond(i, s)),
opt(Whitespace::parse),
|i| s.tag_block_end(i),
cut(|i| Node::many(i, s)),
))),
)),
)),
));
let (i, (_, pws, _, (cond, nws, _, nodes))) = p(i)?;
let (i, (_, pws, (_, (cond, nws, _, nodes)))) = p(i)?;
Ok((
i,
Self {
Expand All @@ -406,18 +413,18 @@ pub struct CondTest<'a> {

impl<'a> CondTest<'a> {
fn parse(i: &'a str, s: &State<'_>) -> ParseResult<'a, Self> {
let mut p = preceded(
ws(keyword("if")),
cut(tuple((
opt(delimited(
ws(alt((keyword("let"), keyword("set")))),
ws(|i| Target::parse(i, s)),
ws(char('=')),
)),
ws(|i| Expr::parse(i, s.level.get())),
))),
);
let (i, (target, expr)) = p(i)?;
preceded(ws(keyword("if")), |i| Self::parse_cond(i, s))(i)
}

fn parse_cond(i: &'a str, s: &State<'_>) -> ParseResult<'a, Self> {
let (i, (target, expr)) = cut(tuple((
opt(delimited(
ws(alt((keyword("let"), keyword("set")))),
ws(|i| Target::parse(i, s)),
ws(char('=')),
)),
ws(|i| Expr::parse(i, s.level.get())),
)))(i)?;
Ok((i, Self { target, expr }))
}
}
Expand Down
2 changes: 2 additions & 0 deletions book/src/template_syntax.md
Original file line number Diff line number Diff line change
Expand Up @@ -459,6 +459,8 @@ and is used as you might expect:
No users
{% else if users.len() == 1 %}
1 user
{% elif users.len() == 2 %}
2 users
{% else %}
{{ users.len() }} users
{% endif %}
Expand Down
26 changes: 26 additions & 0 deletions testing/tests/if.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
use askama::Template;

#[derive(Template)]
#[template(
source = r#"{%- if s == "" -%}
empty
{%- else if s == "b" -%}
b
{%- elif s == "c" -%}
c
{%- else -%}
else
{%- endif -%}"#,
ext = "txt"
)]
struct If<'a> {
s: &'a str,
}

#[test]
fn test_if() {
assert_eq!(If { s: "" }.render().unwrap(), "empty");
assert_eq!(If { s: "b" }.render().unwrap(), "b");
assert_eq!(If { s: "c" }.render().unwrap(), "c");
assert_eq!(If { s: "d" }.render().unwrap(), "else");
}
21 changes: 21 additions & 0 deletions testing/tests/if_let.rs
Original file line number Diff line number Diff line change
Expand Up @@ -107,3 +107,24 @@ fn test_if_let_else() {
};
assert_eq!(s.render().unwrap(), "fail");
}

#[derive(Template)]
#[template(
source = r#"{%- if s.is_none() -%}
empty
{%- elif let Some(a) = s -%}
{{a}}
{%- else -%}
else
{%- endif -%}"#,
ext = "txt"
)]
struct Elif<'a> {
s: Option<&'a str>,
}

#[test]
fn test_elif() {
assert_eq!(Elif { s: None }.render().unwrap(), "empty");
assert_eq!(Elif { s: Some("tada") }.render().unwrap(), "tada");
}
8 changes: 0 additions & 8 deletions testing/tests/ui/elif.rs

This file was deleted.

9 changes: 0 additions & 9 deletions testing/tests/ui/elif.stderr

This file was deleted.

0 comments on commit 22b2a28

Please sign in to comment.