use reqwest::Response as ReqwestResponse;

#[derive(Debug)]
pub struct TextResponse {
    pub status: u16,
    pub content_type: Option<String>,
    pub body: String,
}

impl TextResponse {
    pub async fn from(response: ReqwestResponse) -> Self {
        Self {
            status: response.status().as_u16(),
            content_type: response
                .headers()
                .get("content-type")
                .map(|content_type| content_type.to_str().unwrap().to_owned()),
            body: response.text().await.unwrap(),
        }
    }

    pub fn is_json_and_ok(&self) -> bool {
        self.is_ok() && self.is_json()
    }

    pub fn is_json(&self) -> bool {
        if let Some(content_type) = &self.content_type {
            return content_type == "application/json";
        }
        false
    }

    pub fn is_ok(&self) -> bool {
        self.status == 200
    }
}

#[derive(Debug)]
pub struct BinaryResponse {
    pub status: u16,
    pub content_type: Option<String>,
    pub bytes: Vec<u8>,
}

impl BinaryResponse {
    pub async fn from(response: ReqwestResponse) -> Self {
        Self {
            status: response.status().as_u16(),
            content_type: response
                .headers()
                .get("content-type")
                .map(|content_type| content_type.to_str().unwrap().to_owned()),
            bytes: response.bytes().await.unwrap().to_vec(),
        }
    }
    pub fn is_bittorrent_and_ok(&self) -> bool {
        self.is_ok() && self.is_bittorrent()
    }

    pub fn is_bittorrent(&self) -> bool {
        if let Some(content_type) = &self.content_type {
            return content_type == "application/x-bittorrent";
        }
        false
    }

    pub fn is_ok(&self) -> bool {
        self.status == 200
    }
}