From 6f86ee7f934fa541c4b899b3ac723418a335c9d8 Mon Sep 17 00:00:00 2001
From: Jacob Hoffman-Andrews <github@hoffman-andrews.com>
Date: Sun, 21 Feb 2021 14:26:12 -0800
Subject: [PATCH] Add example "cureq". (#330)

Contrary to smoke-test, this takes full URLs on the commandline and
prints their contents to stdout. This makes it easier to test behavior
with specific URLs. I hope to later add flags for various behaviors like
printing headers, following redirects, enabling / disabling cookies,
and verbose output.

Also add a useful debug line when receiving a cookie header.
---
 Cargo.toml             |   4 ++
 examples/cureq/main.rs | 125 +++++++++++++++++++++++++++++++++++++++++
 src/unit.rs            |   4 ++
 3 files changed, 133 insertions(+)
 create mode 100644 examples/cureq/main.rs

diff --git a/Cargo.toml b/Cargo.toml
index 0ab1838..060a33e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -46,3 +46,7 @@ env_logger = "0.8.1"
 
 [[example]]
 name = "smoke-test"
+
+[[example]]
+name = "cureq"
+required-features = ["charset", "cookies", "socks-proxy"]
diff --git a/examples/cureq/main.rs b/examples/cureq/main.rs
new file mode 100644
index 0000000..8898ca9
--- /dev/null
+++ b/examples/cureq/main.rs
@@ -0,0 +1,125 @@
+use std::env;
+use std::error;
+use std::fmt;
+use std::io;
+use std::time::Duration;
+
+use ureq;
+
+#[derive(Debug)]
+struct StringError(String);
+
+impl error::Error for StringError {}
+
+impl fmt::Display for StringError {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{}", self.0)
+    }
+}
+
+impl From<String> for StringError {
+    fn from(source: String) -> Self {
+        Self(source)
+    }
+}
+
+#[derive(Debug)]
+struct Error {
+    source: Box<dyn error::Error>,
+}
+
+impl error::Error for Error {}
+
+impl fmt::Display for Error {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{}", self.source)
+    }
+}
+
+impl From<StringError> for Error {
+    fn from(source: StringError) -> Self {
+        Error {
+            source: source.into(),
+        }
+    }
+}
+
+impl From<ureq::Error> for Error {
+    fn from(source: ureq::Error) -> Self {
+        Error {
+            source: source.into(),
+        }
+    }
+}
+
+impl From<io::Error> for Error {
+    fn from(source: io::Error) -> Self {
+        Error {
+            source: source.into(),
+        }
+    }
+}
+
+fn get(agent: &ureq::Agent, url: &str, print_headers: bool) -> Result<(), Error> {
+    let response = agent.get(url).call()?;
+    if print_headers {
+        println!(
+            "{} {} {}",
+            response.http_version(),
+            response.status(),
+            response.status_text()
+        );
+        for h in response.headers_names() {
+            println!("{}: {}", h, response.header(&h).unwrap_or_default());
+        }
+        println!();
+    }
+    let mut reader = response.into_reader();
+    io::copy(&mut reader, &mut io::stdout())?;
+    Ok(())
+}
+
+fn main() {
+    match main2() {
+        Ok(()) => {}
+        Err(e) => {
+            eprintln!("{}", e);
+            std::process::exit(1);
+        }
+    }
+}
+
+fn main2() -> Result<(), Error> {
+    let mut args: Vec<String> = env::args().collect();
+    if args.len() == 1 {
+        println!(
+            r##"Usage: {:#?} url [url ...]
+            
+Fetch url and copy it to stdout.
+"##,
+            env::current_exe()?
+        );
+        return Ok(());
+    }
+    args.remove(0);
+    env_logger::init();
+    let agent = ureq::builder()
+        .timeout_connect(Duration::from_secs(30))
+        .timeout(Duration::from_secs(300))
+        .build();
+    let flags: Vec<&String> = args.iter().filter(|s| s.starts_with("-")).collect();
+    let nonflags: Vec<&String> = args.iter().filter(|s| !s.starts_with("-")).collect();
+
+    let mut print_headers: bool = false;
+    for flag in flags {
+        match flag.as_ref() {
+            "-i" => print_headers = true,
+            f => Err(StringError(format!("unrecognized flag '{}'", f)))?,
+        }
+    }
+
+    for url in nonflags {
+        get(&agent, &url, print_headers)?;
+    }
+    Ok(())
+}
diff --git a/src/unit.rs b/src/unit.rs
index 65dc49b..601c90a 100644
--- a/src/unit.rs
+++ b/src/unit.rs
@@ -411,6 +411,10 @@ fn save_cookies(unit: &Unit, resp: &Response) {
         return;
     }
     let cookies = headers.into_iter().flat_map(|header_value| {
+        debug!(
+            "received 'set-cookie: {}' from {} {}",
+            header_value, unit.method, unit.url
+        );
         match Cookie::parse(header_value.to_string()) {
             Err(_) => None,
             Ok(c) => Some(c),