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

Add parameter for remote server to download client

Browse source
This commit is contained in:
Ivan 2024-10-19 19:47:57 +02:00
parent 946ad36258
commit 6d08173159
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
crates/rqbit/src/main.rs
View file

@ -301,6 +301,10 @@ struct DownloadOpts {
#[arg(long = "initial-peers")] #[arg(long = "initial-peers")]
initial_peers: Option<InitialPeers>, initial_peers: Option<InitialPeers>,
#[arg(long = "server-url")]
server_url: Option<String>,
} }
#[derive(Clone)] #[derive(Clone)]
@ -634,7 +638,7 @@ async fn async_main(opts: Opts, cancel: CancellationToken) -> anyhow::Result<()>
if download_opts.torrent_path.is_empty() { if download_opts.torrent_path.is_empty() {
anyhow::bail!("you must provide at least one URL to download") anyhow::bail!("you must provide at least one URL to download")
} }
let http_api_url = format!("http://{}", opts.http_api_listen_addr); let http_api_url = download_opts.server_url.clone().unwrap_or_else(|| format!("http://{}", opts.http_api_listen_addr));
let client = http_api_client::HttpApiClient::new(&http_api_url)?; let client = http_api_client::HttpApiClient::new(&http_api_url)?;
let torrent_opts = |with_output_folder: bool| AddTorrentOptions { let torrent_opts = |with_output_folder: bool| AddTorrentOptions {
@ -662,6 +666,10 @@ async fn async_main(opts: Opts, cancel: CancellationToken) -> anyhow::Result<()>
false false
} }
}; };
if !connect_to_existing && download_opts.server_url.is_some() {
anyhow::bail!("cannot connect to server at {}", client.base_url());
}
if connect_to_existing { if connect_to_existing {
for torrent_url in &download_opts.torrent_path { for torrent_url in &download_opts.torrent_path {
match client match client