97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package entity
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gosimple/slug"
|
|
"github.com/photoprism/photoprism/pkg/txt"
|
|
)
|
|
|
|
// Camera model and make (as extracted from UpdateExif metadata)
|
|
type Camera struct {
|
|
ID uint `gorm:"primary_key" json:"ID" yaml:"ID"`
|
|
CameraSlug string `gorm:"type:varbinary(255);unique_index;" json:"Slug" yaml:"-"`
|
|
CameraModel string `gorm:"type:varchar(255);" json:"Model" yaml:"Model"`
|
|
CameraMake string `gorm:"type:varchar(255);" json:"Make" yaml:"Make"`
|
|
CameraType string `gorm:"type:varchar(255);" json:"Type,omitempty" yaml:"Type,omitempty"`
|
|
CameraDescription string `gorm:"type:text;" json:"Description,omitempty" yaml:"Description,omitempty"`
|
|
CameraNotes string `gorm:"type:text;" json:"Notes,omitempty" yaml:"Notes,omitempty"`
|
|
CreatedAt time.Time `json:"-" yaml:"-"`
|
|
UpdatedAt time.Time `json:"-" yaml:"-"`
|
|
DeletedAt *time.Time `sql:"index" json:"-" yaml:"-"`
|
|
}
|
|
|
|
var UnknownCamera = Camera{
|
|
CameraModel: "Unknown",
|
|
CameraMake: "",
|
|
CameraSlug: "zz",
|
|
}
|
|
|
|
// CreateUnknownCamera initializes the database with an unknown camera if not exists
|
|
func CreateUnknownCamera() {
|
|
FirstOrCreateCamera(&UnknownCamera)
|
|
}
|
|
|
|
// NewCamera creates a camera entity from a model name and a make name.
|
|
func NewCamera(modelName string, makeName string) *Camera {
|
|
modelName = txt.Clip(modelName, txt.ClipDefault)
|
|
makeName = txt.Clip(makeName, txt.ClipDefault)
|
|
|
|
if modelName == "" {
|
|
return &UnknownCamera
|
|
} else if strings.HasPrefix(modelName, makeName) {
|
|
modelName = strings.TrimSpace(modelName[len(makeName):])
|
|
}
|
|
|
|
var cameraSlug string
|
|
|
|
if makeName != "" {
|
|
cameraSlug = slug.Make(txt.Clip(makeName+" "+modelName, txt.ClipSlug))
|
|
} else {
|
|
cameraSlug = slug.Make(txt.Clip(modelName, txt.ClipSlug))
|
|
}
|
|
|
|
result := &Camera{
|
|
CameraModel: modelName,
|
|
CameraMake: makeName,
|
|
CameraSlug: cameraSlug,
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
// Create inserts a new row to the database.
|
|
func (m *Camera) Create() error {
|
|
if err := Db().Create(m).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// FirstOrCreateCamera inserts a new row if not exists.
|
|
func FirstOrCreateCamera(m *Camera) *Camera {
|
|
result := Camera{}
|
|
|
|
if err := Db().Where("camera_model = ? AND camera_make = ?", m.CameraModel, m.CameraMake).First(&result).Error; err == nil {
|
|
return &result
|
|
} else if err := m.Create(); err != nil {
|
|
log.Errorf("camera: %s", err)
|
|
return nil
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
// String returns a string designing the given Camera entity
|
|
func (m *Camera) String() string {
|
|
if m.CameraMake != "" && m.CameraModel != "" {
|
|
return fmt.Sprintf("%s %s", m.CameraMake, m.CameraModel)
|
|
} else if m.CameraModel != "" {
|
|
return m.CameraModel
|
|
}
|
|
|
|
return ""
|
|
}
|