drawbridge_type/
meta.rs

1// SPDX-License-Identifier: Apache-2.0
2
3use crate::digest::ContentDigest;
4
5use mime::Mime;
6use serde::{de::Error as _, Deserialize, Deserializer, Serialize, Serializer};
7
8#[cfg(feature = "axum")]
9use axum::{
10    extract::rejection::{TypedHeaderRejection, TypedHeaderRejectionReason},
11    extract::{FromRequest, RequestParts, TypedHeader},
12    headers::{ContentLength, ContentType},
13    response::{IntoResponseParts, Response, ResponseParts},
14};
15
16#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
17pub struct Meta {
18    #[serde(rename = "digest")]
19    pub hash: ContentDigest<Box<[u8]>>,
20
21    #[serde(rename = "length")]
22    pub size: u64,
23
24    #[serde(deserialize_with = "deserialize")]
25    #[serde(serialize_with = "serialize")]
26    #[serde(rename = "type")]
27    pub mime: Mime,
28}
29
30#[allow(single_use_lifetimes)]
31fn deserialize<'de, D: Deserializer<'de>>(deserializer: D) -> Result<Mime, D::Error> {
32    String::deserialize(deserializer)?
33        .parse()
34        .map_err(|_| D::Error::custom("invalid mime type"))
35}
36
37fn serialize<S: Serializer>(mime: &Mime, serializer: S) -> Result<S::Ok, S::Error> {
38    mime.to_string().serialize(serializer)
39}
40
41#[cfg(feature = "axum")]
42#[axum::async_trait]
43impl<B: Send> FromRequest<B> for Meta {
44    type Rejection = TypedHeaderRejection;
45
46    async fn from_request(req: &mut RequestParts<B>) -> Result<Self, Self::Rejection> {
47        let hash = match req.extract().await {
48            Ok(TypedHeader(hash)) => hash,
49            Err(e) if matches!(e.reason(), TypedHeaderRejectionReason::Missing) => {
50                Default::default()
51            }
52            Err(e) => return Err(e),
53        };
54        let size = req.extract::<TypedHeader<ContentLength>>().await?.0 .0;
55        let mime = req.extract::<TypedHeader<ContentType>>().await?.0.into();
56        Ok(Meta { hash, size, mime })
57    }
58}
59
60#[cfg(feature = "axum")]
61impl IntoResponseParts for Meta {
62    type Error = Response;
63
64    fn into_response_parts(self, res: ResponseParts) -> Result<ResponseParts, Self::Error> {
65        let hash = TypedHeader(self.hash);
66        let size = TypedHeader(ContentLength(self.size));
67        let mime = TypedHeader(ContentType::from(self.mime));
68        (hash, size, mime).into_response_parts(res)
69    }
70}
71
72#[cfg(test)]
73mod tests {
74    use super::*;
75
76    use serde_json::json;
77
78    #[test]
79    fn serialization() {
80        let meta = Meta {
81            hash: "sha-384=:mqVuAfXRKap7bdgcCY5uykM6+R9GqQ8K/uxy9rx7HNQlGYl1kPzQho1wx4JwY8w=:"
82                .parse()
83                .unwrap(),
84            size: 42,
85            mime: "text/plain".parse().unwrap(),
86        };
87
88        let json = json!({
89            "digest": {"sha-384": "mqVuAfXRKap7bdgcCY5uykM6+R9GqQ8K/uxy9rx7HNQlGYl1kPzQho1wx4JwY8w="},
90            "length": 42,
91            "type": "text/plain",
92        });
93
94        assert_eq!(serde_json::to_string(&meta).unwrap(), json.to_string());
95    }
96}
drawbridge_type/meta.rs

drawbridge_type/
meta.rs
