-
Notifications
You must be signed in to change notification settings - Fork 150
/
convert.go
64 lines (53 loc) · 1.07 KB
/
convert.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package nyaafeeds
import (
"strconv"
"strings"
"github.com/NyaaPantsu/nyaa/utils/categories"
)
// ConvertToCat : Convert a torznab cat to our cat
func ConvertToCat(cat string) string {
if cat == "" {
return ""
}
cats := strings.Split(cat, ",")
var returnCat []string
for _, val := range cats {
localeCat := convertCat(val)
if localeCat != "" {
returnCat = append(returnCat, localeCat)
}
}
return strings.Join(returnCat, ",")
}
func convertCat(cat string) string {
if cat == "5070" {
return "3_5"
}
if len(cat) < 6 {
return ""
}
cI, _ := strconv.Atoi(cat[2:4])
subI, _ := strconv.Atoi(cat[4:6])
c := strconv.Itoa(cI)
sub := strconv.Itoa(subI)
if categories.Exists(c + "_" + sub) {
return c + "_" + sub
}
return ""
}
// ConvertFromCat : Convert a cat to a torznab cat
func ConvertFromCat(category string) (cat string) {
c := strings.Split(category, "_")
if len(c[0]) < 2 {
c[0] = "0" + c[0]
}
if len(c) < 2 || c[1] == "" {
cat = "10" + c[0] + "00"
return
}
if len(c[1]) < 2 {
c[1] = "0" + c[1]
}
cat = "10" + c[0] + c[1]
return
}