mirror of
https://github.com/navidrome/navidrome.git
synced 2025-05-24 13:57:44 +03:00
116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
package db_sql
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/cloudsonic/sonic-server/domain"
|
|
"github.com/cloudsonic/sonic-server/persistence"
|
|
)
|
|
|
|
type MediaFile struct {
|
|
ID string `orm:"pk;column(id)"`
|
|
Path string ``
|
|
Title string `orm:"index"`
|
|
Album string ``
|
|
Artist string ``
|
|
ArtistID string `orm:"column(artist_id)"`
|
|
AlbumArtist string ``
|
|
AlbumID string `orm:"column(album_id);index"`
|
|
HasCoverArt bool ``
|
|
TrackNumber int ``
|
|
DiscNumber int ``
|
|
Year int ``
|
|
Size string ``
|
|
Suffix string ``
|
|
Duration int ``
|
|
BitRate int ``
|
|
Genre string ``
|
|
Compilation bool ``
|
|
PlayCount int ``
|
|
PlayDate time.Time `orm:"null"`
|
|
Rating int ``
|
|
Starred bool `orm:"index"`
|
|
StarredAt time.Time `orm:"null"`
|
|
CreatedAt time.Time `orm:"null"`
|
|
UpdatedAt time.Time `orm:"null"`
|
|
}
|
|
|
|
type mediaFileRepository struct {
|
|
sqlRepository
|
|
}
|
|
|
|
func NewMediaFileRepository() domain.MediaFileRepository {
|
|
r := &mediaFileRepository{}
|
|
r.entityName = "media_file"
|
|
return r
|
|
}
|
|
|
|
func (r *mediaFileRepository) Put(m *domain.MediaFile) error {
|
|
tm := MediaFile(*m)
|
|
return r.put(m.ID, &tm)
|
|
}
|
|
|
|
func (r *mediaFileRepository) Get(id string) (*domain.MediaFile, error) {
|
|
tm := MediaFile{ID: id}
|
|
err := Db().Read(&tm)
|
|
if err == orm.ErrNoRows {
|
|
return nil, domain.ErrNotFound
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
a := domain.MediaFile(tm)
|
|
return &a, nil
|
|
}
|
|
|
|
func (r *mediaFileRepository) toMediaFiles(all []MediaFile) (domain.MediaFiles, error) {
|
|
result := make(domain.MediaFiles, len(all))
|
|
for i, m := range all {
|
|
result[i] = domain.MediaFile(m)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (r *mediaFileRepository) FindByAlbum(albumId string) (domain.MediaFiles, error) {
|
|
var mfs []MediaFile
|
|
_, err := r.newQuery(Db()).Filter("album_id", albumId).OrderBy("disc_number", "track_number").All(&mfs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.toMediaFiles(mfs)
|
|
}
|
|
|
|
func (r *mediaFileRepository) GetStarred(options ...domain.QueryOptions) (domain.MediaFiles, error) {
|
|
var starred []MediaFile
|
|
_, err := r.newQuery(Db(), options...).Filter("starred", true).All(&starred)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.toMediaFiles(starred)
|
|
}
|
|
|
|
func (r *mediaFileRepository) GetAllIds() ([]string, error) {
|
|
qs := r.newQuery(Db())
|
|
var values []orm.Params
|
|
num, err := qs.Values(&values, "id")
|
|
if num == 0 {
|
|
return nil, err
|
|
}
|
|
|
|
result := persistence.CollectValue(values, func(item interface{}) string {
|
|
return item.(orm.Params)["ID"].(string)
|
|
})
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func (r *mediaFileRepository) PurgeInactive(activeList domain.MediaFiles) ([]string, error) {
|
|
return r.purgeInactive(activeList, func(item interface{}) string {
|
|
return item.(domain.MediaFile).ID
|
|
})
|
|
}
|
|
|
|
var _ domain.MediaFileRepository = (*mediaFileRepository)(nil)
|
|
var _ = domain.MediaFile(MediaFile{})
|