use std::sync::Arc; use crate::application::dtos::file_dto::FileDto; use crate::application::ports::authorization_ports::AuthorizationEngine; use crate::application::ports::file_lifecycle::FileLifecycleHook; use crate::application::ports::file_ports::FileManagementUseCase; use crate::application::ports::storage_ports::{CopyFolderTreeResult, FileWritePort}; use crate::application::ports::trash_ports::TrashUseCase; use crate::application::services::trash_service::TrashService; use crate::common::errors::DomainError; use crate::domain::services::authorization::{Permission, Resource, Subject}; use crate::domain::services::path_service::validate_storage_name; use crate::infrastructure::repositories::pg::file_blob_read_repository::FileBlobReadRepository; use crate::infrastructure::repositories::pg::file_blob_write_repository::FileBlobWriteRepository; use crate::infrastructure::repositories::pg::folder_db_repository::FolderDbRepository; use crate::infrastructure::services::file_content_cache::FileContentCache; use crate::infrastructure::services::pg_acl_engine::PgAclEngine; use tracing::{error, info, warn}; use uuid::Uuid; /// Service for file management operations (move, delete). /// /// Blob ref_count bookkeeping on deletion is handled by the PG trigger /// `trg_files_decrement_blob_ref` (fires on DELETE FROM storage.files). /// This service only orchestrates trash vs. permanent delete — it never /// touches ref_count directly. pub struct FileManagementService { file_repository: Arc, trash_service: Option>, content_cache: Option>, authz: Arc, /// Lifecycle hook dispatcher — fired on file created (copy) and deleted. file_lifecycle_hook: Option>, } impl FileManagementService { /// Creates a FileManagementService with a trash service, content cache /// and the ReBAC authorization engine. File/folder owner lookups (used /// for owner short-circuit inside the engine) are now the engine's /// responsibility — this service no longer holds direct repo references /// for ownership. pub fn with_trash( file_repository: Arc, trash_service: Option>, _file_read: Option>, _folder_repo: Option>, content_cache: Option>, authz: Arc, ) -> Self { Self { file_repository, trash_service, content_cache, authz, file_lifecycle_hook: None, } } /// Sets the lifecycle hook dispatcher (thumbnails, audio metadata, …). pub fn with_file_lifecycle_hook(mut self, hook: Arc) -> Self { self.file_lifecycle_hook = Some(hook); self } /// Engine check for a file resource. Parses the id into a `Uuid` and /// requires the specified permission. async fn require_file_perm( &self, file_id: &str, perm: Permission, caller_id: Uuid, ) -> Result<(), DomainError> { let uuid = Uuid::parse_str(file_id).map_err(|_| DomainError::not_found("File", file_id))?; self.authz .require(Subject::User(caller_id), perm, Resource::File(uuid)) .await } /// Engine check for a target folder. `None` is allowed (root namespace, /// implicitly owned by the caller). async fn require_target_folder_perm( &self, folder_id: Option<&str>, perm: Permission, caller_id: Uuid, ) -> Result<(), DomainError> { let Some(target) = folder_id else { return Ok(()); }; let uuid = Uuid::parse_str(target).map_err(|_| DomainError::not_found("Folder", target))?; self.authz .require(Subject::User(caller_id), perm, Resource::Folder(uuid)) .await } //impl FileManagementPrivateUseCase for FileManagementService { async fn move_file( &self, file_id: &str, folder_id: Option, caller_id: Uuid, ) -> Result { info!( "Moving file with ID: {} to folder: {:?}", file_id, folder_id ); let moved_file = self .file_repository .move_file(file_id, folder_id, caller_id) .await .map_err(|e| { error!("Error moving file (ID: {}): {}", file_id, e); e })?; info!( "File moved successfully: {} (ID: {}) to folder: {:?}", moved_file.name(), moved_file.id(), moved_file.folder_id() ); Ok(FileDto::from(moved_file)) } async fn copy_file( &self, file_id: &str, target_folder_id: Option, new_name: Option<&str>, caller_id: Uuid, ) -> Result { info!( "Copying file with ID: {} to folder: {:?} as {:?}", file_id, target_folder_id, new_name ); let copied_file = self .file_repository .copy_file(file_id, target_folder_id, new_name, caller_id) .await .map_err(|e| { error!("Error copying file (ID: {}): {}", file_id, e); e })?; info!( "File copied successfully: {} (ID: {}) to folder: {:?}", copied_file.name(), copied_file.id(), copied_file.folder_id() ); let dto = FileDto::from(copied_file); if let Some(hook) = &self.file_lifecycle_hook { hook.on_file_copied(&dto.id, &dto.content_hash, &dto.mime_type, file_id); } Ok(dto) } async fn rename_file( &self, file_id: &str, new_name: &str, caller_id: Uuid, ) -> Result { if let Err(reason) = validate_storage_name(new_name) { return Err(DomainError::validation_error(format!( "Invalid file name '{new_name}': {reason}" ))); } info!("Renaming file with ID: {} to \"{}\"", file_id, new_name); let renamed_file = self .file_repository .rename_file(file_id, new_name, caller_id) .await .map_err(|e| { error!("Error renaming file (ID: {}): {}", file_id, e); e })?; info!( "File renamed successfully: {} (ID: {})", renamed_file.name(), renamed_file.id() ); Ok(FileDto::from(renamed_file)) } async fn delete_file(&self, id: &str) -> Result<(), DomainError> { warn!("Permanently deleting file: {}", id); self.file_repository.delete_file(id).await?; if let Some(cc) = &self.content_cache { cc.invalidate(id).await; } if let Some(hook) = &self.file_lifecycle_hook { hook.on_file_deleted(id); } info!("File permanently deleted: {}", id); Ok(()) } async fn copy_folder_tree( &self, source_folder_id: &str, target_parent_id: Option, dest_name: Option, ) -> Result { info!( "Copying folder tree: source={}, target_parent={:?}, dest_name={:?}", source_folder_id, target_parent_id, dest_name ); let result = self .file_repository .copy_folder_tree(source_folder_id, target_parent_id, dest_name) .await .map_err(|e| { error!( "Error copying folder tree (source: {}): {}", source_folder_id, e ); e })?; info!( "Folder tree copied: {} folders, {} files (new root: {})", result.folders_copied, result.files_copied, result.new_root_folder_id ); Ok(result) } } impl FileManagementUseCase for FileManagementService { async fn require_permission( &self, caller_id: Uuid, permission: Permission, file_id: &str, ) -> Result<(), DomainError> { let uuid = Uuid::parse_str(file_id).map_err(|_| DomainError::not_found("File", file_id))?; self.authz .require(Subject::User(caller_id), permission, Resource::File(uuid)) .await } async fn move_file_with_perms( &self, file_id: &str, caller_id: Uuid, folder_id: Option, ) -> Result { // Move = Update on the file + Create on the target folder (if any). self.require_file_perm(file_id, Permission::Update, caller_id) .await?; self.require_target_folder_perm(folder_id.as_deref(), Permission::Create, caller_id) .await?; self.move_file(file_id, folder_id, caller_id).await } async fn copy_file_with_perms( &self, file_id: &str, caller_id: Uuid, target_folder_id: Option, new_name: Option, ) -> Result { // Copy = Read on the source file + Create on the target folder. self.require_file_perm(file_id, Permission::Read, caller_id) .await?; self.require_target_folder_perm(target_folder_id.as_deref(), Permission::Create, caller_id) .await?; self.copy_file(file_id, target_folder_id, new_name.as_deref(), caller_id) .await } async fn rename_file_with_perms( &self, file_id: &str, caller_id: Uuid, new_name: &str, ) -> Result { self.require_file_perm(file_id, Permission::Update, caller_id) .await?; self.rename_file(file_id, new_name, caller_id).await } async fn delete_file_with_perms(&self, id: &str, caller_id: Uuid) -> Result<(), DomainError> { self.require_file_perm(id, Permission::Delete, caller_id) .await?; self.delete_file(id).await } /// Smart delete: trash-first with dedup reference cleanup. /// /// Blob ref_count bookkeeping is handled entirely by the PG trigger /// `trg_files_decrement_blob_ref` which fires on DELETE FROM storage.files. /// We do NOT decrement here — trashing is a soft-delete (UPDATE, not DELETE) /// so the blob must remain referenced until the file is permanently deleted. async fn delete_and_cleanup_with_perms( &self, id: &str, caller_id: Uuid, ) -> Result { self.require_file_perm(id, Permission::Delete, caller_id) .await?; // Step 1: Try trash (soft delete — file row stays, blob stays referenced) if let Some(trash) = &self.trash_service { info!("Moving file to trash: {}", id); match trash.move_to_trash(id, "file", caller_id).await { Ok(_) => { info!("File successfully moved to trash: {}", id); // Invalidate content cache — trashed files must not be served. if let Some(cc) = &self.content_cache { cc.invalidate(id).await; } // Do NOT decrement blob ref here — the file row still exists // (is_trashed = TRUE). The trigger will decrement when the // row is actually DELETEd during trash emptying. return Ok(true); // trashed } Err(err) => { error!("Could not move file to trash: {:?}", err); warn!("Falling back to permanent delete"); // fall through } } } else { warn!("Trash service not available, using permanent delete"); } // Step 2: Permanent delete — trigger handles blob ref_count self.delete_file(id).await?; Ok(false) // permanently deleted } async fn copy_folder_tree_with_perms( &self, source_folder_id: &str, caller_id: Uuid, target_parent_id: Option, dest_name: Option, ) -> Result { // copy_folder_tree = Read on the source folder + Create on the target parent. self.require_target_folder_perm(Some(source_folder_id), Permission::Read, caller_id) .await?; self.require_target_folder_perm(target_parent_id.as_deref(), Permission::Create, caller_id) .await?; self.copy_folder_tree(source_folder_id, target_parent_id, dest_name) .await } }