1use proc_macro2::TokenStream;
2
3mod definition;
4mod kw;
5mod node;
6mod scalar;
7mod variants;
8
9use definition::Definition;
10use scalar::{Scalar, emit_scalar};
11
12
13fn emit_decoder(def: &Definition) -> syn::Result<TokenStream> {
14 match def {
15 Definition::Struct(s) => node::emit_struct(s, true),
16 Definition::NewType(s) => node::emit_new_type(s),
17 Definition::TupleStruct(s) => node::emit_struct(s, false),
18 Definition::UnitStruct(s) => node::emit_struct(s, true),
19 Definition::Enum(e) => variants::emit_enum(e),
20 }
21}
22
23#[proc_macro_error::proc_macro_error]
24#[proc_macro_derive(Decode, attributes(knuffel))]
25#[doc = include_str!("../derive_decode.md")]
26pub fn decode_derive(input: proc_macro::TokenStream)
27 -> proc_macro::TokenStream
28{
29 let item = syn::parse_macro_input!(input as Definition);
30 match emit_decoder(&item) {
31 Ok(stream) => stream.into(),
32 Err(e) => e.to_compile_error().into(),
33 }
34}
35
36#[proc_macro_error::proc_macro_error]
37#[proc_macro_derive(DecodeScalar, attributes(knuffel))]
38#[doc = include_str!("../derive_decode_scalar.md")]
39pub fn decode_scalar_derive(input: proc_macro::TokenStream)
40 -> proc_macro::TokenStream
41{
42 let item = syn::parse_macro_input!(input as Scalar);
43 match emit_scalar(&item) {
44 Ok(stream) => stream.into(),
45 Err(e) => e.to_compile_error().into(),
46 }
47}