Add `join` filter & tests for it

2 files changed, 71 insertions, 0 deletions

diff --git a/askama_shared/src/filters/mod.rs b/askama_shared/src/filters/mod.rs
index 73b311d..75014fa 100644
--- a/askama_shared/src/filters/mod.rs
+++ b/askama_shared/src/filters/mod.rs
@@ -12,6 +12,7 @@ mod json;
 pub use self::json::json;
 
 use std::fmt;
+
 use super::Result;
 
 
@@ -96,6 +97,27 @@ pub fn trim(s: &fmt::Display) -> Result<String> {
     Ok(s.trim().to_owned())
 }
 
+/// Joins iterable into a string separated by provided argument
+pub fn join<T, I, S>(input: I, separator: S) -> Result<String>
+    where T: fmt::Display,
+          I: Iterator<Item = T>,
+          S: AsRef<str>
+{
+    let separator: &str = separator.as_ref();
+
+    let mut rv = String::new();
+
+    for (num, item) in input.enumerate() {
+        if num > 0 {
+            rv.push_str(separator);
+        }
+
+        rv.push_str(&format!("{}", item));
+    }
+
+    Ok(rv)
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
@@ -127,4 +149,28 @@ mod tests {
     fn test_trim() {
         assert_eq!(trim(&" Hello\tworld\t").unwrap(), "Hello\tworld");
     }
+
+    #[test]
+    fn test_join() {
+        assert_eq!(join((&["hello", "world"]).into_iter(), ", ").unwrap(), "hello, world");
+        assert_eq!(join((&["hello"]).into_iter(), ", ").unwrap(), "hello");
+
+        let empty: &[&str] = &[];
+        assert_eq!(join(empty.into_iter(), ", ").unwrap(), "");
+
+        let input: Vec<String> = vec!["foo".into(), "bar".into(), "bazz".into()];
+        assert_eq!(join((&input).into_iter(), ":".to_string()).unwrap(), "foo:bar:bazz");
+        assert_eq!(join(input.clone().into_iter(), ":").unwrap(), "foo:bar:bazz");
+        assert_eq!(join(input.clone().into_iter(), ":".to_string()).unwrap(), "foo:bar:bazz");
+
+        let input: &[String] = &["foo".into(), "bar".into()];
+        assert_eq!(join(input.into_iter(), ":").unwrap(), "foo:bar");
+        assert_eq!(join(input.into_iter(), ":".to_string()).unwrap(), "foo:bar");
+
+        let real: String = "blah".into();
+        let input: Vec<&str> = vec![&real];
+        assert_eq!(join(input.into_iter(), ";").unwrap(), "blah");
+
+        assert_eq!(join((&&&&&["foo", "bar"]).into_iter(), ", ").unwrap(), "foo, bar");
+    }
 }
diff --git a/askama_shared/src/generator.rs b/askama_shared/src/generator.rs
index e5ffd08..836da34 100644
--- a/askama_shared/src/generator.rs
+++ b/askama_shared/src/generator.rs
@@ -238,10 +238,35 @@ impl<'a> Generator<'a> {
         self.write(")");
     }
 
+    fn _visit_join_filter(&mut self, args: &[Expr]) {
+        // this is a separate handler because we need to generate code like
+        // this to force auto-deref for first argument:
+        // `join((&&&&&&self.s).into_iter(), ", ")`
+        // otherwise, we were getting errors described in PR #39
+        self.write("::askama::filters::join((&");
+
+        for (i, arg) in args.iter().enumerate() {
+            if i > 0 {
+                self.write(", &");
+            }
+
+            self.visit_expr(arg);
+
+            if i == 0 {
+                self.write(").into_iter()");
+            }
+        }
+
+        self.write(")?");
+    }
+
     fn visit_filter(&mut self, name: &str, args: &[Expr]) {
         if name == "format" {
             self._visit_format_filter(args);
             return;
+        } else if name == "join" {
+            self._visit_join_filter(args);
+            return;
         }
 
         if BUILT_IN_FILTERS.contains(&name) {