nyaa-pantsu/router/template_variables.go

package router

import (
	"net/http"
	"net/url"

	"github.com/NyaaPantsu/nyaa/common"
	"github.com/NyaaPantsu/nyaa/config"
	"github.com/NyaaPantsu/nyaa/model"
	"github.com/NyaaPantsu/nyaa/service/user"
	userForms "github.com/NyaaPantsu/nyaa/service/user/form"
	"github.com/NyaaPantsu/nyaa/util/filelist"
	"github.com/NyaaPantsu/nyaa/util/publicSettings"
	"github.com/gorilla/mux"
	"github.com/justinas/nosurf"
)

/* Each Page should have an object to pass to their own template
* Therefore, we put them in a separate file for better maintenance
*
* MAIN Template Variables
 */

type viewTemplateVariables struct {
	commonTemplateVariables
	Torrent    model.TorrentJSON
	RootFolder *filelist.FileListFolder // used for tree view
	CaptchaID  string
	FormErrors map[string][]string
	Infos      map[string][]string
}

type formTemplateVariables struct {
	commonTemplateVariables
	Form       interface{}
	FormErrors map[string][]string
	FormInfos  map[string][]string
}

type modelListVbs struct {
	commonTemplateVariables
	Models interface{}
	Errors map[string][]string
	Infos  map[string][]string
}

type userProfileEditVariables struct {
	commonTemplateVariables
	UserProfile *model.User
	UserForm    userForms.UserForm
	FormErrors  map[string][]string
	FormInfos   map[string][]string
	Languages   map[string]string
}

type userVerifyTemplateVariables struct {
	commonTemplateVariables
	FormErrors map[string][]string
}

type userProfileVariables struct {
	commonTemplateVariables
	UserProfile *model.User
	FormInfos   map[string][]string
}

type databaseDumpTemplateVariables struct {
	commonTemplateVariables
	ListDumps []model.DatabaseDumpJSON
	GPGLink   string
}

type changeLanguageVariables struct {
	commonTemplateVariables
	Language  string
	Languages map[string]string
}

/* MODERATION Variables */

type panelIndexVbs struct {
	commonTemplateVariables
	Torrents       []model.Torrent
	TorrentReports []model.TorrentReportJSON
	Users          []model.User
	Comments       []model.Comment
}

/*
* Variables used by the upper ones
 */

type commonTemplateVariables struct {
	Navigation navigation
	Search     searchForm
	T          publicSettings.TemplateTfunc
	Theme      string
	Mascot     string
	MascotURL  string
	User       *model.User
	URL        *url.URL   // for parsing URL in templates
	Route      *mux.Route // for getting current route in templates
	CsrfToken  string
	Config     *config.Config
}

type navigation struct {
	TotalItem      int
	MaxItemPerPage int // FIXME: shouldn't this be in SearchForm?
	CurrentPage    int
	Route          string
}

type searchForm struct {
	common.SearchParam
	Category         string
	ShowItemsPerPage bool
	SizeType         string
	DateType         string
	MinSize          string
	MaxSize          string
	FromDate         string
	ToDate           string
}

// Some Default Values to ease things out
func newNavigation() navigation {
	return navigation{
		MaxItemPerPage: 50,
	}
}

func newSearchForm(r *http.Request) searchForm {
	sizeType := r.URL.Query().Get("sizeType")
	if sizeType == "" {
		sizeType = "m"
	}

	return searchForm{
		Category:         "_",
		ShowItemsPerPage: true,
		SizeType:         sizeType,
		DateType:         r.URL.Query().Get("dateType"),
		MinSize:          r.URL.Query().Get("minSize"),  // We need to overwrite the value here, since size are formatted
		MaxSize:          r.URL.Query().Get("maxSize"),  // We need to overwrite the value here, since size are formatted
		FromDate:         r.URL.Query().Get("fromDate"), // We need to overwrite the value here, since we can have toDate instead and date are formatted
		ToDate:           r.URL.Query().Get("toDate"),   // We need to overwrite the value here, since date are formatted
	}
}

func newModelList(r *http.Request, models interface{}) modelListVbs {
	return modelListVbs{
		commonTemplateVariables: newCommonVariables(r),
		Models:                  models,
	}
}

func getUser(r *http.Request) *model.User {
	user, _, _ := userService.RetrieveCurrentUser(r)
	return &user
}

func newCommonVariables(r *http.Request) commonTemplateVariables {
	return commonTemplateVariables{
		Navigation: newNavigation(),
		Search:     newSearchForm(r),
		T:          publicSettings.GetTfuncFromRequest(r),
		Theme:      publicSettings.GetThemeFromRequest(r),
		Mascot:     publicSettings.GetMascotFromRequest(r),
		MascotURL:  publicSettings.GetMascotUrlFromRequest(r),
		User:       getUser(r),
		URL:        r.URL,
		Route:      mux.CurrentRoute(r),
		CsrfToken:  nosurf.Token(r),
		Config:     config.Conf,
	}
}