Initial support for playlists. Missing permissions

engine/playlists.go

@@ -3,15 +3,17 @@ package engine
 import (
 	"context"
 
+	"github.com/cloudsonic/sonic-server/consts"
 	"github.com/cloudsonic/sonic-server/model"
+	"github.com/cloudsonic/sonic-server/utils"
 )
 
 type Playlists interface {
 	GetAll() (model.Playlists, error)
 	Get(id string) (*PlaylistInfo, error)
-	Create(ctx context.Context, name string, ids []string) error
+	Create(ctx context.Context, playlistId, name string, ids []string) error
 	Delete(ctx context.Context, playlistId string) error
-	Update(playlistId string, name *string, idsToAdd []string, idxToRemove []int) error
+	Update(ctx context.Context, playlistId string, name *string, idsToAdd []string, idxToRemove []int) error
 }
 
 func NewPlaylists(ds model.DataStore) Playlists {
@@ -22,6 +24,62 @@ type playlists struct {
 	ds model.DataStore
 }
 
+func (p *playlists) Create(ctx context.Context, playlistId, name string, ids []string) error {
+	owner := consts.InitialUserName
+	user, ok := ctx.Value("user").(*model.User)
+	if ok {
+		owner = user.UserName
+	}
+	var pls *model.Playlist
+	var err error
+	// If playlistID is present, override tracks
+	if playlistId != "" {
+		pls, err = p.ds.Playlist().Get(playlistId)
+		if err != nil {
+			return err
+		}
+		pls.Tracks = nil
+	} else {
+		pls = &model.Playlist{
+			Name:  name,
+			Owner: owner,
+		}
+	}
+	for _, id := range ids {
+		pls.Tracks = append(pls.Tracks, model.MediaFile{ID: id})
+	}
+
+	return p.ds.Playlist().Put(pls)
+}
+
+func (p *playlists) Delete(ctx context.Context, playlistId string) error {
+	return p.ds.Playlist().Delete(playlistId)
+}
+
+func (p *playlists) Update(ctx context.Context, playlistId string, name *string, idsToAdd []string, idxToRemove []int) error {
+	pls, err := p.ds.Playlist().Get(playlistId)
+	if err != nil {
+		return err
+	}
+	if name != nil {
+		pls.Name = *name
+	}
+	newTracks := model.MediaFiles{}
+	for i, t := range pls.Tracks {
+		if utils.IntInSlice(i, idxToRemove) {
+			continue
+		}
+		newTracks = append(newTracks, t)
+	}
+
+	for _, id := range idsToAdd {
+		newTracks = append(newTracks, model.MediaFile{ID: id})
+	}
+	pls.Tracks = newTracks
+
+	return p.ds.Playlist().Put(pls)
+}
+
 func (p *playlists) GetAll() (model.Playlists, error) {
 	return p.ds.Playlist().GetAll(model.QueryOptions{})
 }
@@ -37,21 +95,6 @@ type PlaylistInfo struct {
 	Comment   string
 }
 
-func (p *playlists) Create(ctx context.Context, name string, ids []string) error {
-	// TODO
-	return nil
-}
-
-func (p *playlists) Delete(ctx context.Context, playlistId string) error {
-	// TODO
-	return nil
-}
-
-func (p *playlists) Update(playlistId string, name *string, idsToAdd []string, idxToRemove []int) error {
-	// TODO
-	return nil
-}
-
 func (p *playlists) Get(id string) (*PlaylistInfo, error) {
 	pl, err := p.ds.Playlist().Get(id)
 	if err != nil {
@@ -70,8 +113,8 @@ func (p *playlists) Get(id string) (*PlaylistInfo, error) {
 	pinfo.Entries = make(Entries, len(pl.Tracks))
 
 	// TODO Optimize: Get all tracks at once
-	for i, mfId := range pl.Tracks {
-		mf, err := p.ds.MediaFile().Get(mfId)
+	for i, mf := range pl.Tracks {
+		mf, err := p.ds.MediaFile().Get(mf.ID)
 		if err != nil {
 			return nil, err
 		}