grit: Allow if's to be recursive

Not often used in grit files, if at all, but needed when we want
to add support for expanding grit-part in grit structure.

1 files changed, 34 insertions, 20 deletions

diff --git a/server/common/src/grit.rs b/server/common/src/grit.rs
index 273da8a..fab1130 100644
--- a/server/common/src/grit.rs
+++ b/server/common/src/grit.rs
@@ -34,7 +34,7 @@ pub struct Output {
 pub enum IfOutput {
     Output(Output),
 
-    If { expr: String, output: Vec<Output> },
+    If { expr: String, output: Vec<IfOutput> },
 }
 
 #[derive(Debug, PartialEq)]
@@ -52,7 +52,7 @@ pub struct File {
 pub enum IfFile {
     File(File),
 
-    If { expr: String, file: Vec<File> },
+    If { expr: String, file: Vec<IfFile> },
 }
 
 #[derive(Debug, PartialEq)]
@@ -79,18 +79,11 @@ pub enum IfMessagePart {
     If {
         expr: String,
 
-        message: Vec<MessagePart>,
+        message: Vec<IfMessagePart>,
     },
 }
 
 #[derive(Debug, PartialEq)]
-pub enum MessagePart {
-    Message(Message),
-
-    Part(PartRef),
-}
-
-#[derive(Debug, PartialEq)]
 pub struct PartRef {
     pub file: String,
 }
@@ -119,7 +112,10 @@ pub enum TextPlaceholder {
 pub enum IfMessage {
     Message(Message),
 
-    If { expr: String, message: Vec<Message> },
+    If {
+        expr: String,
+        message: Vec<IfMessage>,
+    },
 }
 
 #[derive(Debug, PartialEq)]
@@ -305,7 +301,7 @@ fn parse_if_output_element<R: Read>(
     reader: &mut EventReader<R>,
 ) -> anyhow::Result<IfOutput> {
     let expr = get_attribute(attributes, "expr")?;
-    let mut output = Vec::<Output>::new();
+    let mut output = Vec::<IfOutput>::new();
 
     loop {
         let event = reader.next()?;
@@ -316,7 +312,10 @@ fn parse_if_output_element<R: Read>(
                 namespace: _,
             } => match name.local_name.as_str() {
                 "output" => {
-                    output.push(parse_output_element(&attributes, reader)?);
+                    output.push(IfOutput::Output(parse_output_element(&attributes, reader)?));
+                }
+                "if" => {
+                    output.push(parse_if_output_element(&attributes, reader)?);
                 }
                 _ => {
                     return Err(Error::msg(format!(
@@ -449,7 +448,7 @@ fn parse_if_file_element<R: Read>(
     reader: &mut EventReader<R>,
 ) -> anyhow::Result<IfFile> {
     let expr = get_attribute(attributes, "expr")?;
-    let mut file = Vec::<File>::new();
+    let mut file = Vec::<IfFile>::new();
 
     loop {
         let event = reader.next()?;
@@ -460,7 +459,10 @@ fn parse_if_file_element<R: Read>(
                 namespace: _,
             } => match name.local_name.as_str() {
                 "file" => {
-                    file.push(parse_file_element(&attributes, reader)?);
+                    file.push(IfFile::File(parse_file_element(&attributes, reader)?));
+                }
+                "if" => {
+                    file.push(parse_if_file_element(&attributes, reader)?);
                 }
                 _ => {
                     return Err(Error::msg(format!(
@@ -695,7 +697,7 @@ fn parse_if_message_part_element<R: Read>(
 ) -> anyhow::Result<IfMessagePart> {
     let expr = get_attribute(attributes, "expr")?;
 
-    let mut message = Vec::<MessagePart>::new();
+    let mut message = Vec::<IfMessagePart>::new();
 
     loop {
         let event = reader.next()?;
@@ -706,13 +708,19 @@ fn parse_if_message_part_element<R: Read>(
                 namespace: _,
             } => match name.local_name.as_str() {
                 "message" => {
-                    message.push(MessagePart::Message(parse_message_element(
+                    message.push(IfMessagePart::Message(parse_message_element(
                         &attributes,
                         reader,
                     )?));
                 }
                 "part" => {
-                    message.push(MessagePart::Part(parse_part_element(&attributes, reader)?));
+                    message.push(IfMessagePart::Part(parse_part_element(
+                        &attributes,
+                        reader,
+                    )?));
+                }
+                "if" => {
+                    message.push(parse_if_message_part_element(&attributes, reader)?);
                 }
                 _ => {
                     return Err(Error::msg(format!(
@@ -752,7 +760,7 @@ fn parse_if_message_element<R: Read>(
 ) -> anyhow::Result<IfMessage> {
     let expr = get_attribute(attributes, "expr")?;
 
-    let mut message = Vec::<Message>::new();
+    let mut message = Vec::<IfMessage>::new();
 
     loop {
         let event = reader.next()?;
@@ -763,7 +771,13 @@ fn parse_if_message_element<R: Read>(
                 namespace: _,
             } => match name.local_name.as_str() {
                 "message" => {
-                    message.push(parse_message_element(&attributes, reader)?);
+                    message.push(IfMessage::Message(parse_message_element(
+                        &attributes,
+                        reader,
+                    )?));
+                }
+                "if" => {
+                    message.push(parse_if_message_element(&attributes, reader)?);
                 }
                 _ => {
                     return Err(Error::msg(format!(