photoprism/internal/entity/duplicate.go

103 lines
3.1 KiB
Go
Raw Normal View History

package entity
import (
"fmt"
"github.com/photoprism/photoprism/pkg/txt"
)
type Duplicates []Duplicate
type DuplicatesMap map[string]Duplicate
// Duplicate represents an exact file duplicate.
type Duplicate struct {
FileName string `gorm:"type:VARBINARY(768);primary_key;" json:"Name" yaml:"Name"`
FileRoot string `gorm:"type:VARBINARY(16);primary_key;default:'/';" json:"Root" yaml:"Root,omitempty"`
FileHash string `gorm:"type:VARBINARY(128);default:'';index" json:"Hash" yaml:"Hash,omitempty"`
FileSize int64 `json:"Size" yaml:"Size,omitempty"`
ModTime int64 `json:"ModTime" yaml:"-"`
}
func AddDuplicate(fileName, fileRoot, fileHash string, fileSize, modTime int64) error {
if fileName == "" {
return fmt.Errorf("duplicate: file name must not be empty (add)")
} else if fileHash == "" {
return fmt.Errorf("duplicate: file hash must not be empty (add)")
} else if modTime == 0 {
return fmt.Errorf("duplicate: mod time must not be empty (add)")
} else if fileRoot == "" {
return fmt.Errorf("duplicate: file root must not be empty (add)")
}
duplicate := &Duplicate{
FileName: fileName,
FileRoot: fileRoot,
FileHash: fileHash,
FileSize: fileSize,
ModTime: modTime,
}
if err := duplicate.Create(); err == nil {
return nil
} else if err := duplicate.Save(); err != nil {
return err
}
return nil
}
func RemoveDuplicate(fileName, fileRoot string) error {
if fileName == "" {
return fmt.Errorf("duplicate: file name must not be empty (remove)")
} else if fileRoot == "" {
return fmt.Errorf("duplicate: file root must not be empty (remove)")
}
if err := UnscopedDb().Delete(Duplicate{}, "file_name = ? AND file_root = ?", fileName, fileRoot).Error; err != nil {
log.Errorf("duplicate: %s (remove %s)", err, txt.Quote(fileName))
return err
}
return nil
}
// Find returns a duplicate from the database.
func (m *Duplicate) Find() error {
return UnscopedDb().First(m, "file_name = ? AND file_root = ?", m.FileName, m.FileRoot).Error
}
// Create inserts a new row to the database.
func (m *Duplicate) Create() error {
if m.FileName == "" {
return fmt.Errorf("duplicate: file name must not be empty (create)")
} else if m.FileHash == "" {
return fmt.Errorf("duplicate: file hash must not be empty (create)")
} else if m.ModTime == 0 {
return fmt.Errorf("duplicate: mod time must not be empty (create)")
} else if m.FileRoot == "" {
return fmt.Errorf("duplicate: file root must not be empty (create)")
}
return UnscopedDb().Create(m).Error
}
// Saves the duplicates in the database.
func (m *Duplicate) Save() error {
if m.FileName == "" {
return fmt.Errorf("duplicate: file name must not be empty (save)")
} else if m.FileHash == "" {
return fmt.Errorf("duplicate: file hash must not be empty (save)")
} else if m.ModTime == 0 {
return fmt.Errorf("duplicate: mod time must not be empty (save)")
} else if m.FileRoot == "" {
return fmt.Errorf("duplicate: file root must not be empty (save)")
}
if err := UnscopedDb().Save(m).Error; err != nil {
log.Errorf("duplicate: %s (save %s)", err, txt.Quote(m.FileName))
return err
}
return nil
}