cmod-macros 0.4.6

Build a cross-language module with Rust FFI
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

use super::utils::*;
use proc_macro::TokenStream;
use quote::quote;
use std::mem;
use syn::{parse_macro_input, parse_quote, Ident, ImplItem, ImplItemFn, ItemImpl};

pub fn cmod_methods(_attr: TokenStream, input: TokenStream) -> TokenStream {
    let mut input = parse_macro_input!(input as ItemImpl);
    let mut py_input = input.clone();

    input.attrs.clear();
    input.items.iter_mut().for_each(|ii| {
        if let ImplItem::Fn(md) = ii {
            md.attrs.clear();
        }
    });

    let mut item_record: Vec<ImplItem> = Vec::new();
    py_input.items.iter().for_each(|ii| {
        if let ImplItem::Fn(md) = ii {
            match inner_method_handle(md) {
                Flag::Static => {
                    item_record.push(ImplItem::from(method_static(md.clone())));
                }
                Flag::Class => {
                    item_record.push(ImplItem::from(method_class(md.clone())));
                }
                _ => (),
            }
        }
    });
    {
        let item = &mut py_input.items;
        let _ = mem::replace(item, item_record);
    }
    TokenStream::from(quote!(
        #input

        #[pyo3::pymethods]
        #py_input
    ))
}

enum Flag {
    Empty,
    Static,
    Class,
}

fn inner_method_handle(inner_method: &ImplItemFn) -> Flag {
    for p in inner_method.attrs.iter() {
        if p.meta.path().is_ident("staticmethod") {
            return Flag::Static;
        } else if p.meta.path().is_ident("classmethod") {
            return Flag::Class;
        }
    }
    Flag::Empty
}

pub fn method_static(input: ImplItemFn) -> ImplItemFn {
    let py_input = input.clone();
    let function = Function::parse_impl_fn(py_input);
    let Function {
        name,
        asy,
        input: inp,
        args,
        ret,
        map_ret,
    } = function;
    let after_name = Ident::rename(name.clone());
    let name_str = name.to_string();
    if asy {
        parse_quote!(
            #[staticmethod]
            #[pyo3(name = #name_str)]
            fn #after_name(py: pyo3::Python, #inp)#ret{
                cmod::ffi::py::block_on(py, async move{
                    Self::#name(#args).await.map_err(cmod::ffi::py::map_err)#map_ret
                })
            }
        )
    } else {
        parse_quote!(
            #[staticmethod]
            #[pyo3(name = #name_str)]
            fn #after_name(py: pyo3::Python, #inp)#ret{
                Self::#name(#args).map_err(cmod::ffi::py::map_err)#map_ret
            }
        )
    }
}

pub fn method_class(input: ImplItemFn) -> ImplItemFn {
    let py_input = input.clone();
    let function = Function::parse_impl_fn(py_input);
    let Function {
        name,
        asy,
        input: mut inp,
        args,
        ret,
        map_ret,
    } = function;
    inp = inp.into_iter().skip(1).collect();
    let after_name = Ident::rename(name.clone());
    let name_str = name.to_string();
    if asy {
        parse_quote!(
            #[pyo3(name = #name_str)]
            fn #after_name<'py>(this:pyo3::Py<Self>,py: pyo3::Python<'py>, #inp)#ret{
                let this:Self = this.extract(py)?;
                cmod::ffi::py::block_on(py, async move{
                    this.#name(#args).await.map_err(cmod::ffi::py::map_err)#map_ret
                })
            }
        )
    } else {
        parse_quote!(
            #[pyo3(name = #name_str)]
            fn #after_name<'py>(this: pyo3::Py<Self>,py: pyo3::Python<'py>, #inp)#ret{
                let this:Self = this.extract(py)?;
                this.#name(#args).map_err(cmod::ffi::py::map_err)#map_ret
            }
        )
    }
}