Skip to content

Commit

Permalink
temp fix search
Browse files Browse the repository at this point in the history
  • Loading branch information
trinity-aml committed Jul 5, 2023
1 parent 4ca51ef commit ee00b05
Showing 1 changed file with 30 additions and 24 deletions.
54 changes: 30 additions & 24 deletions db/search.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,12 @@ package db

import (
"NUMParser/db/models"
"NUMParser/db/torrsearch"
"NUMParser/utils"
"github.com/agnivade/levenshtein"
"sort"
"strconv"
"strings"
)

func indexTorrs() {
Expand All @@ -11,28 +17,28 @@ func indexTorrs() {
}

func SearchTorr(query string) []*models.TorrentDetails {
//matchedIDs := torrsearch.Search(query)
//if len(matchedIDs) == 0 {
// return nil
//}
////torrs := GetTorrs()
//var list []*models.TorrentDetails
//for _, id := range matchedIDs {
// list = append(list, torrs[id])
//}
//
//hash := utils.ClearStr(query)
//
//sort.Slice(list, func(i, j int) bool {
// lhash := utils.ClearStr(strings.ToLower(list[i].Name+list[i].GetNames())) + strconv.Itoa(list[i].Year)
// lev1 := levenshtein.ComputeDistance(hash, lhash)
// lhash = utils.ClearStr(strings.ToLower(list[j].Name+list[j].GetNames())) + strconv.Itoa(list[j].Year)
// lev2 := levenshtein.ComputeDistance(hash, lhash)
// if lev1 == lev2 {
// return list[j].CreateDate.Before(list[i].CreateDate)
// }
// return lev1 < lev2
//})
//return list
return nil
matchedIDs := torrsearch.Search(query)
if len(matchedIDs) == 0 {
return nil
}
torrs := GetTorrs()
var list []*models.TorrentDetails
for _, id := range matchedIDs {
list = append(list, torrs[id])
}

hash := utils.ClearStr(query)

sort.Slice(list, func(i, j int) bool {
lhash := utils.ClearStr(strings.ToLower(list[i].Name+list[i].GetNames())) + strconv.Itoa(list[i].Year)
lev1 := levenshtein.ComputeDistance(hash, lhash)
lhash = utils.ClearStr(strings.ToLower(list[j].Name+list[j].GetNames())) + strconv.Itoa(list[j].Year)
lev2 := levenshtein.ComputeDistance(hash, lhash)
if lev1 == lev2 {
return list[j].CreateDate.Before(list[i].CreateDate)
}
return lev1 < lev2
})
return list
//return nil
}

0 comments on commit ee00b05

Please sign in to comment.