prefixed-tsid 0.9.0

Create Stripe-style prefixed database IDs that are type-safe and portable
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

use std::{borrow::Cow, fmt::Write, str::FromStr};

use rocket::{
    data::ToByteUnit,
    form::{FromFormField, error::ErrorKind},
    http::{
        impl_from_uri_param_identity,
        uri::fmt::{Part, Path, UriDisplay},
    },
    request::FromParam,
    tokio::io::AsyncReadExt,
};

use crate::{resources::TSIDResource, tsid::TSIDDatabaseID};

impl<'d, Resource: TSIDResource> FromParam<'d> for TSIDDatabaseID<Resource> {
    type Error = anyhow::Error;
    fn from_param(param: &'d str) -> Result<Self, Self::Error> {
        Self::from_str(param)
    }
}

#[rocket::async_trait]
impl<'d, Resource: TSIDResource> FromFormField<'d> for TSIDDatabaseID<Resource> {
    fn from_value(field: rocket::form::ValueField<'d>) -> rocket::form::Result<'d, Self> {
        Self::from_str(field.value)
            .map_err(|e| ErrorKind::Validation(Cow::Owned(e.to_string())).into())
    }

    async fn from_data(field: rocket::form::DataField<'d, '_>) -> rocket::form::Result<'d, Self> {
        let mut value = "".to_string();
        field
            .data
            .open(5.kibibytes())
            .read_to_string(&mut value)
            .await?;
        Self::from_str(&value).map_err(|e| ErrorKind::Validation(Cow::Owned(e.to_string())).into())
    }
}

impl<Resource: TSIDResource, P: Part> UriDisplay<P> for TSIDDatabaseID<Resource> {
    fn fmt(&self, f: &mut rocket::http::uri::fmt::Formatter<'_, P>) -> std::fmt::Result {
        f.write_str(&self.to_string())
    }
}

impl_from_uri_param_identity!([Path] (Resource: TSIDResource) TSIDDatabaseID<Resource>);