Skip to main content

ftracker_identifiers/country/
serde.rs

1//! `Serialize` and `Deserialize` for [`CountryCode`], gated behind the `serde` feature.
2//!
3//! `CountryCode` (de)serializes as its canonical two letter string, for example `"US"`, so it round
4//! trips as a plain identifier in JSON and config files. Deserializing always re-runs full
5//! validation. An untrusted payload can never produce an invalid `CountryCode`.
6
7use ::serde::de::{self, Visitor};
8use ::serde::{Deserialize, Deserializer, Serialize, Serializer};
9use core::fmt;
10
11use super::CountryCode;
12
13impl Serialize for CountryCode {
14    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
15    where
16        S: Serializer,
17    {
18        serializer.serialize_str(self.as_str())
19    }
20}
21
22struct CountryCodeVisitor;
23
24impl<'de> Visitor<'de> for CountryCodeVisitor {
25    type Value = CountryCode;
26
27    fn expecting(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
28        f.write_str("a 2-character ISO 3166-1 alpha-2 country code, e.g. US")
29    }
30
31    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
32    where
33        E: de::Error,
34    {
35        CountryCode::parse(v).map_err(de::Error::custom)
36    }
37}
38
39impl<'de> Deserialize<'de> for CountryCode {
40    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
41    where
42        D: Deserializer<'de>,
43    {
44        deserializer.deserialize_str(CountryCodeVisitor)
45    }
46}
47
48#[cfg(test)]
49mod tests {
50    use super::super::CountryCode;
51
52    #[test]
53    fn round_trips_through_json() {
54        let code = CountryCode::parse("US").unwrap();
55        let json = serde_json::to_string(&code).unwrap();
56        assert_eq!(json, "\"US\"");
57        let back: CountryCode = serde_json::from_str(&json).unwrap();
58        assert_eq!(code, back);
59    }
60
61    #[test]
62    fn rejects_invalid_json_string() {
63        let err = serde_json::from_str::<CountryCode>("\"ZZ\"").unwrap_err();
64        assert!(err.to_string().contains("ISO 3166-1"));
65    }
66}