feat: Sites feature backend

internal/infrastructure/useCases/DeleteMedia.go

@@ -3,6 +3,7 @@ package useCases
 import (
 	"RenewCMS/internal/domain/audit"
 	"RenewCMS/internal/domain/image"
+	"RenewCMS/internal/domain/site"
 	"context"
 )
 
@@ -10,10 +11,23 @@ type DeleteMediaUseCase struct {
 	imageRepository image.Repository
 	remover         mediaRemover
 	recorder        audit.Recorder
+	sites           site.Repository
 }
 
-func NewDeleteMediaUseCase(imageRepository image.Repository, storage image.Storage, variantFiles image.VariantFileRepository, recorder audit.Recorder) *DeleteMediaUseCase {
-	return &DeleteMediaUseCase{imageRepository, newMediaRemover(imageRepository, storage, variantFiles), recorder}
+func NewDeleteMediaUseCase(imageRepository image.Repository, storage image.Storage, variantFiles image.VariantFileRepository, recorder audit.Recorder, sites site.Repository) *DeleteMediaUseCase {
+	return &DeleteMediaUseCase{imageRepository: imageRepository, remover: newMediaRemover(imageRepository, storage, variantFiles), recorder: recorder, sites: sites}
+}
+
+func (u *DeleteMediaUseCase) removeClearingSites(img image.Image) error {
+	clearedSites, err := u.sites.ClearImage(img.ID)
+	if err != nil {
+		return err
+	}
+	if err := u.remover.remove(img); err != nil {
+		_ = u.sites.AssignImage(clearedSites, img.ID)
+		return err
+	}
+	return nil
 }
 
 func (u *DeleteMediaUseCase) Delete(ctx context.Context, id uint32) error {
@@ -24,7 +38,7 @@ func (u *DeleteMediaUseCase) Delete(ctx context.Context, id uint32) error {
 
 	event := audit.NewEvent(audit.ActionMediaDelete, audit.CategoryMedia)
 	event.Target = imageTarget(img)
-	if err := u.remover.remove(img); err != nil {
+	if err := u.removeClearingSites(img); err != nil {
 		recordFailure(ctx, u.recorder, event, err)
 		return err
 	}