AppFlowy/rust-lib/flowy-derive/src/proto_buf/enum_serde.rs

use flowy_ast::*;
use proc_macro2::TokenStream;

#[allow(dead_code)]
pub fn make_enum_token_stream(_ctxt: &Ctxt, cont: &ASTContainer) -> Option<TokenStream> {
    let enum_ident = &cont.ident;
    let pb_enum = cont.attrs.pb_enum_type()?;
    let build_to_pb_enum = cont.data.all_idents().map(|i| {
        let token_stream: TokenStream = quote! {
            #enum_ident::#i => crate::protobuf::#pb_enum::#i,
        };
        token_stream
    });

    let build_from_pb_enum = cont.data.all_idents().map(|i| {
        let token_stream: TokenStream = quote! {
            crate::protobuf::#pb_enum::#i => #enum_ident::#i,
        };
        token_stream
    });

    Some(quote! {
        impl std::convert::TryFrom<&crate::protobuf::#pb_enum> for #enum_ident {
            type Error = String;
            fn try_from(pb:&crate::protobuf::#pb_enum) -> Result<Self, Self::Error> {
                Ok(match pb {
                    #(#build_from_pb_enum)*
                })
            }
        }

        impl std::convert::TryInto<crate::protobuf::#pb_enum> for #enum_ident {
            type Error = String;
            fn try_into(self) -> Result<crate::protobuf::#pb_enum, Self::Error> {
                Ok(match self {
                    #(#build_to_pb_enum)*
                })
            }
        }
    })
}