96 lignes
1.8 KiB
Go
96 lignes
1.8 KiB
Go
package upload
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/NyaaPantsu/nyaa/models"
|
|
)
|
|
|
|
// Convert automatically our sukebei cats to platform specified Hentai cats
|
|
var sukebeiCategories = []map[string]string{
|
|
ttosho: {
|
|
"1_1": "12",
|
|
"1_2": "12",
|
|
"1_3": "14",
|
|
"1_4": "13",
|
|
"1_5": "4",
|
|
"2_1": "4",
|
|
"2_2": "15",
|
|
},
|
|
nyaasi: {
|
|
"1_1": "1_1",
|
|
"1_2": "1_2",
|
|
"1_3": "1_3",
|
|
"1_4": "1_4",
|
|
"1_5": "1_5",
|
|
"2_1": "2_1",
|
|
"2_2": "2_2",
|
|
},
|
|
}
|
|
|
|
var normalCategories = []map[string]string{
|
|
ttosho: {
|
|
"3_12": "1",
|
|
"3_5": "1",
|
|
"3_13": "10",
|
|
"3_6": "7",
|
|
"2_3": "2",
|
|
"2_4": "2",
|
|
"4_7": "3",
|
|
"4_8": "7",
|
|
"4_14": "10",
|
|
"5_9": "8",
|
|
"5_10": "8",
|
|
"5_18": "10",
|
|
"5_11": "7",
|
|
"6_15": "5",
|
|
"6_16": "5",
|
|
"1_1": "5",
|
|
"1_2": "5",
|
|
},
|
|
nyaasi: {
|
|
"3_12": "1_1",
|
|
"3_5": "1_2",
|
|
"3_13": "1_3",
|
|
"3_6": "1_4",
|
|
"2_3": "2_1",
|
|
"2_4": "2_2",
|
|
"4_7": "3_1",
|
|
"4_8": "3_4",
|
|
"4_14": "3_3",
|
|
"5_9": "4_1",
|
|
"5_10": "4_2",
|
|
"5_18": "4_3",
|
|
"5_11": "4_4",
|
|
"6_15": "5_1",
|
|
"6_16": "5_2",
|
|
"1_1": "6_1",
|
|
"1_2": "6_2",
|
|
},
|
|
}
|
|
|
|
// Category returns the category converted from nyaa one to tosho one
|
|
func Category(platform int, t *models.Torrent) string {
|
|
cat := fmt.Sprintf("%d_%d", t.Category, t.SubCategory)
|
|
// if we are in sukebei, there are some categories
|
|
if config.IsSukebei() {
|
|
// check that platform exist in our map for sukebei categories
|
|
if platform < len(sukebeiCategories) {
|
|
// return the remaped category if it exists
|
|
if val, ok := sukebeiCategories[platform][cat]; ok {
|
|
return val
|
|
}
|
|
}
|
|
}
|
|
// check that platform exist in our map
|
|
if platform >= len(normalCategories) {
|
|
return ""
|
|
}
|
|
// return the remaped category if it exists
|
|
if val, ok := normalCategories[platform][cat]; ok {
|
|
return val
|
|
}
|
|
return ""
|
|
}
|