use crate::error::Result;
use crate::provider::{AwsProvider, CloudProvider};
use crate::store::traits::SigningCertificateStore;
use crate::wami::credentials::signing_certificate::{
builder as cert_builder, DeleteSigningCertificateRequest, ListSigningCertificatesRequest,
SigningCertificate, UpdateSigningCertificateRequest, UploadSigningCertificateRequest,
};
use std::sync::{Arc, RwLock};
pub struct SigningCertificateService<S> {
store: Arc<RwLock<S>>,
provider: Arc<dyn CloudProvider>,
account_id: String,
}
impl<S: SigningCertificateStore> SigningCertificateService<S> {
pub fn new(store: Arc<RwLock<S>>, account_id: String) -> Self {
Self {
store,
provider: Arc::new(AwsProvider::new()),
account_id,
}
}
pub fn with_provider(&self, provider: Arc<dyn CloudProvider>) -> Self {
Self {
store: self.store.clone(),
provider,
account_id: self.account_id.clone(),
}
}
pub async fn upload_signing_certificate(
&self,
request: UploadSigningCertificateRequest,
) -> Result<SigningCertificate> {
let certificate = cert_builder::build_signing_certificate(
request.user_name,
request.certificate_body,
&*self.provider,
&self.account_id,
);
self.store
.write()
.unwrap()
.create_signing_certificate(certificate)
.await
}
pub async fn get_signing_certificate(
&self,
certificate_id: &str,
) -> Result<Option<SigningCertificate>> {
self.store
.read()
.unwrap()
.get_signing_certificate(certificate_id)
.await
}
pub async fn update_signing_certificate(
&self,
request: UpdateSigningCertificateRequest,
) -> Result<SigningCertificate> {
let mut certificate = self
.store
.read()
.unwrap()
.get_signing_certificate(&request.certificate_id)
.await?
.ok_or_else(|| crate::error::AmiError::ResourceNotFound {
resource: format!("SigningCertificate: {}", request.certificate_id),
})?;
certificate.status = request.status;
self.store
.write()
.unwrap()
.update_signing_certificate(certificate)
.await
}
pub async fn delete_signing_certificate(
&self,
request: DeleteSigningCertificateRequest,
) -> Result<()> {
self.store
.write()
.unwrap()
.delete_signing_certificate(&request.certificate_id)
.await
}
pub async fn list_signing_certificates(
&self,
request: ListSigningCertificatesRequest,
) -> Result<Vec<SigningCertificate>> {
self.store
.read()
.unwrap()
.list_signing_certificates(request.user_name.as_deref())
.await
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::store::memory::InMemoryWamiStore;
use crate::wami::credentials::signing_certificate::CertificateStatus;
fn setup_service() -> SigningCertificateService<InMemoryWamiStore> {
let store = Arc::new(RwLock::new(InMemoryWamiStore::default()));
SigningCertificateService::new(store, "123456789012".to_string())
}
#[tokio::test]
async fn test_upload_and_get_signing_certificate() {
let service = setup_service();
let request = UploadSigningCertificateRequest {
user_name: "alice".to_string(),
certificate_body: "-----BEGIN CERTIFICATE-----\ntest\n-----END CERTIFICATE-----"
.to_string(),
};
let certificate = service.upload_signing_certificate(request).await.unwrap();
assert_eq!(certificate.user_name, "alice");
assert!(!certificate.certificate_id.is_empty());
let retrieved = service
.get_signing_certificate(&certificate.certificate_id)
.await
.unwrap();
assert!(retrieved.is_some());
assert_eq!(retrieved.unwrap().user_name, "alice");
}
#[tokio::test]
async fn test_update_signing_certificate_status() {
let service = setup_service();
let upload_req = UploadSigningCertificateRequest {
user_name: "bob".to_string(),
certificate_body: "-----BEGIN CERTIFICATE-----\ntest\n-----END CERTIFICATE-----"
.to_string(),
};
let certificate = service
.upload_signing_certificate(upload_req)
.await
.unwrap();
let update_req = UpdateSigningCertificateRequest {
user_name: "bob".to_string(),
certificate_id: certificate.certificate_id.clone(),
status: CertificateStatus::Inactive,
};
let updated = service
.update_signing_certificate(update_req)
.await
.unwrap();
assert_eq!(updated.status, CertificateStatus::Inactive);
}
#[tokio::test]
async fn test_delete_signing_certificate() {
let service = setup_service();
let upload_req = UploadSigningCertificateRequest {
user_name: "charlie".to_string(),
certificate_body: "-----BEGIN CERTIFICATE-----\ntest\n-----END CERTIFICATE-----"
.to_string(),
};
let certificate = service
.upload_signing_certificate(upload_req)
.await
.unwrap();
let delete_req = DeleteSigningCertificateRequest {
user_name: "charlie".to_string(),
certificate_id: certificate.certificate_id.clone(),
};
service
.delete_signing_certificate(delete_req)
.await
.unwrap();
let retrieved = service
.get_signing_certificate(&certificate.certificate_id)
.await
.unwrap();
assert!(retrieved.is_none());
}
#[tokio::test]
async fn test_list_signing_certificates() {
let service = setup_service();
for _ in 0..3 {
let request = UploadSigningCertificateRequest {
user_name: "david".to_string(),
certificate_body: "-----BEGIN CERTIFICATE-----\ntest\n-----END CERTIFICATE-----"
.to_string(),
};
service.upload_signing_certificate(request).await.unwrap();
}
let list_request = ListSigningCertificatesRequest {
user_name: Some("david".to_string()),
};
let certificates = service
.list_signing_certificates(list_request)
.await
.unwrap();
assert_eq!(certificates.len(), 3);
}
}