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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

//! Serialize and deserialize enum string.
//!
//! ```Cargo.toml
//! [dependencies]
//! serde = "1.0"
//! serde-enum-str = "0.1"
//! ```
//!
//! ```
//! use serde_enum_str::{Deserialize_enum_str, Serialize_enum_str};
//!
//! #[derive(Deserialize_enum_str, Serialize_enum_str, PartialEq, Debug)]
//! #[serde(rename_all = "snake_case")]
//! enum Foo {
//!     A,
//!     #[serde(rename = "B")]
//!     B,
//!     #[serde(other)]
//!     Other(String),
//! }
//!
//! fn main() -> serde_json::Result<()> {
//!     assert_eq!(serde_json::to_string(&Foo::A).unwrap(), r#""a""#);
//!     assert_eq!(serde_json::from_str::<Foo>(r#""a""#).unwrap(), Foo::A);
//!
//!     assert_eq!(
//!         serde_json::from_str::<Foo>(r#""c""#).unwrap(),
//!         Foo::Other("c".to_owned())
//!     );
//!
//!     assert_eq!(Foo::A.to_string(), "a");
//!     assert_eq!("a".parse::<Foo>().unwrap(), Foo::A);
//!
//!     Ok(())
//! }
//! ```

#![allow(clippy::single_match_else)]

extern crate proc_macro;

use proc_macro::TokenStream;
use quote::quote;
use syn::parse_macro_input;

mod attributes;
mod input;
mod output;
mod output_de;
mod output_ser;

#[proc_macro_derive(Serialize_enum_str, attributes(serde))]
pub fn derive_serialize(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as self::input::Input);
    let input = self::output_ser::InputWrapper(input);

    TokenStream::from(quote!(#input))
}

#[proc_macro_derive(Deserialize_enum_str, attributes(serde))]
pub fn derive_deserialize(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as self::input::Input);
    let input = self::output_de::InputWrapper(input);

    TokenStream::from(quote!(#input))
}