trillium_http/
version.rs

1// originally from https://github.com/http-rs/http-types/blob/main/src/version.rs
2
3use crate::Error;
4use std::{
5    fmt::Display,
6    str::{self, FromStr},
7};
8
9/// The version of the HTTP protocol in use.
10#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
11#[non_exhaustive]
12pub enum Version {
13    /// HTTP/0.9
14    Http0_9,
15
16    /// HTTP/1.0
17    Http1_0,
18
19    /// HTTP/1.1
20    Http1_1,
21
22    /// HTTP/2
23    Http2,
24
25    /// HTTP/3
26    Http3,
27}
28
29#[cfg(feature = "serde")]
30impl serde::Serialize for Version {
31    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
32    where
33        S: serde::Serializer,
34    {
35        serializer.collect_str(self)
36    }
37}
38
39#[cfg(feature = "serde")]
40impl<'de> serde::Deserialize<'de> for Version {
41    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
42    where
43        D: serde::Deserializer<'de>,
44    {
45        String::deserialize(deserializer)?
46            .parse()
47            .map_err(serde::de::Error::custom)
48    }
49}
50
51impl PartialEq<&Version> for Version {
52    #[allow(clippy::unconditional_recursion)] // false positive
53    fn eq(&self, other: &&Version) -> bool {
54        self == *other
55    }
56}
57
58impl PartialEq<Version> for &Version {
59    #[allow(clippy::unconditional_recursion)] // false positive
60    fn eq(&self, other: &Version) -> bool {
61        *self == other
62    }
63}
64
65impl Version {
66    /// returns the http version as a static str, such as "HTTP/1.1"
67    pub const fn as_str(&self) -> &'static str {
68        match self {
69            Version::Http0_9 => "HTTP/0.9",
70            Version::Http1_0 => "HTTP/1.0",
71            Version::Http1_1 => "HTTP/1.1",
72            Version::Http2 => "HTTP/2",
73            Version::Http3 => "HTTP/3",
74        }
75    }
76
77    #[cfg(feature = "parse")]
78    pub(crate) fn parse(buf: &[u8]) -> crate::Result<Self> {
79        str::from_utf8(buf)
80            .map_err(|_| Error::InvalidVersion)?
81            .parse()
82    }
83}
84
85impl FromStr for Version {
86    type Err = Error;
87
88    fn from_str(s: &str) -> Result<Self, Self::Err> {
89        match s {
90            "HTTP/0.9" | "http/0.9" | "0.9" => Ok(Self::Http0_9),
91            "HTTP/1.0" | "http/1.0" | "1.0" => Ok(Self::Http1_0),
92            "HTTP/1.1" | "http/1.1" | "1.1" => Ok(Self::Http1_1),
93            "HTTP/2" | "http/2" | "2" => Ok(Self::Http2),
94            "HTTP/3" | "http/3" | "3" => Ok(Self::Http3),
95            _ => Err(Error::InvalidVersion),
96        }
97    }
98}
99
100impl AsRef<str> for Version {
101    fn as_ref(&self) -> &str {
102        self.as_str()
103    }
104}
105
106impl AsRef<[u8]> for Version {
107    fn as_ref(&self) -> &[u8] {
108        self.as_str().as_bytes()
109    }
110}
111
112impl Display for Version {
113    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
114        f.write_str(self.as_ref())
115    }
116}
117
118#[cfg(test)]
119mod test {
120    use super::*;
121    #[test]
122    fn from_str() {
123        let versions = [
124            Version::Http0_9,
125            Version::Http1_0,
126            Version::Http1_1,
127            Version::Http2,
128            Version::Http3,
129        ];
130
131        for version in versions {
132            assert_eq!(version.as_str().parse::<Version>().unwrap(), version);
133            assert_eq!(version.to_string().parse::<Version>().unwrap(), version);
134        }
135
136        assert_eq!(
137            "not a version".parse::<Version>().unwrap_err().to_string(),
138            "Invalid or missing version"
139        );
140    }
141
142    #[test]
143    fn eq() {
144        assert_eq!(Version::Http1_1, Version::Http1_1);
145        assert_eq!(Version::Http1_1, &Version::Http1_1);
146        assert_eq!(&Version::Http1_1, Version::Http1_1);
147    }
148
149    #[test]
150    fn to_string() {
151        let output = format!(
152            "{} {} {} {} {}",
153            Version::Http0_9,
154            Version::Http1_0,
155            Version::Http1_1,
156            Version::Http2,
157            Version::Http3
158        );
159        assert_eq!("HTTP/0.9 HTTP/1.0 HTTP/1.1 HTTP/2 HTTP/3", output);
160    }
161
162    #[test]
163    fn ord() {
164        use Version::{Http0_9, Http1_0, Http1_1, Http2, Http3};
165        assert!(Http3 > Http2);
166        assert!(Http2 > Http1_1);
167        assert!(Http1_1 > Http1_0);
168        assert!(Http1_0 > Http0_9);
169    }
170}
trillium_http/version.rs

trillium_http/
version.rs
