471 lignes
13 KiB
Go
471 lignes
13 KiB
Go
package templates
|
|
|
|
import (
|
|
"html/template"
|
|
"path/filepath"
|
|
"math"
|
|
"math/rand"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
"os"
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
"github.com/CloudyKit/jet"
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/NyaaPantsu/nyaa/models"
|
|
"github.com/NyaaPantsu/nyaa/models/torrents"
|
|
"github.com/NyaaPantsu/nyaa/utils/categories"
|
|
"github.com/NyaaPantsu/nyaa/utils/filelist"
|
|
"github.com/NyaaPantsu/nyaa/utils/format"
|
|
"github.com/NyaaPantsu/nyaa/utils/publicSettings"
|
|
"github.com/NyaaPantsu/nyaa/utils/torrentLanguages"
|
|
)
|
|
|
|
// FuncMap : Functions accessible in templates by {{ $.Function }}
|
|
func templateFunctions(vars jet.VarMap) jet.VarMap {
|
|
vars.Set("getRawQuery", getRawQuery)
|
|
vars.Set("genSearchWithOrdering", genSearchWithOrdering)
|
|
vars.Set("genSearchWithCategory", genSearchWithCategory)
|
|
vars.Set("genSortArrows", genSortArrows)
|
|
vars.Set("genNav", genNav)
|
|
vars.Set("Sukebei", config.IsSukebei)
|
|
vars.Set("getDefaultLanguage", publicSettings.GetDefaultLanguage)
|
|
vars.Set("FlagCode", flagCode)
|
|
vars.Set("getAvatar", getAvatar)
|
|
vars.Set("torrentFileExists", torrentFileExists)
|
|
vars.Set("GetHostname", format.GetHostname)
|
|
vars.Set("GetCategories", categories.GetSelect)
|
|
vars.Set("GetCategory", getCategory)
|
|
vars.Set("CategoryName", categoryName)
|
|
vars.Set("GetCategoryName", GetCategoryName)
|
|
vars.Set("GetTorrentLanguages", torrentLanguages.GetTorrentLanguages)
|
|
vars.Set("GenSearchName", GenSearchName)
|
|
vars.Set("LanguageName", languageName)
|
|
vars.Set("LanguageNameFromCode", languageNameFromCode)
|
|
vars.Set("fileSize", fileSize)
|
|
vars.Set("DefaultUserSettings", defaultUserSettings)
|
|
vars.Set("makeTreeViewData", makeTreeViewData)
|
|
vars.Set("lastID", lastID)
|
|
vars.Set("getReportDescription", getReportDescription)
|
|
vars.Set("genUploaderLink", genUploaderLink)
|
|
vars.Set("genActivityContent", genActivityContent)
|
|
vars.Set("contains", contains)
|
|
vars.Set("strcmp", strcmp)
|
|
vars.Set("strfind", strfind)
|
|
vars.Set("rand", rand.Intn)
|
|
vars.Set("toString", strconv.Itoa)
|
|
vars.Set("getDomainName", getDomainName)
|
|
vars.Set("getThemeList", getThemeList)
|
|
vars.Set("formatThemeName", formatThemeName)
|
|
vars.Set("formatDate", formatDate)
|
|
return vars
|
|
}
|
|
func getRawQuery(currentURL *url.URL) string {
|
|
return currentURL.RawQuery
|
|
}
|
|
func genSearchWithOrdering(currentURL *url.URL, sortBy string, searchRoute string) string {
|
|
values := currentURL.Query()
|
|
order := false //Default is DESC
|
|
sort := "2" //Default is Date (Actually ID, but Date is the same thing)
|
|
|
|
if _, ok := values["order"]; ok {
|
|
order, _ = strconv.ParseBool(values["order"][0])
|
|
}
|
|
if _, ok := values["sort"]; ok {
|
|
sort = values["sort"][0]
|
|
}
|
|
|
|
if sort == sortBy {
|
|
order = !order //Flip order by repeat-clicking
|
|
} else {
|
|
order = false //Default to descending when sorting by something new
|
|
}
|
|
|
|
values.Set("sort", sortBy)
|
|
values.Set("order", strconv.FormatBool(order))
|
|
|
|
u, _ := url.Parse(searchRoute)
|
|
u.RawQuery = values.Encode()
|
|
|
|
return u.String()
|
|
}
|
|
|
|
func genSearchWithCategory(currentURL *url.URL, category string, searchRoute string) string {
|
|
values := currentURL.Query()
|
|
cat := "_" //Default
|
|
|
|
if _, ok := values["c"]; ok {
|
|
cat = values["c"][0]
|
|
}
|
|
|
|
cat = category
|
|
|
|
values.Set("c", cat)
|
|
|
|
u, _ := url.Parse(searchRoute)
|
|
u.RawQuery = values.Encode()
|
|
|
|
return u.String()
|
|
}
|
|
|
|
func genSortArrows(currentURL *url.URL, sortBy string) template.HTML {
|
|
values := currentURL.Query()
|
|
leftclass := "sortarrowdim"
|
|
rightclass := "sortarrowdim"
|
|
|
|
order := false
|
|
sort := "2"
|
|
|
|
if _, ok := values["order"]; ok {
|
|
order, _ = strconv.ParseBool(values["order"][0])
|
|
}
|
|
if _, ok := values["sort"]; ok {
|
|
sort = values["sort"][0]
|
|
}
|
|
|
|
if sort == sortBy {
|
|
if order {
|
|
rightclass = ""
|
|
} else {
|
|
leftclass = ""
|
|
}
|
|
}
|
|
|
|
arrows := "<span class=\"sortarrowleft " + leftclass + "\">▼</span><span class=\"" + rightclass + "\">▲</span>"
|
|
|
|
return template.HTML(arrows)
|
|
}
|
|
|
|
|
|
func genNav(nav Navigation, currentURL *url.URL, pagesSelectable int) template.HTML {
|
|
var ret = ""
|
|
if nav.TotalItem > 0 {
|
|
maxPages := math.Ceil(float64(nav.TotalItem) / float64(nav.MaxItemPerPage))
|
|
|
|
href := ""
|
|
display := " style=\"display:none;\""
|
|
if nav.CurrentPage-1 > 0 {
|
|
display = ""
|
|
href = " href=\"" + "/" + nav.Route + "/1" + "?" + currentURL.RawQuery + "\""
|
|
}
|
|
ret = ret + "<a class=\"page-prev\"" + display + href + " aria-label=\"Previous\"><span aria-hidden=\"true\">«</span></a>"
|
|
|
|
startValue := 1
|
|
if nav.CurrentPage > pagesSelectable/2 {
|
|
startValue = (int(math.Min((float64(nav.CurrentPage)+math.Floor(float64(pagesSelectable)/2)), maxPages)) - pagesSelectable + 1)
|
|
}
|
|
if startValue < 1 {
|
|
startValue = 1
|
|
}
|
|
endValue := (startValue + pagesSelectable - 1)
|
|
if endValue > int(maxPages) {
|
|
endValue = int(maxPages)
|
|
}
|
|
for i := startValue; i <= endValue; i++ {
|
|
pageNum := strconv.Itoa(i)
|
|
url := "/" + nav.Route + "/" + pageNum
|
|
ret = ret + "<a aria-label=\"Page " + strconv.Itoa(i) + "\" href=\"" + url + "?" + currentURL.RawQuery + "\">" + "<span"
|
|
if i == nav.CurrentPage {
|
|
ret = ret + " class=\"active\""
|
|
}
|
|
ret = ret + ">" + strconv.Itoa(i) + "</span></a>"
|
|
}
|
|
|
|
href = ""
|
|
display = " style=\"display:none;\""
|
|
if nav.CurrentPage < int(maxPages) {
|
|
display = ""
|
|
href = " href=\"" + "/" + nav.Route + "/" + strconv.Itoa(int(maxPages)) + "?" + currentURL.RawQuery + "\""
|
|
}
|
|
ret = ret + "<a class=\"page-next\"" + display + href +" aria-label=\"Next\"><span aria-hidden=\"true\">»</span></a>"
|
|
|
|
itemsThisPageStart := nav.MaxItemPerPage*(nav.CurrentPage-1) + 1
|
|
itemsThisPageEnd := nav.MaxItemPerPage * nav.CurrentPage
|
|
if nav.TotalItem < itemsThisPageEnd {
|
|
itemsThisPageEnd = nav.TotalItem
|
|
}
|
|
ret = ret + "<p>" + strconv.Itoa(itemsThisPageStart) + "-" + strconv.Itoa(itemsThisPageEnd) + "/" + strconv.Itoa(nav.TotalItem) + "</p>"
|
|
}
|
|
return template.HTML(ret)
|
|
}
|
|
|
|
func flagCode(languageCode string) string {
|
|
return publicSettings.Flag(languageCode, true)
|
|
}
|
|
|
|
func getAvatar(hash string, size int) string {
|
|
if hash != "" {
|
|
return "https://www.gravatar.com/avatar/" + hash + "?s=" + strconv.Itoa(size)
|
|
} else {
|
|
if config.IsSukebei() {
|
|
return "/img/sukebei_avatar_" + strconv.Itoa(size) + ".jpg"
|
|
}
|
|
return "/img/avatar_" + strconv.Itoa(size) + ".jpg"
|
|
}
|
|
}
|
|
|
|
func getCategory(category string, keepParent bool) categories.Categories {
|
|
cats := categories.GetSelect(true, true)
|
|
found := false
|
|
categoryRet := categories.Categories{}
|
|
for _, v := range cats {
|
|
if v.ID == category+"_" {
|
|
found = true
|
|
if keepParent {
|
|
categoryRet = append(categoryRet, v)
|
|
}
|
|
} else if len(v.ID) <= 2 && len(categoryRet) > 0 {
|
|
break
|
|
} else if found {
|
|
categoryRet = append(categoryRet, v)
|
|
}
|
|
}
|
|
return categoryRet
|
|
}
|
|
func categoryName(category string, subCategory string) string {
|
|
return GetCategoryName( category + "_" + subCategory)
|
|
}
|
|
|
|
func GetCategoryName(category string) string {
|
|
if cat, ok := categories.GetByID(category); ok {
|
|
return cat.Name
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func languageName(lang publicSettings.Language, T publicSettings.TemplateTfunc) string {
|
|
if strings.Contains(lang.Name, ",") {
|
|
langs := strings.Split(lang.Name, ", ")
|
|
tags := strings.Split(lang.Tag, ", ")
|
|
for k := range langs {
|
|
langs[k] = strings.Title(publicSettings.Translate(tags[k], string(T("language_code"))))
|
|
}
|
|
return strings.Join(langs, ", ")
|
|
}
|
|
return strings.Title(lang.Translate(T("language_code")))
|
|
}
|
|
func languageNameFromCode(languageCode string, T publicSettings.TemplateTfunc) string {
|
|
if strings.Contains(languageCode, ",") {
|
|
tags := strings.Split(languageCode, ", ")
|
|
var langs []string
|
|
for k := range tags {
|
|
langs = append(langs, strings.Title(publicSettings.Translate(tags[k], string(T("language_code")))))
|
|
}
|
|
return strings.Join(langs, ", ")
|
|
}
|
|
return strings.Title(publicSettings.Translate(languageCode, string(T("language_code"))))
|
|
}
|
|
func fileSize(filesize int64, T publicSettings.TemplateTfunc, showUnknown bool) template.HTML {
|
|
if showUnknown && filesize == 0 {
|
|
return T("unknown")
|
|
}
|
|
return template.HTML(format.FileSize(filesize))
|
|
}
|
|
|
|
func defaultUserSettings(s string) bool {
|
|
return config.Get().Users.DefaultUserSettings[s]
|
|
}
|
|
|
|
func makeTreeViewData(f *filelist.FileListFolder, nestLevel int, identifierChain string) interface{} {
|
|
return struct {
|
|
Folder *filelist.FileListFolder
|
|
NestLevel int
|
|
IdentifierChain string
|
|
}{f, nestLevel, identifierChain}
|
|
}
|
|
func lastID(currentURL *url.URL, torrents []models.TorrentJSON) int {
|
|
if len(torrents) == 0 {
|
|
return 0
|
|
}
|
|
values := currentURL.Query()
|
|
|
|
order := false
|
|
sort := "2"
|
|
|
|
if _, ok := values["order"]; ok {
|
|
order, _ = strconv.ParseBool(values["order"][0])
|
|
}
|
|
if _, ok := values["sort"]; ok {
|
|
sort = values["sort"][0]
|
|
}
|
|
lastID := 0
|
|
if sort == "2" || sort == "" {
|
|
if order {
|
|
lastID = int(torrents[len(torrents)-1].ID)
|
|
} else if len(torrents) > 0 {
|
|
lastID = int(torrents[0].ID)
|
|
}
|
|
}
|
|
return lastID
|
|
}
|
|
func getReportDescription(d string, T publicSettings.TemplateTfunc) string {
|
|
if d == "illegal" {
|
|
return "Illegal content"
|
|
} else if d == "spam" {
|
|
return "Spam / Garbage"
|
|
} else if d == "wrongcat" {
|
|
return "Wrong category"
|
|
} else if d == "dup" {
|
|
return "Duplicate / Deprecated"
|
|
}
|
|
return string(T(d))
|
|
}
|
|
func genUploaderLink(uploaderID uint, uploaderName template.HTML, torrentHidden bool) template.HTML {
|
|
uploaderID, username := torrents.HideUser(uploaderID, string(uploaderName), torrentHidden)
|
|
if uploaderID == 0 {
|
|
return template.HTML(username)
|
|
}
|
|
url := "/user/" + strconv.Itoa(int(uploaderID)) + "/" + username
|
|
|
|
return template.HTML("<a href=\"" + url + "\">" + username + "</a>")
|
|
}
|
|
func genActivityContent(a models.Activity, T publicSettings.TemplateTfunc) template.HTML {
|
|
return a.ToLocale(T)
|
|
}
|
|
func contains(arr interface{}, comp string) bool {
|
|
switch str := arr.(type) {
|
|
case string:
|
|
if str == comp {
|
|
return true
|
|
}
|
|
case publicSettings.Language:
|
|
if str.Code == comp {
|
|
return true
|
|
}
|
|
default:
|
|
return false
|
|
}
|
|
return false
|
|
}
|
|
|
|
func torrentFileExists(hash string, TorrentLink string) bool {
|
|
if(len(config.Get().Torrents.FileStorage) == 0) {
|
|
//File isn't stored on our servers
|
|
return true
|
|
}
|
|
Openfile, err := os.Open(fmt.Sprintf("%s%c%s.torrent", config.Get().Torrents.FileStorage, os.PathSeparator, hash))
|
|
if err != nil || len(TorrentLink) == 0 {
|
|
//File doesn't exist
|
|
return false
|
|
}
|
|
defer Openfile.Close()
|
|
return true
|
|
}
|
|
|
|
func strcmp(str1 string, str2 string, end int, start int) bool {
|
|
//Compare two strings but has length arguments
|
|
|
|
len1 := len(str1)
|
|
len2 := len(str2)
|
|
|
|
if end == -1 && len1 != len2 {
|
|
return false
|
|
}
|
|
|
|
if end == -1 || end > len1 {
|
|
end = len1
|
|
}
|
|
if end > len2 {
|
|
end = len2
|
|
}
|
|
|
|
if start >= end {
|
|
return false
|
|
}
|
|
|
|
return strings.Compare(str1[start:end], str2[start:end]) == 0
|
|
}
|
|
|
|
func strfind(str1 string, searchfor string, start int) bool {
|
|
//Search a string inside another with start parameter
|
|
//start parameter indicates where we start searching
|
|
|
|
len1 := len(str1)
|
|
|
|
if start >= len1 || searchfor == "" {
|
|
return false
|
|
}
|
|
|
|
|
|
return strings.Contains(str1[start:len1], searchfor)
|
|
}
|
|
|
|
func getDomainName() string {
|
|
domain := config.Get().Cookies.DomainName
|
|
if config.Get().Environment == "DEVELOPMENT" {
|
|
domain = ""
|
|
}
|
|
return domain
|
|
}
|
|
|
|
func getThemeList() ([]string) {
|
|
searchDir := "public/css/themes/"
|
|
|
|
themeList := []string{}
|
|
|
|
filepath.Walk(searchDir, func(path string, f os.FileInfo, err error) error {
|
|
if strfind(path, ".css", len(searchDir)) {
|
|
//we only want .css file
|
|
|
|
fileName := path[len(searchDir):strings.Index(path, ".css")]
|
|
//Remove file extension and path, keep only file name
|
|
|
|
themeList = append(themeList, fileName)
|
|
}
|
|
return nil
|
|
})
|
|
|
|
return themeList
|
|
}
|
|
|
|
func formatThemeName(name string, T publicSettings.TemplateTfunc) string {
|
|
translationString := fmt.Sprintf("themes_%s", name)
|
|
translatedName := string(T(translationString))
|
|
|
|
if translatedName != translationString {
|
|
//Translation string exists
|
|
return translatedName
|
|
}
|
|
|
|
if len(name) == 1 {
|
|
name = fmt.Sprintf("/%c/", name[0])
|
|
} else {
|
|
name = strings.Replace(name, "_", " ", -1)
|
|
name = strings.Title(name)
|
|
//Upper case at each start of word
|
|
}
|
|
return name
|
|
}
|
|
|
|
func formatDate(Date time.Time, short bool) string {
|
|
Date = Date.UTC()
|
|
if short {
|
|
return fmt.Sprintf("%.3s %d, %d", Date.Month(), Date.Day(), Date.Year())
|
|
}
|
|
|
|
if Date.Hour() >= 12 {
|
|
return fmt.Sprintf("%d/%d/%d, %d:%.2d:%.2d PM UTC+0", Date.Month(), Date.Day(), Date.Year(), Date.Hour() - 12, Date.Minute(), Date.Second())
|
|
} else {
|
|
return fmt.Sprintf("%d/%d/%d, %d:%.2d:%.2d AM UTC+0", Date.Month(), Date.Day(), Date.Year(), Date.Hour(), Date.Minute(), Date.Second())
|
|
}
|
|
}
|
|
|
|
func GenSearchName(Search SearchForm, currentURL string, T publicSettings.TemplateTfunc) string {
|
|
if currentURL == "/" {
|
|
return string(T("home"))
|
|
}
|
|
if Search.UserName != "" {
|
|
return Search.UserName
|
|
}
|
|
if Search.NameSearch != "" {
|
|
return Search.NameSearch
|
|
}
|
|
if Search.Category != "" {
|
|
return string(T(GetCategoryName(Search.Category)))
|
|
}
|
|
return string(T("search"))
|
|
}
|