AppFlowy/frontend/rust-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 ast_result = ASTResult::new();
  let cont = match ASTContainer::from_ast(&ast_result, input) {
    Some(cont) => cont,
    None => return Err(ast_result.check().unwrap_err()),
  };

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

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

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

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

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

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

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

  ast_result.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);