tiedcrossing_type/
meta.rs

1// SPDX-FileCopyrightText: 2022 Profian Inc. <opensource@profian.com>
2// SPDX-License-Identifier: AGPL-3.0-only
3
4use crate::digest::ContentDigest;
5
6use mime::Mime;
7use serde::{de::Error as _, Deserialize, Deserializer, Serialize, Serializer};
8
9#[cfg(feature = "axum")]
10use axum::{
11    extract::rejection::{TypedHeaderRejection, TypedHeaderRejectionReason},
12    extract::{FromRequest, RequestParts, TypedHeader},
13    headers::{ContentLength, ContentType},
14    response::{IntoResponseParts, Response, ResponseParts},
15};
16
17#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
18pub struct Meta {
19    #[serde(rename = "digest")]
20    pub hash: ContentDigest<Box<[u8]>>,
21
22    #[serde(rename = "length")]
23    pub size: u64,
24
25    #[serde(deserialize_with = "deserialize")]
26    #[serde(serialize_with = "serialize")]
27    #[serde(rename = "type")]
28    pub mime: Mime,
29}
30
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}
tiedcrossing_type/meta.rs

tiedcrossing_type/
meta.rs
