traitful 0.3.0

A collection of helper macros for trait patterns
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

use syn::{FnArg, Pat, Signature};

/// Remove `mut` references from function arguments in signature.
pub(crate) fn remove_arg_patterns(signature: &mut Signature) {
    for arg in signature.inputs.iter_mut() {
        match arg {
            FnArg::Receiver(ref mut receiver) => {
                if receiver.reference.is_none() {
                    receiver.mutability = None;
                }
            }
            FnArg::Typed(ref mut typed) => remove_pat_patterns(&mut typed.pat),
        }
    }
}

fn remove_pat_patterns(typed: &mut Pat) {
    match typed {
        Pat::Ident(ident) => {
            if ident.by_ref.is_none() {
                ident.mutability = None;
            }
        }
        Pat::Or(or) => {
            for pat in or.cases.iter_mut() {
                remove_pat_patterns(pat);
            }
        }
        Pat::Paren(paren) => remove_pat_patterns(&mut paren.pat),
        Pat::Reference(reference) => remove_pat_patterns(&mut reference.pat),
        Pat::Slice(slice) => {
            for pat in slice.elems.iter_mut() {
                remove_pat_patterns(pat);
            }
        }
        Pat::Struct(tuple) => {
            for pat in tuple.fields.iter_mut() {
                remove_pat_patterns(&mut pat.pat);
            }
        }
        Pat::Tuple(tuple) => {
            for pat in tuple.elems.iter_mut() {
                remove_pat_patterns(pat);
            }
        }
        Pat::TupleStruct(tuple) => {
            for pat in tuple.elems.iter_mut() {
                remove_pat_patterns(pat);
            }
        }
        Pat::Type(ty) => remove_pat_patterns(&mut ty.pat),
        _ => {}
    }
}