subplot: Enable passing of location into the template expansion

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

1 files changed, 17 insertions, 0 deletions

diff --git a/src/codegen.rs b/src/codegen.rs
index 41b5556..5855d8b 100644
--- a/src/codegen.rs
+++ b/src/codegen.rs
@@ -1,3 +1,4 @@
+use crate::html::Location;
 use crate::{resource, Document, SubplotError, TemplateSpec};
 use std::collections::HashMap;
 use std::fs::File;
@@ -57,6 +58,7 @@ fn tera(tmplspec: &TemplateSpec, templatename: &str) -> Result<Tera, SubplotErro
     tera.register_filter("base64", base64);
     tera.register_filter("nameslug", nameslug);
     tera.register_filter("commentsafe", commentsafe);
+    tera.register_filter("location", locationfilter);
     let dirname = tmplspec.template_filename().parent().unwrap();
     for helper in tmplspec.helpers() {
         let helper_path = dirname.join(helper);
@@ -93,6 +95,21 @@ fn base64(v: &Value, _: &HashMap<String, Value>) -> tera::Result<Value> {
     }
 }
 
+fn locationfilter(v: &Value, _: &HashMap<String, Value>) -> tera::Result<Value> {
+    let location: Location = serde_json::from_value(v.clone())?;
+    Ok(Value::String(format!(
+        "{:?}",
+        match location {
+            Location::Known {
+                filename,
+                line,
+                col,
+            } => format!("{}:{}:{}", filename.display(), line, col),
+            Location::Unknown => "unknown".to_string(),
+        }
+    )))
+}
+
 fn nameslug(name: &Value, _: &HashMap<String, Value>) -> tera::Result<Value> {
     match name {
         Value::String(s) => {