Files
RenewCMS/internal/infrastructure/useCases/DeleteMedia.go
T

72 lines
2.3 KiB
Go

package useCases
import (
"RenewCMS/internal/domain/article"
"RenewCMS/internal/domain/audit"
"RenewCMS/internal/domain/author"
"RenewCMS/internal/domain/image"
"RenewCMS/internal/domain/site"
"context"
"log"
)
type DeleteMediaUseCase struct {
imageRepository image.Repository
remover mediaRemover
recorder audit.Recorder
sites site.Repository
authors author.Repository
articles article.Repository
}
func NewDeleteMediaUseCase(imageRepository image.Repository, storage image.Storage, variantFiles image.VariantFileRepository, recorder audit.Recorder, sites site.Repository, authors author.Repository, articles article.Repository) *DeleteMediaUseCase {
return &DeleteMediaUseCase{imageRepository: imageRepository, remover: newMediaRemover(imageRepository, storage, variantFiles), recorder: recorder, sites: sites, authors: authors, articles: articles}
}
func logCompensation(step string, err error) {
if err != nil {
log.Println("media compensation failed to", step+":", err)
}
}
func (u *DeleteMediaUseCase) removeClearingSites(img image.Image) error {
clearedSites, err := u.sites.ClearImage(img.ID)
if err != nil {
return err
}
clearedAuthors, err := u.authors.ClearAvatar(img.ID)
if err != nil {
logCompensation("restore site images", u.sites.AssignImage(clearedSites, img.ID))
return err
}
clearedArticles, err := u.articles.ClearImage(img.ID)
if err != nil {
logCompensation("restore site images", u.sites.AssignImage(clearedSites, img.ID))
logCompensation("restore author avatars", u.authors.AssignAvatar(clearedAuthors, img.ID))
return err
}
if err := u.remover.remove(img); err != nil {
logCompensation("restore site images", u.sites.AssignImage(clearedSites, img.ID))
logCompensation("restore author avatars", u.authors.AssignAvatar(clearedAuthors, img.ID))
logCompensation("restore article images", u.articles.RestoreImage(clearedArticles, img.ID))
return err
}
return nil
}
func (u *DeleteMediaUseCase) Delete(ctx context.Context, id uint32) error {
img, err := u.imageRepository.Get(id)
if err != nil {
return err
}
event := audit.NewEvent(audit.ActionMediaDelete, audit.CategoryMedia)
event.Target = imageTarget(img)
if err := u.removeClearingSites(img); err != nil {
recordFailure(ctx, u.recorder, event, err)
return err
}
u.recorder.Record(ctx, event)
return nil
}