drylib-procs 0.1.5

Rust macro-library for Don't Repeating Yourself
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use proc_macro::{
    Span,
    Ident,
    Group,
    Punct,
    Spacing,
    Delimiter,
    TokenTree,
    TokenStream
};

// IM = (ident -> i, mutability -> m).to_uppercase();
type IM = (Ident, bool);

use crate::clones_prefix::CLONES_PREFIX;

pub fn get_clones_(idents: Vec::<IM>, muts: bool) -> Vec::<TokenTree> {
    let mut ts = Vec::new();
    
    for (ident, mut_) in idents.into_iter() {
        ts.push(TokenTree::Ident(Ident::new("let", Span::call_site())));
        if muts || mut_ { ts.push(TokenTree::Ident(Ident::new("mut", Span::call_site()))); }
        ts.push(TokenTree::Ident(Ident::new(&format!("{CLONES_PREFIX}{ident}"), Span::call_site())));
        ts.push(TokenTree::Punct(Punct::new('=', Spacing::Alone)));
        ts.push(TokenTree::Ident(ident));
        ts.push(TokenTree::Punct(Punct::new('.', Spacing::Alone)));
        ts.push(TokenTree::Ident(Ident::new("clone", Span::call_site())));
        ts.push(TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::new())));
        ts.push(TokenTree::Punct(Punct::new(';', Spacing::Alone)));
    } ts
}

pub fn clones_parse_muts_<I>(iter: I) -> Vec::<IM>
where I: Iterator<Item = TokenTree>
{
    iter.fold((false, Vec::new()), |(next, mut ret), t| {
        if let TokenTree::Ident(ident) = t {
            if next {
                ret.push((ident, true));
                (false, ret)
            } else if ident.to_string().eq("mut") {
                (true, ret)
            } else {
                ret.push((ident, false));
                (false, ret)
            }
        } else {
            if next { panic!("Expected identifier after `mut`") }
            (false, ret)
        }
    }).1
}