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

use super::{
    builtin_type_ext::BuiltinTypeExt, to_ident::ToIdent, TokenStreamPair, TypeDefinitionExtensions,
};
use proc_macro2::TokenStream;
use quote::quote;
use witx::{BuiltinType, Id};

impl TypeDefinitionExtensions for BuiltinType {
    fn get_type_definitions(&self, ident: &Id, docs: TokenStream) -> TokenStreamPair {
        let inner_type = self.to_inner();

        let native_ident = ident.to_ident_native(None);

        let native = quote! {
            #docs
            pub type #native_ident = #inner_type;
        };

        let ident = ident.to_ident();
        let mapped = quote! {
            #docs
            #[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
            pub struct #ident(pub native::#native_ident);

            impl witx_gen::WasiValue for #ident {
                type NativeType = native::#native_ident;
                fn from_native(native: Self::NativeType) -> std::result::Result<Self, witx_gen::WasiValueError<Self>> {
                    Ok(Self(native))
                }
                fn to_native(self) -> Self::NativeType {
                    self.0
                }
            }
        };

        TokenStreamPair::from_streams(native, mapped)
    }

    fn has_rust_value(&self) -> bool {
        true
    }
}