package router import ( "html/template" "log" "math" "net/url" "strconv" "github.com/ewhal/nyaa/config" "github.com/ewhal/nyaa/service/user/permission" "github.com/ewhal/nyaa/util/languages" "github.com/nicksnyder/go-i18n/i18n" ) var FuncMap = template.FuncMap{ "min": math.Min, "genRoute": func(name string, params ...string) string { url, err := Router.Get(name).URL(params...) if err == nil { return url.String() } return "error" }, "genRouteWithQuery": func(name string, currentUrl *url.URL, params ...string) template.URL { url, err := Router.Get(name).URL(params...) if err == nil { return template.URL(url.String() + "?" + currentUrl.RawQuery) } return "error" }, "genViewTorrentRoute": func(torrent_id uint) string { // Helper for when you have an uint while genRoute("view_torrent", ...) takes a string // FIXME better solution? s := strconv.FormatUint(uint64(torrent_id), 10) url, err := Router.Get("view_torrent").URL("id", s) if err == nil { return url.String() } return "error" }, "genSearchWithOrdering": func(currentUrl url.URL, sortBy string) template.URL { values := currentUrl.Query() order := false if _, ok := values["order"]; ok { order, _ = strconv.ParseBool(values["order"][0]) if values["sort"][0] == 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)) url, _ := Router.Get("search").URL() url.RawQuery = values.Encode() return template.URL(url.String()) }, "genSortArrows": func(currentUrl url.URL, sortBy string) template.HTML { values := currentUrl.Query() leftclass := "sortarrowdim" rightclass := "sortarrowdim" if _, ok := values["order"]; ok { if values["sort"][0] == sortBy { if values["order"][0] == "true" { rightclass = "" } else { leftclass = "" } } } arrows := "" return template.HTML(arrows) }, "genNav": func(nav Navigation, currentUrl *url.URL, pagesSelectable int) template.HTML { var ret = "" if nav.TotalItem > 0 { maxPages := math.Ceil(float64(nav.TotalItem) / float64(nav.MaxItemPerPage)) if nav.CurrentPage-1 > 0 { url, _ := Router.Get(nav.Route).URL("page", "1") ret = ret + "
  • «
  • " } startValue := 1 if nav.CurrentPage > pagesSelectable/2 { startValue = (int(math.Min((float64(nav.CurrentPage)+math.Floor(float64(pagesSelectable)/2)), maxPages)) - pagesSelectable + 1) } endValue := (startValue + pagesSelectable - 1) if endValue > int(maxPages) { endValue = int(maxPages) } log.Println(nav.TotalItem) for i := startValue; i <= endValue; i++ { pageNum := strconv.Itoa(i) url, _ := Router.Get(nav.Route).URL("page", pageNum) ret = ret + "" + strconv.Itoa(i) + "" } if nav.CurrentPage < int(maxPages) { url, _ := Router.Get(nav.Route).URL("page", strconv.Itoa(nav.CurrentPage+1)) ret = ret + "
  • »
  • " } } return template.HTML(ret) }, "Sukebei": func() bool { if config.TableName == "sukebei_torrents" { return true } else { return false } }, "T": i18n.IdentityTfunc, "Ts": i18n.IdentityTfunc, "getDefaultLanguage": languages.GetDefaultLanguage, "getAvatar": func(hash string, size int) string { return "https://www.gravatar.com/avatar/" + hash + "?s=" + strconv.Itoa(size) }, "CurrentOrAdmin": userPermission.CurrentOrAdmin, "CurrentUserIdentical": userPermission.CurrentUserIdentical, "HasAdmin": userPermission.HasAdmin, "NeedsCaptcha": userPermission.NeedsCaptcha, "GetRole": userPermission.GetRole, "IsFollower": userPermission.IsFollower, "NoEncode": func(str string) template.HTML { return template.HTML(str) }, "calcWidthSeed": func(seed uint32, leech uint32) float64 { return float64(float64(seed)/(float64(seed)+float64(leech))) * 100 }, "calcWidthLeech": func(seed uint32, leech uint32) float64 { return float64(float64(leech)/(float64(seed)+float64(leech))) * 100 }, }