sark-gen 0.9.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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

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

use crate::util::TypeExt;

pub(super) struct RawBody;

impl RawBody {
    pub(super) fn borrowed_type(ty: &Type) -> Result<Type> {
        if ty.is_bytes_with_storage("Retained") {
            return Ok(syn::parse_quote!(
                ::o3::buffer::Bytes<::o3::buffer::Borrowed<'req>>
            ));
        }
        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() {
            "Shared" => syn::parse_quote!(::o3::buffer::Bytes<::o3::buffer::Borrowed<'req>>),
            _ => return Err(Self::ty_error(ty)),
        })
    }

    pub(super) fn borrowed_field_expr(ty: &Type) -> Result<TokenStream> {
        if ty.is_bytes_with_storage("Retained") {
            return Ok(quote!(req.body_frame()));
        }
        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() {
            "Shared" => quote!(req.body_frame()),
            _ => return Err(Self::ty_error(ty)),
        })
    }

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