simon/hottub
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pimpbunny updates

Browse Source
This commit is contained in:
Simon
2026-03-17 09:53:34 +00:00
parent 3c3af70ed6
commit 0563a7231a
2 changed files with 123 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
src/proxies/pimpbunnythumb.rs
View File

@@ -4,9 +4,15 @@ use ntex::{
web::{self, HttpRequest, error},
};
use url::Url;
use wreq::Version;
use crate::util::requester::Requester;
const FIREFOX_USER_AGENT: &str =
"Mozilla/5.0 (X11; Linux x86_64; rv:147.0) Gecko/20100101 Firefox/147.0";
const IMAGE_ACCEPT: &str =
"image/avif,image/webp,image/png,image/svg+xml,image/*;q=0.8,*/*;q=0.5";
fn is_allowed_thumb_url(url: &str) -> bool {
let Some(url) = Url::parse(url).ok() else {
return false;
@@ -37,17 +43,66 @@ pub async fn get_image(
return Ok(web::HttpResponse::BadRequest().finish());
}
let upstream = match requester
.get_ref()
.clone()
.get_raw_with_headers(
image_url.as_str(),
vec![("Referer".to_string(), "https://pimpbunny.com/".to_string())],
)
let headers = vec![
("Referer".to_string(), "https://pimpbunny.com/".to_string()),
("User-Agent".to_string(), FIREFOX_USER_AGENT.to_string()),
("Accept".to_string(), IMAGE_ACCEPT.to_string()),
("Accept-Language".to_string(), "en-US,en;q=0.9".to_string()),
];
let mut requester = requester.get_ref().clone();
let mut upstream = requester
.get_raw_with_headers(image_url.as_str(), headers.clone())
.await
{
Ok(response) if response.status().is_success() => response,
_ => return Ok(web::HttpResponse::NotFound().finish()),
.ok();
let needs_warmup = upstream
.as_ref()
.map(|response| !response.status().is_success())
.unwrap_or(true);
if needs_warmup {
let _ = requester
.get_with_headers(
"https://pimpbunny.com/",
vec![
("Referer".to_string(), "https://pimpbunny.com/".to_string()),
("User-Agent".to_string(), FIREFOX_USER_AGENT.to_string()),
(
"Accept".to_string(),
"text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8"
.to_string(),
),
("Accept-Language".to_string(), "en-US,en;q=0.9".to_string()),
],
Some(Version::HTTP_11),
)
.await;
upstream = requester
.get_raw_with_headers(image_url.as_str(), headers.clone())
.await
.ok();
}
let needs_image_specific_warmup = upstream
.as_ref()
.map(|response| !response.status().is_success())
.unwrap_or(true);
if needs_image_specific_warmup {
let _ = requester
.get_with_headers(image_url.as_str(), headers.clone(), Some(Version::HTTP_11))
.await;
upstream = requester
.get_raw_with_headers(image_url.as_str(), headers)
.await
.ok();
}
let Some(upstream) = upstream.filter(|response| response.status().is_success()) else {
return Ok(web::HttpResponse::NotFound().finish());
};
let status = upstream.status();