roblox-api 0.1.4

use reqwest::{
    Response,
    header::{self, HeaderMap, HeaderValue},
};
use serde::de::DeserializeOwned;

use crate::{Error, ratelimit::Ratelimit};

#[derive(Default)]
pub struct Cookie(String);

impl std::fmt::Display for Cookie {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

impl From<&str> for Cookie {
    fn from(value: &str) -> Self {
        Self(format!(".ROBLOSECURITY={}", value))
    }
}

#[derive(Default, Debug)]
pub struct ClientRequestor {
    pub(crate) client: reqwest::Client,
    pub(crate) default_headers: HeaderMap,
}

#[derive(Default, Debug)]
pub struct Client {
    pub requestor: ClientRequestor,
    pub(crate) ratelimit: Option<Ratelimit>,
}

impl Client {
    pub fn from_cookie(cookie: Cookie) -> Self {
        let client = reqwest::Client::new();
        let mut default_headers = HeaderMap::new();

        default_headers.insert(
            header::USER_AGENT,
            HeaderValue::from_str("Roblox/WinInet").unwrap(),
        );

        default_headers.insert(
            header::COOKIE,
            HeaderValue::from_str(&cookie.to_string()).unwrap(),
        );

        // For some reason some APIs error if not set
        default_headers.append(
            header::COOKIE,
            HeaderValue::from_str("RBXEventTrackerV2=&browserid=2").unwrap(),
        );

        Client {
            ratelimit: None,
            requestor: ClientRequestor {
                client,
                default_headers,
            },
        }
    }
}

impl ClientRequestor {
    pub(crate) async fn parse_json<T: DeserializeOwned>(
        &self,
        response: Response,
    ) -> Result<T, Error> {
        match response.json::<T>().await {
            Ok(result) => Ok(result),
            Err(_) => Err(Error::BadResponse),
        }
    }
}