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
|
extern crate reqwest;
use reqwest::{
header::{Accept, qitem},
mime,
StatusCode,
};
use std::io::Read;
const ZOTAPI_CHANNEL_STREAM_PATH : &str = "/api/z/1.0/channel/stream";
const ZOTAPI_NETWORK_STREAM_PATH : &str = "/api/z/1.0/network/stream";
#[derive(Debug)]
pub enum Error {
Http(reqwest::Error),
Io(std::io::Error),
Unauthorized,
Unknown,
}
impl From<reqwest::Error> for Error {
fn from(e: reqwest::Error) -> Error {
Error::Http(e)
}
}
impl From<std::io::Error> for Error {
fn from(e: std::io::Error) -> Error {
Error::Io(e)
}
}
pub struct Client {
inner: reqwest::Client,
base_url: String,
user: String,
pw: String,
}
impl Client {
pub fn channel_stream(&self) -> Result<String, Error> {
self.fetch_stream(ZOTAPI_CHANNEL_STREAM_PATH)
}
pub fn network_stream(&self) -> Result<String, Error> {
self.fetch_stream(ZOTAPI_NETWORK_STREAM_PATH)
}
fn url(&self, path: &str) -> String {
self.base_url.clone() + path
}
fn fetch_stream(&self, path: &str) -> Result<String, Error> {
let url = self.url(path);
let mut res = self.inner.get(&url)
.header(Accept(vec![qitem(mime::APPLICATION_JSON)]))
.basic_auth(self.user.clone(), Some(self.pw.clone()))
.send()?;
match res.status() {
StatusCode::Unauthorized => Err(Error::Unauthorized),
StatusCode::Ok => {
let mut body = String::new();
res.read_to_string(&mut body)?;
Ok(body)
},
_ => Err(Error::Unknown)
}
}
}
pub fn client(url: &str, user: &str, pw: &str) -> Client
{
Client {
inner: reqwest::Client::new(),
base_url: String::from(url),
user: String::from(user),
pw: String::from(pw),
}
}
#[cfg(test)]
mod tests {
#[test]
fn it_works() {
assert_eq!(2 + 2, 4);
}
}
|
