94 lines
2.5 KiB
Rust
94 lines
2.5 KiB
Rust
|
use std::collections::{HashMap};
|
||
|
|
|
||
|
|
use base64::{encode, decode};
|
||
|
|
use pulldown_cmark as cmark;
|
||
|
|
use tera::{Value, to_value, Result as TeraResult};
|
||
|
|
|
||
|
|
// fn(Value, HashMap<String, Value>) -> Result<Value>
|
||
|
|
|
||
|
|
pub fn markdown(value: Value, _: HashMap<String, Value>) -> TeraResult<Value> {
|
||
|
|
let s = try_get_value!("markdown", "value", String, value);
|
||
|
|
|
||
|
|
let mut html = String::new();
|
||
|
|
let parser = cmark::Parser::new(&s);
|
||
|
|
cmark::html::push_html(&mut html, parser);
|
||
|
|
|
||
|
|
Ok(to_value(&html).unwrap())
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
pub fn base64_encode(value: Value, _: HashMap<String, Value>) -> TeraResult<Value> {
|
||
|
|
let s = try_get_value!("base64_encode", "value", String, value);
|
||
|
|
Ok(
|
||
|
|
to_value(&encode(s.as_bytes())).unwrap()
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn base64_decode(value: Value, _: HashMap<String, Value>) -> TeraResult<Value> {
|
||
|
|
let s = try_get_value!("base64_decode", "value", String, value);
|
||
|
|
Ok(
|
||
|
|
to_value(
|
||
|
|
&String::from_utf8(
|
||
|
|
decode(s.as_bytes()).unwrap()
|
||
|
|
).unwrap()
|
||
|
|
).unwrap()
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
use tera::{to_value};
|
||
|
|
|
||
|
|
use super::{markdown, base64_decode, base64_encode};
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_markdown() {
|
||
|
|
let result = markdown(to_value(&"# Hey").unwrap(), HashMap::new());
|
||
|
|
assert!(result.is_ok());
|
||
|
|
assert_eq!(result.unwrap(), to_value(&"<h1>Hey</h1>\n").unwrap());
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_base64_encode() {
|
||
|
|
// from https://tools.ietf.org/html/rfc4648#section-10
|
||
|
|
let tests = vec![
|
||
|
|
("", ""),
|
||
|
|
("f", "Zg=="),
|
||
|
|
("fo", "Zm8="),
|
||
|
|
("foo", "Zm9v"),
|
||
|
|
("foob", "Zm9vYg=="),
|
||
|
|
("fooba", "Zm9vYmE="),
|
||
|
|
("foobar", "Zm9vYmFy")
|
||
|
|
];
|
||
|
|
for (input, expected) in tests {
|
||
|
|
let args = HashMap::new();
|
||
|
|
let result = base64_encode(to_value(input).unwrap(), args);
|
||
|
|
assert!(result.is_ok());
|
||
|
|
assert_eq!(result.unwrap(), to_value(expected).unwrap());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_base64_decode() {
|
||
|
|
let tests = vec![
|
||
|
|
("", ""),
|
||
|
|
("Zg==", "f"),
|
||
|
|
("Zm8=", "fo"),
|
||
|
|
("Zm9v", "foo"),
|
||
|
|
("Zm9vYg==", "foob"),
|
||
|
|
("Zm9vYmE=", "fooba"),
|
||
|
|
("Zm9vYmFy", "foobar")
|
||
|
|
];
|
||
|
|
for (input, expected) in tests {
|
||
|
|
let args = HashMap::new();
|
||
|
|
let result = base64_decode(to_value(input).unwrap(), args);
|
||
|
|
assert!(result.is_ok());
|
||
|
|
assert_eq!(result.unwrap(), to_value(expected).unwrap());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|