photoprism/internal/search/labels.go

94 lines
2.1 KiB
Go
Raw Normal View History

package search
import (
"strings"
"github.com/gosimple/slug"
"github.com/photoprism/photoprism/internal/entity"
"github.com/photoprism/photoprism/internal/form"
"github.com/photoprism/photoprism/pkg/txt"
)
// Labels searches labels based on their name.
func Labels(f form.LabelSearch) (results []Label, err error) {
if err := f.ParseQueryString(); err != nil {
return results, err
}
s := UnscopedDb()
// s.LogMode(true)
// Base query.
s = s.Table("labels").
Select(`labels.*`).
Where("labels.deleted_at IS NULL").
Where("labels.photo_count > 0").
Group("labels.id")
// Limit result count.
if f.Count > 0 && f.Count <= MaxResults {
s = s.Limit(f.Count).Offset(f.Offset)
} else {
s = s.Limit(MaxResults).Offset(f.Offset)
}
// Set sort order.
switch f.Order {
case "slug":
s = s.Order("labels.label_favorite DESC, custom_slug ASC")
default:
s = s.Order("labels.label_favorite DESC, custom_slug ASC")
}
if f.ID != "" {
s = s.Where("labels.label_uid IN (?)", strings.Split(f.ID, txt.Or))
if result := s.Scan(&results); result.Error != nil {
return results, result.Error
}
return results, nil
}
if f.Query != "" {
var labelIds []uint
var categories []entity.Category
var label entity.Label
slugString := slug.Make(f.Query)
likeString := "%" + f.Query + "%"
if result := Db().First(&label, "label_slug = ? OR custom_slug = ?", slugString, slugString); result.Error != nil {
log.Infof("search: label %s not found", txt.Quote(f.Query))
s = s.Where("labels.label_name LIKE ?", likeString)
} else {
labelIds = append(labelIds, label.ID)
Db().Where("category_id = ?", label.ID).Find(&categories)
for _, category := range categories {
labelIds = append(labelIds, category.LabelID)
}
log.Infof("search: label %s includes %d categories", txt.Quote(label.LabelName), len(labelIds))
s = s.Where("labels.id IN (?)", labelIds)
}
}
if f.Favorite {
s = s.Where("labels.label_favorite = 1")
}
if !f.All {
s = s.Where("labels.label_priority >= 0 OR labels.label_favorite = 1")
}
if result := s.Scan(&results); result.Error != nil {
return results, result.Error
}
return results, nil
}