Fixing domain checking.

This commit is contained in:
Dessalines 2024-04-22 14:59:06 -04:00
parent e22909697b
commit 0e2b3952bf
2 changed files with 11 additions and 19 deletions

View file

@ -537,23 +537,15 @@ pub async fn get_url_blocklist(context: &LemmyContext) -> LemmyResult<RegexSet>
let urls = LocalSiteUrlBlocklist::get_all(&mut context.pool()).await?;
let regexes = urls.iter().map(|url| {
let url = &url.url;
let parsed = Url::parse(url).expect("Coundln't parse URL.");
if url.ends_with('/') {
format!(
"({}://)?{}{}?",
parsed.scheme(),
escape(parsed.domain().expect("No domain.")),
escape(parsed.path())
)
} else {
format!(
"({}://)?{}{}",
parsed.scheme(),
escape(parsed.domain().expect("No domain.")),
escape(parsed.path())
)
}
// The scheme is removed in the saving,
// so fake it here to build the url.
let url = &format!("https://{}", url.url);
let parsed = Url::parse(url).expect("Couldn't parse URL.");
format!(
"{}{}",
escape(parsed.host_str().expect("No domain.")),
escape(parsed.path())
)
});
let set = RegexSet::new(regexes)?;

View file

@ -624,12 +624,12 @@ mod tests {
"example.com".to_string(),
"http://example.com".to_string(),
"https://example.com".to_string(),
"https://example.blog/test?q=test2&q2=test3#test4".to_string(),
"https://example.com/test?q=test2&q2=test3#test4".to_string(),
])
.unwrap(),
&vec![
"example.com/".to_string(),
"example.blog/test?q=test2&q2=test3#test4".to_string()
"example.com/test?q=test2&q2=test3#test4".to_string()
],
);