diff options
Diffstat (limited to 'askama_shared')
-rw-r--r-- | askama_shared/src/filters/mod.rs | 46 | ||||
-rw-r--r-- | askama_shared/src/generator.rs | 25 |
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) { |