cgp-macro-lib 0.7.0

Context-generic programming core component macros implemented as a library.
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

use proc_macro2::{Ident, TokenStream};
use syn::{FnArg, Receiver, Signature, parse_quote};

pub fn replace_self_receiver_in_signature(
    sig: &mut Signature,
    replaced_var: &Ident,
    replaced_type: TokenStream,
) {
    if let Some(arg) = sig.inputs.first_mut()
        && let FnArg::Receiver(receiver) = arg
    {
        *arg = replace_self_receiver(receiver, replaced_var, replaced_type);
    }
}

pub fn replace_self_receiver(
    receiver: &mut Receiver,
    replaced_var: &Ident,
    replaced_type: TokenStream,
) -> FnArg {
    match (&receiver.reference, &receiver.mutability) {
        (None, None) => {
            parse_quote!(#replaced_var : #replaced_type)
        }
        (Some((_and, None)), None) => {
            parse_quote!(#replaced_var : & #replaced_type)
        }
        (Some((_and, Some(life))), None) => {
            parse_quote!(#replaced_var : & #life #replaced_type)
        }
        (Some((_and, None)), Some(_mut)) => {
            parse_quote!(#replaced_var : &mut #replaced_type)
        }
        (Some((_and, Some(life))), Some(_mut)) => {
            parse_quote!(#replaced_var : & #life mut #replaced_type)
        }
        (None, Some(_mut)) => {
            parse_quote!(#replaced_var : mut #replaced_type)
        }
    }
}