Initial work on Shares

core/share.go

@@ -2,13 +2,19 @@ package core
 
 import (
 	"context"
+	"strings"
+	"time"
 
+	"github.com/Masterminds/squirrel"
 	"github.com/deluan/rest"
-	gonanoid "github.com/matoous/go-nanoid"
+	gonanoid "github.com/matoous/go-nanoid/v2"
+	"github.com/navidrome/navidrome/log"
 	"github.com/navidrome/navidrome/model"
+	"github.com/navidrome/navidrome/utils/slice"
 )
 
 type Share interface {
+	Load(ctx context.Context, id string) (*model.Share, error)
 	NewRepository(ctx context.Context) rest.Repository
 }
 
@@ -22,6 +28,50 @@ type shareService struct {
 	ds model.DataStore
 }
 
+func (s *shareService) Load(ctx context.Context, id string) (*model.Share, error) {
+	repo := s.ds.Share(ctx)
+	entity, err := repo.(rest.Repository).Read(id)
+	if err != nil {
+		return nil, err
+	}
+	share := entity.(*model.Share)
+	now := time.Now()
+	share.LastVisitedAt = &now
+	share.VisitCount++
+
+	err = repo.(rest.Persistable).Update(id, share, "last_visited_at", "visit_count")
+	if err != nil {
+		log.Warn(ctx, "Could not increment visit count for share", "share", share.ID)
+	}
+
+	idList := strings.Split(share.ResourceIDs, ",")
+	switch share.ResourceType {
+	case "album":
+		share.Tracks, err = s.loadMediafiles(ctx, squirrel.Eq{"album_id": idList}, "album")
+	}
+	if err != nil {
+		return nil, err
+	}
+	return entity.(*model.Share), nil
+}
+
+func (s *shareService) loadMediafiles(ctx context.Context, filter squirrel.Eq, sort string) ([]model.ShareTrack, error) {
+	all, err := s.ds.MediaFile(ctx).GetAll(model.QueryOptions{Filters: filter, Sort: sort})
+	if err != nil {
+		return nil, err
+	}
+	return slice.Map(all, func(mf model.MediaFile) model.ShareTrack {
+		return model.ShareTrack{
+			ID:        mf.ID,
+			Title:     mf.Title,
+			Artist:    mf.Artist,
+			Album:     mf.Album,
+			Duration:  mf.Duration,
+			UpdatedAt: mf.UpdatedAt,
+		}
+	}), nil
+}
+
 func (s *shareService) NewRepository(ctx context.Context) rest.Repository {
 	repo := s.ds.Share(ctx)
 	wrapper := &shareRepositoryWrapper{
@@ -38,17 +88,37 @@ type shareRepositoryWrapper struct {
 	rest.Persistable
 }
 
+func (r *shareRepositoryWrapper) newId() (string, error) {
+	for {
+		id, err := gonanoid.Generate("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz", 10)
+		if err != nil {
+			return "", err
+		}
+		exists, err := r.Exists(id)
+		if err != nil {
+			return "", err
+		}
+		if !exists {
+			return id, nil
+		}
+	}
+}
+
 func (r *shareRepositoryWrapper) Save(entity interface{}) (string, error) {
 	s := entity.(*model.Share)
-	id, err := gonanoid.Generate("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz", 9)
+	id, err := r.newId()
 	if err != nil {
 		return "", err
 	}
-	s.Name = id
+	s.ID = id
+	if s.ExpiresAt.IsZero() {
+		exp := time.Now().Add(365 * 24 * time.Hour)
+		s.ExpiresAt = &exp
+	}
 	id, err = r.Persistable.Save(s)
 	return id, err
 }
 
 func (r *shareRepositoryWrapper) Update(id string, entity interface{}, _ ...string) error {
-	return r.Persistable.Update(id, entity, "description")
+	return r.Persistable.Update(id, entity, "description", "expires_at")
 }