iridium/redsunlib
iridium/redsunlib
2
2
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 4 Wiki Activity

Fix obfuscated link handling

Browse Source
This commit is contained in:
Matthew Esposito 2023-12-28 18:21:07 -05:00
parent 878ef8e95e
commit ce0c6eca8a
No known key found for this signature in database
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/client.rs
View File

@ -54,8 +54,14 @@ pub async fn canonical_path(path: String) -> Result<Option<String>, String> {
// If Reddit responds with a 2xx, then the path is already canonical. // If Reddit responds with a 2xx, then the path is already canonical.
200..=299 => Ok(Some(path)), 200..=299 => Ok(Some(path)),
// If Reddit responds with anything other than 3xx (except for the 2xx as // If Reddit responds with a 301, then the path is redirected.
// above), return a None. 301 => match res.headers().get(header::LOCATION) {
Some(val) => Ok(Some(val.to_str().unwrap().to_string())),
None => Ok(None),
},
// If Reddit responds with anything other than 3xx (except for the 2xx and 301
// as above), return a None.
300..=399 => Ok(None), 300..=399 => Ok(None),
_ => Ok( _ => Ok(
@ -337,3 +343,10 @@ async fn test_localization_popular() {
let val = json("/r/popular/hot.json?&raw_json=1&geo_filter=GLOBAL".to_string(), false).await.unwrap(); let val = json("/r/popular/hot.json?&raw_json=1&geo_filter=GLOBAL".to_string(), false).await.unwrap();
assert_eq!("GLOBAL", val["data"]["geo_filter"].as_str().unwrap()); assert_eq!("GLOBAL", val["data"]["geo_filter"].as_str().unwrap());
} }
#[tokio::test(flavor = "multi_thread", worker_threads = 8)]
async fn test_obfuscated_share_link() {
let share_link = "/r/rust/s/kPgq8WNHRK".into();
let canonical_link = "https://www.reddit.com/r/rust/comments/18t5968/why_use_tuple_struct_over_standard_struct/kfbqlbc?share_id=N0wD38nOLSUMMNnWpDRO3&utm_content=2&utm_medium=android_app&utm_name=androidcss&utm_source=share&utm_term=1".into();
assert_eq!(canonical_path(share_link).await, Ok(Some(canonical_link)));
}