use crate::headers::{HeaderName, HeaderValue};
use crate::{Body, Headers, Method, Request, Response, StatusCode, Url, Version};
use std::convert::TryFrom;
use std::str::FromStr;
impl From<http::Method> for Method {
fn from(method: http::Method) -> Self {
Method::from_str(method.as_str()).unwrap()
}
}
impl From<Method> for http::Method {
fn from(method: Method) -> Self {
http::Method::from_str(&format!("{}", method)).unwrap()
}
}
impl From<http::StatusCode> for StatusCode {
fn from(status: http::StatusCode) -> Self {
StatusCode::try_from(status.as_u16()).unwrap()
}
}
impl From<StatusCode> for http::StatusCode {
fn from(status: StatusCode) -> Self {
http::StatusCode::from_u16(status.into()).unwrap()
}
}
impl From<http::Version> for Version {
fn from(version: http::Version) -> Self {
match version {
http::Version::HTTP_09 => Version::Http0_9,
http::Version::HTTP_10 => Version::Http1_0,
http::Version::HTTP_11 => Version::Http1_1,
http::Version::HTTP_2 => Version::Http2_0,
http::Version::HTTP_3 => Version::Http3_0,
_ => panic!("unknown HTTP version conversion"),
}
}
}
impl From<Version> for http::Version {
fn from(version: Version) -> Self {
match version {
Version::Http0_9 => http::Version::HTTP_09,
Version::Http1_0 => http::Version::HTTP_10,
Version::Http1_1 => http::Version::HTTP_11,
Version::Http2_0 => http::Version::HTTP_2,
Version::Http3_0 => http::Version::HTTP_3,
}
}
}
fn hyperium_headers_to_headers(hyperium_headers: http::HeaderMap, headers: &mut Headers) {
for (name, value) in hyperium_headers {
let value = value.as_bytes().to_owned();
let value = unsafe { HeaderValue::from_ascii_unchecked(value) };
if let Some(name) = name {
let name = name.as_str().as_bytes().to_owned();
let name = unsafe { HeaderName::from_ascii_unchecked(name) };
headers.insert(name, value).unwrap();
}
}
}
fn headers_to_hyperium_headers(headers: &mut Headers, hyperium_headers: &mut http::HeaderMap) {
for (name, values) in headers {
let name = format!("{}", name).into_bytes();
let name = http::header::HeaderName::from_bytes(&name).unwrap();
for value in values {
let value = format!("{}", value).into_bytes();
let value = http::header::HeaderValue::from_bytes(&value).unwrap();
hyperium_headers.append(&name, value);
}
}
}
fn from_uri_to_url(uri: http::Uri) -> Result<Url, crate::url::ParseError> {
format!("{}", uri).parse()
}
fn from_url_to_uri(url: &Url) -> http::Uri {
http::Uri::try_from(&format!("{}", url)).unwrap()
}
impl TryFrom<http::Request<Body>> for Request {
type Error = crate::url::ParseError;
fn try_from(req: http::Request<Body>) -> Result<Self, Self::Error> {
let (parts, body) = req.into_parts();
let method = parts.method.into();
let url = from_uri_to_url(parts.uri)?;
let mut req = Request::new(method, url);
req.set_body(body);
req.set_version(Some(parts.version.into()));
hyperium_headers_to_headers(parts.headers, req.as_mut());
Ok(req)
}
}
impl From<Request> for http::Request<Body> {
fn from(mut req: Request) -> Self {
let method: http::Method = req.method().into();
let version = req.version().map(|v| v.into()).unwrap_or_default();
let mut builder = http::request::Builder::new()
.method(method)
.uri(from_url_to_uri(req.url()))
.version(version);
headers_to_hyperium_headers(req.as_mut(), builder.headers_mut().unwrap());
builder.body(req.into()).unwrap()
}
}
impl From<http::Response<Body>> for Response {
fn from(res: http::Response<Body>) -> Self {
let (parts, body) = res.into_parts();
let status = parts.status.into();
let mut res = Response::new(status);
res.set_body(body);
res.set_version(Some(parts.version.into()));
hyperium_headers_to_headers(parts.headers, res.as_mut());
res
}
}
impl From<Response> for http::Response<Body> {
fn from(mut res: Response) -> Self {
let status: u16 = res.status().into();
let version = res.version().map(|v| v.into()).unwrap_or_default();
let mut builder = http::response::Builder::new()
.status(status)
.version(version);
headers_to_hyperium_headers(res.as_mut(), builder.headers_mut().unwrap());
builder.body(res.into()).unwrap()
}
}