various bugfixes
This commit is contained in:
@@ -15,7 +15,7 @@ ntex = { version = "2.15.1", features = ["tokio"] }
|
|||||||
ntex-files = "2.0.0"
|
ntex-files = "2.0.0"
|
||||||
serde = "1.0.228"
|
serde = "1.0.228"
|
||||||
serde_json = "1.0.145"
|
serde_json = "1.0.145"
|
||||||
tokio = { version = "1.47.1", features = ["full"] }
|
tokio = { version = "1.49", features = ["full"] }
|
||||||
wreq = { version = "5.3.0", features = ["full", "cookies", "multipart"] }
|
wreq = { version = "5.3.0", features = ["full", "cookies", "multipart"] }
|
||||||
wreq-util = "2"
|
wreq-util = "2"
|
||||||
percent-encoding = "2.3.2"
|
percent-encoding = "2.3.2"
|
||||||
|
|||||||
@@ -14,10 +14,10 @@ services:
|
|||||||
environment:
|
environment:
|
||||||
- RUST_LOG=info
|
- RUST_LOG=info
|
||||||
- BURP_URL=http://127.0.0.1:8081 # local burpsuite proxy for crawler analysis
|
- BURP_URL=http://127.0.0.1:8081 # local burpsuite proxy for crawler analysis
|
||||||
- PROXY=1 # 1 for enable, else disabled
|
- PROXY=0 # 1 for enable, else disabled
|
||||||
- DATABASE_URL=hottub.db # sqlite db to store hard to get videos for easy access
|
- DATABASE_URL=hottub.db # sqlite db to store hard to get videos for easy access
|
||||||
- FLARE_URL=http://flaresolverr:8191/v1 # flaresolverr to get around cloudflare 403 codes
|
- FLARE_URL=http://flaresolverr:8191/v1 # flaresolverr to get around cloudflare 403 codes
|
||||||
- DOMAIN=hottub.spacemoehre.de # optional for the 302 forward on "/"
|
- DOMAIN=hottub.spacemoehre.de # optional for the 302 forward on "/" to
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
working_dir: /app
|
working_dir: /app
|
||||||
ports:
|
ports:
|
||||||
@@ -34,11 +34,13 @@ services:
|
|||||||
interval: 30s
|
interval: 30s
|
||||||
timeout: 5s
|
timeout: 5s
|
||||||
retries: 3
|
retries: 3
|
||||||
|
start_period: 1s
|
||||||
ulimits:
|
ulimits:
|
||||||
nofile:
|
nofile:
|
||||||
soft: 65536
|
soft: 65536
|
||||||
hard: 65536
|
hard: 65536
|
||||||
|
|
||||||
|
# flaresolverr to bypass cloudflare protections
|
||||||
flaresolverr:
|
flaresolverr:
|
||||||
container_name: flaresolverr
|
container_name: flaresolverr
|
||||||
ports:
|
ports:
|
||||||
@@ -53,9 +55,4 @@ services:
|
|||||||
max-size: "10m" # Maximum size of each log file (e.g., 10MB)
|
max-size: "10m" # Maximum size of each log file (e.g., 10MB)
|
||||||
max-file: "3" # Maximum number of log files to keep
|
max-file: "3" # Maximum number of log files to keep
|
||||||
|
|
||||||
restarter: # restarts the flaresolverr so its always ready for work
|
|
||||||
image: docker:cli
|
|
||||||
container_name: flaresolverr-restarter
|
|
||||||
volumes: ["/var/run/docker.sock:/var/run/docker.sock"]
|
|
||||||
command: ["/bin/sh", "-c", "while true; do sleep 26400; docker restart flaresolverr; done"]
|
|
||||||
restart: unless-stopped
|
|
||||||
|
|||||||
113
src/api.rs
113
src/api.rs
@@ -3,7 +3,7 @@ use ntex::http::header;
|
|||||||
use ntex::web;
|
use ntex::web;
|
||||||
use ntex::web::HttpRequest;
|
use ntex::web::HttpRequest;
|
||||||
use std::cmp::Ordering;
|
use std::cmp::Ordering;
|
||||||
use std::fs;
|
use std::{fs, io};
|
||||||
use tokio::task;
|
use tokio::task;
|
||||||
|
|
||||||
use crate::providers::all::AllProvider;
|
use crate::providers::all::AllProvider;
|
||||||
@@ -15,6 +15,7 @@ use crate::providers::redtube::RedtubeProvider;
|
|||||||
use crate::providers::rule34video::Rule34videoProvider;
|
use crate::providers::rule34video::Rule34videoProvider;
|
||||||
// use crate::providers::spankbang::SpankbangProvider;
|
// use crate::providers::spankbang::SpankbangProvider;
|
||||||
use crate::util::cache::VideoCache;
|
use crate::util::cache::VideoCache;
|
||||||
|
use crate::util::discord::send_discord_error_report;
|
||||||
use crate::util::requester::Requester;
|
use crate::util::requester::Requester;
|
||||||
use crate::{DbPool, db, status::*, videos::*};
|
use crate::{DbPool, db, status::*, videos::*};
|
||||||
use cute::c;
|
use cute::c;
|
||||||
@@ -104,7 +105,12 @@ pub fn config(cfg: &mut web::ServiceConfig) {
|
|||||||
web::resource("/videos")
|
web::resource("/videos")
|
||||||
// .route(web::get().to(videos_get))
|
// .route(web::get().to(videos_get))
|
||||||
.route(web::post().to(videos_post)),
|
.route(web::post().to(videos_post)),
|
||||||
);
|
)
|
||||||
|
.service(
|
||||||
|
web::resource("/test")
|
||||||
|
.route(web::get().to(test))
|
||||||
|
)
|
||||||
|
;
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn status(req: HttpRequest) -> Result<impl web::Responder, web::Error> {
|
async fn status(req: HttpRequest) -> Result<impl web::Responder, web::Error> {
|
||||||
@@ -345,40 +351,6 @@ async fn status(req: HttpRequest) -> Result<impl web::Responder, web::Error> {
|
|||||||
cacheDuration: None,
|
cacheDuration: None,
|
||||||
});
|
});
|
||||||
|
|
||||||
// status.add_channel(Channel {
|
|
||||||
// id: "spankbang".to_string(),
|
|
||||||
// name: "SpankBang".to_string(),
|
|
||||||
// description: "Popular Porn Videos - SpankBang".to_string(),
|
|
||||||
// premium: false,
|
|
||||||
// favicon: "https://www.google.com/s2/favicons?sz=64&domain=spankbang.com".to_string(),
|
|
||||||
// status: "active".to_string(),
|
|
||||||
// categories: vec![],
|
|
||||||
// options: vec![ChannelOption {
|
|
||||||
// id: "sort".to_string(),
|
|
||||||
// title: "Sort".to_string(),
|
|
||||||
// description: "Sort the Videos".to_string(), //"Sort the videos by Date or Name.".to_string(),
|
|
||||||
// systemImage: "list.number".to_string(),
|
|
||||||
// colorName: "blue".to_string(),
|
|
||||||
// options: vec![
|
|
||||||
// FilterOption {
|
|
||||||
// id: "trending_videos".to_string(),
|
|
||||||
// title: "Trending".to_string(),
|
|
||||||
// },
|
|
||||||
// FilterOption {
|
|
||||||
// id: "new_videos".to_string(),
|
|
||||||
// title: "New".to_string(),
|
|
||||||
// },
|
|
||||||
// FilterOption {
|
|
||||||
// id: "most_popular".to_string(),
|
|
||||||
// title: "Popular".to_string(),
|
|
||||||
// },
|
|
||||||
// ],
|
|
||||||
// multiSelect: false,
|
|
||||||
// }],
|
|
||||||
// nsfw: true,
|
|
||||||
//cacheDuration: Some(1800),
|
|
||||||
// });
|
|
||||||
|
|
||||||
// rule34video
|
// rule34video
|
||||||
status.add_channel(Channel {
|
status.add_channel(Channel {
|
||||||
id: "rule34video".to_string(),
|
id: "rule34video".to_string(),
|
||||||
@@ -647,41 +619,6 @@ async fn status(req: HttpRequest) -> Result<impl web::Responder, web::Error> {
|
|||||||
cacheDuration: Some(1800),
|
cacheDuration: Some(1800),
|
||||||
});
|
});
|
||||||
|
|
||||||
// // hentaimoon
|
|
||||||
// status.add_channel(Channel {
|
|
||||||
// id: "hentaimoon".to_string(),
|
|
||||||
// name: "Hentai Moon".to_string(),
|
|
||||||
// description: "Your Hentai Sputnik".to_string(),
|
|
||||||
// premium: false,
|
|
||||||
// favicon: "https://www.google.com/s2/favicons?sz=64&domain=hentai-moon.com".to_string(),
|
|
||||||
// status: "active".to_string(),
|
|
||||||
// categories: vec![],
|
|
||||||
// options: vec![ChannelOption {
|
|
||||||
// id: "sort".to_string(),
|
|
||||||
// title: "Sort".to_string(),
|
|
||||||
// description: "Sort the Videos".to_string(), //"Sort the videos by Date or Name.".to_string(),
|
|
||||||
// systemImage: "list.number".to_string(),
|
|
||||||
// colorName: "blue".to_string(),
|
|
||||||
// options: vec![
|
|
||||||
// FilterOption {
|
|
||||||
// id: "new".to_string(),
|
|
||||||
// title: "New".to_string(),
|
|
||||||
// },
|
|
||||||
// FilterOption {
|
|
||||||
// id: "popular".to_string(),
|
|
||||||
// title: "Popular".to_string(),
|
|
||||||
// },
|
|
||||||
// FilterOption {
|
|
||||||
// id: "top-rated".to_string(),
|
|
||||||
// title: "Top Rated".to_string(),
|
|
||||||
// },
|
|
||||||
// ],
|
|
||||||
// multiSelect: false,
|
|
||||||
// }],
|
|
||||||
// nsfw: true,
|
|
||||||
// cacheDuration: Some(1800),
|
|
||||||
// });
|
|
||||||
|
|
||||||
// xxthots
|
// xxthots
|
||||||
status.add_channel(Channel {
|
status.add_channel(Channel {
|
||||||
id: "xxthots".to_string(),
|
id: "xxthots".to_string(),
|
||||||
@@ -852,20 +789,6 @@ async fn status(req: HttpRequest) -> Result<impl web::Responder, web::Error> {
|
|||||||
cacheDuration: None,
|
cacheDuration: None,
|
||||||
});
|
});
|
||||||
|
|
||||||
// noodlemagazine
|
|
||||||
// status.add_channel(Channel {
|
|
||||||
// id: "noodlemagazine".to_string(),
|
|
||||||
// name: "Noodlemagazine".to_string(),
|
|
||||||
// description: "Discover the Best Adult Videos".to_string(),
|
|
||||||
// premium: false,
|
|
||||||
// favicon: "https://www.google.com/s2/favicons?sz=64&domain=noodlemagazine.com".to_string(),
|
|
||||||
// status: "active".to_string(),
|
|
||||||
// categories: vec![],
|
|
||||||
// options: vec![],
|
|
||||||
// nsfw: true,
|
|
||||||
// cacheDuration: Some(1800),
|
|
||||||
// });
|
|
||||||
|
|
||||||
//missav
|
//missav
|
||||||
status.add_channel(Channel {
|
status.add_channel(Channel {
|
||||||
id: "missav".to_string(),
|
id: "missav".to_string(),
|
||||||
@@ -1065,7 +988,9 @@ async fn status(req: HttpRequest) -> Result<impl web::Responder, web::Error> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for provider in ALL_PROVIDERS.values() {
|
for provider in ALL_PROVIDERS.values() {
|
||||||
status.add_channel(provider.get_channel(clientversion.clone()));
|
if let Some(channel) = provider.get_channel(clientversion.clone()){
|
||||||
|
status.add_channel(channel);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
status.iconUrl = format!("http://{}/favicon.ico", host).to_string();
|
status.iconUrl = format!("http://{}/favicon.ico", host).to_string();
|
||||||
Ok(web::HttpResponse::Ok().json(&status))
|
Ok(web::HttpResponse::Ok().json(&status))
|
||||||
@@ -1261,3 +1186,19 @@ pub fn get_provider(channel: &str) -> Option<DynProvider> {
|
|||||||
x => ALL_PROVIDERS.get(x).cloned(),
|
x => ALL_PROVIDERS.get(x).cloned(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn test() -> Result<impl web::Responder, web::Error> {
|
||||||
|
// Simply await the function instead of blocking the thread
|
||||||
|
let e = io::Error::new(io::ErrorKind::Other, "test error");
|
||||||
|
let _ = send_discord_error_report(
|
||||||
|
e.to_string(),
|
||||||
|
Some("chain_str".to_string()),
|
||||||
|
Some("Context"),
|
||||||
|
Some("xtra info"),
|
||||||
|
file!(),
|
||||||
|
line!(),
|
||||||
|
module_path!(),
|
||||||
|
).await;
|
||||||
|
|
||||||
|
Ok(web::HttpResponse::Ok())
|
||||||
|
}
|
||||||
@@ -94,11 +94,11 @@ impl Provider for AllProvider {
|
|||||||
return video_items;
|
return video_items;
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self,clientversion:ClientVersion) -> Channel {
|
fn get_channel(&self,clientversion:ClientVersion) -> Option<Channel> {
|
||||||
println!("Getting channel for placeholder with client version: {:?}",clientversion);
|
println!("Getting channel for placeholder with client version: {:?}",clientversion);
|
||||||
let _ = clientversion;
|
let _ = clientversion;
|
||||||
Channel {
|
Some(Channel {
|
||||||
id:"placeholder".to_string(),name:"PLACEHOLDER".to_string(),description:"PLACEHOLDER FOR PARENT CLASS".to_string(),premium:false,favicon:"https://www.google.com/s2/favicons?sz=64&domain=missav.ws".to_string(),status:"active".to_string(),categories:vec![],options:vec![],nsfw:true,cacheDuration:None,
|
id:"placeholder".to_string(),name:"PLACEHOLDER".to_string(),description:"PLACEHOLDER FOR PARENT CLASS".to_string(),premium:false,favicon:"https://www.google.com/s2/favicons?sz=64&domain=missav.ws".to_string(),status:"active".to_string(),categories:vec![],options:vec![],nsfw:true,cacheDuration:None,
|
||||||
}
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ use crate::api::ClientVersion;
|
|||||||
use crate::providers::Provider;
|
use crate::providers::Provider;
|
||||||
use crate::util::cache::VideoCache;
|
use crate::util::cache::VideoCache;
|
||||||
use crate::util::parse_abbreviated_number;
|
use crate::util::parse_abbreviated_number;
|
||||||
use crate::util::time::parse_time_to_seconds;
|
|
||||||
use crate::videos::{ServerOptions, VideoItem};
|
use crate::videos::{ServerOptions, VideoItem};
|
||||||
use crate::{status::*, util};
|
use crate::{status::*, util};
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
@@ -311,7 +310,7 @@ impl BeegProvider {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let id = file.get("id").and_then(|v| v.as_i64()).unwrap_or(0).to_string();
|
let id = file.get("id").and_then(|v| v.as_i64()).unwrap_or(0).to_string();
|
||||||
let title = video
|
let title = file
|
||||||
.get("data")
|
.get("data")
|
||||||
.and_then(|v| v.get(0))
|
.and_then(|v| v.get(0))
|
||||||
.and_then(|v| v.get("cd_value"))
|
.and_then(|v| v.get("cd_value"))
|
||||||
@@ -321,8 +320,7 @@ impl BeegProvider {
|
|||||||
|
|
||||||
let duration = file
|
let duration = file
|
||||||
.get("fl_duration")
|
.get("fl_duration")
|
||||||
.and_then(|v| v.as_str())
|
.and_then(|v| v.as_u64())
|
||||||
.and_then(|s| parse_time_to_seconds(s))
|
|
||||||
.unwrap_or(0);
|
.unwrap_or(0);
|
||||||
|
|
||||||
let views = video
|
let views = video
|
||||||
@@ -378,7 +376,7 @@ impl Provider for BeegProvider {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -192,11 +192,11 @@ impl Provider for FreshpornoProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self,clientversion:ClientVersion) -> Channel {
|
fn get_channel(&self,clientversion:ClientVersion) -> Option<Channel> {
|
||||||
println!("Getting channel for placeholder with client version: {:?}",clientversion);
|
println!("Getting channel for placeholder with client version: {:?}",clientversion);
|
||||||
let _ = clientversion;
|
let _ = clientversion;
|
||||||
Channel {
|
Some(Channel {
|
||||||
id:"placeholder".to_string(),name:"PLACEHOLDER".to_string(),description:"PLACEHOLDER FOR PARENT CLASS".to_string(),premium:false,favicon:"https://www.google.com/s2/favicons?sz=64&domain=missav.ws".to_string(),status:"active".to_string(),categories:vec![],options:vec![],nsfw:true,cacheDuration:None,
|
id:"placeholder".to_string(),name:"PLACEHOLDER".to_string(),description:"PLACEHOLDER FOR PARENT CLASS".to_string(),premium:false,favicon:"https://www.google.com/s2/favicons?sz=64&domain=missav.ws".to_string(),status:"active".to_string(),categories:vec![],options:vec![],nsfw:true,cacheDuration:None,
|
||||||
}
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -458,7 +458,7 @@ impl Provider for HqpornerProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> crate::status::Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<crate::status::Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,14 +3,14 @@ use crate::api::ClientVersion;
|
|||||||
use crate::providers::Provider;
|
use crate::providers::Provider;
|
||||||
use crate::status::*;
|
use crate::status::*;
|
||||||
use crate::util::cache::VideoCache;
|
use crate::util::cache::VideoCache;
|
||||||
use crate::util::discord::send_discord_error_report;
|
use crate::util::discord::{format_error_chain, send_discord_error_report};
|
||||||
use crate::util::requester::Requester;
|
use crate::util::requester::Requester;
|
||||||
use crate::util::time::parse_time_to_seconds;
|
use crate::util::time::parse_time_to_seconds;
|
||||||
use crate::videos::{ServerOptions, VideoItem};
|
use crate::videos::{ServerOptions, VideoItem};
|
||||||
|
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use error_chain::error_chain;
|
use error_chain::error_chain;
|
||||||
use htmlentity::entity::{decode, ICodedDataTrait};
|
use htmlentity::entity::{ICodedDataTrait, decode};
|
||||||
use std::sync::{Arc, RwLock};
|
use std::sync::{Arc, RwLock};
|
||||||
use std::{thread, vec};
|
use std::{thread, vec};
|
||||||
use titlecase::Titlecase;
|
use titlecase::Titlecase;
|
||||||
@@ -50,7 +50,7 @@ impl HypnotubeProvider {
|
|||||||
let url = self.url.clone();
|
let url = self.url.clone();
|
||||||
let categories = Arc::clone(&self.categories);
|
let categories = Arc::clone(&self.categories);
|
||||||
|
|
||||||
thread::spawn(move || {
|
thread::spawn(async move || {
|
||||||
let rt = match tokio::runtime::Builder::new_current_thread()
|
let rt = match tokio::runtime::Builder::new_current_thread()
|
||||||
.enable_all()
|
.enable_all()
|
||||||
.build()
|
.build()
|
||||||
@@ -58,14 +58,16 @@ impl HypnotubeProvider {
|
|||||||
Ok(rt) => rt,
|
Ok(rt) => rt,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
eprintln!("tokio runtime failed: {e}");
|
eprintln!("tokio runtime failed: {e}");
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
send_discord_error_report(
|
||||||
&e,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("HypnoTube Provider"),
|
Some("HypnoTube Provider"),
|
||||||
Some("Failed to create tokio runtime"),
|
Some("Failed to create tokio runtime"),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
)
|
||||||
|
.await;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -74,13 +76,15 @@ impl HypnotubeProvider {
|
|||||||
if let Err(e) = Self::load_categories(&url, Arc::clone(&categories)).await {
|
if let Err(e) = Self::load_categories(&url, Arc::clone(&categories)).await {
|
||||||
eprintln!("load_categories failed: {e}");
|
eprintln!("load_categories failed: {e}");
|
||||||
send_discord_error_report(
|
send_discord_error_report(
|
||||||
&e,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("HypnoTube Provider"),
|
Some("HypnoTube Provider"),
|
||||||
Some("Failed to load categories during initial load"),
|
Some("Failed to load categories during initial load"),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
).await;
|
)
|
||||||
|
.await;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
@@ -89,10 +93,7 @@ impl HypnotubeProvider {
|
|||||||
async fn load_categories(base: &str, cats: Arc<RwLock<Vec<FilterOption>>>) -> Result<()> {
|
async fn load_categories(base: &str, cats: Arc<RwLock<Vec<FilterOption>>>) -> Result<()> {
|
||||||
let mut requester = Requester::new();
|
let mut requester = Requester::new();
|
||||||
let text = requester
|
let text = requester
|
||||||
.get(
|
.get(&format!("{base}/channels/"), Some(Version::HTTP_11))
|
||||||
&format!("{base}/channels/"),
|
|
||||||
Some(Version::HTTP_11),
|
|
||||||
)
|
|
||||||
.await
|
.await
|
||||||
.map_err(|e| Error::from(format!("{}", e)))?;
|
.map_err(|e| Error::from(format!("{}", e)))?;
|
||||||
|
|
||||||
@@ -140,8 +141,7 @@ impl HypnotubeProvider {
|
|||||||
.iter()
|
.iter()
|
||||||
.map(|c| c.title.clone())
|
.map(|c| c.title.clone())
|
||||||
.collect(),
|
.collect(),
|
||||||
options: vec![
|
options: vec![ChannelOption {
|
||||||
ChannelOption {
|
|
||||||
id: "sort".to_string(),
|
id: "sort".to_string(),
|
||||||
title: "Sort".to_string(),
|
title: "Sort".to_string(),
|
||||||
description: "Sort the Videos".to_string(),
|
description: "Sort the Videos".to_string(),
|
||||||
@@ -193,10 +193,7 @@ impl HypnotubeProvider {
|
|||||||
"longest" => "longest",
|
"longest" => "longest",
|
||||||
_ => "videos",
|
_ => "videos",
|
||||||
};
|
};
|
||||||
let video_url = format!(
|
let video_url = format!("{}/{}/page{}.html", self.url, sort_string, page);
|
||||||
"{}/{}/page{}.html",
|
|
||||||
self.url, sort_string, page
|
|
||||||
);
|
|
||||||
let old_items = match cache.get(&video_url) {
|
let old_items = match cache.get(&video_url) {
|
||||||
Some((time, items)) => {
|
Some((time, items)) => {
|
||||||
if time.elapsed().unwrap_or_default().as_secs() < 60 * 5 {
|
if time.elapsed().unwrap_or_default().as_secs() < 60 * 5 {
|
||||||
@@ -210,14 +207,15 @@ impl HypnotubeProvider {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
let mut requester = options.requester.clone().unwrap();
|
let mut requester = options.requester.clone().unwrap();
|
||||||
let text = requester.get(&video_url, Some(Version::HTTP_11)).await.unwrap();
|
let text = requester
|
||||||
|
.get(&video_url, Some(Version::HTTP_11))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
if text.contains("Sorry, no results were found.") {
|
if text.contains("Sorry, no results were found.") {
|
||||||
eprint!("Hypnotube query returned no results for page {}", page);
|
eprint!("Hypnotube query returned no results for page {}", page);
|
||||||
return vec![];
|
return vec![];
|
||||||
}
|
}
|
||||||
let video_items: Vec<VideoItem> = self
|
let video_items: Vec<VideoItem> = self.get_video_items_from_html(text.clone()).await;
|
||||||
.get_video_items_from_html(text.clone())
|
|
||||||
.await;
|
|
||||||
if !video_items.is_empty() {
|
if !video_items.is_empty() {
|
||||||
cache.remove(&video_url);
|
cache.remove(&video_url);
|
||||||
cache.insert(video_url.clone(), video_items.clone());
|
cache.insert(video_url.clone(), video_items.clone());
|
||||||
@@ -242,7 +240,10 @@ impl HypnotubeProvider {
|
|||||||
};
|
};
|
||||||
let video_url = format!(
|
let video_url = format!(
|
||||||
"{}/search/videos/{}/{}/page{}.html",
|
"{}/search/videos/{}/{}/page{}.html",
|
||||||
self.url, query.trim().replace(" ","%20"), sort_string, page
|
self.url,
|
||||||
|
query.trim().replace(" ", "%20"),
|
||||||
|
sort_string,
|
||||||
|
page
|
||||||
);
|
);
|
||||||
// Check our Video Cache. If the result is younger than 1 hour, we return it.
|
// Check our Video Cache. If the result is younger than 1 hour, we return it.
|
||||||
let old_items = match cache.get(&video_url) {
|
let old_items = match cache.get(&video_url) {
|
||||||
@@ -260,7 +261,17 @@ impl HypnotubeProvider {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let mut requester = options.requester.clone().unwrap();
|
let mut requester = options.requester.clone().unwrap();
|
||||||
let text = match requester.post(format!("{}/searchgate.php", self.url).as_str(), format!("q={}&type=videos", query.replace(" ","+")).as_str(), vec![("Content-Type", "application/x-www-form-urlencoded")]).await.unwrap().text().await {
|
let text = match requester
|
||||||
|
.post(
|
||||||
|
format!("{}/searchgate.php", self.url).as_str(),
|
||||||
|
format!("q={}&type=videos", query.replace(" ", "+")).as_str(),
|
||||||
|
vec![("Content-Type", "application/x-www-form-urlencoded")],
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.text()
|
||||||
|
.await
|
||||||
|
{
|
||||||
Ok(t) => t,
|
Ok(t) => t,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
eprint!("Hypnotube search POST request failed: {}", e);
|
eprint!("Hypnotube search POST request failed: {}", e);
|
||||||
@@ -273,9 +284,7 @@ impl HypnotubeProvider {
|
|||||||
eprint!("Hypnotube query returned no results for page {}", page);
|
eprint!("Hypnotube query returned no results for page {}", page);
|
||||||
return vec![];
|
return vec![];
|
||||||
}
|
}
|
||||||
let video_items: Vec<VideoItem> = self
|
let video_items: Vec<VideoItem> = self.get_video_items_from_html(text.clone()).await;
|
||||||
.get_video_items_from_html(text.clone())
|
|
||||||
.await;
|
|
||||||
if !video_items.is_empty() {
|
if !video_items.is_empty() {
|
||||||
cache.remove(&video_url);
|
cache.remove(&video_url);
|
||||||
cache.insert(video_url.clone(), video_items.clone());
|
cache.insert(video_url.clone(), video_items.clone());
|
||||||
@@ -285,10 +294,7 @@ impl HypnotubeProvider {
|
|||||||
video_items
|
video_items
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_video_items_from_html(
|
async fn get_video_items_from_html(&self, html: String) -> Vec<VideoItem> {
|
||||||
&self,
|
|
||||||
html: String
|
|
||||||
) -> Vec<VideoItem> {
|
|
||||||
if html.is_empty() || html.contains("404 Not Found") {
|
if html.is_empty() || html.contains("404 Not Found") {
|
||||||
eprint!("Hypnotube returned empty or 404 html");
|
eprint!("Hypnotube returned empty or 404 html");
|
||||||
return vec![];
|
return vec![];
|
||||||
@@ -302,36 +308,43 @@ impl HypnotubeProvider {
|
|||||||
Some(b) => b,
|
Some(b) => b,
|
||||||
None => {
|
None => {
|
||||||
eprint!("Hypnotube Provider: Failed to get block from html");
|
eprint!("Hypnotube Provider: Failed to get block from html");
|
||||||
let err = Error::from(ErrorKind::Parse("html".into()));
|
let e = Error::from(ErrorKind::Parse("html".into()));
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
send_discord_error_report(
|
||||||
&err,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Hypnotube Provider"),
|
Some("Hypnotube Provider"),
|
||||||
Some(&format!("Failed to get block from html:\n```{html}\n```")),
|
Some(&format!("Failed to get block from html:\n```{html}\n```")),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
)
|
||||||
|
.await;
|
||||||
return vec![];
|
return vec![];
|
||||||
},
|
}
|
||||||
};
|
};
|
||||||
let mut items = vec![];
|
let mut items = vec![];
|
||||||
for seg in block.split("<!-- item -->").skip(1) {
|
for seg in block.split("<!-- item -->").skip(1) {
|
||||||
let video_url = match seg
|
let video_url = match seg
|
||||||
.split(" href=\"")
|
.split(" href=\"")
|
||||||
.nth(1)
|
.nth(1)
|
||||||
.and_then(|s| s.split('"').next()) {
|
.and_then(|s| s.split('"').next())
|
||||||
|
{
|
||||||
Some(url) => url.to_string(),
|
Some(url) => url.to_string(),
|
||||||
None => {
|
None => {
|
||||||
eprint!("Hypnotube Provider: Failed to parse video url from segment");
|
eprint!("Hypnotube Provider: Failed to parse video url from segment");
|
||||||
let err = Error::from(ErrorKind::Parse("video url".into()));
|
let e = Error::from(ErrorKind::Parse("video url".into()));
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
send_discord_error_report(
|
||||||
&err,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Hypnotube Provider"),
|
Some("Hypnotube Provider"),
|
||||||
Some(&format!("Failed to parse video url from segment:\n```{seg}\n```")),
|
Some(&format!(
|
||||||
|
"Failed to parse video url from segment:\n```{seg}\n```"
|
||||||
|
)),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
)
|
||||||
|
.await;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -340,9 +353,14 @@ impl HypnotubeProvider {
|
|||||||
.split(" title=\"")
|
.split(" title=\"")
|
||||||
.nth(1)
|
.nth(1)
|
||||||
.and_then(|s| s.split('"').next())
|
.and_then(|s| s.split('"').next())
|
||||||
.unwrap_or_default().trim().to_string();
|
.unwrap_or_default()
|
||||||
|
.trim()
|
||||||
|
.to_string();
|
||||||
|
|
||||||
title = decode(title.clone().as_bytes()).to_string().unwrap_or(title).titlecase();
|
title = decode(title.clone().as_bytes())
|
||||||
|
.to_string()
|
||||||
|
.unwrap_or(title)
|
||||||
|
.titlecase();
|
||||||
let id = video_url
|
let id = video_url
|
||||||
.split('/')
|
.split('/')
|
||||||
.nth(4)
|
.nth(4)
|
||||||
@@ -411,7 +429,7 @@ impl Provider for HypnotubeProvider {
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, v: ClientVersion) -> Channel {
|
fn get_channel(&self, v: ClientVersion) -> Option<Channel> {
|
||||||
self.build_channel(v)
|
Some(self.build_channel(v))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ use crate::api::ClientVersion;
|
|||||||
use crate::providers::Provider;
|
use crate::providers::Provider;
|
||||||
use crate::status::*;
|
use crate::status::*;
|
||||||
use crate::util::cache::VideoCache;
|
use crate::util::cache::VideoCache;
|
||||||
use crate::util::discord::send_discord_error_report;
|
use crate::util::discord::{format_error_chain, send_discord_error_report};
|
||||||
use crate::util::requester::Requester;
|
use crate::util::requester::Requester;
|
||||||
use crate::util::time::parse_time_to_seconds;
|
use crate::util::time::parse_time_to_seconds;
|
||||||
use crate::videos::{ServerOptions, VideoFormat, VideoItem};
|
use crate::videos::{ServerOptions, VideoFormat, VideoItem};
|
||||||
@@ -215,15 +215,16 @@ impl JavtifulProvider {
|
|||||||
Some(b) => b,
|
Some(b) => b,
|
||||||
None => {
|
None => {
|
||||||
eprint!("Javtiful Provider: Failed to get block from html");
|
eprint!("Javtiful Provider: Failed to get block from html");
|
||||||
let err = Error::from(ErrorKind::Parse("html".into()));
|
let e = Error::from(ErrorKind::Parse("html".into()));
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
send_discord_error_report(
|
||||||
&err,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Javtiful Provider"),
|
Some("Javtiful Provider"),
|
||||||
Some(&format!("Failed to get block from html:\n```{html}\n```")),
|
Some(&format!("Failed to get block from html:\n```{html}\n```")),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
).await;
|
||||||
return vec![]
|
return vec![]
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
@@ -239,14 +240,22 @@ impl JavtifulProvider {
|
|||||||
.inspect(|r| {
|
.inspect(|r| {
|
||||||
if let Err(e) = r {
|
if let Err(e) = r {
|
||||||
eprint!("Javtiful Provider: Failed to get video item:{}\n", e);
|
eprint!("Javtiful Provider: Failed to get video item:{}\n", e);
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
// Prepare data to move into the background task
|
||||||
&e,
|
let msg = e.to_string();
|
||||||
|
let chain = format_error_chain(&e);
|
||||||
|
|
||||||
|
// Spawn the report into the background - NO .await here
|
||||||
|
tokio::spawn(async move {
|
||||||
|
let _ = send_discord_error_report(
|
||||||
|
msg,
|
||||||
|
Some(chain),
|
||||||
Some("Javtiful Provider"),
|
Some("Javtiful Provider"),
|
||||||
Some("Failed to get video item"),
|
Some("Failed to get video item"),
|
||||||
file!(),
|
file!(), // Note: these might report the utility line
|
||||||
line!(),
|
line!(), // better to hardcode or pass from outside
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
).await;
|
||||||
|
});
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
.filter_map(Result::ok)
|
.filter_map(Result::ok)
|
||||||
@@ -400,7 +409,7 @@ impl Provider for JavtifulProvider {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, v: ClientVersion) -> Channel {
|
fn get_channel(&self, v: ClientVersion) -> Option<Channel> {
|
||||||
self.build_channel(v)
|
Some(self.build_channel(v))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -82,13 +82,13 @@ pub trait Provider: Send + Sync {
|
|||||||
options: ServerOptions,
|
options: ServerOptions,
|
||||||
) -> Vec<VideoItem>;
|
) -> Vec<VideoItem>;
|
||||||
|
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<Channel> {
|
||||||
println!(
|
println!(
|
||||||
"Getting channel for placeholder with client version: {:?}",
|
"Getting channel for placeholder with client version: {:?}",
|
||||||
clientversion
|
clientversion
|
||||||
);
|
);
|
||||||
let _ = clientversion;
|
let _ = clientversion;
|
||||||
Channel {
|
Some(Channel {
|
||||||
id: "placeholder".to_string(),
|
id: "placeholder".to_string(),
|
||||||
name: "PLACEHOLDER".to_string(),
|
name: "PLACEHOLDER".to_string(),
|
||||||
description: "PLACEHOLDER FOR PARENT CLASS".to_string(),
|
description: "PLACEHOLDER FOR PARENT CLASS".to_string(),
|
||||||
@@ -99,6 +99,6 @@ pub trait Provider: Send + Sync {
|
|||||||
options: vec![],
|
options: vec![],
|
||||||
nsfw: true,
|
nsfw: true,
|
||||||
cacheDuration: None,
|
cacheDuration: None,
|
||||||
}
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -302,7 +302,7 @@ impl Provider for NoodlemagazineProvider {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -621,7 +621,7 @@ impl Provider for OmgxxxProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> crate::status::Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<crate::status::Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ use crate::api::ClientVersion;
|
|||||||
use crate::providers::Provider;
|
use crate::providers::Provider;
|
||||||
use crate::status::*;
|
use crate::status::*;
|
||||||
use crate::util::cache::VideoCache;
|
use crate::util::cache::VideoCache;
|
||||||
use crate::util::discord::send_discord_error_report;
|
use crate::util::discord::{format_error_chain, send_discord_error_report};
|
||||||
use crate::util::requester::Requester;
|
use crate::util::requester::Requester;
|
||||||
use crate::util::time::parse_time_to_seconds;
|
use crate::util::time::parse_time_to_seconds;
|
||||||
use crate::videos::{ServerOptions, VideoFormat, VideoItem};
|
use crate::videos::{ServerOptions, VideoFormat, VideoItem};
|
||||||
@@ -101,7 +101,7 @@ impl PimpbunnyProvider {
|
|||||||
let stars = Arc::clone(&self.stars);
|
let stars = Arc::clone(&self.stars);
|
||||||
let categories = Arc::clone(&self.categories);
|
let categories = Arc::clone(&self.categories);
|
||||||
|
|
||||||
thread::spawn(move || {
|
thread::spawn(async move || {
|
||||||
let rt = match tokio::runtime::Builder::new_current_thread()
|
let rt = match tokio::runtime::Builder::new_current_thread()
|
||||||
.enable_all()
|
.enable_all()
|
||||||
.build()
|
.build()
|
||||||
@@ -109,14 +109,15 @@ impl PimpbunnyProvider {
|
|||||||
Ok(rt) => rt,
|
Ok(rt) => rt,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
eprintln!("tokio runtime failed: {e}");
|
eprintln!("tokio runtime failed: {e}");
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
send_discord_error_report(
|
||||||
&e,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Pimpbunny Provider"),
|
Some("Pimpbunny Provider"),
|
||||||
Some("Failed to create tokio runtime"),
|
Some("Failed to create tokio runtime"),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
).await;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -125,7 +126,8 @@ impl PimpbunnyProvider {
|
|||||||
if let Err(e) = Self::load_stars(&url, Arc::clone(&stars)).await {
|
if let Err(e) = Self::load_stars(&url, Arc::clone(&stars)).await {
|
||||||
eprintln!("load_stars failed: {e}");
|
eprintln!("load_stars failed: {e}");
|
||||||
send_discord_error_report(
|
send_discord_error_report(
|
||||||
&e,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Pimpbunny Provider"),
|
Some("Pimpbunny Provider"),
|
||||||
Some("Failed to load stars during initial load"),
|
Some("Failed to load stars during initial load"),
|
||||||
file!(),
|
file!(),
|
||||||
@@ -136,7 +138,8 @@ impl PimpbunnyProvider {
|
|||||||
if let Err(e) = Self::load_categories(&url, Arc::clone(&categories)).await {
|
if let Err(e) = Self::load_categories(&url, Arc::clone(&categories)).await {
|
||||||
eprintln!("load_categories failed: {e}");
|
eprintln!("load_categories failed: {e}");
|
||||||
send_discord_error_report(
|
send_discord_error_report(
|
||||||
&e,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Pimpbunny Provider"),
|
Some("Pimpbunny Provider"),
|
||||||
Some("Failed to load categories during initial load"),
|
Some("Failed to load categories during initial load"),
|
||||||
file!(),
|
file!(),
|
||||||
@@ -529,7 +532,7 @@ impl Provider for PimpbunnyProvider {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, v: ClientVersion) -> Channel {
|
fn get_channel(&self, v: ClientVersion) -> Option<Channel> {
|
||||||
self.build_channel(v)
|
Some(self.build_channel(v))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ use error_chain::error_chain;
|
|||||||
use htmlentity::entity::{decode, ICodedDataTrait};
|
use htmlentity::entity::{decode, ICodedDataTrait};
|
||||||
use std::sync::{Arc, RwLock};
|
use std::sync::{Arc, RwLock};
|
||||||
use std::vec;
|
use std::vec;
|
||||||
|
use std::fmt::Write;
|
||||||
|
|
||||||
error_chain! {
|
error_chain! {
|
||||||
foreign_links {
|
foreign_links {
|
||||||
@@ -277,20 +278,25 @@ impl Provider for PmvhavenProvider {
|
|||||||
Ok(v) => v,
|
Ok(v) => v,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
eprintln!("pmvhaven error: {e}");
|
eprintln!("pmvhaven error: {e}");
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
let mut chain_str = String::new();
|
||||||
&e,
|
for (i, cause) in e.iter().enumerate() {
|
||||||
|
let _ = writeln!(chain_str, "{}. {}", i + 1, cause);
|
||||||
|
}
|
||||||
|
send_discord_error_report(
|
||||||
|
e.to_string(),
|
||||||
|
Some(chain_str),
|
||||||
Some("PMVHaven Provider"),
|
Some("PMVHaven Provider"),
|
||||||
Some("Failed to load videos from PMVHaven"),
|
Some("Failed to load videos from PMVHaven"),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),
|
module_path!(),
|
||||||
));
|
).await;
|
||||||
vec![]
|
vec![]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -265,7 +265,7 @@ impl Provider for PornxpProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> crate::status::Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<crate::status::Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -282,7 +282,7 @@ impl Provider for Rule34genProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> crate::status::Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<crate::status::Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
use crate::DbPool;
|
use crate::DbPool;
|
||||||
use crate::providers::Provider;
|
use crate::providers::Provider;
|
||||||
use crate::util::cache::VideoCache;
|
use crate::util::cache::VideoCache;
|
||||||
|
use crate::util::discord::format_error_chain;
|
||||||
use crate::util::discord::send_discord_error_report;
|
use crate::util::discord::send_discord_error_report;
|
||||||
use crate::util::requester::Requester;
|
use crate::util::requester::Requester;
|
||||||
use crate::util::time::parse_time_to_seconds;
|
use crate::util::time::parse_time_to_seconds;
|
||||||
@@ -26,14 +27,6 @@ error_chain! {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// fn has_blacklisted_class(element: &ElementRef, blacklist: &[&str]) -> bool {
|
|
||||||
// element
|
|
||||||
// .value()
|
|
||||||
// .attr("class")
|
|
||||||
// .map(|classes| classes.split_whitespace().any(|c| blacklist.contains(&c)))
|
|
||||||
// .unwrap_or(false)
|
|
||||||
// }
|
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct SxyprnProvider {
|
pub struct SxyprnProvider {
|
||||||
url: String,
|
url: String,
|
||||||
@@ -97,6 +90,15 @@ impl SxyprnProvider {
|
|||||||
Ok(items) => items,
|
Ok(items) => items,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
println!("Error parsing video items: {}", e);
|
println!("Error parsing video items: {}", e);
|
||||||
|
send_discord_error_report(
|
||||||
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
|
Some("Sxyprn Provider"),
|
||||||
|
Some(&format!("URL: {}", url_str)),
|
||||||
|
file!(),
|
||||||
|
line!(),
|
||||||
|
module_path!(),
|
||||||
|
).await;
|
||||||
return Ok(old_items);
|
return Ok(old_items);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -159,14 +161,16 @@ impl SxyprnProvider {
|
|||||||
{
|
{
|
||||||
Ok(items) => items,
|
Ok(items) => items,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
println!("Error parsing video items: {}", e);
|
println!("Error parsing video items: {}", e);// 1. Convert the error to a string immediately
|
||||||
let _ = futures::executor::block_on(send_discord_error_report(
|
send_discord_error_report(
|
||||||
&e,
|
e.to_string(),
|
||||||
|
Some(format_error_chain(&e)),
|
||||||
Some("Sxyprn Provider"),
|
Some("Sxyprn Provider"),
|
||||||
Some(format!("Failed to query videos:\nURL: {}\nQuery: {},", url_str, query).as_str()),
|
Some(&format!("URL: {}", url_str)),
|
||||||
file!(),
|
file!(),
|
||||||
line!(),
|
line!(),
|
||||||
module_path!(),));
|
module_path!(),
|
||||||
|
).await;
|
||||||
return Ok(old_items);
|
return Ok(old_items);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -223,7 +227,7 @@ impl SxyprnProvider {
|
|||||||
let title_parts = video_segment
|
let title_parts = video_segment
|
||||||
.split("post_text")
|
.split("post_text")
|
||||||
.nth(1)
|
.nth(1)
|
||||||
.and_then(|s| s.split("style=''>").nth(1))
|
.and_then(|s| s.split("style=''>").nth(100000))
|
||||||
.and_then(|s| s.split("</div>").next())
|
.and_then(|s| s.split("</div>").next())
|
||||||
.ok_or_else(|| ErrorKind::Parse("failed to extract title_parts".into()))?;
|
.ok_or_else(|| ErrorKind::Parse("failed to extract title_parts".into()))?;
|
||||||
|
|
||||||
|
|||||||
@@ -554,7 +554,7 @@ impl Provider for TnaflixProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> crate::status::Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<crate::status::Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -278,7 +278,7 @@ impl Provider for XxdbxProvider {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn get_channel(&self, clientversion: ClientVersion) -> crate::status::Channel {
|
fn get_channel(&self, clientversion: ClientVersion) -> Option<crate::status::Channel> {
|
||||||
self.build_channel(clientversion)
|
Some(self.build_channel(clientversion))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,35 +4,36 @@ use std::time::{SystemTime, UNIX_EPOCH};
|
|||||||
|
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
|
|
||||||
/// Send a detailed error report to a Discord webhook
|
use crate::util::requester;
|
||||||
pub async fn send_discord_error_report<T: Error>(
|
|
||||||
error: &T,
|
pub fn format_error_chain(err: &dyn Error) -> String {
|
||||||
context: Option<&str>, // e.g. provider name, URL, query
|
let mut chain_str = String::new();
|
||||||
extra_info: Option<&str>, // any debug info you want
|
let mut current_err: Option<&dyn Error> = Some(err);
|
||||||
|
let mut index = 1;
|
||||||
|
while let Some(e) = current_err {
|
||||||
|
let _ = writeln!(chain_str, "{}. {}", index, e);
|
||||||
|
current_err = e.source();
|
||||||
|
index += 1;
|
||||||
|
}
|
||||||
|
chain_str
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn send_discord_error_report(
|
||||||
|
error_msg: String,
|
||||||
|
error_chain: Option<String>,
|
||||||
|
context: Option<&str>,
|
||||||
|
extra_info: Option<&str>,
|
||||||
file: &str,
|
file: &str,
|
||||||
line: u32,
|
line: u32,
|
||||||
module: &str,
|
module: &str,
|
||||||
) {
|
) {
|
||||||
|
// 1. Get Webhook from Environment
|
||||||
let webhook_url = match std::env::var("DISCORD_WEBHOOK") {
|
let webhook_url = match std::env::var("DISCORD_WEBHOOK") {
|
||||||
Ok(url) => url,
|
Ok(url) => url,
|
||||||
Err(_) => return,
|
Err(_) => return, // Exit silently if no webhook is configured
|
||||||
};
|
};
|
||||||
|
|
||||||
// Discord embed field limits
|
|
||||||
const MAX_FIELD: usize = 1024;
|
const MAX_FIELD: usize = 1024;
|
||||||
|
|
||||||
let mut error_chain = String::new();
|
|
||||||
let mut current: &dyn Error = error;
|
|
||||||
let mut i = 0;
|
|
||||||
loop {
|
|
||||||
let _ = writeln!(error_chain, "{}. {}", i + 1, current);
|
|
||||||
i += 1;
|
|
||||||
match current.source() {
|
|
||||||
Some(src) => current = src,
|
|
||||||
None => break,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let truncate = |s: &str| {
|
let truncate = |s: &str| {
|
||||||
if s.len() > MAX_FIELD {
|
if s.len() > MAX_FIELD {
|
||||||
format!("{}…", &s[..MAX_FIELD - 1])
|
format!("{}…", &s[..MAX_FIELD - 1])
|
||||||
@@ -53,23 +54,23 @@ pub async fn send_discord_error_report<T: Error>(
|
|||||||
"fields": [
|
"fields": [
|
||||||
{
|
{
|
||||||
"name": "Error",
|
"name": "Error",
|
||||||
"value": truncate(&error.to_string()),
|
"value": format!("```{}```", truncate(&error_msg)),
|
||||||
"inline": false
|
"inline": false
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"name": "Error Chain",
|
"name": "Error Chain",
|
||||||
"value": truncate(&error_chain),
|
"value": truncate(&error_chain.unwrap_or_else(|| "No chain provided".to_string())),
|
||||||
"inline": false
|
"inline": false
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"name": "Location",
|
"name": "Location",
|
||||||
"value": format!("`{}`:{}\n`{}`", file, line, module),
|
"value": format!("`{}`:{}\n`{}`", file, line, module),
|
||||||
"inline": false
|
"inline": true
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"name": "Context",
|
"name": "Context",
|
||||||
"value": truncate(context.unwrap_or("n/a")),
|
"value": truncate(context.unwrap_or("n/a")),
|
||||||
"inline": false
|
"inline": true
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"name": "Extra Info",
|
"name": "Extra Info",
|
||||||
@@ -83,13 +84,7 @@ pub async fn send_discord_error_report<T: Error>(
|
|||||||
}]
|
}]
|
||||||
});
|
});
|
||||||
|
|
||||||
// Send (never panic)
|
let mut requester = requester::Requester::new();
|
||||||
if let Err(e) = wreq::Client::new()
|
let _ = requester.post_json(&webhook_url, &payload, vec![]).await;
|
||||||
.post(webhook_url)
|
println!("Error report sent.");
|
||||||
.json(&payload)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
eprintln!("Failed to send Discord error report: {e}");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
@@ -47,23 +47,6 @@ impl Requester {
|
|||||||
self.proxy = proxy;
|
self.proxy = proxy;
|
||||||
}
|
}
|
||||||
|
|
||||||
// pub fn set_flaresolverr_session(&mut self, session: String) {
|
|
||||||
// self.flaresolverr_session = Some(session);
|
|
||||||
// }
|
|
||||||
|
|
||||||
// fn get_url_from_location_header(&self, prev_url: &str, location: &str) -> String {
|
|
||||||
// if location.starts_with("http://") || location.starts_with("https://") {
|
|
||||||
// location.to_string()
|
|
||||||
// } else if location.starts_with("//") {
|
|
||||||
// format!("{}{}", "https:", location)
|
|
||||||
// } else if location.starts_with('/') {
|
|
||||||
// let base_url = prev_url.split('/').take(3).collect::<Vec<&str>>().join("/");
|
|
||||||
// format!("{}{}", base_url, location)
|
|
||||||
// } else {
|
|
||||||
// format!("{}/{}", prev_url, location)
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
pub async fn get_raw(&mut self, url: &str) -> Result<Response, wreq::Error> {
|
pub async fn get_raw(&mut self, url: &str) -> Result<Response, wreq::Error> {
|
||||||
let client = Client::builder()
|
let client = Client::builder()
|
||||||
.cert_verification(false)
|
.cert_verification(false)
|
||||||
|
|||||||
Reference in New Issue
Block a user