web_push_native/
serde_.rs

1use super::{Auth, WebPushBuilder};
2use base64ct::{Base64UrlUnpadded, Encoding};
3use http::Uri;
4use p256::elliptic_curve::sec1::ToEncodedPoint;
5use serde::{de, Deserialize, Deserializer, Serialize, Serializer};
6use std::{borrow::Cow, time::Duration};
7
8#[derive(Serialize, Deserialize)]
9#[serde(rename = "WebPushBuilder", rename_all = "camelCase")]
10struct WebPushSerde<'a> {
11    #[serde(serialize_with = "url_to_string", deserialize_with = "string_to_url")]
12    endpoint: Cow<'a, Uri>,
13    expiration_time: Option<Duration>,
14    keys: Keys,
15}
16
17#[derive(Serialize, Deserialize)]
18struct Keys {
19    #[serde(serialize_with = "auth_to_bytes", deserialize_with = "bytes_to_auth")]
20    auth: Auth,
21    #[serde(serialize_with = "p256_to_bytes", deserialize_with = "bytes_to_p256")]
22    p256dh: p256::PublicKey,
23}
24
25fn url_to_string<U: AsRef<Uri>, S: Serializer>(url: U, s: S) -> Result<S::Ok, S::Error> {
26    s.serialize_str(&url.as_ref().to_string())
27}
28
29fn string_to_url<'de, D: Deserializer<'de>>(d: D) -> Result<Cow<'static, Uri>, D::Error> {
30    let s: Cow<str> = Deserialize::deserialize(d)?;
31    s.parse().map(Cow::Owned).map_err(de::Error::custom)
32}
33
34fn auth_to_bytes<S: Serializer>(auth: &Auth, s: S) -> Result<S::Ok, S::Error> {
35    s.serialize_str(&Base64UrlUnpadded::encode_string(auth.as_slice()))
36}
37
38fn bytes_to_auth<'de, D: Deserializer<'de>>(d: D) -> Result<Auth, D::Error> {
39    let b64: Cow<str> = Deserialize::deserialize(d)?;
40    Ok(Auth::clone_from_slice(
41        &Base64UrlUnpadded::decode_vec(&b64).map_err(de::Error::custom)?,
42    ))
43}
44
45fn p256_to_bytes<S: Serializer>(auth: &p256::PublicKey, s: S) -> Result<S::Ok, S::Error> {
46    s.serialize_str(&Base64UrlUnpadded::encode_string(
47        auth.to_encoded_point(false).as_bytes(),
48    ))
49}
50
51fn bytes_to_p256<'de, D: Deserializer<'de>>(d: D) -> Result<p256::PublicKey, D::Error> {
52    let b64: Cow<str> = Deserialize::deserialize(d)?;
53    p256::PublicKey::from_sec1_bytes(
54        &Base64UrlUnpadded::decode_vec(&b64).map_err(de::Error::custom)?,
55    )
56    .map_err(de::Error::custom)
57}
58
59impl Serialize for WebPushBuilder {
60    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
61    where
62        S: Serializer,
63    {
64        WebPushSerde {
65            endpoint: Cow::Borrowed(&self.endpoint),
66            expiration_time: None,
67            keys: Keys {
68                auth: self.ua_auth,
69                p256dh: self.ua_public,
70            },
71        }
72        .serialize(serializer)
73    }
74}
75
76impl<'de> Deserialize<'de> for WebPushBuilder {
77    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
78    where
79        D: serde::Deserializer<'de>,
80    {
81        let serde = WebPushSerde::deserialize(deserializer)?;
82        Ok(WebPushBuilder {
83            endpoint: serde.endpoint.into_owned(),
84            valid_duration: Duration::from_secs(12 * 60 * 60),
85            ua_public: serde.keys.p256dh,
86            ua_auth: serde.keys.auth,
87            http_auth: (),
88        })
89    }
90}