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
67
68
use proc_macro2::TokenStream;
mod definition;
mod kw;
mod node;
mod scalar;
mod variants;
use definition::Definition;
use scalar::{Scalar, emit_scalar};
fn emit_decoder(def: &Definition) -> syn::Result<TokenStream> {
match def {
Definition::Struct(s) => node::emit_struct(s, true, false),
Definition::NewType(s) => node::emit_new_type(s),
Definition::TupleStruct(s) => node::emit_struct(s, false, false),
Definition::UnitStruct(s) => node::emit_struct(s, true, false),
Definition::Enum(e) => variants::emit_enum(e),
}
}
fn emit_partial_decoder(def: &Definition) -> syn::Result<TokenStream> {
match def {
Definition::Struct(s) => node::emit_struct(s, true, true),
Definition::NewType(_) => todo!(),
Definition::TupleStruct(s) => node::emit_struct(s, false, true),
Definition::UnitStruct(s) => node::emit_struct(s, true, true),
Definition::Enum(_) => todo!(),
}
}
#[proc_macro_error::proc_macro_error]
#[proc_macro_derive(Decode, attributes(kfl))]
pub fn decode_derive(input: proc_macro::TokenStream)
-> proc_macro::TokenStream
{
let item = syn::parse_macro_input!(input as Definition);
match emit_decoder(&item) {
Ok(stream) => stream.into(),
Err(e) => e.to_compile_error().into(),
}
}
#[proc_macro_error::proc_macro_error]
#[proc_macro_derive(DecodePartial, attributes(kfl))]
pub fn decode_partial_derive(input: proc_macro::TokenStream)
-> proc_macro::TokenStream
{
let item = syn::parse_macro_input!(input as Definition);
match emit_partial_decoder(&item) {
Ok(stream) => stream.into(),
Err(e) => e.to_compile_error().into(),
}
}
#[proc_macro_error::proc_macro_error]
#[proc_macro_derive(DecodeScalar, attributes(kfl))]
#[doc = include_str!("../derive_decode_scalar.md")]
pub fn decode_scalar_derive(input: proc_macro::TokenStream)
-> proc_macro::TokenStream
{
let item = syn::parse_macro_input!(input as Scalar);
match emit_scalar(&item) {
Ok(stream) => stream.into(),
Err(e) => e.to_compile_error().into(),
}
}