leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

👌 Add parse file:// urls

Browse source
This commit is contained in:
Alexander WB 2025-02-25 03:42:41 +01:00
parent c19ea3979a
commit 4c6e19ceab
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
crates/librqbit/src/blocklist.rs
View file

@ -9,6 +9,7 @@ use tokio::io::AsyncRead;
use tokio::{io::AsyncBufReadExt, io::BufReader}; use tokio::{io::AsyncBufReadExt, io::BufReader};
use tokio_util::io::StreamReader; use tokio_util::io::StreamReader;
use tracing::{debug, info, trace}; use tracing::{debug, info, trace};
use url::Url;
pub struct Blocklist { pub struct Blocklist {
// ipv4 and ipv6 do not overlap // ipv4 and ipv6 do not overlap
@ -28,7 +29,16 @@ impl Blocklist {
} }
pub async fn load_from_url(url: &str) -> Result<Self> { pub async fn load_from_url(url: &str) -> Result<Self> {
let response = reqwest::get(url) let parsed_url = Url::parse(url).context("Failed to parse URL")?;
if parsed_url.scheme() == "file" {
let path = parsed_url
.to_file_path()
.map_err(|_| anyhow::anyhow!("Failed to convert file URL to path"))?;
return Self::load_from_file(path.to_str().unwrap()).await;
}
let response = reqwest::get(parsed_url)
.await .await
.context("Failed to send request for blocklist")?; .context("Failed to send request for blocklist")?;
if response.status() != 200 { if response.status() != 200 {