kavin/sponsorblock-mirror
1
0
Fork 0
mirror of https://github.com/TeamPiped/sponsorblock-mirror.git synced 2024-12-13 05:40:36 +05:30
Code Issues Packages Projects Releases Wiki Activity

Unwrap the segments for the video ID endpoint

Browse Source
This commit is contained in:
Adam Novak 2022-10-30 13:06:44 -04:00
parent a9683b08a0
commit c1ef37ad77
1 changed files with 41 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
src/routes.rs
View File

@ -48,26 +48,25 @@ pub async fn skip_segments(
return content::RawJson("Hash prefix does not match format requirements.".to_string()); return content::RawJson("Hash prefix does not match format requirements.".to_string());
} }
let search_result = find_skip_segments(VideoName::ByHashPrefix(hash.clone()), categories, db).await; let sponsors = find_skip_segments(VideoName::ByHashPrefix(hash.clone()), categories, db).await;
match search_result { if sponsors.is_empty() {
Some(segments) => return segments, // Fall back to central Sponsorblock server
None => { let resp = reqwest::get(format!(
// Fall back to central Sponsorblock server "https://sponsor.ajay.app/api/skipSegments/{}?categories={}",
let resp = reqwest::get(format!( hash,
"https://sponsor.ajay.app/api/skipSegments/{}?categories={}", categories.unwrap_or("[]"),
hash, ))
categories.unwrap_or("[]"), .await
)) .unwrap()
.await .text()
.unwrap() .await
.text() .unwrap();
.await
.unwrap();
return content::RawJson(resp); return content::RawJson(resp);
}
} }
return content::RawJson(serde_json::to_string(&sponsors).unwrap());
} }
#[get("/api/skipSegments?<videoID>&<categories>")] #[get("/api/skipSegments?<videoID>&<categories>")]
@ -82,40 +81,39 @@ pub async fn skip_segments_by_id(
return content::RawJson("videoID is missing".to_string()); return content::RawJson("videoID is missing".to_string());
} }
let search_result = find_skip_segments(VideoName::ByID(videoID.clone()), categories, db).await; let sponsors = find_skip_segments(VideoName::ByID(videoID.clone()), categories, db).await;
match search_result { if sponsors.is_empty() {
Some(segments) => return segments, // Fall back to central Sponsorblock server
None => { let resp = reqwest::get(format!(
// Fall back to central Sponsorblock server "https://sponsor.ajay.app/api/skipSegments?videoID={}&categories={}",
let resp = reqwest::get(format!( videoID,
"https://sponsor.ajay.app/api/skipSegments?videoID={}&categories={}", categories.unwrap_or("[]"),
videoID, ))
categories.unwrap_or("[]"), .await
)) .unwrap()
.await .text()
.unwrap() .await
.text() .unwrap();
.await
.unwrap();
return content::RawJson(resp); return content::RawJson(resp);
}
} }
// Doing a lookup by video ID should return only one Sponsor object with
// one list of segments. We need to return just the list of segments.
return content::RawJson(serde_json::to_string(&sponsors[0].segments).unwrap());
} }
async fn find_skip_segments( async fn find_skip_segments(
name: VideoName, name: VideoName,
categories: Option<&str>, categories: Option<&str>,
db: Db, db: Db,
) -> Option<content::RawJson<String>> { ) -> Vec<Sponsor> {
let cat: Vec<String> = serde_json::from_str(categories.unwrap_or("[\"sponsor\"]")).unwrap(); let cat: Vec<String> = serde_json::from_str(categories.unwrap_or("[\"sponsor\"]")).unwrap();
if cat.is_empty() { if cat.is_empty() {
return Some(content::RawJson( return Vec::new();
"[]".to_string(),
));
} }
let results: Vec<SponsorTime> = db.run(move |conn| { let results: Vec<SponsorTime> = db.run(move |conn| {
@ -196,13 +194,8 @@ async fn find_skip_segments(
for sponsor in sponsors.values_mut() { for sponsor in sponsors.values_mut() {
sponsor.segments.sort_by(|a, b| a.partial_cmp(b).unwrap()); sponsor.segments.sort_by(|a, b| a.partial_cmp(b).unwrap());
} }
if !sponsors.is_empty() { return sponsors.into_values().collect();
let sponsors: Vec<&Sponsor> = sponsors.values().collect();
return Some(content::RawJson(serde_json::to_string(&sponsors).unwrap()));
}
return None;
} }
fn similar_segments(segment: &Segment, hash: &str, segments: &Vec<SponsorTime>) -> Vec<Segment> { fn similar_segments(segment: &Segment, hash: &str, segments: &Vec<SponsorTime>) -> Vec<Segment> {
@ -272,9 +265,9 @@ fn best_segment(segments: &Vec<Segment>) -> Segment {
best_segment best_segment
} }
// We might need some more routes to support ReVanced. These are faked for now. // These additional routes are faked to protect ReVanced from seeing errors. We
// Sadly not even these are sufficient to make it work for me, maybe it is just // don't *need* to do this to support ReVanced, but it gets rid of the
// broken? // perpetual "Loading..." in the settings.
// This would take a userID // This would take a userID
#[get("/api/isUserVIP")] #[get("/api/isUserVIP")]