1use serde::{de::Error, de::Visitor, Deserialize, Deserializer, Serialize, Serializer};
2#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
4#[non_exhaustive]
5pub enum Version {
6 Http0_9,
8
9 Http1_0,
11
12 Http1_1,
14
15 Http2_0,
17
18 Http3_0,
20}
21
22impl Serialize for Version {
23 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
24 where
25 S: Serializer,
26 {
27 serializer.serialize_str(&self.to_string())
28 }
29}
30
31struct VersionVisitor;
32
33impl<'de> Visitor<'de> for VersionVisitor {
34 type Value = Version;
35
36 fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
37 write!(formatter, "a HTTP version as &str")
38 }
39
40 fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
41 where
42 E: Error,
43 {
44 match v {
45 "HTTP/0.9" => Ok(Version::Http0_9),
46 "HTTP/1.0" => Ok(Version::Http1_0),
47 "HTTP/1.1" => Ok(Version::Http1_1),
48 "HTTP/2" => Ok(Version::Http2_0),
49 "HTTP/3" => Ok(Version::Http3_0),
50 _ => Err(Error::invalid_value(serde::de::Unexpected::Str(v), &self)),
51 }
52 }
53
54 fn visit_string<E>(self, v: String) -> Result<Self::Value, E>
55 where
56 E: Error,
57 {
58 self.visit_str(&v)
59 }
60}
61
62impl<'de> Deserialize<'de> for Version {
63 fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
64 where
65 D: Deserializer<'de>,
66 {
67 deserializer.deserialize_str(VersionVisitor)
68 }
69}
70
71impl AsRef<str> for Version {
72 fn as_ref(&self) -> &'static str {
73 match self {
74 Version::Http0_9 => "HTTP/0.9",
75 Version::Http1_0 => "HTTP/1.0",
76 Version::Http1_1 => "HTTP/1.1",
77 Version::Http2_0 => "HTTP/2",
78 Version::Http3_0 => "HTTP/3",
79 }
80 }
81}
82
83impl std::fmt::Display for Version {
84 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
85 f.write_str(self.as_ref())
86 }
87}
88
89#[cfg(test)]
90mod test {
91 use super::*;
92
93 #[test]
94 fn as_ref() {
95 assert_eq!(Version::Http0_9.as_ref(), "HTTP/0.9");
96 assert_eq!(Version::Http1_0.as_ref(), "HTTP/1.0");
97 assert_eq!(Version::Http1_1.as_ref(), "HTTP/1.1");
98 assert_eq!(Version::Http2_0.as_ref(), "HTTP/2");
99 assert_eq!(Version::Http3_0.as_ref(), "HTTP/3");
100 }
101
102 #[test]
103 fn to_string() {
104 let output = format!(
105 "{} {} {} {} {}",
106 Version::Http0_9,
107 Version::Http1_0,
108 Version::Http1_1,
109 Version::Http2_0,
110 Version::Http3_0
111 );
112 assert_eq!("HTTP/0.9 HTTP/1.0 HTTP/1.1 HTTP/2 HTTP/3", output);
113 }
114
115 #[test]
116 fn ord() {
117 use Version::*;
118 assert!(Http3_0 > Http2_0);
119 assert!(Http2_0 > Http1_1);
120 assert!(Http1_1 > Http1_0);
121 assert!(Http1_0 > Http0_9);
122 }
123
124 #[test]
125 fn serde() -> Result<(), serde_json::Error> {
126 assert_eq!("\"HTTP/3\"", serde_json::to_string(&Version::Http3_0)?);
127 assert_eq!(Version::Http1_1, serde_json::from_str("\"HTTP/1.1\"")?);
128 Ok(())
129 }
130}