use std::convert::TryFrom;
use url::Url;
use crate::{
pike::store::{PikeStore, PikeStoreError},
rest_api::resources::{error::ErrorResponse, paging::v1::Paging},
};
use super::payloads::{OrganizationListSlice, OrganizationSlice};
pub fn list_organizations<'a>(
url: Url,
store: Box<dyn PikeStore + 'a>,
service_id: Option<&str>,
offset: u64,
limit: u16,
) -> Result<OrganizationListSlice, ErrorResponse> {
let offset = i64::try_from(offset).unwrap_or(i64::MAX);
let limit = i64::try_from(limit).unwrap_or(10);
let organization_list = store
.list_organizations(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 = organization_list
.data
.into_iter()
.map(OrganizationSlice::from)
.collect();
let paging = Paging::new(url, organization_list.paging, service_id);
Ok(OrganizationListSlice { data, paging })
}
pub fn get_organization<'a>(
store: Box<dyn PikeStore + 'a>,
org_id: String,
service_id: Option<&str>,
) -> Result<OrganizationSlice, ErrorResponse> {
let organization = store
.get_organization(&org_id, 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!("Organization {} not found", org_id))
}
})?;
Ok(OrganizationSlice::from(organization.ok_or_else(|| {
ErrorResponse::new(404, &format!("Organization {} not found", org_id))
})?))
}