package entity

import (
	"crypto/sha1"
	"encoding/base32"
	"time"
)

type Faces []Face

// Face represents the face of a Person.
type Face struct {
	ID        string     `gorm:"type:VARBINARY(42);primary_key;auto_increment:false;" json:"ID" yaml:"ID"`
	PersonUID string     `gorm:"type:VARBINARY(42);index;" json:"PersonUID" yaml:"PersonUID,omitempty"`
	Embedding string     `gorm:"type:LONGTEXT;" json:"Embedding" yaml:"Embedding,omitempty"`
	CreatedAt time.Time  `json:"CreatedAt" yaml:"CreatedAt,omitempty"`
	UpdatedAt time.Time  `json:"UpdatedAt" yaml:"UpdatedAt,omitempty"`
	DeletedAt *time.Time `sql:"index" json:"DeletedAt,omitempty" yaml:"-"`
}

// UnknownFace can be used as a placeholder for unknown faces.
var UnknownFace = Face{
	ID:        "zz",
	PersonUID: UnknownPerson.PersonUID,
	Embedding: "",
}

// CreateUnknownFace initializes the database with a placeholder for unknown faces.
func CreateUnknownFace() {
	_ = UnknownFace.Create()
}

// TableName returns the entity database table name.
func (Face) TableName() string {
	return "faces_dev2"
}

// NewFace returns a new face.
func NewFace(personUID, embedding string) *Face {
	timeStamp := Timestamp()
	s := sha1.Sum([]byte(embedding))

	result := &Face{
		ID:        base32.StdEncoding.EncodeToString(s[:]),
		PersonUID: personUID,
		Embedding: embedding,
		CreatedAt: timeStamp,
		UpdatedAt: timeStamp,
	}

	return result
}

// UnmarshalEmbedding parses the face embedding JSON string.
func (m *Face) UnmarshalEmbedding() (result Embedding) {
	return UnmarshalEmbedding(m.Embedding)
}

// Save updates the existing or inserts a new face.
func (m *Face) Save() error {
	peopleMutex.Lock()
	defer peopleMutex.Unlock()

	return Save(m, "ID")
}

// Create inserts the face to the database.
func (m *Face) Create() error {
	peopleMutex.Lock()
	defer peopleMutex.Unlock()

	return Db().Create(m).Error
}

// Delete removes the face from the database.
func (m *Face) Delete() error {
	return Db().Delete(m).Error
}

// Deleted returns true if the face is deleted.
func (m *Face) Deleted() bool {
	return m.DeletedAt != nil
}

// Restore restores the face in the database.
func (m *Face) Restore() error {
	if m.Deleted() {
		return UnscopedDb().Model(m).Update("DeletedAt", nil).Error
	}

	return nil
}

// Update a face property in the database.
func (m *Face) Update(attr string, value interface{}) error {
	return UnscopedDb().Model(m).Update(attr, value).Error
}

// Updates face properties in the database.
func (m *Face) Updates(values interface{}) error {
	return UnscopedDb().Model(m).Updates(values).Error
}