Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/util/search/search.go

159 lignes
4,1 Kio
Go
Brut Vue normale Historique

package search
2017-05-06 12:41:48 +02:00
import (
"net/http"
"strconv"
"strings"
2017-05-08 15:50:18 +02:00
"unicode"
"unicode/utf8"
2017-05-10 11:03:49 +02:00
"github.com/ewhal/nyaa/cache"
"github.com/ewhal/nyaa/common"
"github.com/ewhal/nyaa/db"
"github.com/ewhal/nyaa/model"
2017-05-10 20:42:20 +02:00
"github.com/ewhal/nyaa/service"
2017-05-10 11:03:49 +02:00
"github.com/ewhal/nyaa/service/torrent"
"github.com/ewhal/nyaa/util/log"
)
2017-05-10 11:03:49 +02:00
func SearchByQuery(r *http.Request, pagenum int) (search common.SearchParam, tor []model.Torrent, count int, err error) {
2017-05-09 17:07:42 +02:00
search, tor, count, err = searchByQuery(r, pagenum, true)
return
}
2017-05-10 11:03:49 +02:00
func SearchByQueryNoCount(r *http.Request, pagenum int) (search common.SearchParam, tor []model.Torrent, err error) {
2017-05-09 17:07:42 +02:00
search, tor, _, err = searchByQuery(r, pagenum, false)
return
}
2017-05-10 11:03:49 +02:00
func searchByQuery(r *http.Request, pagenum int, countAll bool) (
search common.SearchParam, tor []model.Torrent, count int, err error,
) {
max, err := strconv.ParseUint(r.URL.Query().Get("max"), 10, 32)
if err != nil {
max = 50 // default Value maxPerPage
} else if max > 300 {
max = 300
}
search.Max = uint(max)
search.Page = pagenum
search.Query = r.URL.Query().Get("q")
2017-05-06 12:41:48 +02:00
switch s := r.URL.Query().Get("s"); s {
case "1":
2017-05-10 11:03:49 +02:00
search.Status = common.FilterRemakes
case "2":
2017-05-10 11:03:49 +02:00
search.Status = common.Trusted
case "3":
2017-05-10 11:03:49 +02:00
search.Status = common.APlus
2017-05-06 12:41:48 +02:00
}
catString := r.URL.Query().Get("c")
if s := catString; len(s) > 1 && s != "_" {
var tmp uint64
tmp, err = strconv.ParseUint(string(s[0]), 10, 8)
if err != nil {
return
}
search.Category.Main = uint8(tmp)
if len(s) == 3 {
tmp, err = strconv.ParseUint(string(s[2]), 10, 8)
if err != nil {
return
}
search.Category.Sub = uint8(tmp)
}
}
2017-05-08 16:39:41 +02:00
orderBy := ""
switch s := r.URL.Query().Get("sort"); s {
case "1":
2017-05-10 11:03:49 +02:00
search.Sort = common.Name
orderBy += "torrent_name"
case "2":
2017-05-10 11:03:49 +02:00
search.Sort = common.Date
orderBy += "date"
case "3":
2017-05-10 11:03:49 +02:00
search.Sort = common.Downloads
orderBy += "downloads"
case "4":
2017-05-10 11:03:49 +02:00
search.Sort = common.Size
orderBy += "filesize"
2017-05-08 16:39:41 +02:00
default:
orderBy += "torrent_id"
}
2017-05-08 16:39:41 +02:00
orderBy += " "
switch s := r.URL.Query().Get("order"); s {
case "true":
search.Order = true
orderBy += "asc"
2017-05-08 16:39:41 +02:00
default:
orderBy += "desc"
}
2017-05-08 16:39:41 +02:00
tor, count, err = cache.Get(search, func() (tor []model.Torrent, count int, err error) {
2017-05-10 20:42:20 +02:00
parameters := serviceBase.WhereParams{
2017-05-10 11:03:49 +02:00
Params: make([]interface{}, 0, 64),
}
2017-05-10 11:03:49 +02:00
conditions := make([]string, 0, 64)
if search.Category.Main != 0 {
conditions = append(conditions, "category = ?")
parameters.Params = append(parameters.Params, string(catString[0]))
}
if search.Category.Sub != 0 {
conditions = append(conditions, "sub_category = ?")
parameters.Params = append(parameters.Params, string(catString[2]))
}
if search.Status != 0 {
if search.Status == common.FilterRemakes {
2017-05-10 11:03:49 +02:00
conditions = append(conditions, "status != ?")
} else {
conditions = append(conditions, "status = ?")
}
parameters.Params = append(parameters.Params, strconv.Itoa(int(search.Status)+1))
}
searchQuerySplit := strings.Fields(search.Query)
for i, word := range searchQuerySplit {
firstRune, _ := utf8.DecodeRuneInString(word)
if len(word) == 1 && unicode.IsPunct(firstRune) {
// some queries have a single punctuation character
// which causes a full scan instead of using the index
// and yields no meaningful results.
// due to len() == 1 we're just looking at 1-byte/ascii
// punctuation characters.
continue
}
2017-05-10 11:03:49 +02:00
// SQLite has case-insensitive LIKE, but no ILIKE
var operator string
if db.ORM.Dialect().GetName() == "sqlite3" {
operator = "LIKE ?"
} else {
operator = "ILIKE ?"
}
// TODO: make this faster ?
conditions = append(conditions, "torrent_name "+operator)
parameters.Params = append(parameters.Params, "%"+searchQuerySplit[i]+"%")
2017-05-08 15:50:18 +02:00
}
2017-05-10 11:03:49 +02:00
parameters.Conditions = strings.Join(conditions[:], " AND ")
log.Infof("SQL query is :: %s\n", parameters.Conditions)
if countAll {
tor, count, err = torrentService.GetTorrentsOrderBy(&parameters, orderBy, int(search.Max), int(search.Max)*(search.Page-1))
} else {
tor, err = torrentService.GetTorrentsOrderByNoCount(&parameters, orderBy, int(search.Max), int(search.Max)*(search.Page-1))
}
2017-05-10 11:03:49 +02:00
return
})
return
}