zapreq 0.1.4

A fast, friendly HTTP client for the terminal
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use anyhow::{anyhow, Result};
use reqwest::blocking::{RequestBuilder, Response};

use super::AuthPlugin;

/// CAUS-SESSIONAUT-41, CAUS-SESSIONAUT-42:
/// Basic authentication plugin with explicit username/password state.
#[derive(Debug)]
pub struct BasicAuth {
    username: String,
    password: Option<String>,
}

impl BasicAuth {
    /// CAUS-SESSIONAUT-41:
    /// Parses `user:pass` (or `user`) into auth credentials.
    pub fn new(credentials: &str) -> Result<Self> {
        let raw = credentials.trim();
        if raw.is_empty() {
            return Err(anyhow!("basic auth credentials cannot be empty"));
        }

        if let Some((user, pass)) = raw.split_once(':') {
            if user.is_empty() {
                return Err(anyhow!("basic auth username cannot be empty"));
            }
            return Ok(Self {
                username: user.to_string(),
                password: Some(pass.to_string()),
            });
        }

        Ok(Self {
            username: raw.to_string(),
            password: None,
        })
    }

    pub(crate) fn placeholder() -> Self {
        Self {
            username: "registry".to_string(),
            password: None,
        }
    }
}

impl AuthPlugin for BasicAuth {
    fn name(&self) -> &'static str {
        "basic"
    }

    fn apply(&self, req: RequestBuilder) -> RequestBuilder {
        req.basic_auth(&self.username, self.password.as_deref())
    }

    fn handle_401(&self, _req: RequestBuilder, _res: &Response) -> Option<RequestBuilder> {
        None
    }
}

#[cfg(test)]
mod tests {
    use super::BasicAuth;
    use crate::auth::AuthPlugin;

    #[test]
    fn new_user_pass() {
        let auth = BasicAuth::new("user:pass").expect("basic credentials should parse");
        let req = auth
            .apply(reqwest::blocking::Client::new().get("http://example.com"))
            .build()
            .expect("request should build");
        let header = req
            .headers()
            .get(reqwest::header::AUTHORIZATION)
            .expect("authorization header should exist")
            .to_str()
            .expect("header should be utf8");
        assert_eq!(header, "Basic dXNlcjpwYXNz");
    }

    #[test]
    fn new_user_empty_pass() {
        let auth = BasicAuth::new("user:").expect("empty password should be allowed");
        let req = auth
            .apply(reqwest::blocking::Client::new().get("http://example.com"))
            .build()
            .expect("request should build");
        let header = req
            .headers()
            .get(reqwest::header::AUTHORIZATION)
            .expect("authorization header should exist")
            .to_str()
            .expect("header should be utf8");
        assert_eq!(header, "Basic dXNlcjo=");
    }

    #[test]
    fn new_user_without_pass() {
        let auth = BasicAuth::new("user").expect("username-only should be accepted");
        let req = auth
            .apply(reqwest::blocking::Client::new().get("http://example.com"))
            .build()
            .expect("request should build");
        let header = req
            .headers()
            .get(reqwest::header::AUTHORIZATION)
            .expect("authorization header should exist")
            .to_str()
            .expect("header should be utf8");
        assert!(header.starts_with("Basic "));
    }

    #[test]
    fn new_empty_error() {
        let err = BasicAuth::new("").expect_err("empty credentials must fail");
        assert!(err.to_string().contains("cannot be empty"));
    }
}