352 lines
11 KiB
Plaintext
352 lines
11 KiB
Plaintext
#[allow(unused_imports)]
|
|
use progenitor_client::{encode_path, RequestBuilderExt};
|
|
pub use progenitor_client::{ByteStream, Error, ResponseValue};
|
|
#[allow(unused_imports)]
|
|
use reqwest::header::{HeaderMap, HeaderValue};
|
|
pub mod types {
|
|
use serde::{Deserialize, Serialize};
|
|
#[allow(unused_imports)]
|
|
use std::convert::TryFrom;
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct EnrolBody {
|
|
pub host: String,
|
|
pub key: String,
|
|
}
|
|
|
|
impl From<&EnrolBody> for EnrolBody {
|
|
fn from(value: &EnrolBody) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct GlobalJobsResult {
|
|
pub summary: Vec<ReportSummary>,
|
|
}
|
|
|
|
impl From<&GlobalJobsResult> for GlobalJobsResult {
|
|
fn from(value: &GlobalJobsResult) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct OutputRecord {
|
|
pub msg: String,
|
|
pub stream: String,
|
|
pub time: chrono::DateTime<chrono::offset::Utc>,
|
|
}
|
|
|
|
impl From<&OutputRecord> for OutputRecord {
|
|
fn from(value: &OutputRecord) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct PingResult {
|
|
pub host: String,
|
|
pub ok: bool,
|
|
}
|
|
|
|
impl From<&PingResult> for PingResult {
|
|
fn from(value: &PingResult) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct ReportFinishBody {
|
|
pub duration_millis: i32,
|
|
pub end_time: chrono::DateTime<chrono::offset::Utc>,
|
|
pub exit_status: i32,
|
|
pub id: ReportId,
|
|
}
|
|
|
|
impl From<&ReportFinishBody> for ReportFinishBody {
|
|
fn from(value: &ReportFinishBody) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct ReportId {
|
|
pub host: String,
|
|
pub job: String,
|
|
pub pid: u64,
|
|
pub time: chrono::DateTime<chrono::offset::Utc>,
|
|
pub uuid: String,
|
|
}
|
|
|
|
impl From<&ReportId> for ReportId {
|
|
fn from(value: &ReportId) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct ReportOutputBody {
|
|
pub id: ReportId,
|
|
pub record: OutputRecord,
|
|
}
|
|
|
|
impl From<&ReportOutputBody> for ReportOutputBody {
|
|
fn from(value: &ReportOutputBody) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct ReportResult {
|
|
pub existed_already: bool,
|
|
}
|
|
|
|
impl From<&ReportResult> for ReportResult {
|
|
fn from(value: &ReportResult) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct ReportStartBody {
|
|
pub id: ReportId,
|
|
pub script: String,
|
|
pub start_time: chrono::DateTime<chrono::offset::Utc>,
|
|
}
|
|
|
|
impl From<&ReportStartBody> for ReportStartBody {
|
|
fn from(value: &ReportStartBody) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize)]
|
|
pub struct ReportSummary {
|
|
pub age_seconds: i32,
|
|
pub duration_seconds: i32,
|
|
pub host: String,
|
|
pub job: String,
|
|
pub status: i32,
|
|
pub when: chrono::DateTime<chrono::offset::Utc>,
|
|
}
|
|
|
|
impl From<&ReportSummary> for ReportSummary {
|
|
fn from(value: &ReportSummary) -> Self {
|
|
value.clone()
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
///Client for Keeper API
|
|
///
|
|
///report execution of cron jobs through a mechanism other than mail
|
|
///
|
|
///Version: 1.0
|
|
pub struct Client {
|
|
pub(crate) baseurl: String,
|
|
pub(crate) client: reqwest::Client,
|
|
}
|
|
|
|
impl Client {
|
|
/// Create a new client.
|
|
///
|
|
/// `baseurl` is the base URL provided to the internal
|
|
/// `reqwest::Client`, and should include a scheme and hostname,
|
|
/// as well as port and a path stem if applicable.
|
|
pub fn new(baseurl: &str) -> Self {
|
|
let dur = std::time::Duration::from_secs(15);
|
|
let client = reqwest::ClientBuilder::new()
|
|
.connect_timeout(dur)
|
|
.timeout(dur)
|
|
.build()
|
|
.unwrap();
|
|
Self::new_with_client(baseurl, client)
|
|
}
|
|
|
|
/// Construct a new client with an existing `reqwest::Client`,
|
|
/// allowing more control over its configuration.
|
|
///
|
|
/// `baseurl` is the base URL provided to the internal
|
|
/// `reqwest::Client`, and should include a scheme and hostname,
|
|
/// as well as port and a path stem if applicable.
|
|
pub fn new_with_client(baseurl: &str, client: reqwest::Client) -> Self {
|
|
Self {
|
|
baseurl: baseurl.to_string(),
|
|
client,
|
|
}
|
|
}
|
|
|
|
/// Get the base URL to which requests are made.
|
|
pub fn baseurl(&self) -> &String {
|
|
&self.baseurl
|
|
}
|
|
|
|
/// Get the internal `reqwest::Client` used to make requests.
|
|
pub fn client(&self) -> &reqwest::Client {
|
|
&self.client
|
|
}
|
|
|
|
/// Get the version of this API.
|
|
///
|
|
/// This string is pulled directly from the source OpenAPI
|
|
/// document and may be in any format the API selects.
|
|
pub fn api_version(&self) -> &'static str {
|
|
"1.0"
|
|
}
|
|
}
|
|
|
|
impl Client {
|
|
///Sends a `POST` request to `/enrol`
|
|
///
|
|
///Arguments:
|
|
/// - `authorization`: Authorization header (bearer token)
|
|
/// - `body`
|
|
pub async fn enrol<'a>(
|
|
&'a self,
|
|
authorization: &'a str,
|
|
body: &'a types::EnrolBody,
|
|
) -> Result<ResponseValue<()>, Error<()>> {
|
|
let url = format!("{}/enrol", self.baseurl,);
|
|
let mut header_map = HeaderMap::with_capacity(1usize);
|
|
header_map.append("Authorization", HeaderValue::try_from(authorization)?);
|
|
let request = self
|
|
.client
|
|
.post(url)
|
|
.json(&body)
|
|
.headers(header_map)
|
|
.build()?;
|
|
let result = self.client.execute(request).await;
|
|
let response = result?;
|
|
match response.status().as_u16() {
|
|
201u16 => Ok(ResponseValue::empty(response)),
|
|
_ => Err(Error::UnexpectedResponse(response)),
|
|
}
|
|
}
|
|
|
|
///Sends a `GET` request to `/global/jobs`
|
|
///
|
|
///Arguments:
|
|
/// - `authorization`: Authorization header (bearer token)
|
|
pub async fn global_jobs<'a>(
|
|
&'a self,
|
|
authorization: &'a str,
|
|
) -> Result<ResponseValue<types::GlobalJobsResult>, Error<()>> {
|
|
let url = format!("{}/global/jobs", self.baseurl,);
|
|
let mut header_map = HeaderMap::with_capacity(1usize);
|
|
header_map.append("Authorization", HeaderValue::try_from(authorization)?);
|
|
let request = self.client.get(url).headers(header_map).build()?;
|
|
let result = self.client.execute(request).await;
|
|
let response = result?;
|
|
match response.status().as_u16() {
|
|
201u16 => ResponseValue::from_response(response).await,
|
|
_ => Err(Error::UnexpectedResponse(response)),
|
|
}
|
|
}
|
|
|
|
///Sends a `GET` request to `/ping`
|
|
///
|
|
///Arguments:
|
|
/// - `authorization`: Authorization header (bearer token)
|
|
pub async fn ping<'a>(
|
|
&'a self,
|
|
authorization: &'a str,
|
|
) -> Result<ResponseValue<types::PingResult>, Error<()>> {
|
|
let url = format!("{}/ping", self.baseurl,);
|
|
let mut header_map = HeaderMap::with_capacity(1usize);
|
|
header_map.append("Authorization", HeaderValue::try_from(authorization)?);
|
|
let request = self.client.get(url).headers(header_map).build()?;
|
|
let result = self.client.execute(request).await;
|
|
let response = result?;
|
|
match response.status().as_u16() {
|
|
201u16 => ResponseValue::from_response(response).await,
|
|
_ => Err(Error::UnexpectedResponse(response)),
|
|
}
|
|
}
|
|
|
|
///Sends a `POST` request to `/report/finish`
|
|
///
|
|
///Arguments:
|
|
/// - `authorization`: Authorization header (bearer token)
|
|
/// - `body`
|
|
pub async fn report_finish<'a>(
|
|
&'a self,
|
|
authorization: &'a str,
|
|
body: &'a types::ReportFinishBody,
|
|
) -> Result<ResponseValue<types::ReportResult>, Error<()>> {
|
|
let url = format!("{}/report/finish", self.baseurl,);
|
|
let mut header_map = HeaderMap::with_capacity(1usize);
|
|
header_map.append("Authorization", HeaderValue::try_from(authorization)?);
|
|
let request = self
|
|
.client
|
|
.post(url)
|
|
.json(&body)
|
|
.headers(header_map)
|
|
.build()?;
|
|
let result = self.client.execute(request).await;
|
|
let response = result?;
|
|
match response.status().as_u16() {
|
|
201u16 => ResponseValue::from_response(response).await,
|
|
_ => Err(Error::UnexpectedResponse(response)),
|
|
}
|
|
}
|
|
|
|
///Sends a `POST` request to `/report/output`
|
|
///
|
|
///Arguments:
|
|
/// - `authorization`: Authorization header (bearer token)
|
|
/// - `body`
|
|
pub async fn report_output<'a>(
|
|
&'a self,
|
|
authorization: &'a str,
|
|
body: &'a types::ReportOutputBody,
|
|
) -> Result<ResponseValue<types::ReportResult>, Error<()>> {
|
|
let url = format!("{}/report/output", self.baseurl,);
|
|
let mut header_map = HeaderMap::with_capacity(1usize);
|
|
header_map.append("Authorization", HeaderValue::try_from(authorization)?);
|
|
let request = self
|
|
.client
|
|
.post(url)
|
|
.json(&body)
|
|
.headers(header_map)
|
|
.build()?;
|
|
let result = self.client.execute(request).await;
|
|
let response = result?;
|
|
match response.status().as_u16() {
|
|
201u16 => ResponseValue::from_response(response).await,
|
|
_ => Err(Error::UnexpectedResponse(response)),
|
|
}
|
|
}
|
|
|
|
///Sends a `POST` request to `/report/start`
|
|
///
|
|
///Arguments:
|
|
/// - `authorization`: Authorization header (bearer token)
|
|
/// - `body`
|
|
pub async fn report_start<'a>(
|
|
&'a self,
|
|
authorization: &'a str,
|
|
body: &'a types::ReportStartBody,
|
|
) -> Result<ResponseValue<types::ReportResult>, Error<()>> {
|
|
let url = format!("{}/report/start", self.baseurl,);
|
|
let mut header_map = HeaderMap::with_capacity(1usize);
|
|
header_map.append("Authorization", HeaderValue::try_from(authorization)?);
|
|
let request = self
|
|
.client
|
|
.post(url)
|
|
.json(&body)
|
|
.headers(header_map)
|
|
.build()?;
|
|
let result = self.client.execute(request).await;
|
|
let response = result?;
|
|
match response.status().as_u16() {
|
|
201u16 => ResponseValue::from_response(response).await,
|
|
_ => Err(Error::UnexpectedResponse(response)),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub mod prelude {
|
|
pub use super::Client;
|
|
}
|