commit

backend/internal/albums/service.go

@@ -17,15 +17,20 @@ type MemberLookup interface {
 	Role(ctx context.Context, albumID, userID string) (string, error)
 }
 
+type PhotoLookup interface {
+	PhotoInAlbum(ctx context.Context, photoID, albumID string) (bool, error)
+}
+
 type Service struct {
 	store   *Store
 	members MemberLookup
+	photos  PhotoLookup
 	hub     *sse.Hub
 	storage *storage.Client
 }
 
-func NewService(store *Store, members MemberLookup, hub *sse.Hub, storage *storage.Client) *Service {
-	return &Service{store: store, members: members, hub: hub, storage: storage}
+func NewService(store *Store, members MemberLookup, hub *sse.Hub, storage *storage.Client, photos PhotoLookup) *Service {
+	return &Service{store: store, members: members, photos: photos, hub: hub, storage: storage}
 }
 
 type CreateInput struct {
@@ -51,7 +56,23 @@ func (s *Service) Create(ctx context.Context, userID string, in CreateInput) (*m
 }
 
 func (s *Service) List(ctx context.Context, userID string) ([]models.Album, error) {
-	return s.store.ListForUser(ctx, userID)
+	list, err := s.store.ListForUser(ctx, userID)
+	if err != nil {
+		return nil, err
+	}
+	for i := range list {
+		s.attachCoverURL(ctx, &list[i])
+	}
+	return list, nil
+}
+
+func (s *Service) attachCoverURL(ctx context.Context, a *models.Album) {
+	if a == nil || a.CoverThumbKey == "" || s.storage == nil {
+		return
+	}
+	if url, err := s.storage.PresignGet(ctx, a.CoverThumbKey); err == nil {
+		a.CoverURL = url
+	}
 }
 
 func (s *Service) Access(ctx context.Context, albumID, userID string) (*models.Album, string, error) {
@@ -62,6 +83,7 @@ func (s *Service) Access(ctx context.Context, albumID, userID string) (*models.A
 		}
 		return nil, "", err
 	}
+	s.attachCoverURL(ctx, a)
 	if a.OwnerID == userID {
 		return a, roles.Owner, nil
 	}
@@ -107,11 +129,23 @@ func (s *Service) Update(ctx context.Context, albumID, userID string, in UpdateI
 	if in.ClearCover {
 		a.CoverPhotoID = nil
 	} else if in.CoverPhotoID != nil {
+		ok, err := s.photos.PhotoInAlbum(ctx, *in.CoverPhotoID, albumID)
+		if err != nil {
+			return nil, err
+		}
+		if !ok {
+			return nil, httpx.BadRequest("invalid_cover", "cover photo must belong to the album")
+		}
 		a.CoverPhotoID = in.CoverPhotoID
 	}
 	if err := s.store.Update(ctx, a); err != nil {
 		return nil, err
 	}
+	if fresh, err := s.store.GetByID(ctx, albumID); err == nil {
+		a.CoverThumbKey = fresh.CoverThumbKey
+		a.CoverURL = ""
+		s.attachCoverURL(ctx, a)
+	}
 	s.hub.Publish(albumID, "album.updated", a)
 	return a, nil
 }