odra-ir 0.7.1

Data structures for Odra IR (intermediate representation).
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

use syn::{parse_quote, FnArg, PatType};

pub fn is_mut(sig: &syn::Signature) -> bool {
    let receiver = sig.inputs.iter().find_map(|input| match input {
        syn::FnArg::Receiver(receiver) => Some(receiver),
        syn::FnArg::Typed(_) => None
    });
    receiver.and_then(|r| r.mutability).is_some()
}

pub fn is_ref(ty: &PatType) -> bool {
    matches!(&*ty.ty, syn::Type::Reference(_))
}

pub fn ty(ty: &PatType) -> (syn::Type, bool) {
    let ty = deref_ty(&ty.ty);
    match ty {
        syn::Type::Slice(t) => {
            let slice_ty = &*t.elem;
            (parse_quote!(odra::prelude::vec::Vec<#slice_ty>), true)
        }
        _ => (ty.clone(), false)
    }
}

fn deref_ty(ty: &syn::Type) -> &syn::Type {
    match ty {
        syn::Type::Reference(r) => deref_ty(&r.elem),
        other => other
    }
}

pub fn typed_arg(arg: &FnArg) -> Option<PatType> {
    match arg {
        syn::FnArg::Receiver(_) => None,
        syn::FnArg::Typed(pat) => Some(pat.clone())
    }
}

pub fn extract_typed_inputs(
    sig: &syn::Signature
) -> syn::punctuated::Punctuated<syn::PatType, syn::token::Comma> {
    sig.inputs
        .iter()
        .filter_map(typed_arg)
        .collect::<syn::punctuated::Punctuated<syn::PatType, syn::token::Comma>>()
}