use std::slice::Iter;
use crate::comment::Comment;
use crate::fragment::Fragment;
#[derive(Debug, PartialEq, Clone)]
pub struct Comments {
comments: Vec<Comment>,
}
impl Comments {
#[must_use]
pub fn iter(&self) -> Iter<'_, Comment> {
self.comments.iter()
}
}
impl From<Vec<Comment>> for Comments {
fn from(comments: Vec<Comment>) -> Self {
Comments { comments }
}
}
impl From<Comments> for String {
fn from(comments: Comments) -> Self {
comments
.comments
.into_iter()
.map(String::from)
.collect::<Vec<_>>()
.join("\n\n")
}
}
impl From<Vec<Fragment>> for Comments {
fn from(ast: Vec<Fragment>) -> Self {
ast.iter()
.filter_map(|values| {
if let Fragment::Comment(comment) = values {
Some(comment.clone())
} else {
None
}
})
.collect::<Vec<Comment>>()
.into()
}
}
#[cfg(test)]
mod tests {
use pretty_assertions::assert_eq;
use indoc::indoc;
use crate::body::Body;
use crate::comment::Comment;
use crate::fragment::Fragment;
use super::Comments;
#[test]
fn implements_iterator() {
use crate::Comment;
use crate::Comments;
let trailers = Comments::from(vec![
Comment::from("# Comment 1"),
Comment::from("# Comment 2"),
Comment::from("# Comment 3"),
]);
let mut iterator = trailers.iter();
assert_eq!(iterator.next(), Some(&Comment::from("# Comment 1")));
assert_eq!(iterator.next(), Some(&Comment::from("# Comment 2")));
assert_eq!(iterator.next(), Some(&Comment::from("# Comment 3")));
assert_eq!(iterator.next(), None);
}
#[test]
fn it_can_give_me_it_as_a_string() {
let comments = Comments::from(vec![
Comment::from("# Message Body"),
Comment::from("# Another Message Body"),
]);
assert_eq!(
String::from(comments),
String::from(indoc!(
"
# Message Body
# Another Message Body"
))
)
}
#[test]
fn it_can_create_itself_from_an_ast() {
let comments = Comments::from(vec![
Fragment::Comment(Comment::from("# Message Body")),
Fragment::Body(Body::from("Some body content")),
Fragment::Comment(Comment::from("# Another Message Body")),
]);
assert_eq!(
comments,
Comments::from(vec![
Comment::from("# Message Body"),
Comment::from("# Another Message Body"),
])
)
}
}