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

use super::utils::Utils;
use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, parse_quote, Ident, Item, ItemFn, ItemMod, Stmt};

pub fn cmod_module(_attr: TokenStream, input: TokenStream) -> TokenStream {
    let mut input = parse_macro_input!(input as ItemMod);
    let name = input.ident.clone();
    let name_str = name.to_string();
    let after_name = Ident::rename_module(name.clone());
    let item = input.content.clone();
    let mut ident_record = Vec::new();
    if let Some((_b, it)) = item {
        it.into_iter().for_each(|i| {
            if let Item::Fn(ifn) = i {
                if ifn.attrs.iter().any(|attr| attr.meta.path().segments.last().unwrap().ident == "function") {
                    ident_record.push(ifn.sig.ident.clone());
                }
            }
        });
    }
    let mut add_func = ident_record
        .into_iter()
        .map(|ident| {
            let ident = Ident::rename(ident);
            let semi: Stmt = parse_quote!(
                m.add_function(pyo3::wrap_pyfunction!(#ident, m)?)?;
            );
            semi
        })
        .collect::<Vec<Stmt>>();
    add_func.push(parse_quote!(
        return father.add_submodule(m);
    ));
    if let Some(ref mut ct) = input.content {
        let mut ifn: ItemFn = parse_quote!(
            pub fn #after_name(py: pyo3::Python<'_>, father: &pyo3::types::PyModule) -> pyo3::PyResult<()>{
                let m = pyo3::types::PyModule::new(py, #name_str)?;
            }
        );
        ifn.block.stmts.append(&mut add_func);
        ct.1.push(Item::Fn(ifn));
    }
    TokenStream::from(quote!(
        #input
    ))
}