2019-12-11 15:55:18 +00:00
|
|
|
package entity
|
2018-07-18 13:17:56 +00:00
|
|
|
|
|
|
|
import (
|
2019-06-18 04:37:10 +00:00
|
|
|
"strings"
|
2019-12-04 11:11:11 +00:00
|
|
|
"time"
|
2019-06-18 04:37:10 +00:00
|
|
|
|
|
|
|
"github.com/gosimple/slug"
|
2018-07-18 13:17:56 +00:00
|
|
|
"github.com/jinzhu/gorm"
|
2020-04-20 08:38:01 +00:00
|
|
|
"github.com/photoprism/photoprism/internal/form"
|
2020-01-12 13:00:56 +00:00
|
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
2020-04-20 08:38:01 +00:00
|
|
|
"github.com/ulule/deepcopier"
|
2018-07-18 13:17:56 +00:00
|
|
|
)
|
|
|
|
|
2020-02-21 00:14:45 +00:00
|
|
|
// Album represents a photo album
|
2018-07-18 13:17:56 +00:00
|
|
|
type Album struct {
|
2019-12-27 04:18:52 +00:00
|
|
|
ID uint `gorm:"primary_key"`
|
|
|
|
CoverUUID string `gorm:"type:varbinary(36);"`
|
|
|
|
AlbumUUID string `gorm:"type:varbinary(36);unique_index;"`
|
2020-01-06 01:14:17 +00:00
|
|
|
AlbumSlug string `gorm:"type:varbinary(128);index;"`
|
2019-12-04 14:14:04 +00:00
|
|
|
AlbumName string `gorm:"type:varchar(128);"`
|
2018-09-27 06:59:53 +00:00
|
|
|
AlbumDescription string `gorm:"type:text;"`
|
|
|
|
AlbumNotes string `gorm:"type:text;"`
|
2020-02-01 23:31:09 +00:00
|
|
|
AlbumOrder string `gorm:"type:varbinary(32);"`
|
2020-04-08 11:24:06 +00:00
|
|
|
AlbumTemplate string `gorm:"type:varbinary(256);"`
|
|
|
|
AlbumFavorite bool
|
|
|
|
Links []Link `gorm:"foreignkey:ShareUUID;association_foreignkey:AlbumUUID"`
|
2019-12-27 04:18:52 +00:00
|
|
|
CreatedAt time.Time
|
|
|
|
UpdatedAt time.Time
|
|
|
|
DeletedAt *time.Time `sql:"index"`
|
2018-07-18 13:17:56 +00:00
|
|
|
}
|
2019-06-04 16:26:35 +00:00
|
|
|
|
2020-02-21 00:14:45 +00:00
|
|
|
// BeforeCreate computes a random UUID when a new album is created in database
|
2019-06-04 16:26:35 +00:00
|
|
|
func (m *Album) BeforeCreate(scope *gorm.Scope) error {
|
2020-01-12 11:32:24 +00:00
|
|
|
if err := scope.SetColumn("AlbumUUID", rnd.PPID('a')); err != nil {
|
2019-12-06 09:26:57 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2019-06-04 16:26:35 +00:00
|
|
|
}
|
2019-06-18 04:37:10 +00:00
|
|
|
|
2020-02-21 00:14:45 +00:00
|
|
|
// NewAlbum creates a new album; default name is current month and year
|
2019-06-18 04:37:10 +00:00
|
|
|
func NewAlbum(albumName string) *Album {
|
|
|
|
albumName = strings.TrimSpace(albumName)
|
|
|
|
|
|
|
|
if albumName == "" {
|
2019-12-04 11:11:11 +00:00
|
|
|
albumName = time.Now().Format("January 2006")
|
2019-06-18 04:37:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
albumSlug := slug.Make(albumName)
|
|
|
|
|
|
|
|
result := &Album{
|
2020-04-20 10:53:58 +00:00
|
|
|
AlbumSlug: albumSlug,
|
|
|
|
AlbumName: albumName,
|
|
|
|
AlbumOrder: SortOrderOldest,
|
2019-06-18 04:37:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return result
|
|
|
|
}
|
2019-12-03 22:55:24 +00:00
|
|
|
|
2020-02-21 00:14:45 +00:00
|
|
|
// Rename an existing album
|
2019-12-03 22:55:24 +00:00
|
|
|
func (m *Album) Rename(albumName string) {
|
2019-12-04 11:11:11 +00:00
|
|
|
if albumName == "" {
|
|
|
|
albumName = m.CreatedAt.Format("January 2006")
|
|
|
|
}
|
|
|
|
|
2019-12-03 22:55:24 +00:00
|
|
|
m.AlbumName = strings.TrimSpace(albumName)
|
|
|
|
m.AlbumSlug = slug.Make(m.AlbumName)
|
|
|
|
}
|
2020-04-20 08:38:01 +00:00
|
|
|
|
|
|
|
// Save updates the entity using form data and stores it in the database.
|
|
|
|
func (m *Album) Save(f form.Album, db *gorm.DB) error {
|
|
|
|
if err := deepcopier.Copy(m).From(f); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if f.AlbumName != "" {
|
|
|
|
m.Rename(f.AlbumName)
|
|
|
|
}
|
|
|
|
|
|
|
|
return db.Save(m).Error
|
|
|
|
}
|