bitmagnet/internal/torznab/adapter/search_options.go

package adapter

import (
	"fmt"
	"strings"

	"github.com/bitmagnet-io/bitmagnet/internal/database/query"
	"github.com/bitmagnet-io/bitmagnet/internal/database/search"
	"github.com/bitmagnet-io/bitmagnet/internal/model"
	"github.com/bitmagnet-io/bitmagnet/internal/torznab"
)

func searchRequestToQueryOptions(r torznab.SearchRequest) ([]query.Option, error) {
	var options []query.Option
	switch r.Type {
	case torznab.FunctionSearch:
		break
	case torznab.FunctionMovie:
		options = append(options, query.Where(search.TorrentContentTypeCriteria(model.ContentTypeMovie)))
	case torznab.FunctionTV:
		options = append(options, query.Where(search.TorrentContentTypeCriteria(model.ContentTypeTvShow)))
		if r.Season.Valid {
			episodes := make(model.Episodes)
			if r.Episode.Valid {
				episodes = episodes.AddEpisode(r.Season.Int, r.Episode.Int)
			} else {
				episodes = episodes.AddSeason(r.Season.Int)
			}
			options = append(options, query.Where(search.TorrentContentEpisodesCriteria(episodes)))
		}
	case torznab.FunctionMusic:
		options = append(options, query.Where(search.TorrentContentTypeCriteria(model.ContentTypeMusic)))
	case torznab.FunctionBook:
		options = append(options, query.Where(search.TorrentContentTypeCriteria(
			model.ContentTypeEbook,
			model.ContentTypeComic,
			model.ContentTypeAudiobook,
		)))
	default:
		return nil, torznab.Error{
			Code:        202,
			Description: fmt.Sprintf("no such function (%s)", r.Type),
		}
	}
	if r.Query != "" {
		order := search.TorrentContentOrderByRelevance
		if r.Profile.DisableOrderByRelevance {
			order = search.TorrentContentOrderByPublishedAt
		}
		options = append(options, query.QueryString(r.Query), query.OrderBy(order.Clauses(search.OrderDirectionDescending)...))
	}
	var catsCriteria []query.Criteria
	for _, cat := range r.Cats {
		var catCriteria []query.Criteria
		if torznab.CategoryMovies.Has(cat) {
			if r.Type != torznab.FunctionMovie || torznab.CategoryMovies.ID == cat {
				catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeMovie))
			}
			if torznab.CategoryMoviesSD.ID == cat {
				catCriteria = append(catCriteria, search.VideoResolutionCriteria(model.VideoResolutionV480p))
			} else if torznab.CategoryMoviesHD.ID == cat {
				catCriteria = append(catCriteria, search.VideoResolutionCriteria(
					model.VideoResolutionV720p,
					model.VideoResolutionV1080p,
					model.VideoResolutionV1440p,
					model.VideoResolutionV2160p,
				))
			} else if torznab.CategoryMoviesUHD.ID == cat {
				catCriteria = append(catCriteria, search.VideoResolutionCriteria(model.VideoResolutionV2160p))
			} else if torznab.CategoryMovies3D.ID == cat {
				catCriteria = append(catCriteria, search.Video3DCriteria(
					model.Video3DV3D,
					model.Video3DV3DSBS,
					model.Video3DV3DOU,
				))
			}
		} else if torznab.CategoryTV.Has(cat) {
			if r.Type != torznab.FunctionTV || torznab.CategoryTV.ID == cat {
				catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeTvShow))
			}
			if torznab.CategoryTVSD.ID == cat {
				catCriteria = append(catCriteria, search.VideoResolutionCriteria(model.VideoResolutionV480p))
			} else if torznab.CategoryTVHD.ID == cat {
				catCriteria = append(catCriteria, search.VideoResolutionCriteria(
					model.VideoResolutionV720p,
					model.VideoResolutionV1080p,
					model.VideoResolutionV1440p,
					model.VideoResolutionV2160p,
				))
			} else if torznab.CategoryTVUHD.ID == cat {
				catCriteria = append(catCriteria, search.VideoResolutionCriteria(model.VideoResolutionV2160p))
			}
		} else if torznab.CategoryXXX.Has(cat) {
			catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeXxx))
		} else if torznab.CategoryPC.Has(cat) {
			catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeSoftware, model.ContentTypeGame))
		} else if torznab.CategoryAudioAudiobook.Has(cat) {
			catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeAudiobook))
		} else if torznab.CategoryAudio.Has(cat) {
			catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeMusic))
		} else if torznab.CategoryBooksComics.Has(cat) {
			catCriteria = append(catCriteria, search.TorrentContentTypeCriteria(model.ContentTypeComic))
		} else if torznab.CategoryBooks.Has(cat) {
			options = append(options, query.Where(search.TorrentContentTypeCriteria(
				model.ContentTypeEbook,
				model.ContentTypeComic,
				model.ContentTypeAudiobook,
			)))
		}
		if len(catCriteria) > 0 {
			catsCriteria = append(catsCriteria, query.And(catCriteria...))
		}
	}
	if len(catsCriteria) > 0 {
		options = append(options, query.Where(query.Or(catsCriteria...)))
	}
	if r.IMDBID.Valid {
		imdbId := r.IMDBID.String
		if !strings.HasPrefix(imdbId, "tt") {
			imdbId = "tt" + imdbId
		}
		var ct model.ContentType
		if r.Type != torznab.FunctionTV {
			ct = model.ContentTypeMovie
		} else if r.Type != torznab.FunctionMovie {
			ct = model.ContentTypeTvShow
		}
		options = append(options, query.Where(search.ContentAlternativeIdentifierCriteria(model.ContentRef{
			Type:   ct,
			Source: "imdb",
			ID:     imdbId,
		})))
	}
	if r.TMDBID.Valid {
		tmdbId := r.TMDBID.String
		var ct model.ContentType
		if r.Type != torznab.FunctionTV {
			ct = model.ContentTypeMovie
		} else if r.Type != torznab.FunctionMovie {
			ct = model.ContentTypeTvShow
		}
		options = append(options, query.Where(search.ContentCanonicalIdentifierCriteria(model.ContentRef{
			Type:   ct,
			Source: "tmdb",
			ID:     tmdbId,
		})))
	}
	limit := r.Profile.DefaultLimit
	if r.Limit.Valid {
		limit = r.Limit.Uint
		if limit > r.Profile.MaxLimit {
			limit = r.Profile.MaxLimit
		}
	}
	options = append(options, query.Limit(limit))
	if r.Offset.Valid {
		options = append(options, query.Offset(r.Offset.Uint))
	}

	if len(r.Profile.Tags) > 0 {
		options = append(options, query.Where(search.TorrentTagCriteria(r.Profile.Tags...)))
	}
	return options, nil
}