feat: Initial rust templates

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

2 files changed, 68 insertions, 0 deletions

diff --git a/templates/rust/template.rs.tera b/templates/rust/template.rs.tera
new file mode 100644
index 0000000..845cbc7
--- /dev/null
+++ b/templates/rust/template.rs.tera
@@ -0,0 +1,66 @@
+use subplotlib::prelude::*;
+
+{% for func in functions %}
+
+// --------------------------------
+// This came from {{ func.source }}
+
+{{ func.code }}
+
+{% endfor %}
+
+// --------------------------------
+
+lazy_static! {
+    static ref SUBPLOT_EMBEDDED_FILES: Vec<SubplotDataFile> = vec![
+{% for file in files %}
+        SubplotDataFile::new("{{ file.filename | base64 }}",
+                             "{{ file.contents | base64 }}"),
+{% endfor %}
+    ];
+}
+
+{% for scenario in scenarios %}
+
+// ---------------------------------
+
+// {{ scenario.title }}
+#[test]
+fn scenario_{{ loop.index }}() {
+    let mut scenario = Scenario::new(&base64_decode("{{scenario.title | base64}}"));
+    {% for step in scenario.steps %}
+    let step = BUILDER_{{step.function}}::default()
+    {% for part in step.parts %}{% if part.CapturedText is defined -%}
+        {%- set name = part.CapturedText.name -%}
+        {%- set text = part.CapturedText.text -%}
+        {%- set type = step.types[name] | default(value='string') -%}
+        .{{name}}(
+            {% if type in ['number', 'raw', 'literal'] %}{{text}}
+            {% elif type in ['string']%}&base64_decode("{{text | base64}}")
+            {% elif type in ['file'] %}
+            {
+                use std::path::PathBuf;
+                let target_name: PathBuf = base64_decode("{{ text | base64 }}").into();
+                SUBPLOT_EMBEDDED_FILES
+                    .iter()
+                    .find(|df| df.name() == target_name)
+                    .expect("Unable to find file at runtime")
+                    .clone()
+            }
+            {% else %} /* WOAH */ {{text}}
+            {%- endif %}
+        )
+    {% endif -%}
+    {% endfor -%}
+        .build();
+    {%- if step.cleanup %}
+    let cleanup = BUILDER_{{step.cleanup}}::default().build();
+    scenario.add_step(step, Some(cleanup));
+    {%- else %}
+    scenario.add_step(step, None);
+    {%- endif %}
+    {% endfor %}
+
+    scenario.run().unwrap();
+}
+{% endfor %}
\ No newline at end of file
diff --git a/templates/rust/template.yaml b/templates/rust/template.yaml
new file mode 100644
index 0000000..110f5df
--- /dev/null
+++ b/templates/rust/template.yaml
@@ -0,0 +1,2 @@
+template: template.rs.tera
+run: cargo test