use async_trait::async_trait;
use anyhow::Result;
use uuid::Uuid;
use crate::domain::entity::{UploadSession, StorageBackend, UploadStatus};
#[derive(Debug, Clone, Default)]
pub struct UploadSessionPaginationParams {
pub page: u32,
pub per_page: u32,
}
impl UploadSessionPaginationParams {
pub fn new(page: u32, per_page: u32) -> Self {
Self { page, per_page }
}
pub fn offset(&self) -> u64 {
((self.page.saturating_sub(1)) * self.per_page) as u64
}
pub fn limit(&self) -> u64 {
self.per_page as u64
}
}
#[derive(Debug, Clone)]
pub struct UploadSessionPaginatedResult {
pub data: Vec<UploadSession>,
pub total: u64,
pub page: u32,
pub per_page: u32,
}
#[derive(Debug, Clone, Default)]
pub struct UploadSessionFilter {
pub bucket_id: Option<Uuid>,
pub user_id: Option<Uuid>,
pub path: Option<String>,
pub filename: Option<String>,
pub mime_type: Option<String>,
pub file_size: Option<i64>,
pub status: Option<UploadStatus>,
pub storage_backend: Option<StorageBackend>,
}
impl UploadSessionFilter {
pub fn has_filters(&self) -> bool {
self.bucket_id.is_some() || self.user_id.is_some() || self.path.is_some() || self.filename.is_some() || self.mime_type.is_some() || self.file_size.is_some() || self.status.is_some() || self.storage_backend.is_some()
}
}
#[async_trait]
pub trait UploadSessionRepository: Send + Sync {
async fn save(&self, entity: &UploadSession) -> Result<UploadSession>;
async fn find_by_id(&self, id: &str) -> Result<Option<UploadSession>>;
async fn find_all(&self) -> Result<Vec<UploadSession>>;
async fn update(&self, id: &str, entity: &UploadSession) -> Result<Option<UploadSession>>;
async fn delete(&self, id: &str) -> Result<bool>;
async fn list(&self, params: UploadSessionPaginationParams) -> Result<UploadSessionPaginatedResult>;
async fn list_with_filters(&self, params: UploadSessionPaginationParams, filters: UploadSessionFilter) -> Result<UploadSessionPaginatedResult>;
async fn count(&self) -> Result<u64>;
async fn count_with_filters(&self, filters: UploadSessionFilter) -> Result<u64>;
async fn exists(&self, id: &str) -> Result<bool>;
async fn soft_delete(&self, id: &str) -> Result<bool>;
async fn restore(&self, id: &str) -> Result<Option<UploadSession>>;
async fn list_deleted(&self, params: UploadSessionPaginationParams) -> Result<UploadSessionPaginatedResult>;
async fn empty_trash(&self) -> Result<u64>;
async fn bulk_save(&self, entities: &[UploadSession]) -> Result<Vec<UploadSession>>;
async fn bulk_delete(&self, ids: &[&str]) -> Result<u64>;
}