AppFlowy/shared-lib/flowy-derive/src/proto_buf/mod.rs

mod deserialize;
mod enum_serde;
mod serialize;
mod util;

use crate::proto_buf::{
    deserialize::make_de_token_steam, enum_serde::make_enum_token_stream, serialize::make_se_token_stream,
};
use flowy_ast::*;
use proc_macro2::TokenStream;
use std::default::Default;

pub fn expand_derive(input: &syn::DeriveInput) -> Result<TokenStream, Vec<syn::Error>> {
    let ctxt = Ctxt::new();
    let cont = match ASTContainer::from_ast(&ctxt, input) {
        Some(cont) => cont,
        None => return Err(ctxt.check().unwrap_err()),
    };

    let mut token_stream: TokenStream = TokenStream::default();

    if let Some(de_token_stream) = make_de_token_steam(&ctxt, &cont) {
        token_stream.extend(de_token_stream);
    }

    if let Some(se_token_stream) = make_se_token_stream(&ctxt, &cont) {
        token_stream.extend(se_token_stream);
    }

    ctxt.check()?;
    Ok(token_stream)
}

pub fn expand_enum_derive(input: &syn::DeriveInput) -> Result<TokenStream, Vec<syn::Error>> {
    let ctxt = Ctxt::new();
    let cont = match ASTContainer::from_ast(&ctxt, input) {
        Some(cont) => cont,
        None => return Err(ctxt.check().unwrap_err()),
    };

    let mut token_stream: TokenStream = TokenStream::default();

    if let Some(enum_token_stream) = make_enum_token_stream(&ctxt, &cont) {
        token_stream.extend(enum_token_stream);
    }

    ctxt.check()?;
    Ok(token_stream)
}
// #[macro_use]
// macro_rules! impl_try_for_primitive_type {
//     ($target:ident) => {
//         impl std::convert::TryFrom<&$target> for $target {
//             type Error = String;
//             fn try_from(val: &$target) -> Result<Self, Self::Error> {
// Ok(val.clone()) }         }
//
//         impl std::convert::TryInto<$target> for $target {
//             type Error = String;
//
//             fn try_into(self) -> Result<Self, Self::Error> { Ok(self) }
//         }
//     };
// }
//
// impl_try_for_primitive_type!(String);
// impl_try_for_primitive_type!(i64);
// impl_try_for_primitive_type!(i32);
// impl_try_for_primitive_type!(i16);
// impl_try_for_primitive_type!(u64);
// impl_try_for_primitive_type!(u32);
// impl_try_for_primitive_type!(u16);
// impl_try_for_primitive_type!(bool);
// impl_try_for_primitive_type!(f64);
// impl_try_for_primitive_type!(f32);