feat: Sites feature backend

internal/infrastructure/useCases/ListMedia.go

@@ -3,6 +3,7 @@ package useCases
 import (
 	"RenewCMS/internal/domain/article"
 	"RenewCMS/internal/domain/image"
+	"RenewCMS/internal/domain/site"
 	"RenewCMS/internal/domain/user"
 	"strings"
 )
@@ -36,10 +37,11 @@ type ListMediaUseCase struct {
 	articleRepository article.Repository
 	userRepository    user.Repository
 	variantFiles      image.VariantFileRepository
+	sites             site.Repository
 }
 
-func NewListMediaUseCase(imageRepository image.Repository, articleRepository article.Repository, userRepository user.Repository, variantFiles image.VariantFileRepository) *ListMediaUseCase {
-	return &ListMediaUseCase{imageRepository, articleRepository, userRepository, variantFiles}
+func NewListMediaUseCase(imageRepository image.Repository, articleRepository article.Repository, userRepository user.Repository, variantFiles image.VariantFileRepository, sites site.Repository) *ListMediaUseCase {
+	return &ListMediaUseCase{imageRepository: imageRepository, articleRepository: articleRepository, userRepository: userRepository, variantFiles: variantFiles, sites: sites}
 }
 
 func (l *ListMediaUseCase) List(query image.ListQuery) (MediaPage, error) {
@@ -106,6 +108,10 @@ func (l *ListMediaUseCase) Describe(img image.Image) (MediaItem, error) {
 	if err != nil {
 		return MediaItem{}, err
 	}
+	siteUsage, err := siteImageUsage(l.sites)
+	if err != nil {
+		return MediaItem{}, err
+	}
 	uploadedBy, err := newUploaderNames(l.userRepository).nameOf(img.UploadedByID)
 	if err != nil {
 		return MediaItem{}, err
@@ -129,7 +135,7 @@ func (l *ListMediaUseCase) Describe(img image.Image) (MediaItem, error) {
 		variantSize += row.Size
 	}
 
-	return MediaItem{Image: img, UploadedBy: uploadedBy, UsedIn: len(articles), Duplicate: duplicate, VariantCount: len(rows), VariantSize: variantSize}, nil
+	return MediaItem{Image: img, UploadedBy: uploadedBy, UsedIn: len(articles) + siteUsage[img.ID], Duplicate: duplicate, VariantCount: len(rows), VariantSize: variantSize}, nil
 }
 
 func (l *ListMediaUseCase) usageCounter() (func(image.Image) int, error) {
@@ -137,9 +143,13 @@ func (l *ListMediaUseCase) usageCounter() (func(image.Image) int, error) {
 	if err != nil {
 		return nil, err
 	}
+	siteUsage, err := siteImageUsage(l.sites)
+	if err != nil {
+		return nil, err
+	}
 	return func(img image.Image) int {
 		storedName := image.StoredName(img.Path)
-		used := 0
+		used := siteUsage[img.ID]
 		for _, a := range articles {
 			if strings.Contains(a.Body, storedName) {
 				used++