From a7e38c97a6019db435919750e3d461428dbdfbd3 Mon Sep 17 00:00:00 2001 From: Simon Date: Mon, 6 Apr 2026 06:51:43 +0000 Subject: [PATCH] header fix --- src/providers/pimpbunny.rs | 12 +++++++++++- src/proxies/pimpbunny.rs | 2 +- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/src/providers/pimpbunny.rs b/src/providers/pimpbunny.rs index 267d4c2..3a13f74 100644 --- a/src/providers/pimpbunny.rs +++ b/src/providers/pimpbunny.rs @@ -799,7 +799,8 @@ impl PimpbunnyProvider { let formats = vec![ VideoFormat::new(proxy_url.clone(), "auto".into(), "video/mp4".into()) .format_id("auto".into()) - .format_note("proxied".into()), + .format_note("proxied".into()) + .http_header("Referer".to_string(), video_url.clone()), ]; Ok( @@ -969,6 +970,15 @@ mod tests { assert_eq!(items[0].duration, 754); assert_eq!(items[0].views, Some(1200)); assert_eq!(items[0].formats.as_ref().map(|f| f.len()), Some(1)); + assert_eq!( + items[0].formats.as_ref().and_then(|formats| formats.first()).map( + |format| format.http_headers_pairs().contains(&( + "Referer".to_string(), + "https://pimpbunny.com/videos/example-video/".to_string(), + )) + ), + Some(true) + ); } #[test] diff --git a/src/proxies/pimpbunny.rs b/src/proxies/pimpbunny.rs index 4b12615..4610d25 100644 --- a/src/proxies/pimpbunny.rs +++ b/src/proxies/pimpbunny.rs @@ -172,7 +172,7 @@ impl crate::proxies::Proxy for PimpbunnyProxy { let mut requester = requester.get_ref().clone(); Self::warm_root_session(&mut requester).await; - let headers = Self::headers_with_cookies(&requester, &detail_url, Self::root_referer()); + let headers = Self::headers_with_cookies(&requester, &detail_url, &detail_url); let text = match requester .get_with_headers(&detail_url, headers, Some(Version::HTTP_2)) .await