62e00434b0
todo: need an unsafe method for casting pointers in order to validate contents via recursive descent
39 lines
1.3 KiB
Rust
39 lines
1.3 KiB
Rust
use proc_macro2::TokenStream;
|
|
use quote::quote;
|
|
|
|
use crate::names::Names;
|
|
use witx::Module;
|
|
|
|
pub fn define_module_trait(names: &Names, m: &Module) -> TokenStream {
|
|
let traitname = names.trait_name(&m.name);
|
|
let traitmethods = m.funcs().map(|f| {
|
|
let funcname = names.func(&f.name);
|
|
let args = f.params.iter().map(|arg| {
|
|
let arg_name = names.func_param(&arg.name);
|
|
let arg_typename = names.type_ref(&arg.tref);
|
|
let arg_type = match arg.tref.type_().passed_by() {
|
|
witx::TypePassedBy::Value { .. } => quote!(#arg_typename),
|
|
witx::TypePassedBy::Pointer { .. } => quote!(&#arg_typename),
|
|
witx::TypePassedBy::PointerLengthPair { .. } => unimplemented!(),
|
|
};
|
|
quote!(#arg_name: #arg_type)
|
|
});
|
|
let rets = f
|
|
.results
|
|
.iter()
|
|
.skip(1)
|
|
.map(|ret| names.type_ref(&ret.tref));
|
|
let err = f
|
|
.results
|
|
.get(0)
|
|
.map(|err_result| names.type_ref(&err_result.tref))
|
|
.unwrap_or(quote!(()));
|
|
quote!(fn #funcname(&mut self, #(#args),*) -> Result<(#(#rets),*), #err>;)
|
|
});
|
|
quote! {
|
|
pub trait #traitname {
|
|
#(#traitmethods)*
|
|
}
|
|
}
|
|
}
|