Skip to main content

web_push_native/
serde_.rs

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