lib.rs

//! # Derive macros for traits in `tls_codec`
//!
//! Derive macros can be used to automatically implement the
//! [`Serialize`](../tls_codec::Serialize),
//! [`SerializeBytes`](../tls_codec::SerializeBytes),
//! [`Deserialize`](../tls_codec::Deserialize),
//! [`DeserializeBytes`](../tls_codec::DeserializeBytes), and
//! [`Size`](../tls_codec::Size) traits for structs and enums. Note that the
//! functions of the [`Serialize`](../tls_codec::Serialize) and
//! [`Deserialize`](../tls_codec::Deserialize) traits (and thus the
//! corresponding derive macros) require the `"std"` feature to work.
//!
//! ## Warning
//!
//! The derive macros support deriving the `tls_codec` traits for enumerations
//! and the resulting serialized format complies with [the "variants" section of
//! the TLS RFC](https://datatracker.ietf.org/doc/html/rfc8446#section-3.8).
//! However support is limited to enumerations that are serialized with their
//! discriminant immediately followed by the variant data. If this is not
//! appropriate (e.g. the format requires other fields between the discriminant
//! and variant data), the `tls_codec` traits can be implemented manually.
//!
//! ## Parsing unknown values
//!
//! In many cases it is necessary to deserialize structs with unknown values,
//! e.g. when receiving unknown TLS extensions. In this case the deserialize
//! function returns an `Error::UnknownValue` with a `u64` value of the unknown
//! type.
//!
//! ```
//! # #[cfg(feature = "std")]
//! # {
//! use tls_codec_derive::{TlsDeserialize, TlsSerialize, TlsSize};
//!
//! #[derive(TlsDeserialize, TlsSerialize, TlsSize)]
//! #[repr(u16)]
//! enum TypeWithUnknowns {
//!     First = 1,
//!     Second = 2,
//! }
//!
//! #[test]
//! fn type_with_unknowns() {
//!     let incoming = [0x00u8, 0x03]; // This must be parsed into TypeWithUnknowns into an unknown
//!     let deserialized = TypeWithUnknowns::tls_deserialize_exact(incoming);
//!     assert!(matches!(deserialized, Err(Error::UnknownValue(3))));
//! }
//! # }
//! ```
//!
//! ## Available attributes
//!
//! Attributes can be used to control serialization and deserialization on a
//! per-field basis.
//!
//! ### with
//!
//! ```text
//! #[tls_codec(with = "prefix")]
//! ```
//!
//! This attribute may be applied to a struct field. It indicates that deriving
//! any of the `tls_codec` traits for the containing struct calls the following
//! functions:
//! - `prefix::tls_deserialize` when deriving `Deserialize`
//! - `prefix::tls_serialize` when deriving `Serialize`
//! - `prefix::tls_serialized_len` when deriving `Size`
//!
//! `prefix` can be a path to a module, type or trait where the functions are
//! defined.
//!
//! Their expected signatures match the corresponding methods in the traits.
//!
//! ```
//! # #[cfg(feature = "std")]
//! # {
//! use tls_codec_derive::{TlsSerialize, TlsSize};
//!
//! #[derive(TlsSerialize, TlsSize)]
//! struct Bytes {
//!     #[tls_codec(with = "bytes")]
//!     values: Vec<u8>,
//! }
//!
//! mod bytes {
//!     use std::io::Write;
//!     use tls_codec::{Serialize, Size, TlsByteSliceU32};
//!
//!     pub fn tls_serialized_len(v: &[u8]) -> usize {
//!         TlsByteSliceU32(v).tls_serialized_len()
//!     }
//!
//!     pub fn tls_serialized_len_checked(v: &[u8]) -> Option<usize> {
//!         TlsByteSliceU32(v).tls_serialized_len_checked()
//!     }
//!
//!     pub fn tls_serialize<W: Write>(v: &[u8], writer: &mut W) -> Result<usize, tls_codec::Error> {
//!         TlsByteSliceU32(v).tls_serialize(writer)
//!     }
//! }
//! # }
//! ```
//!
//! ### discriminant
//!
//! ```text
//! #[tls_codec(discriminant = 123)]
//! #[tls_codec(discriminant = "path::to::const::or::enum::Variant")]
//! ```
//!
//! This attribute may be applied to an enum variant to specify the discriminant
//! to use when serializing it. If all variants are units (e.g. they do not have
//! any data), this attribute must not be used and the desired discriminants
//! should be assigned to the variants using standard Rust syntax (`Variant =
//! Discriminant`).
//!
//! For enumerations with non-unit variants, if no variant has this attribute,
//! the serialization discriminants will start from zero. If this attribute is
//! used on a variant and the following variant does not have it, its
//! discriminant will be equal to the previous variant discriminant plus 1. This
//! behavior is referred to as "implicit discriminants".
//!
//! You can also provide paths that lead to `const` definitions or enum
//! Variants. The important thing is that any of those path expressions must
//! resolve to something that can be coerced to the `#[repr(enum_repr)]` of the
//! enum. Please note that there are checks performed at compile time to check
//! if the provided value fits within the bounds of the `enum_repr` to avoid
//! misuse.
//!
//! Note: When using paths *once* in your enum discriminants, as we do not have
//! enough information to deduce the next implicit discriminant (the constant
//! expressions those paths resolve is only evaluated at a later compilation
//! stage than macros), you will be forced to use explicit discriminants for all
//! the other Variants of your enum.
//!
//! ```
//! # #[cfg(feature = "std")]
//! # {
//! use tls_codec_derive::{TlsSerialize, TlsSize};
//!
//! const CONST_DISCRIMINANT: u8 = 5;
//! #[repr(u8)]
//! enum TokenType {
//!     Constant = 3,
//!     Variant = 4,
//! }
//!
//! #[derive(TlsSerialize, TlsSize)]
//! #[repr(u8)]
//! enum TokenImplicit {
//!     #[tls_codec(discriminant = 5)]
//!     Int(u32),
//!     // This will have the discriminant 6 as it's implicitly determined
//!     Bytes([u8; 16]),
//! }
//!
//! #[derive(TlsSerialize, TlsSize)]
//! #[repr(u8)]
//! enum TokenExplicit {
//!     #[tls_codec(discriminant = "TokenType::Constant")]
//!     Constant(u32),
//!     #[tls_codec(discriminant = "TokenType::Variant")]
//!     Variant(Vec<u8>),
//!     #[tls_codec(discriminant = "CONST_DISCRIMINANT")]
//!     StaticConstant(u8),
//! }
//! # }
//! ```
//!
//! ### skip
//!
//! ```text
//! #[tls_codec(skip)]
//! ```
//!
//! This attribute may be applied to a struct field to specify that it should be
//! skipped. Skipping means that the field at hand will neither be serialized
//! into TLS bytes nor deserialized from TLS bytes. For deserialization, it is
//! required to populate the field with a known value. Thus, when `skip` is
//! used, the field type needs to implement the [Default] trait so it can be
//! populated with a default value.
//!
//! ```
//! # #[cfg(feature = "std")]
//! # {
//! use tls_codec_derive::{TlsSerialize, TlsDeserialize, TlsSize};
//!
//! struct CustomStruct;
//!
//! impl Default for CustomStruct {
//!     fn default() -> Self {
//!         CustomStruct {}
//!     }
//! }
//!
//! #[derive(TlsSerialize, TlsDeserialize, TlsSize)]
//! struct StructWithSkip {
//!     a: u8,
//!     #[tls_codec(skip)]
//!     b: CustomStruct,
//!     c: u8,
//! }
//! # }
//! ```
//!
//! ## Conditional deserialization via the `conditionally_deserializable` attribute macro
//!
//! In some cases, it can be useful to have two variants of a struct, where one
//! is deserializable and one isn't. For example, the deserializable variant of
//! the struct could represent an unverified message, where only verification
//! produces the verified variant. Further processing could then be restricted
//! to the undeserializable struct variant.
//!
//! A pattern like this can be created via the `conditionally_deserializable`
//! attribute macro (requires the `conditional_deserialization` feature flag).
//!
//! The macro adds a boolean const generic to the struct and creates two
//! aliases, one for the deserializable variant (with a "`Deserializable`"
//! prefix) and one for the undeserializable one (with an "`Undeserializable`"
//! prefix).
//!
//! ```
//! # #[cfg(all(feature = "conditional_deserialization", feature = "std"))]
//! # {
//! use tls_codec::{Serialize, Deserialize};
//! use tls_codec_derive::{TlsSerialize, TlsSize, conditionally_deserializable};
//!
//! #[conditionally_deserializable]
//! #[derive(TlsSize, TlsSerialize, PartialEq, Debug)]
//! struct ExampleStruct {
//!     a: u8,
//!     b: u16,
//! }
//!
//! let undeserializable_struct = UndeserializableExampleStruct { a: 1, b: 2 };
//! let serialized = undeserializable_struct.tls_serialize_detached().unwrap();
//! let deserializable_struct =
//!     DeserializableExampleStruct::tls_deserialize(&mut serialized.as_slice()).unwrap();
//! # }
//! ```
//!
//! The helper macro `#[tls_codec(cd_field)]` can be used to mark a field as
//! conditionally deserializable, thus allowing nested conditionally
//! deserializable structs.
//!
//! ```
//! # #[cfg(all(feature = "conditional_deserialization", feature = "std"))]
//! # {
//! use tls_codec::{Serialize, Deserialize};
//! use tls_codec_derive::{TlsSerialize, TlsSize, conditionally_deserializable};
//!
//! #[conditionally_deserializable]
//! #[derive(TlsSize, TlsSerialize, PartialEq, Debug)]
//! struct ExampleStruct {
//!     a: u8,
//!     b: u16,
//! }
//!
//! #[conditionally_deserializable]
//! #[derive(TlsSize, TlsSerialize, PartialEq, Debug)]
//! struct NestedExampleStruct {
//!    #[tls_codec(cd_field)]
//!    example_struct: ExampleStruct,
//! }
//! # }
//! ```

extern crate proc_macro;
extern crate proc_macro2;

use proc_macro::TokenStream;
use proc_macro2::{Span, TokenStream as TokenStream2};
use quote::quote;
use syn::{
    self, Attribute, Data, DeriveInput, Expr, ExprLit, ExprPath, Field, Generics, Ident, Lit,
    Member, Meta, Result, Token, Type, parse_macro_input, punctuated::Punctuated, token::Comma,
};

#[cfg(feature = "conditional_deserialization")]
use syn::{ConstParam, ImplGenerics, ItemStruct, TypeGenerics, parse_quote};

/// Attribute name to identify attributes to be processed by derive-macros in this crate.
const ATTR_IDENT: &str = "tls_codec";

/// Prefix to add to `tls_codec` functions
///
/// This is either `<Type as Trait>` or a custom module containing the functions.
#[derive(Clone)]
enum Prefix {
    Type(Type),
    Custom(ExprPath),
}

impl Prefix {
    /// Returns the path prefix to use for functions from the given trait.
    fn for_trait(&self, trait_name: &str) -> TokenStream2 {
        let trait_name = Ident::new(trait_name, Span::call_site());
        match self {
            Prefix::Type(ty) => quote! { <#ty as tls_codec::#trait_name> },
            Prefix::Custom(p) => quote! { #p },
        }
    }
}

#[derive(Clone)]
struct Struct {
    call_site: Span,
    ident: Ident,
    generics: Generics,
    members: Vec<Member>,
    member_prefixes: Vec<Prefix>,
    member_skips: Vec<bool>,
}

#[derive(Clone)]
struct Enum {
    call_site: Span,
    ident: Ident,
    generics: Generics,
    repr: Ident,
    variants: Vec<Variant>,
    discriminant_constants: TokenStream2,
}

#[derive(Clone)]
struct Variant {
    ident: Ident,
    members: Vec<Member>,
    member_prefixes: Vec<Prefix>,
}

#[derive(Clone)]
enum DiscriminantValue {
    Literal(usize),
    Path(ExprPath),
}

#[derive(Clone)]
enum TlsStruct {
    Struct(Struct),
    Enum(Enum),
}

/// Attributes supported by derive-macros in this crate
#[derive(Clone)]
enum TlsAttr {
    /// Prefix for custom serialization functions
    With(ExprPath),
    /// Custom literal discriminant for an enum variant
    Discriminant(DiscriminantValue),
    /// Skip this attribute during (de)serialization.
    ///
    /// Note: The type of the attribute needs to implement [Default].
    ///       This is required to populate the field with a known
    ///       value during deserialization.
    Skip,
    #[cfg(feature = "conditional_deserialization")]
    CdField,
}

impl TlsAttr {
    fn name(&self) -> &'static str {
        match self {
            TlsAttr::With(_) => "with",
            TlsAttr::Discriminant(_) => "discriminant",
            TlsAttr::Skip => "skip",
            #[cfg(feature = "conditional_deserialization")]
            TlsAttr::CdField => "cd_field",
        }
    }

    /// Parses attributes of the form, `#[tls_codec(with = <string>)]`,
    /// `#[tls_codec(discriminant = <number>)]`, and `#[tls_codec(skip)]`.
    fn parse(attr: &Attribute) -> Result<Vec<TlsAttr>> {
        fn lit(e: &Expr) -> Result<&Lit> {
            if let Expr::Lit(ExprLit { lit, .. }) = e {
                Ok(lit)
            } else {
                Err(syn::Error::new_spanned(e, "expected literal"))
            }
        }

        if attr.path().get_ident().is_none_or(|id| id != ATTR_IDENT) {
            return Ok(Vec::new());
        }
        attr.parse_args_with(Punctuated::<Meta, Token![,]>::parse_terminated)?
            .iter()
            .map(|item| match item {
                Meta::NameValue(kv) => kv
                    .path
                    .get_ident()
                    .map(|ident| {
                        let ident_str = ident.to_string();
                        match &*ident_str {
                            "discriminant" => match lit(&kv.value)? {
                                Lit::Int(i) => i
                                    .base10_parse::<usize>()
                                    .map(DiscriminantValue::Literal)
                                    .map(TlsAttr::Discriminant),
                                Lit::Str(raw_path) => raw_path
                                    .parse::<ExprPath>()
                                    .map(DiscriminantValue::Path)
                                    .map(TlsAttr::Discriminant),
                                _ => Err(syn::Error::new_spanned(
                                    &kv.value,
                                    "Expected integer literal",
                                )),
                            },
                            "with" => match lit(&kv.value)? {
                                Lit::Str(s) => s.parse::<ExprPath>().map(TlsAttr::With),
                                _ => Err(syn::Error::new_spanned(
                                    &kv.value,
                                    "Expected string literal",
                                )),
                            },
                            _ => Err(syn::Error::new_spanned(
                                ident,
                                format!("Unexpected identifier {ident}"),
                            )),
                        }
                    })
                    .unwrap_or_else(|| {
                        Err(syn::Error::new_spanned(&kv.path, "Expected identifier"))
                    }),
                Meta::Path(path) => {
                    if let Some(ident) = path.get_ident() {
                        match ident.to_string().to_ascii_lowercase().as_ref() {
                            "skip" => Ok(TlsAttr::Skip),
                            #[cfg(feature = "conditional_deserialization")]
                            "cd_field" => Ok(TlsAttr::CdField),
                            _ => Err(syn::Error::new_spanned(
                                ident,
                                format!("Unexpected identifier {ident}"),
                            )),
                        }
                    } else {
                        Err(syn::Error::new_spanned(path, "Expected identifier"))
                    }
                }
                _ => Err(syn::Error::new_spanned(item, "Invalid attribute syntax")),
            })
            .collect()
    }

    /// Parses attributes of the form:
    /// ```text
    /// #[tls_codec(with = "module", ...)]
    /// ```
    fn parse_multi(attrs: &[Attribute]) -> Result<Vec<TlsAttr>> {
        attrs.iter().try_fold(Vec::new(), |mut acc, attr| {
            acc.extend(TlsAttr::parse(attr)?);
            Ok(acc)
        })
    }
}

/// Gets the [`Prefix`] for a field, i.e. the type itself or a path to prepend to the `tls_codec`
/// functions (e.g. a module or type).
fn function_prefix(field: &Field) -> Result<Prefix> {
    let prefix = TlsAttr::parse_multi(&field.attrs)?
        .into_iter()
        .try_fold(None, |path, attr| match (path, attr) {
            (None, TlsAttr::With(p)) => Ok(Some(p)),
            (Some(_), TlsAttr::With(p)) => Err(syn::Error::new_spanned(
                p,
                "Attribute `with` specified more than once",
            )),
            (path, _) => Ok(path),
        })?
        .map(Prefix::Custom)
        .unwrap_or_else(|| Prefix::Type(field.ty.clone()));
    Ok(prefix)
}

/// Process all attributes of a field and return a single, true or false, `skip` value.
/// This function will return an error in the case of multiple `skip` attributes.
fn function_skip(field: &Field) -> Result<bool> {
    let skip = TlsAttr::parse_multi(&field.attrs)?
        .into_iter()
        .try_fold(None, |skip, attr| match (skip, attr) {
            (None, TlsAttr::Skip) => Ok(Some(true)),
            (Some(_), TlsAttr::Skip) => Err(syn::Error::new(
                Span::call_site(),
                "Attribute `skip` specified more than once",
            )),
            (skip, _) => Ok(skip),
        })?
        .unwrap_or(false);

    Ok(skip)
}

/// Process all attributes of a field and return a single, true or false, `cd_field` value.
/// This function will return an error in the case of multiple `cd` attributes.
#[cfg(feature = "conditional_deserialization")]
fn function_cd(field: &Field) -> Result<bool> {
    let skip = TlsAttr::parse_multi(&field.attrs)?
        .into_iter()
        .try_fold(None, |skip, attr| match (skip, attr) {
            (None, TlsAttr::CdField) => Ok(Some(true)),
            (Some(_), TlsAttr::CdField) => Err(syn::Error::new(
                Span::call_site(),
                "Attribute `cd_field` specified more than once",
            )),
            (skip, _) => Ok(skip),
        })?
        .unwrap_or(false);

    Ok(skip)
}

/// Gets the serialization discriminant if specified.
fn discriminant_value(attrs: &[Attribute]) -> Result<Option<DiscriminantValue>> {
    TlsAttr::parse_multi(attrs)?
        .into_iter()
        .try_fold(None, |discriminant, attr| match (discriminant, attr) {
            (None, TlsAttr::Discriminant(d)) => Ok(Some(d)),
            (Some(_), TlsAttr::Discriminant(_)) => Err(syn::Error::new(
                Span::call_site(),
                "Attribute `discriminant` specified more than once",
            )),
            (_, attr) => Err(syn::Error::new(
                Span::call_site(),
                format!("Unrecognized variant attribute `{}`", attr.name()),
            )),
        })
}

fn fields_to_members(fields: &syn::Fields) -> Vec<Member> {
    fields
        .iter()
        .enumerate()
        .map(|(i, field)| {
            field
                .ident
                .clone()
                .map_or_else(|| Member::Unnamed(syn::Index::from(i)), Member::Named)
        })
        .collect()
}

/// Gets the [`Prefix`]es for all fields, i.e. the types themselves or paths to prepend to the
/// `tls_codec` functions (e.g. a module or type).
fn fields_to_member_prefixes(fields: &syn::Fields) -> Result<Vec<Prefix>> {
    fields.iter().map(function_prefix).collect()
}

fn fields_to_member_skips(fields: &syn::Fields) -> Result<Vec<bool>> {
    fields.iter().map(function_skip).collect()
}

fn parse_ast(ast: DeriveInput) -> Result<TlsStruct> {
    let call_site = Span::call_site();
    let ident = ast.ident.clone();
    let generics = ast.generics.clone();
    match ast.data {
        Data::Struct(st) => {
            let members = fields_to_members(&st.fields);
            let member_prefixes = fields_to_member_prefixes(&st.fields)?;
            let member_skips = fields_to_member_skips(&st.fields)?;
            Ok(TlsStruct::Struct(Struct {
                call_site,
                ident,
                generics,
                members,
                member_prefixes,
                member_skips,
            }))
        }
        // Enums.
        // Note that they require a repr attribute.
        Data::Enum(syn::DataEnum { variants, .. }) => {
            let mut repr = None;
            for attr in ast.attrs {
                if attr.path().is_ident("repr") {
                    let ty = attr.parse_args()?;
                    repr = Some(ty);
                    break;
                }
            }
            let repr =
                repr.ok_or_else(|| syn::Error::new(call_site, "missing #[repr(...)] attribute"))?;
            let discriminant_constants = define_discriminant_constants(&ident, &repr, &variants)?;
            let variants = variants
                .into_iter()
                .map(|variant| {
                    Ok(Variant {
                        ident: variant.ident,
                        members: fields_to_members(&variant.fields),
                        member_prefixes: fields_to_member_prefixes(&variant.fields)?,
                    })
                })
                .collect::<Result<Vec<_>>>()?;

            Ok(TlsStruct::Enum(Enum {
                call_site,
                ident,
                generics,
                repr,
                variants,
                discriminant_constants,
            }))
        }
        Data::Union(_) => unimplemented!(),
    }
}

enum SerializeVariant {
    Write,
    Bytes,
}

#[proc_macro_derive(TlsSize, attributes(tls_codec))]
pub fn size_macro_derive(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let parsed_ast = match parse_ast(ast) {
        Ok(ast) => ast,
        Err(err_ts) => return err_ts.into_compile_error().into(),
    };
    impl_tls_size(parsed_ast).into()
}

#[proc_macro_derive(TlsSerialize, attributes(tls_codec))]
pub fn serialize_macro_derive(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let parsed_ast = match parse_ast(ast) {
        Ok(ast) => ast,
        Err(err_ts) => return err_ts.into_compile_error().into(),
    };
    impl_serialize(parsed_ast, SerializeVariant::Write).into()
}

#[proc_macro_derive(TlsDeserialize, attributes(tls_codec))]
pub fn deserialize_macro_derive(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let parsed_ast = match parse_ast(ast) {
        Ok(ast) => ast,
        Err(err_ts) => return err_ts.into_compile_error().into(),
    };
    impl_deserialize(parsed_ast).into()
}

#[proc_macro_derive(TlsDeserializeBytes, attributes(tls_codec))]
pub fn deserialize_bytes_macro_derive(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let parsed_ast = match parse_ast(ast) {
        Ok(ast) => ast,
        Err(err_ts) => return err_ts.into_compile_error().into(),
    };
    impl_deserialize_bytes(parsed_ast).into()
}

#[proc_macro_derive(TlsSerializeBytes, attributes(tls_codec))]
pub fn serialize_bytes_macro_derive(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let parsed_ast = match parse_ast(ast) {
        Ok(ast) => ast,
        Err(err_ts) => return err_ts.into_compile_error().into(),
    };
    impl_serialize(parsed_ast, SerializeVariant::Bytes).into()
}

/// Returns identifiers to use as bindings in generated code
fn make_n_ids(n: usize) -> Vec<Ident> {
    (0..n)
        .map(|i| Ident::new(&format!("__arg{i}"), Span::call_site()))
        .collect()
}

/// Returns identifier to define a constant equal to the discriminant of a variant
fn discriminant_id(variant: &Ident) -> Ident {
    Ident::new(&format!("__TLS_CODEC_{variant}"), Span::call_site())
}

/// Returns definitions of constants equal to the discriminants of each variant
fn define_discriminant_constants(
    enum_ident: &Ident,
    repr: &Ident,
    variants: &Punctuated<syn::Variant, Comma>,
) -> Result<TokenStream2> {
    let all_variants_are_unit = variants
        .iter()
        .all(|variant| matches!(variant.fields, syn::Fields::Unit));
    let discriminant_constants = if all_variants_are_unit {
        variants
            .iter()
            .map(|variant| {
                let variant_id = &variant.ident;
                let constant_id = discriminant_id(variant_id);
                if discriminant_value(&variant.attrs)?.is_some() {
                    Err(syn::Error::new(
                        Span::call_site(),
                        "The tls_codec discriminant attribute must only be used in enumerations \
                        with at least one non-unit variant. When all variants are units, \
                        discriminants can be assigned to variants directly.",
                    ))
                } else {
                    Ok(quote! {
                        #[allow(non_upper_case_globals)]
                        const #constant_id: #repr = #enum_ident::#variant_id as #repr;
                    })
                }
            })
            .collect::<Result<Vec<_>>>()?
    } else {
        let mut spans = Vec::with_capacity(variants.len());
        let mut implicit_discriminant = 0usize;
        let mut discriminant_has_paths = false;
        for variant in variants.iter() {
            let constant_id = discriminant_id(&variant.ident);

            let tokens = if let Some(value) = discriminant_value(&variant.attrs)? {
                match value {
                    DiscriminantValue::Literal(value) => {
                        implicit_discriminant = value;
                        quote! {
                            #[allow(non_upper_case_globals)]
                            const #constant_id: #repr = {
                                if #value < #repr::MIN as usize || #value > #repr::MAX as usize {
                                    panic!("The value corresponding to that expression is outside the bounds of the enum representation");
                                }
                                #value as #repr
                            };
                        }
                    }
                    DiscriminantValue::Path(pathexpr) => {
                        discriminant_has_paths = true;
                        quote! {
                            #[allow(clippy::unnecessary_cast, non_upper_case_globals)]
                            const #constant_id: #repr = {
                                let pathexpr_usize = #pathexpr as usize;
                                if pathexpr_usize < #repr::MIN as usize || pathexpr_usize > #repr::MAX as usize {
                                    panic!("The value corresponding to that expression is outside the bounds of the enum representation");
                                }
                                #pathexpr as #repr
                            };
                        }
                    }
                }
            } else if discriminant_has_paths {
                return Err(syn::Error::new(
                    Span::call_site(),
                    "The tls_codec discriminant attribute is missing. \
                    Once you start using paths in #[tls_codec(discriminant = \"path::to::const::or::enum::variant\"], \
                    You **have** to provide the discriminant attribute on every single variant.",
                ));
            } else {
                quote! {
                    #[allow(non_upper_case_globals)]
                    const #constant_id: #repr = {
                        if #implicit_discriminant > #repr::MAX as usize {
                            panic!("The value corresponding to that expression is outside the bounds of the enum representation");
                        }
                        #implicit_discriminant as #repr
                    };
                }
            };

            implicit_discriminant += 1;

            spans.push(tokens);
        }
        spans
    };
    Ok(quote! { #(#discriminant_constants)* })
}

#[allow(unused_variables)]
fn impl_tls_size(parsed_ast: TlsStruct) -> TokenStream2 {
    match parsed_ast {
        TlsStruct::Struct(Struct {
            call_site,
            ident,
            generics,
            members,
            member_prefixes,
            member_skips,
        }) => {
            let ((members, member_prefixes), _) =
                partition_skipped(members, member_prefixes, member_skips);

            let prefixes = member_prefixes
                .iter()
                .map(|p| p.for_trait("Size"))
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
            quote! {
                impl #impl_generics tls_codec::Size for #ident #ty_generics #where_clause {
                    #[inline]
                    fn tls_serialized_len(&self) -> usize {
                        #(#prefixes::tls_serialized_len(&self.#members) + )*
                        0
                    }
                    #[inline]
                    #[allow(clippy::needless_question_mark)]
                    fn tls_serialized_len_checked(&self) -> Option<usize> {
                        Some(0usize#(.checked_add(#prefixes::tls_serialized_len_checked(&self.#members)?)?)*)
                    }
                }

                impl #impl_generics tls_codec::Size for &#ident #ty_generics #where_clause {
                    #[inline]
                    fn tls_serialized_len(&self) -> usize {
                        tls_codec::Size::tls_serialized_len(*self)
                    }
                    #[inline]
                    fn tls_serialized_len_checked(&self) -> Option<usize> {
                        tls_codec::Size::tls_serialized_len_checked(*self)
                    }
                }
            }
        }
        TlsStruct::Enum(Enum {
            call_site,
            ident,
            generics,
            repr,
            variants,
            ..
        }) => {
            let field_arms = variants
                .iter()
                .map(|variant| {
                    let variant_id = &variant.ident;
                    let members = &variant.members;
                    let bindings = make_n_ids(members.len());
                    let prefixes = variant.member_prefixes.iter().map(|p| p.for_trait("Size")).collect::<Vec<_>>();
                    quote! {
                        #ident::#variant_id { #(#members: #bindings,)* } => 0 #(+ #prefixes::tls_serialized_len(#bindings))*,
                    }
                })
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
            quote! {
                impl #impl_generics tls_codec::Size for #ident #ty_generics #where_clause {
                    #[inline]
                    fn tls_serialized_len(&self) -> usize {
                        let field_len = match self {
                            #(#field_arms)*
                        };
                        core::mem::size_of::<#repr>() + field_len
                    }
                    #[inline]
                    fn tls_serialized_len_checked(&self) -> Option<usize> {
                        let field_len = match self {
                            #(#field_arms)*
                        };
                        core::mem::size_of::<#repr>().checked_add(field_len)
                    }
                }

                impl #impl_generics tls_codec::Size for &#ident #ty_generics #where_clause {
                    #[inline]
                    fn tls_serialized_len(&self) -> usize {
                        tls_codec::Size::tls_serialized_len(*self)
                    }
                    #[inline]
                    fn tls_serialized_len_checked(&self) -> Option<usize> {
                        tls_codec::Size::tls_serialized_len_checked(*self)
                    }
                }
            }
        }
    }
}

#[allow(unused_variables)]
fn impl_serialize(parsed_ast: TlsStruct, svariant: SerializeVariant) -> TokenStream2 {
    match parsed_ast {
        TlsStruct::Struct(Struct {
            call_site,
            ident,
            generics,
            members,
            member_prefixes,
            member_skips,
        }) => {
            let ((members, member_prefixes), _) =
                partition_skipped(members, member_prefixes, member_skips);

            let prefixes = member_prefixes
                .iter()
                .map(|p| match svariant {
                    SerializeVariant::Write => p.for_trait("Serialize"),
                    SerializeVariant::Bytes => p.for_trait("SerializeBytes"),
                })
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();

            match svariant {
                SerializeVariant::Write => {
                    if cfg!(feature = "std") {
                        quote! {
                            impl #impl_generics tls_codec::Serialize for #ident #ty_generics #where_clause {
                                fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                    let mut written = 0usize;
                                    #(
                                        written += #prefixes::tls_serialize(&self.#members, writer)?;
                                    )*
                                    if cfg!(debug_assertions) {
                                        let expected_written = tls_codec::Size::tls_serialized_len(&self);
                                        debug_assert_eq!(written, expected_written, "Expected to serialize {} bytes but only {} were generated.", expected_written, written);
                                        if written != expected_written {
                                            Err(tls_codec::Error::EncodingError(format!("Expected to serialize {} bytes but only {} were generated.", expected_written, written)))
                                        } else {
                                            Ok(written)
                                        }
                                    } else {
                                        Ok(written)
                                    }
                                }
                            }

                            impl #impl_generics tls_codec::Serialize for &#ident #ty_generics #where_clause {
                                fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                    tls_codec::Serialize::tls_serialize(*self, writer)
                                }
                            }
                        }
                    } else {
                        quote! {
                            impl #impl_generics tls_codec::Serialize for #ident #ty_generics #where_clause {}
                            impl #impl_generics tls_codec::Serialize for &#ident #ty_generics #where_clause {}
                        }
                    }
                }
                SerializeVariant::Bytes => {
                    quote! {
                        impl #impl_generics tls_codec::SerializeBytes for #ident #ty_generics #where_clause {
                            fn tls_serialize(&self) -> core::result::Result<Vec<u8>, tls_codec::Error> {
                                let expected_out = tls_codec::Size::tls_serialized_len(&self);
                                let mut out = Vec::with_capacity(expected_out);

                                #(
                                    out.append(&mut #prefixes::tls_serialize(&self.#members)?);
                                )*
                                if cfg!(debug_assertions) {
                                    debug_assert_eq!(out.len(), expected_out, "Expected to serialize {} bytes but only {} were generated.", expected_out, out.len());
                                    if out.len() != expected_out {
                                        Err(tls_codec::Error::EncodingError(format!("Expected to serialize {} bytes but only {} were generated.", expected_out, out.len())))
                                    } else {
                                        Ok(out)
                                    }
                                } else {
                                    Ok(out)
                                }
                            }
                        }

                        impl #impl_generics tls_codec::SerializeBytes for &#ident #ty_generics #where_clause {
                            fn tls_serialize(&self) -> core::result::Result<Vec<u8>, tls_codec::Error> {
                                tls_codec::SerializeBytes::tls_serialize(*self)
                            }
                        }
                    }
                }
            }
        }
        TlsStruct::Enum(Enum {
            call_site,
            ident,
            generics,
            repr,
            variants,
            discriminant_constants,
        }) => {
            let arms: Vec<TokenStream2> = variants
                .iter()
                .map(|variant| {
                    let variant_id = &variant.ident;
                    let discriminant = discriminant_id(variant_id);
                    let members = &variant.members;
                    let bindings = make_n_ids(members.len());
                    match svariant {
                        SerializeVariant::Write => {
                            let prefixes = variant
                                .member_prefixes
                                .iter()
                                .map(|p| p.for_trait("Serialize"))
                                .collect::<Vec<_>>();
                            quote! {
                                #ident::#variant_id { #(#members: #bindings,)* } => Ok(
                                    tls_codec::Serialize::tls_serialize(&#discriminant, writer)?
                                    #(+ #prefixes::tls_serialize(#bindings, writer)?)*
                                ),
                            }
                        }
                        SerializeVariant::Bytes => {
                            let prefixes = variant
                                .member_prefixes
                                .iter()
                                .map(|p| p.for_trait("SerializeBytes"))
                                .collect::<Vec<_>>();
                            quote! {
                                #ident::#variant_id { #(#members: #bindings,)* } => {
                                    let mut discriminant_out = tls_codec::SerializeBytes::tls_serialize(&#discriminant)?;
                                    #(discriminant_out.append(&mut #prefixes::tls_serialize(#bindings)?);)*
                                    Ok(discriminant_out)
                                },
                            }
                        }
                    }
                })
                .collect();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();

            match svariant {
                SerializeVariant::Write => {
                    if cfg!(feature = "std") {
                        quote! {
                            impl #impl_generics tls_codec::Serialize for #ident #ty_generics #where_clause {
                                fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                    #discriminant_constants
                                    match self {
                                        #(#arms)*
                                    }
                                }
                            }

                            impl #impl_generics tls_codec::Serialize for &#ident #ty_generics #where_clause {
                                fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                    tls_codec::Serialize::tls_serialize(*self, writer)
                                }
                            }
                        }
                    } else {
                        quote! {
                            impl #impl_generics tls_codec::Serialize for #ident #ty_generics #where_clause {}
                            impl #impl_generics tls_codec::Serialize for &#ident #ty_generics #where_clause {}
                        }
                    }
                }
                SerializeVariant::Bytes => {
                    quote! {
                        impl #impl_generics tls_codec::SerializeBytes for #ident #ty_generics #where_clause {
                            fn tls_serialize(&self) -> core::result::Result<Vec<u8>, tls_codec::Error> {
                                #discriminant_constants
                                match self {
                                    #(#arms)*
                                }
                            }
                        }

                        impl #impl_generics tls_codec::SerializeBytes for &#ident #ty_generics #where_clause {
                            fn tls_serialize(&self) -> core::result::Result<Vec<u8>, tls_codec::Error> {
                                tls_codec::SerializeBytes::tls_serialize(*self)
                            }
                        }
                    }
                }
            }
        }
    }
}

#[cfg(feature = "conditional_deserialization")]
fn restrict_conditional_generic(
    impl_generics: ImplGenerics,
    ty_generics: TypeGenerics,
    deserializable: bool,
) -> (TokenStream2, TokenStream2) {
    let impl_generics = quote! { #impl_generics }
        .to_string()
        // Make string replacement easier by replacing newlines with spaces.
        .replace('\n', " ")
        .replace(" const IS_DESERIALIZABLE : bool ", "")
        .replace("<>", "")
        .parse()
        .unwrap();
    let deserializable_string = if deserializable { "true" } else { "false" };
    let ty_generics = quote! { #ty_generics }
        .to_string()
        .replace("IS_DESERIALIZABLE", deserializable_string)
        .parse()
        .unwrap();
    (impl_generics, ty_generics)
}

#[allow(unused_variables)]
fn impl_deserialize(parsed_ast: TlsStruct) -> TokenStream2 {
    match parsed_ast {
        TlsStruct::Struct(Struct {
            call_site,
            ident,
            generics,
            members,
            member_prefixes,
            member_skips,
        }) => {
            let ((members, member_prefixes), (members_default, member_prefixes_default)) =
                partition_skipped(members, member_prefixes, member_skips);

            let prefixes = member_prefixes
                .iter()
                .map(|p| p.for_trait("Deserialize"))
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
            #[cfg(feature = "conditional_deserialization")]
            let (impl_generics, ty_generics) =
                restrict_conditional_generic(impl_generics, ty_generics, true);
            if cfg!(feature = "std") {
                quote! {
                    impl #impl_generics tls_codec::Deserialize for #ident #ty_generics #where_clause {
                        fn tls_deserialize<R: std::io::Read>(bytes: &mut R) -> core::result::Result<Self, tls_codec::Error> {
                            Ok(Self {
                                #(#members: #prefixes::tls_deserialize(bytes)?,)*
                                #(#members_default: Default::default(),)*
                            })
                        }
                    }
                }
            } else {
                quote! {
                    impl #impl_generics tls_codec::Deserialize for #ident #ty_generics #where_clause {}
                }
            }
        }
        TlsStruct::Enum(Enum {
            call_site,
            ident,
            generics,
            repr,
            variants,
            discriminant_constants,
        }) => {
            let arms = variants
                .iter()
                .map(|variant| {
                    let variant_id = &variant.ident;
                    let discriminant = discriminant_id(variant_id);
                    let members = &variant.members;
                    let prefixes = variant
                        .member_prefixes
                        .iter()
                        .map(|p| p.for_trait("Deserialize"))
                        .collect::<Vec<_>>();
                    quote! {
                        #discriminant => Ok(#ident::#variant_id {
                            #(#members: #prefixes::tls_deserialize(bytes)?,)*
                        }),
                    }
                })
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
            if cfg!(feature = "std") {
                quote! {
                    impl #impl_generics tls_codec::Deserialize for #ident #ty_generics #where_clause {
                        fn tls_deserialize<R: std::io::Read>(bytes: &mut R) -> core::result::Result<Self, tls_codec::Error> {
                            #discriminant_constants
                            let discriminant = <#repr as tls_codec::Deserialize>::tls_deserialize(bytes)?;
                            match discriminant {
                                #(#arms)*
                                _ => {
                                    Err(tls_codec::Error::UnknownValue(discriminant.into()))
                                },
                            }
                        }
                    }
                }
            } else {
                quote! {
                    impl #impl_generics tls_codec::Deserialize for #ident #ty_generics #where_clause {}
                }
            }
        }
    }
}

#[allow(unused_variables)]
fn impl_deserialize_bytes(parsed_ast: TlsStruct) -> TokenStream2 {
    match parsed_ast {
        TlsStruct::Struct(Struct {
            call_site,
            ident,
            generics,
            members,
            member_prefixes,
            member_skips,
        }) => {
            let ((members, member_prefixes), (members_default, member_prefixes_default)) =
                partition_skipped(members, member_prefixes, member_skips);
            let members_values = members
                .iter()
                .map(|m| match m {
                    Member::Named(named) => {
                        Member::Named(Ident::new(&format!("value_{named}"), Span::call_site()))
                    }
                    Member::Unnamed(unnamed) => Member::Named(Ident::new(
                        &format!("value_{}", unnamed.index),
                        Span::call_site(),
                    )),
                })
                .collect::<Vec<_>>();

            let prefixes = member_prefixes
                .iter()
                .map(|p| p.for_trait("DeserializeBytes"))
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
            #[cfg(feature = "conditional_deserialization")]
            let (impl_generics, ty_generics) =
                restrict_conditional_generic(impl_generics, ty_generics, true);
            quote! {
                impl #impl_generics tls_codec::DeserializeBytes for #ident #ty_generics #where_clause {
                    fn tls_deserialize_bytes(bytes: &[u8]) -> core::result::Result<(Self, &[u8]), tls_codec::Error> {
                        #(let (#members_values, bytes) = #prefixes::tls_deserialize_bytes(bytes)?;)*
                        Ok((Self {
                            #(#members: #members_values,)*
                            #(#members_default: Default::default(),)*
                        }, bytes))
                    }
                }
            }
        }
        TlsStruct::Enum(Enum {
            call_site,
            ident,
            generics,
            repr,
            variants,
            discriminant_constants,
        }) => {
            let arms = variants
                .iter()
                .map(|variant| {
                    let variant_id = &variant.ident;
                    let discriminant = discriminant_id(variant_id);
                    let members = &variant.members;
                    let member_values = members
                        .iter()
                        .map(|m| match m {
                            Member::Named(named) => Member::Named(Ident::new(
                                &format!("value_{named}"),
                                Span::call_site(),
                            )),
                            Member::Unnamed(unnamed) => Member::Named(Ident::new(
                                &format!("value_{}", unnamed.index),
                                Span::call_site(),
                            )),
                        })
                        .collect::<Vec<_>>();
                    let prefixes = variant
                        .member_prefixes
                        .iter()
                        .map(|p| p.for_trait("DeserializeBytes"))
                        .collect::<Vec<_>>();
                    quote! {
                        #discriminant => {
                            #(let (#member_values, remainder) = #prefixes::tls_deserialize_bytes(remainder)?;)*
                            let result = #ident::#variant_id { #(#members: #member_values,)* };
                            Ok((result, remainder))
                        },
                    }
                })
                .collect::<Vec<_>>();
            let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
            quote! {
                impl #impl_generics tls_codec::DeserializeBytes for #ident #ty_generics #where_clause {
                    fn tls_deserialize_bytes(bytes: &[u8]) -> core::result::Result<(Self, &[u8]), tls_codec::Error> {
                        #discriminant_constants
                        let (discriminant, remainder) = #repr::tls_deserialize_bytes(bytes)?;
                        match discriminant {
                            #(#arms)*
                            _ => {
                                Err(tls_codec::Error::UnknownValue(discriminant.into()))
                            },
                        }
                    }
                }
            }
        }
    }
}

#[allow(clippy::type_complexity)]
fn partition_skipped(
    members: Vec<Member>,
    member_prefixes: Vec<Prefix>,
    member_skips: Vec<bool>,
) -> ((Vec<Member>, Vec<Prefix>), (Vec<Member>, Vec<Prefix>)) {
    let mut members_not_skip: Vec<Member> = Vec::new();
    let mut member_prefixes_not_skip: Vec<Prefix> = Vec::new();

    let mut members_skip: Vec<Member> = Vec::new();
    let mut member_prefixes_skip: Vec<Prefix> = Vec::new();

    for ((member, prefix), skip) in members.into_iter().zip(member_prefixes).zip(member_skips) {
        if !skip {
            members_not_skip.push(member);
            member_prefixes_not_skip.push(prefix);
        } else {
            members_skip.push(member);
            member_prefixes_skip.push(prefix);
        }
    }

    (
        (members_not_skip, member_prefixes_not_skip),
        (members_skip, member_prefixes_skip),
    )
}

/// The `conditionally_deserializable` attribute macro creates two versions of
/// the affected struct: One that implements the
/// [`Deserialize`](../tls_codec::Deserialize) and
/// [`DeserializeBytes`](../tls_codec::DeserializeBytes) traits and one that
/// does not. It does so by introducing a boolean const generic that indicates
/// if the struct can be deserialized or not.
///
/// This conditional deserialization can be used, for example, to implement a
/// simple state machine, where after deserialization, the user must first
/// complete an additional transition (e.g. verification) to the
/// undeserializable version of the struct, which might then implement functions
/// for further processing.
///
/// For ease of use, the macro creates type aliases for the deserializable and
/// undeserializable variant of the struct, where the alias is the name of the
/// struct prefixed with `Deserializable` or `Undeserializable` respectively.
///
/// Due to the way that the macro rewrites the struct, it must be placed before
/// any `#[derive(...)]` statements.
///
/// The `conditionally_deserializable` attribute macro is only available if the
/// `conditional_deserialization` feature is enabled.
#[cfg_attr(
    feature = "conditional_deserialization",
    doc = r##"
```compile_fail
use tls_codec_derive::{TlsSerialize, TlsDeserialize, TlsSize, conditionally_deserializable};

#[conditionally_deserializable(Bytes)]
#[derive(TlsDeserialize, TlsSerialize, TlsSize)]
struct ExampleStruct {
    pub a: u16,
}

impl UndeserializableExampleStruct {
    #[cfg(feature = "conditional_deserialization")]
    fn deserialize(bytes: &[u8]) -> Result<Self, tls_codec::Error> {
        Self::tls_deserialize_exact(bytes)
    }
}
```
"##
)]
#[cfg(feature = "conditional_deserialization")]
#[proc_macro_attribute]
pub fn conditionally_deserializable(
    _input: TokenStream,
    annotated_item: TokenStream,
) -> TokenStream {
    let annotated_item = parse_macro_input!(annotated_item as ItemStruct);
    impl_conditionally_deserializable(annotated_item).into()
}

#[cfg(feature = "conditional_deserialization")]
fn set_cd_fields_generic(
    mut item_struct: ItemStruct,
    value: proc_macro2::TokenStream,
) -> ItemStruct {
    use syn::{
        AngleBracketedGenericArguments, PathArguments,
        parse::{Parse, Parser},
    };

    item_struct.fields.iter_mut().for_each(|field| {
        if function_cd(field).unwrap() {
            // We only do this if it's a simple (Path) type
            if let Type::Path(path) = &mut field.ty {
                // If there is already an AngleBracketedGenericArguments, we just add the const generic at the end.
                if let Some(segment) = path.path.segments.last_mut() {
                    if let PathArguments::AngleBracketed(argument) = &mut segment.arguments {
                        argument.args.push(parse_quote! {#value});
                    } else {
                        // If there is no AngleBracketedGenericArguments, we create one and add the const generic.
                        let parser = AngleBracketedGenericArguments::parse;
                        let angle_bracketed = parser.parse(TokenStream::from(quote!(<#value>)));
                        let angle_bracketed = angle_bracketed.unwrap();
                        segment.arguments = PathArguments::AngleBracketed(angle_bracketed);
                    }
                }
            } else {
                panic!("Only simple types are supported for conditional deserialization.");
            }
        }
    });
    item_struct
}

#[cfg(feature = "conditional_deserialization")]
fn impl_conditionally_deserializable(mut annotated_item: ItemStruct) -> TokenStream2 {
    let deserializable_const_generic: ConstParam = parse_quote! {const IS_DESERIALIZABLE: bool};
    // Get all the original generics of the annotated item before we modify
    // them.
    let original_annotated_item = annotated_item.clone();
    let (_, original_ty_generics, _) = original_annotated_item.generics.split_for_impl();
    // Add the DESERIALIZABLE const generic to the struct
    annotated_item
        .generics
        .params
        .push(deserializable_const_generic.into());
    // Look through struct fields and if a field has an `cd_field` attribute,
    // add the `IS_DESERIALIZABLE` const generic to the field type (set to true
    // for the deserialize implementation).
    let item_for_deserialize_impl = set_cd_fields_generic(annotated_item.clone(), quote!(true));
    // Look through struct fields and if a field has an `cd_field` attribute,
    // add the `IS_DESERIALIZABLE` const generic to the field type (set to the
    // value IS_DESERIALIZABLE from the struct definition).
    let annotated_item = set_cd_fields_generic(annotated_item, quote!(IS_DESERIALIZABLE));

    // Derive both TlsDeserialize and TlsDeserializeBytes
    let deserialize_bytes_implementation =
        impl_deserialize_bytes(parse_ast(item_for_deserialize_impl.clone().into()).unwrap());
    let deserialize_implementation =
        impl_deserialize(parse_ast(item_for_deserialize_impl.clone().into()).unwrap());
    let (impl_generics, ty_generics, _) = annotated_item.generics.split_for_impl();
    // Patch generics for use by the type aliases
    let (_deserializable_impl_generics, deserializable_ty_generics) =
        restrict_conditional_generic(impl_generics.clone(), ty_generics.clone(), true);
    let (_undeserializable_impl_generics, undeserializable_ty_generics) =
        restrict_conditional_generic(impl_generics.clone(), ty_generics.clone(), false);
    let annotated_item_ident = annotated_item.ident.clone();
    // Create Alias Idents by adding prefixes
    let deserializable_ident = Ident::new(
        &format!("Deserializable{annotated_item_ident}"),
        Span::call_site(),
    );
    let undeserializable_ident = Ident::new(
        &format!("Undeserializable{annotated_item_ident}"),
        Span::call_site(),
    );
    let annotated_item_visibility = annotated_item.vis.clone();
    let doc_string_deserializable =
        format!("Alias for the deserializable version of the [`{annotated_item_ident}`].");
    let doc_string_undeserializable = format!(
        "Alias for the version of the [`{annotated_item_ident}`] that cannot be deserialized."
    );
    quote! {
        #annotated_item

        #[doc = #doc_string_deserializable]
        #annotated_item_visibility type #undeserializable_ident #original_ty_generics = #annotated_item_ident #undeserializable_ty_generics;
        #[doc = #doc_string_undeserializable]
        #annotated_item_visibility type #deserializable_ident #original_ty_generics = #annotated_item_ident #deserializable_ty_generics;

        #deserialize_implementation

        #deserialize_bytes_implementation
    }
}