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

use std::path::Path;

use reqwest::ClientBuilder;
use reqwest::Response;

use crate::Error;
use crate::util::load_pem_pair;

mod model;
pub use model::Plots;
use model::GetPlotsResponse;

pub struct Client {
	host: String,
	port: u16,
	http: reqwest::Client
}

impl Client {
	pub async fn new(host: &str, port: u16, key_file: impl AsRef<Path>, cert_file: impl AsRef<Path>) -> Result<Self, Error> {
		let identity = load_pem_pair(key_file, cert_file).await?;
		let http = ClientBuilder::new()
			.danger_accept_invalid_certs(true)
			//.danger_accept_invalid_hostnames(true)
			.identity(identity)
			.build()?;
		Ok(Self{
			host: host.to_string(),
			port: port,
			http: http
		})
	}

	pub async fn get_plots(&self) -> Result<Plots, Error> {
		let response = self.cmd("get_plots").await?;
		let response: GetPlotsResponse = response.json().await?;
		Ok(response.into())
	}

	async fn cmd(&self, command: &str) -> Result<Response, reqwest::Error> {
		let url = self.make_url(command);
		self.http.post(&url).header("Content-Type", "application/json").body("{}").send().await
	}

	fn make_url(&self, command: &str) -> String {
		format!("https://{}:{}/{}", &self.host, self.port, &command)
	}
}