plux-codegen 0.3.0

A code generator for plux-rs
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 super::{generate_struct::generate_struct, utils::get_externals};

use proc_macro2::{Ident, TokenStream};
use quote::quote;
use syn::{ItemFn, Result, Type};

pub(crate) fn generate(ast: &ItemFn, _: &TokenStream) -> Result<TokenStream> {
    let sig = &ast.sig;
    let ident = &sig.ident;

    let exts = get_externals(&sig.inputs[0]);
    let exts_args = generate_externals_atributes(&exts);
    let externals = generate_externals(&exts);

    let structure = generate_struct(ast, sig, ident, &exts)?;

    Ok(quote! {
        pub fn #ident(#exts_args) -> impl plux_rs::function::Function<Output = plux_rs::function::FunctionOutput> {
            #structure

            Function { #externals }
        }
    })
}

fn generate_externals_atributes(exts: &Vec<(Ident, &Type)>) -> TokenStream {
    let exts: Vec<_> = exts
        .iter()
        .map(|(name, ty)| {
            quote! { #name: #ty }
        })
        .collect();

    quote! { #(#exts),* }
}

fn generate_externals(exts: &Vec<(Ident, &Type)>) -> TokenStream {
    let exts = exts.iter().map(|(name, _)| name).collect::<Vec<_>>();

    quote! { #(#exts),* }
}