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
|
use crate::types::*;
type Result<T> = std::result::Result<T, HydrusError>;
pub struct HydrusClient {
apikey: Option<String>,
sessionkey: Option<String>,
url: String,
}
impl HydrusClient {
pub fn new(url: &str) -> HydrusClient {
HydrusClient {
apikey: None,
sessionkey: None,
url: url.to_string(),
}
}
pub fn set_api_key(&mut self, key: &str) {
self.apikey = Some(key.to_owned())
}
pub fn set_session_key(&mut self, key: &str) {
self.sessionkey = Some(key.to_owned())
}
pub fn request_new_permissions(
&self,
name: &str,
permissions: &[HydrusPermissions],
) -> Result<String> {
let mut req_url = self.url.to_owned();
req_url.push_str("request_new_permissions?name=");
req_url.push_str(&urlencoding::encode(&name));
if permissions.is_empty() {
req_url.push_str("&permit_everything=true");
} else {
req_url.push_str("&basic_permissions=");
let json_string = musli::json::to_string(&permissions)?;
req_url.push_str(&urlencoding::encode(&json_string));
};
let response = ureq::get(req_url).call()?.body_mut().read_to_vec()?;
let key: AccessKey = musli::json::decode(response.as_slice())?;
Ok(key.access_key)
}
pub fn get_session_key(&self) -> Result<String> {
let mut req_url = self.url.to_owned();
req_url.push_str("session_key");
let mut request = ureq::get(req_url);
if let Some(key) = &self.apikey {
request = request.header("Hydrus-Client-API-Access-Key", key);
}
let response = request.call()?.body_mut().read_to_vec()?;
let key: SessionKey = musli::json::decode(response.as_slice())?;
Ok(key.session_key)
}
pub fn verify_access_key(&self, key: &str) -> Result<KeyInfo> {
let mut req_url = self.url.to_owned();
req_url.push_str("verify_access_key");
let response = ureq::get(req_url)
.header("Hydrus-Client-API-Access-Key", key)
.call()?
.body_mut()
.read_to_vec()?;
let data: KeyInfo = musli::json::decode(response.as_slice())?;
Ok(data)
}
pub fn get_service_name(&self, name: &str) -> Result<Service> {
let mut req_url = self.url.to_owned();
req_url.push_str("get_service?service_name=");
req_url.push_str(&urlencoding::encode(&name));
let response = if let Some(key) = &self.sessionkey {
ureq::get(req_url)
.header("Hydrus-Client-API-Access-Key", key)
.call()?
.body_mut()
.read_to_vec()?
} else if let Some(key) = &self.apikey {
ureq::get(req_url)
.header("Hydrus-Client-API-Access-Key", key)
.call()?
.body_mut()
.read_to_vec()?
} else {
return Err(HydrusError::KeyNotSupplied);
};
let service: Service = musli::json::decode(response.as_slice())?;
Ok(service)
}
}
|
