45 lines
1.3 KiB
Rust
45 lines
1.3 KiB
Rust
|
use anyhow::Result;
|
||
|
|
use chrono::DateTime;
|
||
|
|
use chrono::Duration;
|
||
|
|
use chrono::Utc;
|
||
|
|
use reqwest::Response;
|
||
|
|
|
||
|
|
pub mod download_link;
|
||
|
|
pub mod files;
|
||
|
|
|
||
|
|
pub static GAME_NAME: &str = "skyrimspecialedition";
|
||
|
|
pub const GAME_ID: u32 = 1704;
|
||
|
|
pub static USER_AGENT: &str = "mod-mapper/0.1";
|
||
|
|
|
||
|
|
pub fn rate_limit_wait_duration(res: &Response) -> Result<Option<std::time::Duration>> {
|
||
|
|
let daily_remaining = res
|
||
|
|
.headers()
|
||
|
|
.get("x-rl-daily-remaining")
|
||
|
|
.expect("No daily remaining in response headers");
|
||
|
|
let hourly_remaining = res
|
||
|
|
.headers()
|
||
|
|
.get("x-rl-hourly-remaining")
|
||
|
|
.expect("No hourly limit in response headers");
|
||
|
|
let hourly_reset = res
|
||
|
|
.headers()
|
||
|
|
.get("x-rl-hourly-reset")
|
||
|
|
.expect("No hourly reset in response headers");
|
||
|
|
dbg!(daily_remaining);
|
||
|
|
dbg!(hourly_remaining);
|
||
|
|
|
||
|
|
if hourly_remaining == "0" {
|
||
|
|
let hourly_reset = hourly_reset.to_str()?.trim();
|
||
|
|
let hourly_reset: DateTime<Utc> =
|
||
|
|
(DateTime::parse_from_str(hourly_reset, "%Y-%m-%d %H:%M:%S %z")?
|
||
|
|
+ Duration::seconds(5))
|
||
|
|
.into();
|
||
|
|
dbg!(hourly_reset);
|
||
|
|
let duration = (hourly_reset - Utc::now()).to_std()?;
|
||
|
|
dbg!(duration);
|
||
|
|
|
||
|
|
return Ok(Some(duration));
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(None)
|
||
|
|
}
|