package entity import ( "fmt" "path/filepath" "strings" "time" "github.com/dustin/go-humanize/english" "github.com/photoprism/photoprism/internal/classify" "github.com/photoprism/photoprism/pkg/clean" "github.com/photoprism/photoprism/pkg/fs" "github.com/photoprism/photoprism/pkg/txt" ) // HasTitle checks if the photo has a title. func (m *Photo) HasTitle() bool { return m.PhotoTitle != "" } // NoTitle checks if the photo has no Title func (m *Photo) NoTitle() bool { return m.PhotoTitle == "" } // SetTitle changes the photo title and clips it to 300 characters. func (m *Photo) SetTitle(title, source string) { title = strings.Trim(title, "_&|{}<>: \n\r\t\\") title = strings.ReplaceAll(title, "\"", "'") title = txt.Shorten(title, txt.ClipLongName, txt.Ellipsis) if title == "" { return } if (SrcPriority[source] < SrcPriority[m.TitleSrc]) && m.HasTitle() { return } m.PhotoTitle = title m.TitleSrc = source } // UpdateTitle updated the photo title based on location and labels. func (m *Photo) UpdateTitle(labels classify.Labels) error { if m.TitleSrc != SrcAuto && m.HasTitle() { return fmt.Errorf("photo: %s keeps existing %s title", m.String(), SrcString(m.TitleSrc)) } var names string var knownLocation bool start := time.Now() oldTitle := m.PhotoTitle fileTitle := m.FileTitle() people := m.SubjectNames() m.UpdateDescription(people) if n := len(people); n > 0 && n < 4 { names = txt.JoinNames(people, true) } if m.LocationLoaded() && m.TrustedLocation() { knownLocation = true loc := m.Cell // TODO: User defined title format if names != "" { log.Debugf("photo: %s title based on %s (%s)", m.String(), english.Plural(len(people), "person", "people"), clean.Log(names)) if l := len([]rune(names)); l > 35 { m.SetTitle(names, SrcAuto) } else if l > 20 && (loc.NoCity() || loc.LongCity()) { m.SetTitle(fmt.Sprintf("%s / %s", names, m.TakenAt.Format("2006")), SrcAuto) } else if l > 20 { m.SetTitle(fmt.Sprintf("%s / %s", names, loc.City()), SrcAuto) } else if loc.NoCity() || loc.LongCity() { m.SetTitle(fmt.Sprintf("%s / %s / %s", names, loc.CountryName(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", names, loc.City(), m.TakenAt.Format("2006")), SrcAuto) } } else if title := labels.Title(loc.Name()); title != "" { log.Debugf("photo: %s title based on label %s", m.String(), clean.Log(title)) if loc.NoCity() || loc.LongCity() || loc.CityContains(title) { m.SetTitle(fmt.Sprintf("%s / %s / %s", txt.Title(title), loc.CountryName(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", txt.Title(title), loc.City(), m.TakenAt.Format("2006")), SrcAuto) } } else if loc.Name() != "" && loc.City() != "" { if len(loc.Name()) > 45 { m.SetTitle(txt.Title(loc.Name()), SrcAuto) } else if len(loc.Name()) > 20 || len(loc.City()) > 16 || strings.Contains(loc.Name(), loc.City()) { m.SetTitle(fmt.Sprintf("%s / %s", loc.Name(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", loc.Name(), loc.City(), m.TakenAt.Format("2006")), SrcAuto) } } else if loc.City() != "" && loc.CountryName() != "" { if len(loc.City()) > 20 { m.SetTitle(fmt.Sprintf("%s / %s", loc.City(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", loc.City(), loc.CountryName(), m.TakenAt.Format("2006")), SrcAuto) } } } else if m.PlaceLoaded() { knownLocation = true if names != "" { log.Debugf("photo: %s title based on %s (%s)", m.String(), english.Plural(len(people), "person", "people"), clean.Log(names)) if l := len([]rune(names)); l > 35 { m.SetTitle(names, SrcAuto) } else if l > 20 && (m.Place.NoCity() || m.Place.LongCity()) { m.SetTitle(fmt.Sprintf("%s / %s", names, m.TakenAt.Format("2006")), SrcAuto) } else if l > 20 { m.SetTitle(fmt.Sprintf("%s / %s", names, m.Place.City()), SrcAuto) } else if m.Place.NoCity() || m.Place.LongCity() { m.SetTitle(fmt.Sprintf("%s / %s / %s", names, m.Place.CountryName(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", names, m.Place.City(), m.TakenAt.Format("2006")), SrcAuto) } } else if title := labels.Title(fileTitle); title != "" { log.Debugf("photo: %s title based on label %s", m.String(), clean.Log(title)) if m.Place.NoCity() || m.Place.LongCity() || m.Place.CityContains(title) { m.SetTitle(fmt.Sprintf("%s / %s / %s", txt.Title(title), m.Place.CountryName(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", txt.Title(title), m.Place.City(), m.TakenAt.Format("2006")), SrcAuto) } } else if m.Place.City() != "" && m.Place.CountryName() != "" { if len(m.Place.City()) > 20 { m.SetTitle(fmt.Sprintf("%s / %s", m.Place.City(), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(fmt.Sprintf("%s / %s / %s", m.Place.City(), m.Place.CountryName(), m.TakenAt.Format("2006")), SrcAuto) } } } if !knownLocation || m.NoTitle() { if names != "" { if len([]rune(names)) <= 35 && m.TakenSrc != SrcAuto { m.SetTitle(fmt.Sprintf("%s / %s", names, m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(names, SrcAuto) } } else if fileTitle == "" && len(labels) > 0 && labels[0].Priority >= -1 && labels[0].Uncertainty <= 85 && labels[0].Name != "" { if m.TakenSrc != SrcAuto { m.SetTitle(fmt.Sprintf("%s / %s", txt.Title(labels[0].Name), m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(txt.Title(labels[0].Name), SrcAuto) } } else if fileTitle != "" && len(fileTitle) <= 20 && !m.TakenAtLocal.IsZero() && m.TakenSrc != SrcAuto { m.SetTitle(fmt.Sprintf("%s / %s", fileTitle, m.TakenAtLocal.Format("2006")), SrcAuto) } else if fileTitle != "" { m.SetTitle(fileTitle, SrcAuto) } else { if m.TakenSrc != SrcAuto { m.SetTitle(fmt.Sprintf("%s / %s", UnknownTitle, m.TakenAt.Format("2006")), SrcAuto) } else { m.SetTitle(UnknownTitle, SrcAuto) } } } if m.PhotoTitle != oldTitle { log.Debugf("photo: %s has new title %s [%s]", m.String(), clean.Log(m.PhotoTitle), time.Since(start)) } return nil } // UpdateAndSaveTitle updates the photo title and saves it. func (m *Photo) UpdateAndSaveTitle() error { if !m.HasID() { return fmt.Errorf("cannot save photo whithout id") } m.PhotoFaces = m.FaceCount() labels := m.ClassifyLabels() m.UpdateDateFields() if err := m.UpdateTitle(labels); err != nil { log.Info(err) } details := m.GetDetails() w := txt.UniqueWords(txt.Words(details.Keywords)) w = append(w, labels.Keywords()...) details.Keywords = strings.Join(txt.UniqueWords(w), ", ") if err := m.IndexKeywords(); err != nil { log.Errorf("photo: %s", err.Error()) } if err := m.Save(); err != nil { return err } return nil } // UpdateDescription updates the photo descriptions based on available metadata. func (m *Photo) UpdateDescription(people []string) { if m.DescriptionSrc != SrcAuto { return } // Add subject names to description when there's more than one person. if len(people) > 3 { m.PhotoDescription = txt.JoinNames(people, false) } else { m.PhotoDescription = "" } } // FileTitle returns a photo title based on the file name and/or path. func (m *Photo) FileTitle() string { // Generate title based on photo name, if not generated: if !fs.IsGenerated(m.PhotoName) { if title := txt.FileTitle(m.PhotoName); title != "" { return title } } // Generate title based on original file name, if any: if m.OriginalName != "" { if title := txt.FileTitle(m.OriginalName); !fs.IsGenerated(m.OriginalName) && title != "" { return title } else if title := txt.FileTitle(filepath.Dir(m.OriginalName)); title != "" { return title } } // Generate title based on photo path, if any: if m.PhotoPath != "" && !fs.IsGenerated(m.PhotoPath) { return txt.FileTitle(m.PhotoPath) } return "" } // SubjectNames returns all known subject names. func (m *Photo) SubjectNames() []string { if f, err := m.PrimaryFile(); err == nil { return f.SubjectNames() } return nil } // SubjectKeywords returns keywords for all known subject names. func (m *Photo) SubjectKeywords() []string { return txt.Words(strings.Join(m.SubjectNames(), " ")) }