1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
use std::borrow::Cow;
use crate::{parse::TryParse, error::HTTPError};
pub struct Status(u16, Cow<'static, str>);
impl Status {
pub fn new(code: u16, msg: impl Into<String>) -> Self {
Self(code, Cow::Owned(msg.into()))
}
pub fn code(&self) -> u16 {
self.0
}
pub fn msg(&self) -> &str {
self.1.as_ref()
}
}
impl ToString for Status {
fn to_string(&self) -> String {
if self.1.len() < 1 {
format!("{}", self.0)
} else {
format!("{} {}", self.0, self.1)
}
}
}
impl TryParse for Status {
fn try_parse(s: impl Into<String>) -> Result<Self, HTTPError> {
let s = s.into();
let mut parts = s.split(" ");
let Some(code_str) = parts.next() else {
return Err(HTTPError::InvalidStatus(s))
};
let Ok(code) = code_str.parse::<u16>() else {
return Err(HTTPError::InvalidStatus(s))
};
let msg: String = parts.collect();
Ok(Self::new(code, msg))
}
}
impl PartialEq for Status {
fn eq(&self, other: &Self) -> bool {
self.0 == other.0
}
}
macro_rules! status_from {
($type:ty) => {
impl From<$type> for Status {
fn from(value: $type) -> Self {
Self(value as u16, Cow::Owned(String::new()))
}
}
impl<T> From<($type, T)> for Status
where
T: Into<String>
{
fn from(value: ($type, T)) -> Self {
Self(value.0 as u16, Cow::Owned(value.1.into()))
}
}
};
}
status_from!(u8);
status_from!(u16);
status_from!(u32);
status_from!(u64);
status_from!(u128);
macro_rules! status {
($name:ident, $code:literal, $msg:literal) => {
pub const $name: Status = Status($code, Cow::Borrowed($msg));
};
}
status!(SUCCESS, 200, "Ok");
|
