2021-09-02 18:09:37 +02:00
|
|
|
package actions
|
|
|
|
|
|
|
|
import (
|
2021-09-02 18:20:10 +02:00
|
|
|
"strings"
|
|
|
|
|
2021-11-06 15:41:25 +01:00
|
|
|
"github.com/photoview/photoview/api/database/drivers"
|
2021-09-02 18:09:37 +02:00
|
|
|
"github.com/photoview/photoview/api/graphql/models"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"gorm.io/gorm/clause"
|
|
|
|
)
|
|
|
|
|
|
|
|
func Search(db *gorm.DB, query string, userID int, _limitMedia *int, _limitAlbums *int) (*models.SearchResult, error) {
|
|
|
|
limitMedia := 10
|
|
|
|
limitAlbums := 10
|
|
|
|
|
|
|
|
if _limitMedia != nil {
|
|
|
|
limitMedia = *_limitMedia
|
|
|
|
}
|
|
|
|
|
|
|
|
if _limitAlbums != nil {
|
|
|
|
limitAlbums = *_limitAlbums
|
|
|
|
}
|
|
|
|
|
2021-09-02 18:20:10 +02:00
|
|
|
wildQuery := "%" + strings.ToLower(query) + "%"
|
2021-09-02 18:09:37 +02:00
|
|
|
|
|
|
|
var media []*models.Media
|
|
|
|
|
|
|
|
userSubquery := db.Table("user_albums").Where("user_id = ?", userID)
|
2021-11-06 15:41:25 +01:00
|
|
|
if drivers.POSTGRES.MatchDatabase(db) {
|
2021-09-02 18:09:37 +02:00
|
|
|
userSubquery = userSubquery.Where("album_id = \"Album\".id")
|
|
|
|
} else {
|
|
|
|
userSubquery = userSubquery.Where("album_id = Album.id")
|
|
|
|
}
|
|
|
|
|
|
|
|
err := db.Joins("Album").
|
|
|
|
Where("EXISTS (?)", userSubquery).
|
2021-09-02 18:20:10 +02:00
|
|
|
Where("LOWER(media.title) LIKE ? OR LOWER(media.path) LIKE ?", wildQuery, wildQuery).
|
2021-09-02 18:09:37 +02:00
|
|
|
Clauses(clause.OrderBy{
|
|
|
|
Expression: clause.Expr{
|
2021-09-02 18:20:10 +02:00
|
|
|
SQL: "(CASE WHEN LOWER(media.title) LIKE ? THEN 2 WHEN LOWER(media.path) LIKE ? THEN 1 END) DESC",
|
2021-09-02 18:09:37 +02:00
|
|
|
Vars: []interface{}{wildQuery, wildQuery},
|
|
|
|
WithoutParentheses: true},
|
|
|
|
}).
|
|
|
|
Limit(limitMedia).Find(&media).Error
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrapf(err, "searching media")
|
|
|
|
}
|
|
|
|
|
|
|
|
var albums []*models.Album
|
|
|
|
|
|
|
|
err = db.
|
|
|
|
Where("EXISTS (?)", db.Table("user_albums").Where("user_id = ?", userID).Where("album_id = albums.id")).
|
|
|
|
Where("albums.title LIKE ? OR albums.path LIKE ?", wildQuery, wildQuery).
|
|
|
|
Clauses(clause.OrderBy{
|
|
|
|
Expression: clause.Expr{
|
|
|
|
SQL: "(CASE WHEN albums.title LIKE ? THEN 2 WHEN albums.path LIKE ? THEN 1 END) DESC",
|
|
|
|
Vars: []interface{}{wildQuery, wildQuery},
|
|
|
|
WithoutParentheses: true},
|
|
|
|
}).
|
|
|
|
Limit(limitAlbums).
|
|
|
|
Find(&albums).Error
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrapf(err, "searching albums")
|
|
|
|
}
|
|
|
|
|
|
|
|
result := models.SearchResult{
|
|
|
|
Query: query,
|
|
|
|
Media: media,
|
|
|
|
Albums: albums,
|
|
|
|
}
|
|
|
|
|
|
|
|
return &result, nil
|
|
|
|
}
|