tl_proto_proc/
lib.rs

1use proc_macro::TokenStream;
2use quote::quote;
3
4use self::id::*;
5use self::tl_read::*;
6use self::tl_write::*;
7
8mod bound;
9mod dummy;
10mod id;
11mod internals;
12mod scheme_loader;
13mod tl_read;
14mod tl_write;
15
16#[derive(Copy, Clone, Eq, PartialEq)]
17enum Derive {
18    Write,
19    Read,
20}
21
22/// Computes TL id at compile time using the given variant name and scheme.
23///
24/// ### Example
25/// ```rust
26/// const ID1: u32 = tl_proto::id!("boolTrue", scheme = "my.tl");
27///
28/// // or with inline scheme
29/// const ID2: u32 = tl_proto::id!(
30///     "boolTrue",
31///     scheme_inline = r##"
32///         boolTrue = Bool;
33///         boolFalse = Bool;
34///     "##
35/// );
36/// ```
37#[proc_macro]
38pub fn id(input: TokenStream) -> TokenStream {
39    let meta = {
40        let input = input.clone();
41        syn::parse_macro_input!(input as IdMacroInput)
42    };
43    impl_id(&input.into(), &meta)
44        .unwrap_or_else(to_compile_errors)
45        .into()
46}
47
48/// Implements `TlWrite` for the type.
49#[proc_macro_derive(TlWrite, attributes(tl))]
50pub fn derive_tl_write(input: TokenStream) -> TokenStream {
51    let input = syn::parse_macro_input!(input as syn::DeriveInput);
52    impl_derive_tl_write(input)
53        .unwrap_or_else(to_compile_errors)
54        .into()
55}
56
57/// Implements `TlRead` for the type.
58#[proc_macro_derive(TlRead, attributes(tl))]
59pub fn derive_tl_read(input: TokenStream) -> TokenStream {
60    let input = syn::parse_macro_input!(input as syn::DeriveInput);
61    impl_derive_tl_read(input)
62        .unwrap_or_else(to_compile_errors)
63        .into()
64}
65
66fn to_compile_errors(errors: Vec<syn::Error>) -> proc_macro2::TokenStream {
67    let compile_errors = errors.iter().map(syn::Error::to_compile_error);
68    quote!(#(#compile_errors)*)
69}