quintodrome/engine/common.go

package engine

import (
	"context"
	"fmt"
	"time"

	"github.com/cloudsonic/sonic-server/model"
)

type Entry struct {
	Id          string
	Title       string
	IsDir       bool
	Parent      string
	Album       string
	Year        int
	Artist      string
	Genre       string
	CoverArt    string
	Starred     time.Time
	Track       int
	Duration    int
	Size        string
	Suffix      string
	BitRate     int
	ContentType string
	Path        string
	PlayCount   int32
	DiscNumber  int
	Created     time.Time
	AlbumId     string
	ArtistId    string
	Type        string
	UserRating  int
	SongCount   int

	UserName   string
	MinutesAgo int
	PlayerId   int
	PlayerName string
	AlbumCount int

	AbsolutePath string
}

type Entries []Entry

func FromArtist(ar *model.Artist, ann *model.Annotation) Entry {
	e := Entry{}
	e.Id = ar.ID
	e.Title = ar.Name
	e.AlbumCount = ar.AlbumCount
	e.IsDir = true
	if ann != nil {
		e.Starred = ann.StarredAt
	}
	return e
}

func FromAlbum(al *model.Album, ann *model.Annotation) Entry {
	e := Entry{}
	e.Id = al.ID
	e.Title = al.Name
	e.IsDir = true
	e.Parent = al.ArtistID
	e.Album = al.Name
	e.Year = al.Year
	e.Artist = al.AlbumArtist
	e.Genre = al.Genre
	e.CoverArt = al.CoverArtId
	e.Created = al.CreatedAt
	e.AlbumId = al.ID
	e.ArtistId = al.ArtistID
	e.Duration = al.Duration
	e.SongCount = al.SongCount
	if ann != nil {
		e.Starred = ann.StarredAt
		e.PlayCount = int32(ann.PlayCount)
		e.UserRating = ann.Rating
	}
	return e
}

func FromMediaFile(mf *model.MediaFile, ann *model.Annotation) Entry {
	e := Entry{}
	e.Id = mf.ID
	e.Title = mf.Title
	e.IsDir = false
	e.Parent = mf.AlbumID
	e.Album = mf.Album
	e.Year = mf.Year
	e.Artist = mf.Artist
	e.Genre = mf.Genre
	e.Track = mf.TrackNumber
	e.Duration = mf.Duration
	e.Size = mf.Size
	e.Suffix = mf.Suffix
	e.BitRate = mf.BitRate
	if mf.HasCoverArt {
		e.CoverArt = mf.ID
	}
	e.ContentType = mf.ContentType()
	e.AbsolutePath = mf.Path
	// Creates a "pseudo" Path, to avoid sending absolute paths to the client
	if mf.Path != "" {
		e.Path = fmt.Sprintf("%s/%s/%s.%s", realArtistName(mf), mf.Album, mf.Title, mf.Suffix)
	}
	e.DiscNumber = mf.DiscNumber
	e.Created = mf.CreatedAt
	e.AlbumId = mf.AlbumID
	e.ArtistId = mf.ArtistID
	e.Type = "music" // TODO Hardcoded for now
	if ann != nil {
		e.PlayCount = int32(ann.PlayCount)
		e.Starred = ann.StarredAt
		e.UserRating = ann.Rating
	}
	return e
}

func realArtistName(mf *model.MediaFile) string {
	switch {
	case mf.Compilation:
		return "Various Artists"
	case mf.AlbumArtist != "":
		return mf.AlbumArtist
	}

	return mf.Artist
}

func FromAlbums(albums model.Albums, annMap model.AnnotationMap) Entries {
	entries := make(Entries, len(albums))
	for i, al := range albums {
		ann := annMap[al.ID]
		entries[i] = FromAlbum(&al, &ann)
	}
	return entries
}

func FromMediaFiles(mfs model.MediaFiles, annMap model.AnnotationMap) Entries {
	entries := make(Entries, len(mfs))
	for i, mf := range mfs {
		ann := annMap[mf.ID]
		entries[i] = FromMediaFile(&mf, &ann)
	}
	return entries
}

func FromArtists(ars model.Artists, annMap model.AnnotationMap) Entries {
	entries := make(Entries, len(ars))
	for i, ar := range ars {
		ann := annMap[ar.ID]
		entries[i] = FromArtist(&ar, &ann)
	}
	return entries
}

func getUserID(ctx context.Context) string {
	user, ok := ctx.Value("user").(*model.User)
	if ok {
		return user.ID
	}
	return ""
}