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

use reqwest::header::{ContentType, Headers, Host};

use client::*;

#[derive(Debug)]
pub struct Token<'a> {
    client: &'a Client,
    token:  String,
}

impl<'a> Token<'a> {
    pub fn new(client: &Client, token: String) -> Token {
        Token {
            client: client,
            token:  token,
        }
    }

    pub fn get_headers(&self) -> Headers {
        let mut headers = Headers::new();
        headers.set(ContentType::plaintext());
        headers.set(Host::new(
            self.client
                .url()
                .host_str()
                .unwrap_or("localhost")
                .to_string(),
            self.client.url().port(),
        ));
        headers.set_raw("X-Warp10-Token", vec![self.token.as_bytes().to_vec()]);
        headers
    }
}