use std::fmt;
use bytes::Bytes;
use http::{HeaderMap, StatusCode};
use js_sys::Uint8Array;
use url::Url;
#[cfg(feature = "json")]
use serde::de::DeserializeOwned;
pub struct Response {
http: http::Response<web_sys::Response>,
url: Box<Url>,
}
impl Response {
pub(super) fn new(res: http::Response<web_sys::Response>, url: Url) -> Response {
Response {
http: res,
url: Box::new(url),
}
}
#[inline]
pub fn status(&self) -> StatusCode {
self.http.status()
}
#[inline]
pub fn headers(&self) -> &HeaderMap {
self.http.headers()
}
#[inline]
pub fn headers_mut(&mut self) -> &mut HeaderMap {
self.http.headers_mut()
}
pub fn content_length(&self) -> Option<u64> {
self.headers()
.get(http::header::CONTENT_LENGTH)?
.to_str()
.ok()?
.parse()
.ok()
}
#[inline]
pub fn url(&self) -> &Url {
&self.url
}
#[cfg(feature = "json")]
#[cfg_attr(docsrs, doc(cfg(feature = "json")))]
pub async fn json<T: DeserializeOwned>(self) -> crate::reqwest::Result<T> {
let full = self.bytes().await?;
serde_json::from_slice(&full).map_err(crate::reqwest::error::decode)
}
pub async fn text(self) -> crate::reqwest::Result<String> {
let p = self
.http
.body()
.text()
.map_err(crate::reqwest::error::wasm)
.map_err(crate::reqwest::error::decode)?;
let js_val = super::promise::<wasm_bindgen::JsValue>(p)
.await
.map_err(crate::reqwest::error::decode)?;
if let Some(s) = js_val.as_string() {
Ok(s)
} else {
Err(crate::reqwest::error::decode("response.text isn't string"))
}
}
pub async fn bytes(self) -> crate::reqwest::Result<Bytes> {
let p = self
.http
.body()
.array_buffer()
.map_err(crate::reqwest::error::wasm)
.map_err(crate::reqwest::error::decode)?;
let buf_js = super::promise::<wasm_bindgen::JsValue>(p)
.await
.map_err(crate::reqwest::error::decode)?;
let buffer = Uint8Array::new(&buf_js);
let mut bytes = vec![0; buffer.length() as usize];
buffer.copy_to(&mut bytes);
Ok(bytes.into())
}
pub fn error_for_status(self) -> crate::reqwest::Result<Self> {
let status = self.status();
if status.is_client_error() || status.is_server_error() {
Err(crate::reqwest::error::status_code(*self.url, status))
} else {
Ok(self)
}
}
pub fn error_for_status_ref(&self) -> crate::reqwest::Result<&Self> {
let status = self.status();
if status.is_client_error() || status.is_server_error() {
Err(crate::reqwest::error::status_code(*self.url.clone(), status))
} else {
Ok(self)
}
}
}
impl fmt::Debug for Response {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("Response")
.field("status", &self.status())
.field("headers", self.headers())
.finish()
}
}