photoprism/internal/entity/photo_user.go

53 lines
1.6 KiB
Go
Raw Normal View History

package entity
import "github.com/photoprism/photoprism/internal/event"
// PhotoUser represents the user and group ownership of a Photo and the corresponding permissions.
type PhotoUser struct {
UID string `gorm:"type:VARBINARY(42);primary_key;auto_increment:false" json:"UID" yaml:"UID"`
UserUID string `gorm:"type:VARBINARY(42);primary_key;auto_increment:false;index" json:"UserUID,omitempty" yaml:"UserUID,omitempty"`
TeamUID string `gorm:"type:VARBINARY(42);index" json:"TeamUID,omitempty" yaml:"TeamUID,omitempty"`
Perm uint `json:"Perm,omitempty" yaml:"Perm,omitempty"`
}
// TableName returns the database table name.
func (PhotoUser) TableName() string {
return "photos_users"
}
// NewPhotoUser creates a new entity model.
func NewPhotoUser(uid, userUid, teamUid string, perm uint) *PhotoUser {
result := &PhotoUser{
UID: uid,
UserUID: userUid,
TeamUID: teamUid,
Perm: perm,
}
return result
}
// Create inserts a new record into the database.
func (m *PhotoUser) Create() error {
return Db().Create(m).Error
}
// Save updates the record in the database or inserts a new record if it does not already exist.
func (m *PhotoUser) Save() error {
return Db().Save(m).Error
}
// FirstOrCreatePhotoUser returns the existing record or inserts a new record if it does not already exist.
func FirstOrCreatePhotoUser(m *PhotoUser) *PhotoUser {
found := PhotoUser{}
if err := Db().Where("uid = ?", m.UID).First(&found).Error; err == nil {
return &found
} else if err = m.Create(); err != nil {
event.AuditErr([]string{"photo %s", "failed to set owner and permissions", "%s"}, m.UID, err)
return nil
}
return m
}