Add function new_mut_list returns a list mutable from the minijinja context

Author: svix-mman

src/cli_v2/mod.rs

@@ -0,0 +1 @@
+pub(crate) mod mut_list;

src/cli_v2/mut_list.rs

@@ -0,0 +1,82 @@
+use std::sync::{Arc, RwLock};
+
+use minijinja::{ErrorKind, value::DynObject};
+
+pub(crate) fn new_mut_list() -> DynObject {
+    DynObject::new(Arc::new(MutList(RwLock::new(vec![]))))
+}
+
+impl minijinja::value::Object for MutList {
+    fn repr(self: &Arc<Self>) -> minijinja::value::ObjectRepr {
+        minijinja::value::ObjectRepr::Iterable
+    }
+
+    fn call_method(
+        self: &Arc<Self>,
+        _state: &minijinja::State<'_, '_>,
+        method: &str,
+        args: &[minijinja::Value],
+    ) -> Result<minijinja::Value, minijinja::Error> {
+        match method {
+            "append" => self.append(args),
+            _ => Err(minijinja::Error::new(
+                ErrorKind::UnknownMethod,
+                format!("Unexpected method {method}"),
+            )),
+        }
+    }
+
+    fn enumerate(self: &Arc<Self>) -> minijinja::value::Enumerator {
+        let vals = self
+            .0
+            .read()
+            .expect("Unable to read from MutList, RwLock was poisoned")
+            .iter()
+            .map(|s| s.to_owned())
+            .collect::<Vec<_>>();
+        minijinja::value::Enumerator::Iter(Box::new(vals.into_iter()))
+    }
+}
+
+#[derive(Debug)]
+/// list of minijinja::Value, a work-around the minijinja mutability limitations.
+struct MutList(RwLock<Vec<minijinja::Value>>);
+
+impl MutList {
+    fn append(
+        self: &Arc<Self>,
+        args: &[minijinja::Value],
+    ) -> Result<minijinja::Value, minijinja::Error> {
+        ensure_n_args("append", 1, args)?;
+        {
+            let mut a = self
+                .0
+                .try_write()
+                .map_err(|e| minijinja::Error::new(ErrorKind::InvalidOperation, e.to_string()))?;
+            a.push(args[0].clone());
+        }
+        Ok(minijinja::Value::UNDEFINED)
+    }
+}
+
+fn ensure_n_args(
+    method: &str,
+    n: usize,
+    args: &[minijinja::Value],
+) -> Result<(), minijinja::Error> {
+    let err = |kind| -> Result<(), minijinja::Error> {
+        Err(minijinja::Error::new(
+            kind,
+            format!(
+                "{method} | Expected: {n} args, got {} arguments",
+                args.len()
+            ),
+        ))
+    };
+
+    match args.len().cmp(&n) {
+        std::cmp::Ordering::Less => err(ErrorKind::MissingArgument),
+        std::cmp::Ordering::Greater => err(ErrorKind::TooManyArguments),
+        std::cmp::Ordering::Equal => Ok(()),
+    }
+}

src/lib.rs

@@ -1,5 +1,6 @@
 mod api;
 mod cli_v1;
+pub(crate) mod cli_v2;
 mod codesamples;
 mod generator;
 mod postprocessing;

src/template.rs

@@ -21,6 +21,9 @@ pub fn env_with_dir(
 pub fn populate_env(
     mut env: minijinja::Environment<'static>,
 ) -> Result<minijinja::Environment<'static>, minijinja::Error> {
+    // === Custom functions ===
+    env.add_function("new_mut_list", crate::cli_v2::mut_list::new_mut_list);
+
     // === Custom filters ===
 
     // --- Case conversion ---