twitter_v2/id/
numeric.rs

1use serde::de::Visitor;
2use serde::{Deserialize, Serialize};
3use std::fmt;
4use std::num::ParseIntError;
5use std::str::FromStr;
6
7#[derive(Debug, Clone, Copy, Eq, PartialEq, Ord, PartialOrd, Hash)]
8pub struct NumericId(u64);
9
10impl NumericId {
11    pub fn new(id: u64) -> Self {
12        Self(id)
13    }
14    pub fn as_u64(self) -> u64 {
15        self.0
16    }
17}
18
19impl From<NumericId> for u64 {
20    fn from(id: NumericId) -> Self {
21        id.0
22    }
23}
24
25impl From<u64> for NumericId {
26    fn from(id: u64) -> Self {
27        NumericId(id)
28    }
29}
30
31impl<'a> From<&'a u64> for NumericId {
32    fn from(id: &'a u64) -> Self {
33        NumericId(*id)
34    }
35}
36
37impl FromStr for NumericId {
38    type Err = ParseIntError;
39    fn from_str(s: &str) -> Result<Self, Self::Err> {
40        Ok(Self(s.parse()?))
41    }
42}
43
44impl fmt::Display for NumericId {
45    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
46        f.write_fmt(format_args!("{}", self.0))
47    }
48}
49
50impl PartialEq<u64> for NumericId {
51    fn eq(&self, other: &u64) -> bool {
52        &self.0 == other
53    }
54}
55impl PartialEq<&u64> for NumericId {
56    fn eq(&self, other: &&u64) -> bool {
57        &self.0 == *other
58    }
59}
60
61impl Serialize for NumericId {
62    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
63    where
64        S: serde::Serializer,
65    {
66        serializer.serialize_str(&self.to_string())
67    }
68}
69
70struct NumericIdVisitor;
71
72impl<'de> Visitor<'de> for NumericIdVisitor {
73    type Value = NumericId;
74    fn expecting(&self, f: &mut fmt::Formatter) -> fmt::Result {
75        f.write_str("a valid unsigned integer or string representing an id")
76    }
77    fn visit_u64<E>(self, v: u64) -> Result<Self::Value, E>
78    where
79        E: serde::de::Error,
80    {
81        Ok(NumericId(v))
82    }
83    fn visit_i64<E>(self, v: i64) -> Result<Self::Value, E>
84    where
85        E: serde::de::Error,
86    {
87        if v < 0 {
88            Err(E::custom("value must be an unsigned integer"))
89        } else {
90            Ok(NumericId(v as u64))
91        }
92    }
93    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
94    where
95        E: serde::de::Error,
96    {
97        v.parse().map_err(E::custom)
98    }
99}
100
101impl<'de> Deserialize<'de> for NumericId {
102    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
103    where
104        D: serde::Deserializer<'de>,
105    {
106        deserializer.deserialize_any(NumericIdVisitor)
107    }
108}
109
110pub trait IntoNumericId: fmt::Display {
111    fn into_id(self) -> NumericId;
112}
113
114impl IntoNumericId for NumericId {
115    fn into_id(self) -> NumericId {
116        self
117    }
118}
119
120impl IntoNumericId for u64 {
121    fn into_id(self) -> NumericId {
122        NumericId(self)
123    }
124}
125
126impl<'a> IntoNumericId for &'a u64 {
127    fn into_id(self) -> NumericId {
128        NumericId(*self)
129    }
130}