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 }