59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package entity
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/gosimple/slug"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/photoprism/photoprism/internal/util"
|
|
)
|
|
|
|
// Labels for photo, album and location categorization
|
|
type Label struct {
|
|
Model
|
|
LabelUUID string `gorm:"unique_index;"`
|
|
LabelSlug string `gorm:"type:varchar(128);index;"`
|
|
LabelName string `gorm:"type:varchar(128);"`
|
|
LabelPriority int
|
|
LabelFavorite bool
|
|
LabelDescription string `gorm:"type:text;"`
|
|
LabelNotes string `gorm:"type:text;"`
|
|
LabelCategories []*Label `gorm:"many2many:categories;association_jointable_foreignkey:category_id"`
|
|
New bool `gorm:"-"`
|
|
}
|
|
|
|
func (m *Label) BeforeCreate(scope *gorm.Scope) error {
|
|
if err := scope.SetColumn("LabelUUID", util.UUID()); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func NewLabel(labelName string, labelPriority int) *Label {
|
|
labelName = strings.TrimSpace(labelName)
|
|
|
|
if labelName == "" {
|
|
labelName = "Unknown"
|
|
}
|
|
|
|
labelSlug := slug.Make(labelName)
|
|
|
|
result := &Label{
|
|
LabelName: labelName,
|
|
LabelSlug: labelSlug,
|
|
LabelPriority: labelPriority,
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func (m *Label) FirstOrCreate(db *gorm.DB) *Label {
|
|
db.FirstOrCreate(m, "label_slug = ?", m.LabelSlug)
|
|
|
|
return m
|
|
}
|
|
|
|
func (m *Label) AfterCreate(scope *gorm.Scope) error {
|
|
return scope.SetColumn("New", true)
|
|
}
|