use crate::tokens::{utils::root_crate, Tokenize, TokenizeContext};
use proc_macro2::TokenStream;
use quote::quote;
use std::borrow::Cow;
use vimwiki_core::{Comment, LineComment, MultiLineComment};
impl_tokenize!(tokenize_comment, Comment<'a>, 'a);
fn tokenize_comment(ctx: &TokenizeContext, comment: &Comment) -> TokenStream {
let root = root_crate();
match comment {
Comment::Line(x) => {
let t = do_tokenize!(ctx, &x);
quote! { #root::Comment::Line(#t) }
}
Comment::MultiLine(x) => {
let t = do_tokenize!(ctx, &x);
quote! { #root::Comment::MultiLine(#t) }
}
}
}
impl_tokenize!(tokenize_line_comment, LineComment<'a>, 'a);
fn tokenize_line_comment(
ctx: &TokenizeContext,
line_comment: &LineComment,
) -> TokenStream {
let root = root_crate();
let t = do_tokenize!(ctx, Cow::Borrowed(line_comment.as_str()));
quote! {
#root::LineComment::new(#t)
}
}
impl_tokenize!(tokenize_multi_line_comment, MultiLineComment<'a>, 'a);
fn tokenize_multi_line_comment(
ctx: &TokenizeContext,
multi_line_comment: &MultiLineComment,
) -> TokenStream {
let root = root_crate();
let t = multi_line_comment
.iter()
.map(|x| do_tokenize!(ctx, Cow::Borrowed(x)));
quote! {
#root::MultiLineComment::new(::std::vec![#(#t),*])
}
}