Use "step" instead of "element", and get tables working

1 files changed, 14 insertions, 12 deletions

diff --git a/src/random_tables.rs b/src/random_tables.rs
index f801fc9..c1ffdcb 100644
--- a/src/random_tables.rs
+++ b/src/random_tables.rs
@@ -14,13 +14,13 @@ struct RandomTable {
 #[derive(Debug, Deserialize)]
 struct TableRow {
     roll: String,
-    elements: Vec<TableRowElement>,
+    steps: Vec<TableRowStep>,
 }
 
 #[derive(Debug, Deserialize)]
-enum TableRowElement {
-    Text(String),
-    Table(String),
+struct TableRowStep {
+    text: Option<String>,
+    table: Option<String>,
 }
 
 pub struct RandomTables {
@@ -43,15 +43,17 @@ impl RandomTables {
                 if let Some((start, end)) = RandomTables::parse_roll(&table_row.roll) {
                     if start <= roll_result.total() && roll_result.total() <= end {
                         let mut output_text = String::new();
-                        for element in &table_row.elements {
-                            match element {
-                                TableRowElement::Text(text) => output_text.push_str(text),
-                                TableRowElement::Table(inner_table) => {
-                                    let inner_output = self.roll_table(inner_table);
-                                    output_text.push_str(&inner_output);
-                                }
+                        for step in &table_row.steps {
+                            if let Some(text) = &step.text {
+                                output_text.push_str(text);
                             }
-                            output_text.push_str(" ");
+
+                            if let Some(table) = &step.table {
+                                let inner_output = self.roll_table(table);
+                                output_text.push_str(&inner_output);
+                            }
+
+                            output_text.push_str("\n");
                         }
                         return output_text.trim().to_string();
                     }