use std::convert::TryFrom;
use url::Url;
use crate::{
pike::store::{PikeStore, PikeStoreError},
rest_api::resources::{error::ErrorResponse, paging::v1::Paging},
};
use super::payloads::{AgentListSlice, AgentSlice};
pub fn list_agents<'a>(
url: Url,
store: Box<dyn PikeStore + 'a>,
service_id: Option<&str>,
offset: u64,
limit: u16,
) -> Result<AgentListSlice, ErrorResponse> {
let offset = i64::try_from(offset).unwrap_or(i64::MAX);
let limit = i64::try_from(limit).unwrap_or(10);
let agent_list = store
.list_agents(service_id, offset, limit)
.map_err(|err| match err {
PikeStoreError::InternalError(err) => ErrorResponse::internal_error(Box::new(err)),
PikeStoreError::ConstraintViolationError(err) => {
ErrorResponse::new(400, &format!("{}", err))
}
PikeStoreError::ResourceTemporarilyUnavailableError(_) => {
ErrorResponse::new(503, "Service Unavailable")
}
PikeStoreError::NotFoundError(_) => ErrorResponse::new(404, "Resource not found"),
})?;
let data = agent_list
.data
.into_iter()
.map(AgentSlice::try_from)
.collect::<Result<Vec<AgentSlice>, ErrorResponse>>()?;
let paging = Paging::new(url, agent_list.paging, service_id);
Ok(AgentListSlice { data, paging })
}
pub fn get_agent<'a>(
store: Box<dyn PikeStore + 'a>,
public_key: String,
service_id: Option<&str>,
) -> Result<AgentSlice, ErrorResponse> {
let agent = store
.get_agent(&public_key, service_id)
.map_err(|err| match err {
PikeStoreError::InternalError(err) => ErrorResponse::internal_error(Box::new(err)),
PikeStoreError::ConstraintViolationError(err) => {
ErrorResponse::new(400, &format!("{}", err))
}
PikeStoreError::ResourceTemporarilyUnavailableError(_) => {
ErrorResponse::new(503, "Service Unavailable")
}
PikeStoreError::NotFoundError(_) => {
ErrorResponse::new(404, &format!("Agent {} not found", public_key))
}
})?;
AgentSlice::try_from(
agent.ok_or_else(|| ErrorResponse::new(404, &format!("Agent {} not found", public_key)))?,
)
}