fleet 0.2.0

A client library for CoreOS's fleet.
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

use std::error::Error;
use std::fmt;

use hyper::client::Response;
use hyper::error::HttpError;
use rustc_serialize::json::Json;

pub type FleetResult<T> = Result<T, FleetError>;

pub struct FleetError {
    code: Option<u16>,
    message: Option<String>,
}

impl FleetError {
    pub fn from_hyper_error(error: &HttpError) -> FleetError {
        FleetError {
            code: None,
            message: Some(error.description().to_string()),
        }
    }

    pub fn from_hyper_response(response: &mut Response) -> FleetError {
        FleetError {
            code: Some(response.status.to_u16()),
            message: extract_message(response),
        }
    }
}

impl fmt::Display for FleetError {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(
            f,
            "{}: {}",
            self.code.unwrap_or(0),
            self.message.clone().unwrap_or("Unknown error".to_string()),
       )
    }
}

fn extract_message(response: &mut Response) -> Option<String> {
   match Json::from_reader(response) {
       Ok(json) => {
           match json.find_path(&["error", "message"]) {
               Some(message_json) => match message_json.as_string() {
                   Some(message) => {
                       if message.len() == 0 {
                           Some("Error in JSON response from Fleet was empty".to_string())
                       } else {
                           Some(message.to_string())
                       }
                   },
                   None => Some("Error in JSON response from Fleet was empty".to_string()),
               },
               None => Some("Error parsing JSON response from Fleet".to_string()),
           }
       },
       Err(error) => Some(error.description().to_string()),
   }
}