Add new function new_mut_list returns a list mutable from the minijinja context

Author: svix-mman

src/cli_v2/mod.rs

@@ -1,6 +1,9 @@
 //! A WIP new cli, to the openapi-codegen
 
 mod frontmatter;
+mod mut_list;
+
+pub(crate) use mut_list::new_mut_list;
 
 #[derive(Debug, thiserror::Error)]
 pub(super) enum Error {

src/cli_v2/mut_list.rs

@@ -0,0 +1,81 @@
+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(state, 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)]
+/// A mutable list of minijinja::Value, a work-around the minijinja limitations.
+struct MutList(RwLock<Vec<minijinja::Value>>);
+
+impl MutList {
+    fn append(
+        self: &Arc<Self>,
+        _state: &minijinja::State<'_, '_>,
+        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 args_len = args.len();
+    let err = |kind| -> Result<(), minijinja::Error> {
+        Err(minijinja::Error::new(
+            kind,
+            format!("{method} | Expected: {n} args, got {args_len} arguments",),
+        ))
+    };
+
+    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

@@ -12,3 +12,4 @@ pub use crate::{
     codesamples::{CodeSample, CodesampleTemplates, generate_codesamples},
     postprocessing::CodegenLanguage,
 };
+pub(crate) use cli_v2::new_mut_list;

src/template.rs

@@ -10,6 +10,8 @@ use itertools::Itertools as _;
 use minijinja::{State, Value, path_loader, value::Kwargs};
 use serde::Deserialize;
 
+use crate::new_mut_list;
+
 pub fn env_with_dir(
     tpl_dir: &Utf8Path,
 ) -> Result<minijinja::Environment<'static>, minijinja::Error> {
@@ -21,6 +23,7 @@ pub fn env_with_dir(
 pub fn populate_env(
     mut env: minijinja::Environment<'static>,
 ) -> Result<minijinja::Environment<'static>, minijinja::Error> {
+    env.add_function("new_mut_list", new_mut_list);
     // === Custom filters ===
 
     // --- Case conversion ---