redis-lua-macro 0.4.0

Macro component of Redis Lua scripting helper
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 crate::{
    proc_macro::{Ident, Span, TokenStream as TokenStream1, TokenTree},
    script::{Arg, ArgType, Script},
};
use proc_macro2::TokenStream;
use quote::quote;

pub fn new_ident(s: String) -> TokenStream {
    let s: TokenStream1 = TokenTree::Ident(Ident::new(&s, Span::call_site())).into();
    s.into()
}

pub fn all(script: &Script) -> impl Iterator<Item = (usize, &Arg)> {
    script.args().iter().enumerate()
}

pub fn caps(script: &Script) -> impl Iterator<Item = (usize, &Arg)> {
    all(script).filter(|(_, arg)| arg.atype() == ArgType::Cap)
}

pub fn vars(script: &Script) -> impl Iterator<Item = (usize, &Arg)> {
    all(script).filter(|(_, arg)| arg.atype() == ArgType::Var)
}

pub fn to_name((_index, arg): (usize, &Arg)) -> TokenStream {
    new_ident(arg.as_rust().to_string())
}

pub fn to_type((index, _): (usize, &Arg)) -> TokenStream {
    new_ident(format!("A{}", index))
}

pub fn to_param((index, _): (usize, &Arg)) -> TokenStream {
    new_ident(format!("a{}", index))
}

pub fn to_mem(t: (usize, &Arg)) -> TokenStream {
    let p = to_param(t);
    let t = to_type(t);
    quote! { #p: Option<#t> }
}

pub fn to_arg(t: (usize, &Arg)) -> TokenStream {
    let p = to_param(t);
    let t = to_type(t);
    quote! { #p: #t }
}

pub fn to_bound(t: (usize, &Arg)) -> TokenStream {
    let t = to_type(t);
    quote! { #t: redis_lua::serde::Serialize }
}

pub fn to_bound_life(t: (usize, &Arg)) -> TokenStream {
    let t = to_type(t);
    quote! { #t: redis_lua::serde::Serialize + 'a }
}

pub fn varlen(script: &Script) -> usize {
    vars(script).count()
}