drylib-procs 0.1.13

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
53
54
55
56
57
58
59
60
61

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

use crate::parse::IT;
use crate::parse::TokenType;
use crate::ams_prefix::AMS_PREFIX;

pub fn get_ams_(idents: Vec::<IT>, muts: bool) -> Vec::<TokenTree> {
    let mut ts = Vec::new();
    
    for (ident, ttype) in idents.into_iter() {
        ts.push(TokenTree::Ident(Ident::new("let", Span::call_site())));
        if muts || ttype.eq(&TokenType::Mutable) || ttype.eq(&TokenType::MutableReference) {
            ts.push(TokenTree::Ident(Ident::new("mut", Span::call_site())));
        }
        ts.push(TokenTree::Ident(Ident::new(&format!("{AMS_PREFIX}{ident}"), Span::call_site())));
        ts.push(TokenTree::Punct(Punct::new('=', Spacing::Alone)));

        ts.push(TokenTree::Ident(Ident::new("std", Span::call_site())));
        ts.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        ts.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        ts.push(TokenTree::Ident(Ident::new("sync", Span::call_site())));
        ts.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        ts.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        ts.push(TokenTree::Ident(Ident::new("Arc", Span::call_site())));
        ts.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        ts.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        ts.push(TokenTree::Ident(Ident::new("new", Span::call_site())));

        let mut group = Vec::new();
        group.push(TokenTree::Ident(Ident::new("std", Span::call_site())));
        group.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        group.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        group.push(TokenTree::Ident(Ident::new("sync", Span::call_site())));
        group.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        group.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        group.push(TokenTree::Ident(Ident::new("Mutex", Span::call_site())));
        group.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        group.push(TokenTree::Punct(Punct::new(':', Spacing::Joint)));
        group.push(TokenTree::Ident(Ident::new("new", Span::call_site())));

        let ident = if ttype.eq(&TokenType::ImmutableReference) || ttype.eq(&TokenType::MutableReference) {
            vec![TokenTree::Punct(Punct::new('&', Spacing::Joint)), TokenTree::Ident(ident)]
        } else {
            vec![TokenTree::Ident(ident)]
        };
        group.push(TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::from_iter(ident))));

        ts.push(TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::from_iter(group))));
        ts.push(TokenTree::Punct(Punct::new(';', Spacing::Alone)));
    } ts
}