sark-gen 0.8.0

Sark proc-macro generators
Documentation 
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

use proc_macro2::TokenStream;
use quote::quote;
use syn::{Error, Result, Type};

pub(super) struct RawBody;

impl RawBody {
    pub(super) fn field_expr(ty: &Type) -> Result<TokenStream> {
        let Type::Path(path) = ty else {
            return Err(Self::ty_error(ty));
        };
        let Some(seg) = path.path.segments.last() else {
            return Err(Self::ty_error(ty));
        };
        Ok(match seg.ident.to_string().as_str() {
            "Body" => quote!(raw_body),
            "LocalFrameBytes" => quote!(raw_body.into_local()),
            "Shared" => quote!(raw_body.into_bytes()),
            _ => return Err(Self::ty_error(ty)),
        })
    }

    fn ty_error(ty: &Type) -> Error {
        Error::new_spanned(
            ty,
            "#[raw_body] field type must be request::Body, LocalFrameBytes, or Shared",
        )
    }
}