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

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

use super::AuthPlugin;

/// CAUS-SESSIONAUT-41, CAUS-SESSIONAUT-42:
/// Bearer authentication plugin with token state ownership.
#[derive(Debug)]
pub struct BearerAuth {
    token: String,
}

impl BearerAuth {
    /// CAUS-SESSIONAUT-41:
    /// Validates and stores bearer token.
    pub fn new(credentials: &str) -> Result<Self> {
        let token = credentials.trim();
        if token.is_empty() {
            return Err(anyhow!("bearer token cannot be empty"));
        }

        Ok(Self {
            token: token.to_string(),
        })
    }

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

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

    fn apply(&self, req: RequestBuilder) -> RequestBuilder {
        req.header("Authorization", format!("Bearer {}", self.token))
    }

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

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

    #[test]
    fn new_token_ok() {
        let auth = BearerAuth::new("mytoken").expect("token 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, "Bearer mytoken");
    }

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