aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/download.rs19
-rw-r--r--src/main.rs4
2 files changed, 17 insertions, 6 deletions
diff --git a/src/download.rs b/src/download.rs
index fa99ea5..c59eec3 100644
--- a/src/download.rs
+++ b/src/download.rs
@@ -1,10 +1,10 @@
use std::path::{Path, PathBuf};
-use anyhow::{bail, Error, Context};
-use hyper::Uri;
+use anyhow::{bail, Context, Error, Result};
+use hyper::{Body, Request, Uri};
use hyper::body::HttpBody;
use hyper::client::{Client, HttpConnector};
-use hyper::header::{CONTENT_TYPE, LOCATION};
+use hyper::header::{CONTENT_TYPE, LOCATION, USER_AGENT};
use hyper_rustls::HttpsConnector;
use tokio::fs;
use tokio::io::AsyncWriteExt;
@@ -13,7 +13,14 @@ use super::Args;
type HttpsClient = Client<HttpsConnector<HttpConnector>>;
-pub(crate) async fn download(url: Uri, args: &Args) -> Result<(), Error> {
+fn request(url: Uri, args: &Args) -> Result<Request<Body>> {
+ Request::get(url)
+ .header(USER_AGENT, args.user_agent.clone())
+ .body(Body::empty())
+ .map_err(Error::new)
+}
+
+pub(crate) async fn download(url: Uri, args: &Args) -> Result<()> {
let output_file_dir = &args.directory_prefix;
let output_file_path = if let Some(output) = &args.output_document {
@@ -27,13 +34,13 @@ pub(crate) async fn download(url: Uri, args: &Args) -> Result<(), Error> {
};
let client: HttpsClient = Client::builder().build(HttpsConnector::with_native_roots());
- let mut response = client.get(url.clone()).await?;
+ let mut response = client.request(request(url.clone(), args)?).await?;
while response.status().is_redirection() {
let location = response.headers()
.get(LOCATION).context("no Location header in redirect")?
.to_str().context("malformed Location header in redirect")?
.parse().context("non-URL Location header in redirect")?;
- response = client.get(location).await?;
+ response = client.request(request(location, args)?).await?;
}
if !response.status().is_success() {
let status = response.status();
diff --git a/src/main.rs b/src/main.rs
index a622afd..5de67ca 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -24,6 +24,10 @@ struct Args {
#[structopt(short = "P", long, default_value = ".")]
directory_prefix: PathBuf,
+ /// User-Agent header to include
+ #[structopt(short = "U", long, default_value = concat!("webget/", env!("CARGO_PKG_VERSION")))]
+ user_agent: String,
+
/// The URLs to download
urls: Vec<Uri>,
}