serai/common/request/src/response.rs

use hyper::{
  StatusCode,
  header::{HeaderValue, HeaderMap},
  body::{Buf, Incoming},
};
use http_body_util::BodyExt;

use crate::{Client, Error};

// Borrows the client so its async task lives as long as this response exists.
#[allow(dead_code)]
#[derive(Debug)]
pub struct Response<'a>(pub(crate) hyper::Response<Incoming>, pub(crate) &'a Client);
impl Response<'_> {
  pub fn status(&self) -> StatusCode {
    self.0.status()
  }
  pub fn headers(&self) -> &HeaderMap<HeaderValue> {
    self.0.headers()
  }
  pub async fn body(self) -> Result<impl std::io::Read, Error> {
    Ok(self.0.into_body().collect().await.map_err(Error::Hyper)?.aggregate().reader())
  }
}