From 786d132c1deb295b1fc77441d85201b556685b82 Mon Sep 17 00:00:00 2001
From: progmboy <progmboy@gmail.com>
Date: Tue, 16 Jan 2024 17:39:49 +0800
Subject: [PATCH] add else chain support

---
 src/grammar.pest         |   6 +-
 src/grammar.rs           |   1 +
 src/helpers/helper_if.rs |  36 +++++++++++
 src/render.rs            |   6 ++
 src/template.rs          | 135 ++++++++++++++++++++++++++++++++++++---
 5 files changed, 174 insertions(+), 10 deletions(-)

diff --git a/src/grammar.pest b/src/grammar.pest
index ac6776ee2..235dd884b 100644
--- a/src/grammar.pest
+++ b/src/grammar.pest
@@ -53,7 +53,7 @@ subexpression = { "(" ~ ((identifier ~ (hash|param)+) | reference)  ~ ")" }
 pre_whitespace_omitter = { "~" }
 pro_whitespace_omitter = { "~" }
 
-expression = { !invert_tag ~ "{{" ~ pre_whitespace_omitter? ~
+expression = { !(invert_tag|invert_chain_tag) ~ "{{" ~ pre_whitespace_omitter? ~
               ((identifier ~ (hash|param)+) | name )
               ~ pro_whitespace_omitter? ~ "}}" }
 html_expression_triple_bracket_legacy = _{ "{{{" ~ pre_whitespace_omitter? ~
@@ -76,12 +76,14 @@ partial_expression = { "{{" ~ pre_whitespace_omitter? ~ ">" ~ partial_exp_line
 invert_tag_item = { "else"|"^" }
 invert_tag = { !escape ~ "{{" ~ pre_whitespace_omitter? ~ invert_tag_item
              ~ pro_whitespace_omitter? ~ "}}" }
+invert_chain_tag = { !escape ~ "{{" ~ pre_whitespace_omitter? ~ invert_tag_item 
+                     ~ exp_line ~ pro_whitespace_omitter? ~ "}}" }
 helper_block_start = { "{{" ~ pre_whitespace_omitter? ~ "#" ~ exp_line ~
                      pro_whitespace_omitter? ~ "}}" }
 helper_block_end = { "{{" ~ pre_whitespace_omitter? ~ "/" ~ identifier ~
                    pro_whitespace_omitter? ~ "}}" }
 helper_block = _{ helper_block_start ~ template ~
-                  (invert_tag ~ template)? ~ helper_block_end }
+                  (invert_chain_tag ~ template)* ~ (invert_tag ~ template)? ~ helper_block_end }
 
 decorator_block_start = { "{{" ~ pre_whitespace_omitter? ~ "#" ~ "*"
                         ~ exp_line ~ pro_whitespace_omitter? ~ "}}" }
diff --git a/src/grammar.rs b/src/grammar.rs
index 6e61cd4e4..f58909984 100644
--- a/src/grammar.rs
+++ b/src/grammar.rs
@@ -261,6 +261,7 @@ mod test {
             "{{#if}}hello{{else~}}world{{/if}}",
             "{{#if}}hello{{~^~}}world{{/if}}",
             "{{#if}}{{/if}}",
+            "{{#if}}hello{{else if}}world{{else}}test{{/if}}"
         ];
         for i in s.iter() {
             assert_rule!(Rule::helper_block, i);
diff --git a/src/helpers/helper_if.rs b/src/helpers/helper_if.rs
index 60aee45ee..554db45a3 100644
--- a/src/helpers/helper_if.rs
+++ b/src/helpers/helper_if.rs
@@ -96,6 +96,42 @@ mod test {
         assert_eq!(r1.unwrap(), "hello 99".to_string());
     }
 
+    #[test]
+    fn test_if_else_chain() {
+        let handlebars = Registry::new();
+
+        assert_eq!(
+            "0".to_owned(),
+            handlebars
+                .render_template("{{#if a}}1{{else if b}}2{{else}}0{{/if}}", &json!({"d": 1}))
+                .unwrap()
+        );
+    }
+
+    #[test]
+    fn test_if_else_chain2() {
+        let handlebars = Registry::new();
+
+        assert_eq!(
+            "3".to_owned(),
+            handlebars
+                .render_template("{{#if a}}1{{else if b}}2{{else if c}}3{{else if d}}4{{else}}0{{/if}}", &json!({"c": 1, "d":1}))
+                .unwrap()
+        );
+    }
+
+    #[test]
+    fn test_if_else_chain3() {
+        let handlebars = Registry::new();
+
+        assert_eq!(
+            "4".to_owned(),
+            handlebars
+                .render_template("{{#if a}}1{{else if b}}2{{else if c}}3{{else if d}}4{{/if}}", &json!({"d":1}))
+                .unwrap()
+        );
+    }
+
     #[test]
     fn test_if_include_zero() {
         use std::f64;
diff --git a/src/render.rs b/src/render.rs
index 4216dd4e7..ba5c2cb89 100644
--- a/src/render.rs
+++ b/src/render.rs
@@ -665,6 +665,12 @@ impl Parameter {
     }
 }
 
+impl Renderable for HelperTemplate {
+    fn render<'reg: 'rc, 'rc>(&'rc self, registry: &'reg Registry<'reg>, ctx: &'rc Context, rc: &mut RenderContext<'reg, 'rc>, out: &mut dyn Output) -> Result<(), RenderError> {
+        render_helper(self, registry, ctx, rc, out)
+    }
+}
+
 impl Renderable for Template {
     fn render<'reg: 'rc, 'rc>(
         &'rc self,
diff --git a/src/template.rs b/src/template.rs
index d9697b81b..eb500f2b1 100644
--- a/src/template.rs
+++ b/src/template.rs
@@ -62,6 +62,7 @@ impl Subexpression {
                 inverse: None,
                 block_param: None,
                 block: false,
+                chain: false,
             }))),
         }
     }
@@ -135,6 +136,7 @@ pub struct HelperTemplate {
     pub template: Option<Template>,
     pub inverse: Option<Template>,
     pub block: bool,
+    pub chain: bool,
 }
 
 impl HelperTemplate {
@@ -147,6 +149,20 @@ impl HelperTemplate {
             block,
             template: None,
             inverse: None,
+            chain: false
+        }
+    }
+
+    pub fn new_chain(exp: ExpressionSpec, block: bool) -> HelperTemplate {
+        HelperTemplate {
+            name: exp.name,
+            params: exp.params,
+            hash: exp.hash,
+            block_param: exp.block_param,
+            block,
+            template: None,
+            inverse: None,
+            chain: true
         }
     }
 
@@ -160,12 +176,103 @@ impl HelperTemplate {
             template: None,
             inverse: None,
             block: false,
+            chain: false
         }
     }
 
     pub(crate) fn is_name_only(&self) -> bool {
         !self.block && self.params.is_empty() && self.hash.is_empty()
     }
+
+    fn insert_inverse_node(&mut self, mut node: Box<HelperTemplate>)
+    {
+        // Create a list in "inverse" member to hold the else-chain.
+        // Here we create the new template to save the else-chain node.
+        // The template render could render it successfully without any code add.
+        let mut new_chain_template = Template::new();
+        node.inverse = self.inverse.take();
+        new_chain_template.elements.push(HelperBlock(node));
+        self.inverse = Some(new_chain_template);
+
+    }
+    
+
+    fn ref_chain_head_mut(&mut self) -> Option<&mut Box<HelperTemplate>>
+    {
+        if self.chain {
+            if let Some(inverse_tmpl) = &mut self.inverse {
+                assert_eq!(inverse_tmpl.elements.len(), 1);
+                if let HelperBlock(helper) = &mut inverse_tmpl.elements[0] {
+                    return Some(helper);
+                }
+            }
+        }
+        None
+    }
+
+    fn set_chain_template(&mut self, tmpl: Option<Template>)
+    {
+        if let Some(hepler) = self.ref_chain_head_mut() {
+            hepler.template = tmpl;
+        }else{
+            self.template = tmpl;
+        }
+
+    }
+
+    fn revert_chain_and_set(&mut self, inverse: Option<Template>)
+    {
+        if self.chain{
+
+            let mut prev = None;
+
+            if let Some(head) = self.ref_chain_head_mut() {
+                if head.template.is_some() {
+
+                    // Here the prev will hold the head inverse template.
+                    // It will be set when reverse the chain.
+                    prev = inverse;
+                }else{
+
+                    // If the head already has template. set the inverse template.
+                    head.template = inverse;
+                }
+            }
+
+            // Reverse the else chain, to the normal list order.
+            while let Some(mut node) = self.inverse.take() {
+                assert_eq!(node.elements.len(), 1);
+                if let HelperBlock(c) = &mut node.elements[0] {
+                    self.inverse = c.inverse.take();
+                    c.inverse = prev;
+                    prev = Some(node);
+                }
+            }
+
+            self.inverse = prev;
+
+        }else{
+
+            // If the helper has no else chain.
+            // set the template to self.
+            if self.template.is_some() {
+                self.inverse = inverse;
+            }else{
+                self.template = inverse;
+            }
+        }
+    }
+
+    fn set_chained(&mut self) 
+    {
+        self.chain = true;
+    }
+
+    pub fn is_chained(&self) -> bool
+    {
+        self.chain
+    }
+
 }
 
 #[derive(PartialEq, Eq, Clone, Debug)]
@@ -674,9 +781,13 @@ impl Template {
                         let t = template_stack.front_mut().unwrap();
                         t.mapping.push(TemplateMapping(line_no, col_no));
                     }
-                    Rule::invert_tag => {
+                    Rule::invert_tag | Rule::invert_chain_tag => {
                         // hack: invert_tag structure is similar to ExpressionSpec, so I
                         // use it here to represent the data
+
+                        if rule == Rule::invert_chain_tag {
+                            let _ = Template::parse_name(source, &mut it, span.end())?;
+                        }
                         let exp = Template::parse_expression(source, it.by_ref(), span.end())?;
 
                         if exp.omit_pre_ws {
@@ -695,8 +806,18 @@ impl Template {
 
                         let t = template_stack.pop_front().unwrap();
                         let h = helper_stack.front_mut().unwrap();
-                        h.template = Some(t);
+                        
+                        if rule == Rule::invert_chain_tag {
+                            h.set_chained();
+                        }
+
+                        h.set_chain_template(Some(t));
+                        if rule == Rule::invert_chain_tag {
+                            h.insert_inverse_node(Box::new(HelperTemplate::new_chain(exp, true)));
+                        }
+
                     }
+
                     Rule::raw_block_text => {
                         let mut t = Template::new();
                         t.push_element(
@@ -783,12 +904,10 @@ impl Template {
                                 let mut h = helper_stack.pop_front().unwrap();
                                 let close_tag_name = exp.name.as_name();
                                 if h.name.as_name() == close_tag_name {
+
                                     let prev_t = template_stack.pop_front().unwrap();
-                                    if h.template.is_some() {
-                                        h.inverse = Some(prev_t);
-                                    } else {
-                                        h.template = Some(prev_t);
-                                    }
+                                    h.revert_chain_and_set(Some(prev_t));
+
                                     let t = template_stack.front_mut().unwrap();
                                     t.elements.push(HelperBlock(Box::new(h)));
                                 } else {
@@ -1331,6 +1450,6 @@ mod test {
         let s = "{{#>(X)}}{{/X}}";
         let result = Template::compile(s);
         assert!(result.is_err());
-        assert_eq!("decorator \"Subexpression(Subexpression { element: Expression(HelperTemplate { name: Path(Relative(([Named(\\\"X\\\")], \\\"X\\\"))), params: [], hash: {}, block_param: None, template: None, inverse: None, block: false }) })\" was opened, but \"X\" is closing", format!("{}", result.unwrap_err().reason()));
+        assert_eq!("decorator \"Subexpression(Subexpression { element: Expression(HelperTemplate { name: Path(Relative(([Named(\\\"X\\\")], \\\"X\\\"))), params: [], hash: {}, block_param: None, template: None, inverse: None, block: false, chain: false }) })\" was opened, but \"X\" is closing", format!("{}", result.unwrap_err().reason()));
     }
 }